mirror of
https://github.com/pelgraine/Meck.git
synced 2026-07-06 01:31:17 +02:00
pelgraine
261 lines
8.0 KiB
C++
261 lines
8.0 KiB
C++
#pragma once
|
||
|
||
#include <helpers/ui/UIScreen.h>
|
||
#include <helpers/ui/DisplayDriver.h>
|
||
#include <helpers/AdvertDataHelpers.h>
|
||
#include <MeshCore.h>
|
||
|
||
// Forward declarations
|
||
class UITask;
|
||
class MyMesh;
|
||
extern MyMesh the_mesh;
|
||
|
||
class DiscoveryScreen : public UIScreen {
|
||
UITask* _task;
|
||
mesh::RTCClock* _rtc;
|
||
int _scrollPos;
|
||
int _rowsPerPage;
|
||
|
||
static char typeChar(uint8_t adv_type) {
|
||
switch (adv_type) {
|
||
case ADV_TYPE_CHAT: return 'C';
|
||
case ADV_TYPE_REPEATER: return 'R';
|
||
case ADV_TYPE_ROOM: return 'S';
|
||
case ADV_TYPE_SENSOR: return 'N';
|
||
default: return '?';
|
||
}
|
||
}
|
||
|
||
static const char* typeLabel(uint8_t adv_type) {
|
||
switch (adv_type) {
|
||
case ADV_TYPE_CHAT: return "Chat";
|
||
case ADV_TYPE_REPEATER: return "Rptr";
|
||
case ADV_TYPE_ROOM: return "Room";
|
||
case ADV_TYPE_SENSOR: return "Sens";
|
||
default: return "?";
|
||
}
|
||
}
|
||
|
||
public:
|
||
DiscoveryScreen(UITask* task, mesh::RTCClock* rtc)
|
||
: _task(task), _rtc(rtc), _scrollPos(0), _rowsPerPage(5) {}
|
||
|
||
void resetScroll() { _scrollPos = 0; }
|
||
|
||
int getSelectedIdx() const { return _scrollPos; }
|
||
|
||
// Tap-to-select: given virtual Y, select discovered node row.
|
||
// Returns: 0=miss, 1=moved, 2=tapped current row.
|
||
int selectRowAtVY(int vy) {
|
||
int count = the_mesh.getDiscoveredCount();
|
||
if (count == 0) return 0;
|
||
const int headerH = 14, footerH = 14, lineH = the_mesh.getNodePrefs()->smallLineH();
|
||
#if defined(LilyGo_T5S3_EPaper_Pro)
|
||
const int bodyTop = headerH;
|
||
#else
|
||
const int bodyTop = headerH + the_mesh.getNodePrefs()->smallHighlightOff();
|
||
#endif
|
||
if (vy < bodyTop || vy >= 128 - footerH) return 0;
|
||
|
||
int maxVisible = (128 - headerH - footerH) / lineH;
|
||
if (maxVisible < 3) maxVisible = 3;
|
||
int startIdx = max(0, min(_scrollPos - maxVisible / 2,
|
||
count - maxVisible));
|
||
|
||
int tappedRow = startIdx + (vy - bodyTop) / lineH;
|
||
if (tappedRow < 0 || tappedRow >= count) return 0;
|
||
|
||
if (tappedRow == _scrollPos) return 2;
|
||
_scrollPos = tappedRow;
|
||
return 1;
|
||
}
|
||
|
||
int render(DisplayDriver& display) override {
|
||
int count = the_mesh.getDiscoveredCount();
|
||
bool active = the_mesh.isDiscoveryActive();
|
||
|
||
// === Header ===
|
||
display.setTextSize(1);
|
||
display.setColor(DisplayDriver::GREEN);
|
||
display.setCursor(0, 0);
|
||
|
||
char hdr[32];
|
||
if (active) {
|
||
snprintf(hdr, sizeof(hdr), "Scanning... %d found", count);
|
||
} else {
|
||
snprintf(hdr, sizeof(hdr), "Scan done: %d found", count);
|
||
}
|
||
display.print(hdr);
|
||
|
||
// Divider
|
||
display.drawRect(0, 11, display.width(), 1);
|
||
|
||
// === Body — discovered node rows ===
|
||
display.setTextSize(the_mesh.getNodePrefs()->smallTextSize()); // tiny font for compact rows
|
||
int lineHeight = the_mesh.getNodePrefs()->smallLineH();
|
||
int headerHeight = 14;
|
||
int footerHeight = 14;
|
||
int maxY = display.height() - footerHeight;
|
||
int y = headerHeight;
|
||
int rowsDrawn = 0;
|
||
|
||
if (count == 0) {
|
||
display.setColor(DisplayDriver::LIGHT);
|
||
display.setCursor(4, 28);
|
||
display.print(active ? "Listening for adverts..." : "No nodes found");
|
||
if (!active) {
|
||
display.setCursor(4, 38);
|
||
#if defined(LilyGo_T5S3_EPaper_Pro)
|
||
display.print("Long press: Rescan");
|
||
#else
|
||
display.print("F: Scan again Sh+Del: Back");
|
||
#endif
|
||
}
|
||
} else {
|
||
// Center visible window around selected item
|
||
int maxVisible = (maxY - headerHeight) / lineHeight;
|
||
if (maxVisible < 3) maxVisible = 3;
|
||
int startIdx = max(0, min(_scrollPos - maxVisible / 2,
|
||
count - maxVisible));
|
||
int endIdx = min(count, startIdx + maxVisible);
|
||
|
||
for (int i = startIdx; i < endIdx && y + lineHeight <= maxY; i++) {
|
||
const DiscoveredNode& node = the_mesh.getDiscovered(i);
|
||
bool selected = (i == _scrollPos);
|
||
|
||
// Highlight selected row
|
||
if (selected) {
|
||
display.setColor(DisplayDriver::LIGHT);
|
||
#if defined(LilyGo_T5S3_EPaper_Pro)
|
||
display.fillRect(0, y, display.width(), lineHeight);
|
||
#else
|
||
display.fillRect(0, y + the_mesh.getNodePrefs()->smallHighlightOff(), display.width(), lineHeight);
|
||
#endif
|
||
display.setColor(DisplayDriver::DARK);
|
||
} else {
|
||
display.setColor(DisplayDriver::LIGHT);
|
||
}
|
||
|
||
display.setCursor(0, y);
|
||
|
||
// Prefix: cursor + type
|
||
char prefix[4];
|
||
if (selected) {
|
||
snprintf(prefix, sizeof(prefix), ">%c", typeChar(node.contact.type));
|
||
} else {
|
||
snprintf(prefix, sizeof(prefix), " %c", typeChar(node.contact.type));
|
||
}
|
||
display.print(prefix);
|
||
|
||
// Build right-side info: SNR or hop count + status
|
||
char rightStr[16];
|
||
if (node.snr != 0) {
|
||
// Active discovery result — show SNR in dB (value is ×4 scaled)
|
||
int snr_db = node.snr / 4;
|
||
if (node.already_in_contacts) {
|
||
snprintf(rightStr, sizeof(rightStr), "%ddB [+]", snr_db);
|
||
} else {
|
||
snprintf(rightStr, sizeof(rightStr), "%ddB", snr_db);
|
||
}
|
||
} else {
|
||
// Pre-seeded from cache — show hop count
|
||
if (node.already_in_contacts) {
|
||
snprintf(rightStr, sizeof(rightStr), "%dh [+]", node.path_len & 63);
|
||
} else {
|
||
snprintf(rightStr, sizeof(rightStr), "%dh", node.path_len & 63);
|
||
}
|
||
}
|
||
int rightWidth = display.getTextWidth(rightStr) + 2;
|
||
|
||
// Name (truncated with ellipsis)
|
||
char filteredName[32];
|
||
display.translateUTF8ToBlocks(filteredName, node.contact.name, sizeof(filteredName));
|
||
int nameX = display.getTextWidth(prefix) + 2;
|
||
int nameMaxW = display.width() - nameX - rightWidth - 2;
|
||
display.drawTextEllipsized(nameX, y, nameMaxW, filteredName);
|
||
|
||
// Right-aligned info
|
||
display.setCursor(display.width() - rightWidth, y);
|
||
display.print(rightStr);
|
||
|
||
y += lineHeight;
|
||
rowsDrawn++;
|
||
}
|
||
_rowsPerPage = (rowsDrawn > 0) ? rowsDrawn : 1;
|
||
}
|
||
|
||
display.setTextSize(1); // restore for footer
|
||
|
||
// === Footer ===
|
||
int footerY = display.height() - 12;
|
||
display.drawRect(0, footerY - 2, display.width(), 1);
|
||
display.setColor(DisplayDriver::YELLOW);
|
||
|
||
display.setCursor(0, footerY);
|
||
#if defined(LilyGo_T5S3_EPaper_Pro)
|
||
display.print("Swipe:Scroll");
|
||
|
||
const char* mid = "Tap:Add";
|
||
display.setCursor((display.width() - display.getTextWidth(mid)) / 2, footerY);
|
||
display.print(mid);
|
||
|
||
const char* right = "Hold:Rescan";
|
||
display.setCursor(display.width() - display.getTextWidth(right) - 2, footerY);
|
||
display.print(right);
|
||
#else
|
||
display.print("Sh+Del:X F:Scan");
|
||
|
||
const char* right = "Tap/Ent:Add";
|
||
display.setCursor(display.width() - display.getTextWidth(right) - 2, footerY);
|
||
display.print(right);
|
||
#endif
|
||
|
||
// Faster refresh while actively scanning
|
||
return active ? 1000 : 5000;
|
||
}
|
||
|
||
bool handleInput(char c) override {
|
||
int count = the_mesh.getDiscoveredCount();
|
||
|
||
// Shift+W: page up
|
||
if (c == 'W') {
|
||
int pageSize = (128 - 14 - 14) / the_mesh.getNodePrefs()->smallLineH();
|
||
if (pageSize < 3) pageSize = 3;
|
||
_scrollPos = max(0, _scrollPos - pageSize);
|
||
return true;
|
||
}
|
||
// W - scroll up
|
||
if (c == 'w' || c == 0xF2) {
|
||
if (_scrollPos > 0) {
|
||
_scrollPos--;
|
||
return true;
|
||
}
|
||
}
|
||
|
||
// Shift+S: page down
|
||
if (c == 'S') {
|
||
int pageSize = (128 - 14 - 14) / the_mesh.getNodePrefs()->smallLineH();
|
||
if (pageSize < 3) pageSize = 3;
|
||
_scrollPos = min(count - 1, _scrollPos + pageSize);
|
||
return true;
|
||
}
|
||
// S - scroll down
|
||
if (c == 's' || c == 0xF1) {
|
||
if (_scrollPos < count - 1) {
|
||
_scrollPos++;
|
||
return true;
|
||
}
|
||
}
|
||
|
||
// F - rescan (handled here as well as in main.cpp for consistency)
|
||
if (c == 'f') {
|
||
the_mesh.startDiscovery();
|
||
_scrollPos = 0;
|
||
return true;
|
||
}
|
||
|
||
// Enter - handled by main.cpp for alert feedback
|
||
|
||
return false; // Q/back and Enter handled by main.cpp
|
||
}
|
||
}; |