UITask.cpp 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800
  1. #include "UITask.h"
  2. #include <helpers/TxtDataHelpers.h>
  3. #include "../MyMesh.h"
  4. #include "target.h"
  5. #ifndef AUTO_OFF_MILLIS
  6. #define AUTO_OFF_MILLIS 15000 // 15 seconds
  7. #endif
  8. #define BOOT_SCREEN_MILLIS 3000 // 3 seconds
  9. #ifdef PIN_STATUS_LED
  10. #define LED_ON_MILLIS 20
  11. #define LED_ON_MSG_MILLIS 200
  12. #define LED_CYCLE_MILLIS 4000
  13. #endif
  14. #define LONG_PRESS_MILLIS 1200
  15. #ifndef UI_RECENT_LIST_SIZE
  16. #define UI_RECENT_LIST_SIZE 4
  17. #endif
  18. #define PRESS_LABEL "long press"
  19. #include "icons.h"
  20. class SplashScreen : public UIScreen {
  21. UITask* _task;
  22. unsigned long dismiss_after;
  23. char _version_info[12];
  24. public:
  25. SplashScreen(UITask* task) : _task(task) {
  26. // strip off dash and commit hash by changing dash to null terminator
  27. // e.g: v1.2.3-abcdef -> v1.2.3
  28. const char *ver = FIRMWARE_VERSION;
  29. const char *dash = strchr(ver, '-');
  30. int len = dash ? dash - ver : strlen(ver);
  31. if (len >= sizeof(_version_info)) len = sizeof(_version_info) - 1;
  32. memcpy(_version_info, ver, len);
  33. _version_info[len] = 0;
  34. dismiss_after = millis() + BOOT_SCREEN_MILLIS;
  35. }
  36. int render(DisplayDriver& display) override {
  37. // meshcore logo
  38. display.setColor(DisplayDriver::BLUE);
  39. int logoWidth = 128;
  40. display.drawXbm((display.width() - logoWidth) / 2, 3, meshcore_logo, logoWidth, 13);
  41. // version info
  42. display.setColor(DisplayDriver::LIGHT);
  43. display.setTextSize(2);
  44. display.drawTextCentered(display.width()/2, 22, _version_info);
  45. display.setTextSize(1);
  46. display.drawTextCentered(display.width()/2, 42, FIRMWARE_BUILD_DATE);
  47. return 1000;
  48. }
  49. void poll() override {
  50. if (millis() >= dismiss_after) {
  51. _task->gotoHomeScreen();
  52. }
  53. }
  54. };
  55. class HomeScreen : public UIScreen {
  56. enum HomePage {
  57. FIRST,
  58. RECENT,
  59. RADIO,
  60. BLUETOOTH,
  61. ADVERT,
  62. #if UI_SENSORS_PAGE == 1
  63. SENSORS,
  64. #endif
  65. SHUTDOWN,
  66. Count // keep as last
  67. };
  68. UITask* _task;
  69. mesh::RTCClock* _rtc;
  70. SensorManager* _sensors;
  71. NodePrefs* _node_prefs;
  72. uint8_t _page;
  73. bool _shutdown_init;
  74. AdvertPath recent[UI_RECENT_LIST_SIZE];
  75. void renderBatteryIndicator(DisplayDriver& display, uint16_t batteryMilliVolts) {
  76. // Convert millivolts to percentage
  77. const int minMilliVolts = 3000; // Minimum voltage (e.g., 3.0V)
  78. const int maxMilliVolts = 4200; // Maximum voltage (e.g., 4.2V)
  79. int batteryPercentage = ((batteryMilliVolts - minMilliVolts) * 100) / (maxMilliVolts - minMilliVolts);
  80. if (batteryPercentage < 0) batteryPercentage = 0; // Clamp to 0%
  81. if (batteryPercentage > 100) batteryPercentage = 100; // Clamp to 100%
  82. // battery icon
  83. int iconWidth = 24;
  84. int iconHeight = 10;
  85. int iconX = display.width() - iconWidth - 5; // Position the icon near the top-right corner
  86. int iconY = 0;
  87. display.setColor(DisplayDriver::GREEN);
  88. // battery outline
  89. display.drawRect(iconX, iconY, iconWidth, iconHeight);
  90. // battery "cap"
  91. display.fillRect(iconX + iconWidth, iconY + (iconHeight / 4), 3, iconHeight / 2);
  92. // fill the battery based on the percentage
  93. int fillWidth = (batteryPercentage * (iconWidth - 4)) / 100;
  94. display.fillRect(iconX + 2, iconY + 2, fillWidth, iconHeight - 4);
  95. }
  96. CayenneLPP sensors_lpp;
  97. int sensors_nb = 0;
  98. bool sensors_scroll = false;
  99. int sensors_scroll_offset = 0;
  100. int next_sensors_refresh = 0;
  101. void refresh_sensors() {
  102. if (millis() > next_sensors_refresh) {
  103. sensors_lpp.reset();
  104. sensors_nb = 0;
  105. sensors_lpp.addVoltage(TELEM_CHANNEL_SELF, (float)board.getBattMilliVolts() / 1000.0f);
  106. sensors.querySensors(0xFF, sensors_lpp);
  107. LPPReader reader (sensors_lpp.getBuffer(), sensors_lpp.getSize());
  108. uint8_t channel, type;
  109. while(reader.readHeader(channel, type)) {
  110. reader.skipData(type);
  111. sensors_nb ++;
  112. }
  113. sensors_scroll = sensors_nb > UI_RECENT_LIST_SIZE;
  114. #if AUTO_OFF_MILLIS > 0
  115. next_sensors_refresh = millis() + 5000; // refresh sensor values every 5 sec
  116. #else
  117. next_sensors_refresh = millis() + 60000; // refresh sensor values every 1 min
  118. #endif
  119. }
  120. }
  121. public:
  122. HomeScreen(UITask* task, mesh::RTCClock* rtc, SensorManager* sensors, NodePrefs* node_prefs)
  123. : _task(task), _rtc(rtc), _sensors(sensors), _node_prefs(node_prefs), _page(0),
  124. _shutdown_init(false), sensors_lpp(200) { }
  125. void poll() override {
  126. if (_shutdown_init && !_task->isButtonPressed()) { // must wait for USR button to be released
  127. _task->shutdown();
  128. }
  129. }
  130. int render(DisplayDriver& display) override {
  131. char tmp[80];
  132. // node name
  133. display.setCursor(0, 0);
  134. display.setTextSize(1);
  135. display.setColor(DisplayDriver::GREEN);
  136. display.print(_node_prefs->node_name);
  137. // battery voltage
  138. renderBatteryIndicator(display, _task->getBattMilliVolts());
  139. // curr page indicator
  140. int y = 14;
  141. int x = display.width() / 2 - 25;
  142. for (uint8_t i = 0; i < HomePage::Count; i++, x += 10) {
  143. if (i == _page) {
  144. display.fillRect(x-1, y-1, 3, 3);
  145. } else {
  146. display.fillRect(x, y, 1, 1);
  147. }
  148. }
  149. if (_page == HomePage::FIRST) {
  150. display.setColor(DisplayDriver::YELLOW);
  151. display.setTextSize(2);
  152. sprintf(tmp, "MSG: %d", _task->getMsgCount());
  153. display.drawTextCentered(display.width() / 2, 20, tmp);
  154. if (_task->hasConnection()) {
  155. display.setColor(DisplayDriver::GREEN);
  156. display.setTextSize(1);
  157. display.drawTextCentered(display.width() / 2, 43, "< Connected >");
  158. } else if (the_mesh.getBLEPin() != 0) { // BT pin
  159. display.setColor(DisplayDriver::RED);
  160. display.setTextSize(2);
  161. sprintf(tmp, "Pin:%d", the_mesh.getBLEPin());
  162. display.drawTextCentered(display.width() / 2, 43, tmp);
  163. }
  164. } else if (_page == HomePage::RECENT) {
  165. the_mesh.getRecentlyHeard(recent, UI_RECENT_LIST_SIZE);
  166. display.setColor(DisplayDriver::GREEN);
  167. int y = 20;
  168. for (int i = 0; i < UI_RECENT_LIST_SIZE; i++, y += 11) {
  169. auto a = &recent[i];
  170. if (a->name[0] == 0) continue; // empty slot
  171. display.setCursor(0, y);
  172. display.print(a->name);
  173. int secs = _rtc->getCurrentTime() - a->recv_timestamp;
  174. if (secs < 60) {
  175. sprintf(tmp, "%ds", secs);
  176. } else if (secs < 60*60) {
  177. sprintf(tmp, "%dm", secs / 60);
  178. } else {
  179. sprintf(tmp, "%dh", secs / (60*60));
  180. }
  181. display.setCursor(display.width() - display.getTextWidth(tmp) - 1, y);
  182. display.print(tmp);
  183. }
  184. } else if (_page == HomePage::RADIO) {
  185. display.setColor(DisplayDriver::YELLOW);
  186. display.setTextSize(1);
  187. // freq / sf
  188. display.setCursor(0, 20);
  189. sprintf(tmp, "FQ: %06.3f SF: %d", _node_prefs->freq, _node_prefs->sf);
  190. display.print(tmp);
  191. display.setCursor(0, 31);
  192. sprintf(tmp, "BW: %03.2f CR: %d", _node_prefs->bw, _node_prefs->cr);
  193. display.print(tmp);
  194. // tx power, noise floor
  195. display.setCursor(0, 42);
  196. sprintf(tmp, "TX: %ddBm", _node_prefs->tx_power_dbm);
  197. display.print(tmp);
  198. display.setCursor(0, 53);
  199. sprintf(tmp, "Noise floor: %d", radio_driver.getNoiseFloor());
  200. display.print(tmp);
  201. } else if (_page == HomePage::BLUETOOTH) {
  202. display.setColor(DisplayDriver::GREEN);
  203. display.drawXbm((display.width() - 32) / 2, 18,
  204. _task->isSerialEnabled() ? bluetooth_on : bluetooth_off,
  205. 32, 32);
  206. display.setTextSize(1);
  207. display.drawTextCentered(display.width() / 2, 64 - 11, "toggle: " PRESS_LABEL);
  208. } else if (_page == HomePage::ADVERT) {
  209. display.setColor(DisplayDriver::GREEN);
  210. display.drawXbm((display.width() - 32) / 2, 18, advert_icon, 32, 32);
  211. display.drawTextCentered(display.width() / 2, 64 - 11, "advert: " PRESS_LABEL);
  212. #if UI_SENSORS_PAGE == 1
  213. } else if (_page == HomePage::SENSORS) {
  214. int y = 18;
  215. refresh_sensors();
  216. char buf[30];
  217. char name[30];
  218. LPPReader r(sensors_lpp.getBuffer(), sensors_lpp.getSize());
  219. for (int i = 0; i < sensors_scroll_offset; i++) {
  220. uint8_t channel, type;
  221. r.readHeader(channel, type);
  222. r.skipData(type);
  223. }
  224. for (int i = 0; i < (sensors_scroll?UI_RECENT_LIST_SIZE:sensors_nb); i++) {
  225. uint8_t channel, type;
  226. if (!r.readHeader(channel, type)) { // reached end, reset
  227. r.reset();
  228. r.readHeader(channel, type);
  229. }
  230. display.setCursor(0, y);
  231. float v;
  232. switch (type) {
  233. case LPP_GPS: // GPS
  234. float lat, lon, alt;
  235. r.readGPS(lat, lon, alt);
  236. strcpy(name, "gps"); sprintf(buf, "%.4f %.4f", lat, lon);
  237. break;
  238. case LPP_VOLTAGE:
  239. r.readVoltage(v);
  240. strcpy(name, "voltage"); sprintf(buf, "%6.2f", v);
  241. break;
  242. case LPP_CURRENT:
  243. r.readCurrent(v);
  244. strcpy(name, "current"); sprintf(buf, "%.3f", v);
  245. break;
  246. case LPP_TEMPERATURE:
  247. r.readTemperature(v);
  248. strcpy(name, "temperature"); sprintf(buf, "%.2f", v);
  249. break;
  250. case LPP_RELATIVE_HUMIDITY:
  251. r.readRelativeHumidity(v);
  252. strcpy(name, "humidity"); sprintf(buf, "%.2f", v);
  253. break;
  254. case LPP_BAROMETRIC_PRESSURE:
  255. r.readPressure(v);
  256. strcpy(name, "pressure"); sprintf(buf, "%.2f", v);
  257. break;
  258. case LPP_ALTITUDE:
  259. r.readAltitude(v);
  260. strcpy(name, "altitude"); sprintf(buf, "%.0f", v);
  261. break;
  262. case LPP_POWER:
  263. r.readPower(v);
  264. strcpy(name, "power"); sprintf(buf, "%6.2f", v);
  265. break;
  266. default:
  267. r.skipData(type);
  268. strcpy(name, "unk"); sprintf(buf, "");
  269. }
  270. display.setCursor(0, y);
  271. display.print(name);
  272. display.setCursor(
  273. display.width()-display.getTextWidth(buf)-1, y
  274. );
  275. display.print(buf);
  276. y = y + 12;
  277. }
  278. if (sensors_scroll) sensors_scroll_offset = (sensors_scroll_offset+1)%sensors_nb;
  279. else sensors_scroll_offset = 0;
  280. #endif
  281. } else if (_page == HomePage::SHUTDOWN) {
  282. display.setColor(DisplayDriver::GREEN);
  283. display.setTextSize(1);
  284. if (_shutdown_init) {
  285. display.drawTextCentered(display.width() / 2, 34, "hibernating...");
  286. } else {
  287. display.drawXbm((display.width() - 32) / 2, 18, power_icon, 32, 32);
  288. display.drawTextCentered(display.width() / 2, 64 - 11, "hibernate: " PRESS_LABEL);
  289. }
  290. }
  291. return 5000; // next render after 5000 ms
  292. }
  293. bool handleInput(char c) override {
  294. if (c == KEY_LEFT || c == KEY_PREV) {
  295. _page = (_page + HomePage::Count - 1) % HomePage::Count;
  296. return true;
  297. }
  298. if (c == KEY_NEXT || c == KEY_RIGHT) {
  299. _page = (_page + 1) % HomePage::Count;
  300. if (_page == HomePage::RECENT) {
  301. _task->showAlert("Recent adverts", 800);
  302. }
  303. return true;
  304. }
  305. if (c == KEY_ENTER && _page == HomePage::BLUETOOTH) {
  306. if (_task->isSerialEnabled()) { // toggle Bluetooth on/off
  307. _task->disableSerial();
  308. } else {
  309. _task->enableSerial();
  310. }
  311. return true;
  312. }
  313. if (c == KEY_ENTER && _page == HomePage::ADVERT) {
  314. _task->notify(UIEventType::ack);
  315. if (the_mesh.advert()) {
  316. _task->showAlert("Advert sent!", 1000);
  317. } else {
  318. _task->showAlert("Advert failed..", 1000);
  319. }
  320. return true;
  321. }
  322. #if UI_SENSORS_PAGE == 1
  323. if (c == KEY_ENTER && _page == HomePage::SENSORS) {
  324. _task->toggleGPS();
  325. next_sensors_refresh=0;
  326. return true;
  327. }
  328. #endif
  329. if (c == KEY_ENTER && _page == HomePage::SHUTDOWN) {
  330. _shutdown_init = true; // need to wait for button to be released
  331. return true;
  332. }
  333. return false;
  334. }
  335. };
  336. class MsgPreviewScreen : public UIScreen {
  337. UITask* _task;
  338. mesh::RTCClock* _rtc;
  339. struct MsgEntry {
  340. uint32_t timestamp;
  341. char origin[62];
  342. char msg[78];
  343. };
  344. #define MAX_UNREAD_MSGS 32
  345. int num_unread;
  346. MsgEntry unread[MAX_UNREAD_MSGS];
  347. public:
  348. MsgPreviewScreen(UITask* task, mesh::RTCClock* rtc) : _task(task), _rtc(rtc) { num_unread = 0; }
  349. void addPreview(uint8_t path_len, const char* from_name, const char* msg) {
  350. if (num_unread >= MAX_UNREAD_MSGS) return; // full
  351. auto p = &unread[num_unread++];
  352. p->timestamp = _rtc->getCurrentTime();
  353. if (path_len == 0xFF) {
  354. sprintf(p->origin, "(D) %s:", from_name);
  355. } else {
  356. sprintf(p->origin, "(%d) %s:", (uint32_t) path_len, from_name);
  357. }
  358. StrHelper::strncpy(p->msg, msg, sizeof(p->msg));
  359. }
  360. int render(DisplayDriver& display) override {
  361. char tmp[16];
  362. display.setCursor(0, 0);
  363. display.setTextSize(1);
  364. display.setColor(DisplayDriver::GREEN);
  365. sprintf(tmp, "Unread: %d", num_unread);
  366. display.print(tmp);
  367. auto p = &unread[0];
  368. int secs = _rtc->getCurrentTime() - p->timestamp;
  369. if (secs < 60) {
  370. sprintf(tmp, "%ds", secs);
  371. } else if (secs < 60*60) {
  372. sprintf(tmp, "%dm", secs / 60);
  373. } else {
  374. sprintf(tmp, "%dh", secs / (60*60));
  375. }
  376. display.setCursor(display.width() - display.getTextWidth(tmp) - 2, 0);
  377. display.print(tmp);
  378. display.drawRect(0, 11, display.width(), 1); // horiz line
  379. display.setCursor(0, 14);
  380. display.setColor(DisplayDriver::YELLOW);
  381. display.print(p->origin);
  382. display.setCursor(0, 25);
  383. display.setColor(DisplayDriver::LIGHT);
  384. display.printWordWrap(p->msg, display.width());
  385. #if AUTO_OFF_MILLIS==0 // probably e-ink
  386. return 10000; // 10 s
  387. #else
  388. return 1000; // next render after 1000 ms
  389. #endif
  390. }
  391. bool handleInput(char c) override {
  392. if (c == KEY_NEXT || c == KEY_RIGHT) {
  393. num_unread--;
  394. if (num_unread == 0) {
  395. _task->gotoHomeScreen();
  396. } else {
  397. // delete first/curr item from unread queue
  398. for (int i = 0; i < num_unread; i++) {
  399. unread[i] = unread[i + 1];
  400. }
  401. }
  402. return true;
  403. }
  404. if (c == KEY_ENTER) {
  405. num_unread = 0; // clear unread queue
  406. _task->gotoHomeScreen();
  407. return true;
  408. }
  409. return false;
  410. }
  411. };
  412. void UITask::begin(DisplayDriver* display, SensorManager* sensors, NodePrefs* node_prefs) {
  413. _display = display;
  414. _sensors = sensors;
  415. _auto_off = millis() + AUTO_OFF_MILLIS;
  416. #if defined(PIN_USER_BTN)
  417. user_btn.begin();
  418. #endif
  419. #if defined(PIN_USER_BTN_ANA)
  420. analog_btn.begin();
  421. #endif
  422. _node_prefs = node_prefs;
  423. if (_display != NULL) {
  424. _display->turnOn();
  425. }
  426. #ifdef PIN_BUZZER
  427. buzzer.begin();
  428. #endif
  429. #ifdef PIN_VIBRATION
  430. vibration.begin();
  431. #endif
  432. ui_started_at = millis();
  433. _alert_expiry = 0;
  434. splash = new SplashScreen(this);
  435. home = new HomeScreen(this, &rtc_clock, sensors, node_prefs);
  436. msg_preview = new MsgPreviewScreen(this, &rtc_clock);
  437. setCurrScreen(splash);
  438. }
  439. void UITask::showAlert(const char* text, int duration_millis) {
  440. strcpy(_alert, text);
  441. _alert_expiry = millis() + duration_millis;
  442. }
  443. void UITask::notify(UIEventType t) {
  444. #if defined(PIN_BUZZER)
  445. switch(t){
  446. case UIEventType::contactMessage:
  447. // gemini's pick
  448. buzzer.play("MsgRcv3:d=4,o=6,b=200:32e,32g,32b,16c7");
  449. break;
  450. case UIEventType::channelMessage:
  451. buzzer.play("kerplop:d=16,o=6,b=120:32g#,32c#");
  452. break;
  453. case UIEventType::ack:
  454. buzzer.play("ack:d=32,o=8,b=120:c");
  455. break;
  456. case UIEventType::roomMessage:
  457. case UIEventType::newContactMessage:
  458. case UIEventType::none:
  459. default:
  460. break;
  461. }
  462. #endif
  463. #ifdef PIN_VIBRATION
  464. // Trigger vibration for all UI events except none
  465. if (t != UIEventType::none) {
  466. vibration.trigger();
  467. }
  468. #endif
  469. }
  470. void UITask::msgRead(int msgcount) {
  471. _msgcount = msgcount;
  472. if (msgcount == 0) {
  473. gotoHomeScreen();
  474. }
  475. }
  476. void UITask::newMsg(uint8_t path_len, const char* from_name, const char* text, int msgcount) {
  477. _msgcount = msgcount;
  478. ((MsgPreviewScreen *) msg_preview)->addPreview(path_len, from_name, text);
  479. setCurrScreen(msg_preview);
  480. if (_display != NULL) {
  481. if (!_display->isOn()) _display->turnOn();
  482. _auto_off = millis() + AUTO_OFF_MILLIS; // extend the auto-off timer
  483. _next_refresh = 100; // trigger refresh
  484. }
  485. }
  486. void UITask::userLedHandler() {
  487. #ifdef PIN_STATUS_LED
  488. int cur_time = millis();
  489. if (cur_time > next_led_change) {
  490. if (led_state == 0) {
  491. led_state = 1;
  492. if (_msgcount > 0) {
  493. last_led_increment = LED_ON_MSG_MILLIS;
  494. } else {
  495. last_led_increment = LED_ON_MILLIS;
  496. }
  497. next_led_change = cur_time + last_led_increment;
  498. } else {
  499. led_state = 0;
  500. next_led_change = cur_time + LED_CYCLE_MILLIS - last_led_increment;
  501. }
  502. digitalWrite(PIN_STATUS_LED, led_state);
  503. }
  504. #endif
  505. }
  506. void UITask::setCurrScreen(UIScreen* c) {
  507. curr = c;
  508. _next_refresh = 100;
  509. }
  510. /*
  511. hardware-agnostic pre-shutdown activity should be done here
  512. */
  513. void UITask::shutdown(bool restart){
  514. #ifdef PIN_BUZZER
  515. /* note: we have a choice here -
  516. we can do a blocking buzzer.loop() with non-deterministic consequences
  517. or we can set a flag and delay the shutdown for a couple of seconds
  518. while a non-blocking buzzer.loop() plays out in UITask::loop()
  519. */
  520. buzzer.shutdown();
  521. uint32_t buzzer_timer = millis(); // fail-safe shutdown
  522. while (buzzer.isPlaying() && (millis() - 2500) < buzzer_timer)
  523. buzzer.loop();
  524. #endif // PIN_BUZZER
  525. if (restart) {
  526. _board->reboot();
  527. } else {
  528. _display->turnOff();
  529. _board->powerOff();
  530. }
  531. }
  532. bool UITask::isButtonPressed() const {
  533. #ifdef PIN_USER_BTN
  534. return user_btn.isPressed();
  535. #else
  536. return false;
  537. #endif
  538. }
  539. void UITask::loop() {
  540. char c = 0;
  541. #if defined(PIN_USER_BTN)
  542. int ev = user_btn.check();
  543. if (ev == BUTTON_EVENT_CLICK) {
  544. c = checkDisplayOn(KEY_NEXT);
  545. } else if (ev == BUTTON_EVENT_LONG_PRESS) {
  546. c = handleLongPress(KEY_ENTER);
  547. } else if (ev == BUTTON_EVENT_DOUBLE_CLICK) {
  548. c = handleDoubleClick(KEY_PREV);
  549. } else if (ev == BUTTON_EVENT_TRIPLE_CLICK) {
  550. c = handleTripleClick(KEY_SELECT);
  551. }
  552. #endif
  553. #if defined(WIO_TRACKER_L1)
  554. ev = joystick_left.check();
  555. if (ev == BUTTON_EVENT_CLICK) {
  556. c = checkDisplayOn(KEY_LEFT);
  557. } else if (ev == BUTTON_EVENT_LONG_PRESS) {
  558. c = handleLongPress(KEY_LEFT);
  559. }
  560. ev = joystick_right.check();
  561. if (ev == BUTTON_EVENT_CLICK) {
  562. c = checkDisplayOn(KEY_RIGHT);
  563. } else if (ev == BUTTON_EVENT_LONG_PRESS) {
  564. c = handleLongPress(KEY_RIGHT);
  565. }
  566. #endif
  567. #if defined(PIN_USER_BTN_ANA)
  568. ev = analog_btn.check();
  569. if (ev == BUTTON_EVENT_CLICK) {
  570. c = checkDisplayOn(KEY_NEXT);
  571. } else if (ev == BUTTON_EVENT_LONG_PRESS) {
  572. c = handleLongPress(KEY_ENTER);
  573. } else if (ev == BUTTON_EVENT_DOUBLE_CLICK) {
  574. c = handleDoubleClick(KEY_PREV);
  575. } else if (ev == BUTTON_EVENT_TRIPLE_CLICK) {
  576. c = handleTripleClick(KEY_SELECT);
  577. }
  578. #endif
  579. #if defined(DISP_BACKLIGHT) && defined(BACKLIGHT_BTN)
  580. if (millis() > next_backlight_btn_check) {
  581. bool touch_state = digitalRead(PIN_BUTTON2);
  582. digitalWrite(DISP_BACKLIGHT, !touch_state);
  583. next_backlight_btn_check = millis() + 300;
  584. }
  585. #endif
  586. if (c != 0 && curr) {
  587. curr->handleInput(c);
  588. _auto_off = millis() + AUTO_OFF_MILLIS; // extend auto-off timer
  589. _next_refresh = 100; // trigger refresh
  590. }
  591. userLedHandler();
  592. #ifdef PIN_BUZZER
  593. if (buzzer.isPlaying()) buzzer.loop();
  594. #endif
  595. if (curr) curr->poll();
  596. if (_display != NULL && _display->isOn()) {
  597. if (millis() >= _next_refresh && curr) {
  598. _display->startFrame();
  599. int delay_millis = curr->render(*_display);
  600. if (millis() < _alert_expiry) { // render alert popup
  601. _display->setTextSize(1);
  602. int y = _display->height() / 3;
  603. int p = _display->height() / 32;
  604. _display->setColor(DisplayDriver::DARK);
  605. _display->fillRect(p, y, _display->width() - p*2, y);
  606. _display->setColor(DisplayDriver::LIGHT); // draw box border
  607. _display->drawRect(p, y, _display->width() - p*2, y);
  608. _display->drawTextCentered(_display->width() / 2, y + p*3, _alert);
  609. _next_refresh = _alert_expiry; // will need refresh when alert is dismissed
  610. } else {
  611. _next_refresh = millis() + delay_millis;
  612. }
  613. _display->endFrame();
  614. }
  615. #if AUTO_OFF_MILLIS > 0
  616. if (millis() > _auto_off) {
  617. _display->turnOff();
  618. }
  619. #endif
  620. }
  621. #ifdef PIN_VIBRATION
  622. vibration.loop();
  623. #endif
  624. #ifdef AUTO_SHUTDOWN_MILLIVOLTS
  625. if (millis() > next_batt_chck) {
  626. uint16_t milliVolts = getBattMilliVolts();
  627. if (milliVolts > 0 && milliVolts < AUTO_SHUTDOWN_MILLIVOLTS) {
  628. // show low battery shutdown alert
  629. // we should only do this for eink displays, which will persist after power loss
  630. #if defined(THINKNODE_M1) || defined(LILYGO_TECHO)
  631. if (_display != NULL) {
  632. _display->startFrame();
  633. _display->setTextSize(2);
  634. _display->setColor(DisplayDriver::RED);
  635. _display->drawTextCentered(_display->width() / 2, 20, "Low Battery.");
  636. _display->drawTextCentered(_display->width() / 2, 40, "Shutting Down!");
  637. _display->endFrame();
  638. }
  639. #endif
  640. shutdown();
  641. }
  642. next_batt_chck = millis() + 8000;
  643. }
  644. #endif
  645. }
  646. char UITask::checkDisplayOn(char c) {
  647. if (_display != NULL) {
  648. if (!_display->isOn()) {
  649. _display->turnOn(); // turn display on and consume event
  650. c = 0;
  651. }
  652. _auto_off = millis() + AUTO_OFF_MILLIS; // extend auto-off timer
  653. _next_refresh = 0; // trigger refresh
  654. }
  655. return c;
  656. }
  657. char UITask::handleLongPress(char c) {
  658. if (millis() - ui_started_at < 8000) { // long press in first 8 seconds since startup -> CLI/rescue
  659. the_mesh.enterCLIRescue();
  660. c = 0; // consume event
  661. }
  662. return c;
  663. }
  664. char UITask::handleDoubleClick(char c) {
  665. MESH_DEBUG_PRINTLN("UITask: double click triggered");
  666. checkDisplayOn(c);
  667. return c;
  668. }
  669. char UITask::handleTripleClick(char c) {
  670. MESH_DEBUG_PRINTLN("UITask: triple click triggered");
  671. checkDisplayOn(c);
  672. toggleBuzzer();
  673. c = 0;
  674. return c;
  675. }
  676. void UITask::toggleGPS() {
  677. if (_sensors != NULL) {
  678. // toggle GPS on/off
  679. int num = _sensors->getNumSettings();
  680. for (int i = 0; i < num; i++) {
  681. if (strcmp(_sensors->getSettingName(i), "gps") == 0) {
  682. if (strcmp(_sensors->getSettingValue(i), "1") == 0) {
  683. _sensors->setSettingValue("gps", "0");
  684. notify(UIEventType::ack);
  685. showAlert("GPS: Disabled", 800);
  686. } else {
  687. _sensors->setSettingValue("gps", "1");
  688. notify(UIEventType::ack);
  689. showAlert("GPS: Enabled", 800);
  690. }
  691. _next_refresh = 0;
  692. break;
  693. }
  694. }
  695. }
  696. }
  697. void UITask::toggleBuzzer() {
  698. // Toggle buzzer quiet mode
  699. #ifdef PIN_BUZZER
  700. if (buzzer.isQuiet()) {
  701. buzzer.quiet(false);
  702. notify(UIEventType::ack);
  703. showAlert("Buzzer: ON", 800);
  704. } else {
  705. buzzer.quiet(true);
  706. showAlert("Buzzer: OFF", 800);
  707. }
  708. _next_refresh = 0; // trigger refresh
  709. #endif
  710. }