restructure to use rust source tree primarily

1 files changed, 151 insertions, 0 deletions

diff --git a/src-arduino/utils/device_check.cpp b/src-arduino/utils/device_check.cpp
new file mode 100644
index 0000000..17613b3
--- /dev/null
+++ b/src-arduino/utils/device_check.cpp
@@ -0,0 +1,151 @@
+#include "ELECHOUSE_CC1101_SRC_DRV.h"
+#include <Arduino.h>
+#include <RF24.h>
+#include <SPI.h>
+#include <Wire.h>
+
+#include "config.h"
+#include "ui/display.h"
+#include "buttons.h"
+
+// ===== EXTERNALS =====
+extern RF24 radio1;
+extern RF24 radio2;
+extern SPIClass *RADIO_SPI;
+extern U8G2_SSD1306_128X64_NONAME_F_HW_I2C u8g2;
+
+// ===== RESULTS =====
+struct DeviceStatus {
+    bool nrf1 = false;
+    bool nrf2 = false;
+    bool cc1101 = false;
+    bool oled = true;
+};
+
+// ===== NRF CHECK =====
+// bool checkNRF(RF24 &radio)
+//{
+//    // safer: only init if needed
+//    if (!radio.isChipConnected()) {
+//        if (!radio.begin(RADIO_SPI))
+//            return false;
+//    }
+//
+//    return radio.isChipConnected();
+//}
+
+bool checkNRF(RF24 &radio) {
+    radio.powerDown();
+    delay(5);
+
+    if (!radio.begin(RADIO_SPI))
+        return false;
+
+    delay(5);
+
+    return radio.isChipConnected();
+}
+
+// ===== CC1101 CHECK =====
+bool checkCC1101(uint8_t csPin) {
+    ELECHOUSE_cc1101.setSpiPin(cc1101_SCK, cc1101_MISO, cc1101_MOSI, csPin);
+
+    delay(5);
+
+    return ELECHOUSE_cc1101.getCC1101();
+}
+
+// ===== DRAW =====
+#define MAX_ITEMS 4
+#define VISIBLE_ROWS 5
+
+const char *labels[MAX_ITEMS] = {"NRF1", "NRF2", "CC1101", "OLED"};
+
+bool values[MAX_ITEMS];
+
+int selectedIndex = 0;
+int offset = 0;
+
+void drawStatus(DeviceStatus &s) {
+    values[0] = s.nrf1;
+    values[1] = s.nrf2;
+    values[2] = s.cc1101;
+    values[3] = s.oled;
+
+    u8g2.clearBuffer();
+    u8g2.setFont(u8g2_font_6x10_tr);
+
+    // scrolling logic
+    if (selectedIndex < offset)
+        offset = selectedIndex;
+
+    if (selectedIndex >= offset + VISIBLE_ROWS)
+        offset = selectedIndex - VISIBLE_ROWS + 1;
+
+    for (int i = 0; i < VISIBLE_ROWS; i++) {
+        int item = offset + i;
+        if (item >= MAX_ITEMS)
+            break;
+
+        int y = 12 + i * 10;
+
+        if (item == selectedIndex) {
+            u8g2.drawBox(0, y - 9, 128, 10);
+            u8g2.setDrawColor(0);
+        }
+
+        u8g2.drawStr(2, y, labels[item]);
+
+        if (values[item])
+            u8g2.drawStr(80, y, "OK");
+        else
+            u8g2.drawStr(80, y, "FAIL");
+
+        if (item == selectedIndex)
+            u8g2.setDrawColor(1);
+    }
+
+    u8g2.sendBuffer();
+}
+
+// ===== MAIN =====
+void device_check_run() {
+    DeviceStatus status;
+
+    Serial.println("Running device diagnostics...");
+
+    // NRF
+    status.nrf1 = checkNRF(radio1);
+    status.nrf2 = checkNRF(radio2);
+
+    // CC1101
+    status.cc1101 = checkCC1101(CC1101_CS);
+    // status.cc1101 = true;
+
+    drawStatus(status);
+
+    Serial.println("Diagnostics complete");
+
+    while (1) {
+        drawStatus(status);
+
+        if (btnUp()) {
+            selectedIndex--;
+            if (selectedIndex < 0)
+                selectedIndex = MAX_ITEMS - 1;
+            delay(150);
+        }
+
+        if (btnDown()) {
+            selectedIndex++;
+            if (selectedIndex >= MAX_ITEMS)
+                selectedIndex = 0;
+            delay(150);
+        }
+
+        if (btnBack()) {
+            delay(150);
+            break;
+        }
+    }
+}