Compare commits
18 Commits
27597bceab
...
debug
| Author | SHA1 | Date | |
|---|---|---|---|
| 23834d660c | |||
| a0821fed80 | |||
| 014346c442 | |||
| cf545cb849 | |||
| 9af28acb51 | |||
| af42a1c4de | |||
| ffd807ff45 | |||
| c57e0d8ae2 | |||
| a294ce1fd9 | |||
| 179fb77b95 | |||
| cf36a2c07f | |||
| 70a6719553 | |||
| cba7d907e9 | |||
| ee0d85fa73 | |||
| 0b4f285ddb | |||
| 415310b1f9 | |||
| b1afa2f06c | |||
| 423b9f91be |
@@ -58,7 +58,7 @@ The sketch controls `GAUGE_COUNT` stepper-motor gauges using a trapezoidal veloc
|
|||||||
|
|
||||||
### Key data structures
|
### Key data structures
|
||||||
|
|
||||||
- `GaugePins` — hardware pin mapping per gauge (dir, step, enable, active-high/low polarity flags, `ledOrder` string). Declared `constexpr` so `TOTAL_LEDS` can be computed from it at compile time. Configured in the `gaugePins[]` array at the top.
|
- `GaugePins` — hardware pin mapping per gauge (dir, step, enable, active-high/low polarity flags, `ledCount`). Declared `constexpr` so `TOTAL_LEDS` can be computed from it at compile time. Configured in the `gaugePins[]` array at the top.
|
||||||
- `Gauge` — per-gauge runtime state: position, target, velocity, accel, homing state machine, sweep mode.
|
- `Gauge` — per-gauge runtime state: position, target, velocity, accel, homing state machine, sweep mode.
|
||||||
|
|
||||||
### Motion control (`updateGauge`)
|
### Motion control (`updateGauge`)
|
||||||
@@ -76,7 +76,7 @@ When `sweepEnabled`, `updateSweepTarget` bounces `targetPos` between `minPos` an
|
|||||||
|
|
||||||
### LED strip
|
### LED strip
|
||||||
|
|
||||||
One shared WS2812B strip is driven from `LED_DATA_PIN` (currently 22). Each gauge owns a contiguous segment of the strip; `gaugePins[i].ledOrder` is a per-LED type string (one char per LED, `'G'` = GRB-ordered, `'R'` = RGB-ordered) and its length defines the segment length (empty string = no LEDs). `TOTAL_LEDS` is computed at compile time via `constexpr sumLedCounts()` — no manual constant to keep in sync. Per-gauge offsets and counts are cached in `setup()` into `gaugeLedOffset[]` and `gaugeLedCount[]`. The strip is initialised as `GRB`; writes to RGB-ordered LEDs are R/G-swapped via the `writeLed`/`readLed` helpers so callers always work in logical RGB. LED commands and effects mark the strip dirty, and `FastLED.show()` is called once per main-loop iteration if anything changed.
|
One shared WS2812B strip is driven from `LED_DATA_PIN` (currently 22). Each gauge owns a contiguous segment of the strip; `gaugePins[i].ledCount` sets the segment length (0 = no LEDs). `TOTAL_LEDS` is computed at compile time via `constexpr sumLedCounts()` — no manual constant to keep in sync. Per-gauge offsets into the flat `leds[]` array are computed once in `setup()` into `gaugeLedOffset[]`. LED commands and effects mark the strip dirty, and `FastLED.show()` is called once per main-loop iteration if anything changed.
|
||||||
|
|
||||||
### Serial command protocol
|
### Serial command protocol
|
||||||
|
|
||||||
@@ -104,6 +104,6 @@ All commands reply `OK` or `ERR BAD_ID` / `ERR BAD_CMD` etc.
|
|||||||
### Adding gauges
|
### Adding gauges
|
||||||
|
|
||||||
1. Increment `GAUGE_COUNT`.
|
1. Increment `GAUGE_COUNT`.
|
||||||
2. Add a `constexpr GaugePins` entry to `gaugePins[]` (including the `ledOrder` string — one char per LED, `'G'` for GRB or `'R'` for RGB).
|
2. Add a `constexpr GaugePins` entry to `gaugePins[]` (including `ledCount`).
|
||||||
3. Tune `maxPos` and `homingBackoffSteps` in the corresponding `Gauge` default or at runtime.
|
3. Tune `maxPos` and `homingBackoffSteps` in the corresponding `Gauge` default or at runtime.
|
||||||
4. `TOTAL_LEDS`, `gaugeLedOffset[]`, and `gaugeLedCount[]` update automatically — no manual changes needed.
|
4. `TOTAL_LEDS` and `gaugeLedOffset[]` update automatically — no manual changes needed.
|
||||||
|
|||||||
@@ -3,7 +3,7 @@
|
|||||||
#include <math.h>
|
#include <math.h>
|
||||||
#include <FastLED.h>
|
#include <FastLED.h>
|
||||||
|
|
||||||
static const uint8_t GAUGE_COUNT = 4;
|
static const uint8_t GAUGE_COUNT = 3;
|
||||||
|
|
||||||
// One shared WS2812B strip, split into per-gauge segments.
|
// One shared WS2812B strip, split into per-gauge segments.
|
||||||
static const uint8_t LED_DATA_PIN = 22;
|
static const uint8_t LED_DATA_PIN = 22;
|
||||||
@@ -226,23 +226,18 @@ struct GaugePins {
|
|||||||
bool dirInverted;
|
bool dirInverted;
|
||||||
bool stepActiveHigh;
|
bool stepActiveHigh;
|
||||||
bool enableActiveLow;
|
bool enableActiveLow;
|
||||||
const char* ledOrder; // one char per LED: 'G' = GRB, 'R' = RGB; length defines ledCount
|
uint8_t ledCount; // LEDs assigned to this gauge
|
||||||
};
|
};
|
||||||
|
|
||||||
constexpr GaugePins gaugePins[GAUGE_COUNT] = {
|
constexpr GaugePins gaugePins[GAUGE_COUNT] = {
|
||||||
// dir, step, en, dirInv, stepHigh, enActiveLow, ledOrder
|
// dir, step, en, dirInv, stepHigh, enActiveLow, leds
|
||||||
{50, 51, -1, false, true, true, "RRRGGRR"}, // Gauge 0
|
{50, 51, -1, false, true, true, 7}, // Gauge 0
|
||||||
{8, 9, -1, true, true, true, "GGGRRRR"}, // Gauge 1
|
{8, 9, -1, true, true, true, 7}, // Gauge 1
|
||||||
{52, 53, -1, false, true, true, "GGGRRRR"}, // Gauge 2
|
{52, 53, -1, false, true, true, 7}, // Gauge 2
|
||||||
{48, 49, -1, false, true, true, "GGGRRRR"}, // Gauge 3
|
|
||||||
};
|
};
|
||||||
|
|
||||||
constexpr uint8_t cstrLen(const char* s) {
|
|
||||||
return *s ? uint8_t(1 + cstrLen(s + 1)) : uint8_t(0);
|
|
||||||
}
|
|
||||||
|
|
||||||
constexpr uint8_t sumLedCounts(uint8_t i = 0) {
|
constexpr uint8_t sumLedCounts(uint8_t i = 0) {
|
||||||
return i >= GAUGE_COUNT ? 0 : cstrLen(gaugePins[i].ledOrder) + sumLedCounts(i + 1);
|
return i >= GAUGE_COUNT ? 0 : gaugePins[i].ledCount + sumLedCounts(i + 1);
|
||||||
}
|
}
|
||||||
static const uint8_t TOTAL_LEDS = sumLedCounts();
|
static const uint8_t TOTAL_LEDS = sumLedCounts();
|
||||||
|
|
||||||
@@ -302,40 +297,9 @@ String rxLine;
|
|||||||
|
|
||||||
CRGB leds[TOTAL_LEDS];
|
CRGB leds[TOTAL_LEDS];
|
||||||
uint8_t gaugeLedOffset[GAUGE_COUNT];
|
uint8_t gaugeLedOffset[GAUGE_COUNT];
|
||||||
uint8_t gaugeLedCount[GAUGE_COUNT];
|
|
||||||
BlinkState blinkState[TOTAL_LEDS];
|
BlinkState blinkState[TOTAL_LEDS];
|
||||||
bool ledsDirty = false;
|
bool ledsDirty = false;
|
||||||
|
|
||||||
// FastLED drives the shared strip as RGB. Each gauge's ledOrder string marks per-LED
|
|
||||||
// type ('R' = RGB, 'G' = GRB); writes to GRB-ordered LEDs pre-swap R and G to compensate.
|
|
||||||
inline bool ledNeedsRgSwap(uint8_t globalIdx) {
|
|
||||||
for (uint8_t i = 0; i < GAUGE_COUNT; i++) {
|
|
||||||
uint8_t off = gaugeLedOffset[i];
|
|
||||||
if (globalIdx >= off && globalIdx < off + gaugeLedCount[i]) {
|
|
||||||
char c = gaugePins[i].ledOrder[globalIdx - off];
|
|
||||||
return c == 'G' || c == 'g';
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
inline CRGB encodeForStrip(uint8_t globalIdx, CRGB color) {
|
|
||||||
if (ledNeedsRgSwap(globalIdx)) {
|
|
||||||
uint8_t tmp = color.r;
|
|
||||||
color.r = color.g;
|
|
||||||
color.g = tmp;
|
|
||||||
}
|
|
||||||
return color;
|
|
||||||
}
|
|
||||||
|
|
||||||
inline void writeLed(uint8_t globalIdx, CRGB color) {
|
|
||||||
leds[globalIdx] = encodeForStrip(globalIdx, color);
|
|
||||||
}
|
|
||||||
|
|
||||||
inline CRGB readLed(uint8_t globalIdx) {
|
|
||||||
return encodeForStrip(globalIdx, leds[globalIdx]);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Sends one-line command replies back over the control port.
|
// Sends one-line command replies back over the control port.
|
||||||
//
|
//
|
||||||
// Serial protocol summary.
|
// Serial protocol summary.
|
||||||
@@ -830,24 +794,6 @@ bool parsePosQuery(const String& line) {
|
|||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Answers `CFG?` with speed and acceleration for every gauge.
|
|
||||||
// Emits one `CFG <id> <maxSpeed> <accel>` line per gauge, then replies `OK`.
|
|
||||||
bool parseCfgQuery(const String& line) {
|
|
||||||
if (line == "CFG?") {
|
|
||||||
for (uint8_t i = 0; i < GAUGE_COUNT; i++) {
|
|
||||||
CMD_PORT.print("CFG ");
|
|
||||||
CMD_PORT.print(i);
|
|
||||||
CMD_PORT.print(' ');
|
|
||||||
CMD_PORT.print((int)gauges[i].maxSpeed);
|
|
||||||
CMD_PORT.print(' ');
|
|
||||||
CMD_PORT.println((int)gauges[i].accel);
|
|
||||||
}
|
|
||||||
sendReply("OK");
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Answers the mandatory life question: are you there?
|
// Answers the mandatory life question: are you there?
|
||||||
// Reply: `PONG`.
|
// Reply: `PONG`.
|
||||||
bool parsePing(const String& line) {
|
bool parsePing(const String& line) {
|
||||||
@@ -889,8 +835,8 @@ bool parseVfd(const String& line) {
|
|||||||
bool parseLedQuery(const String& line) {
|
bool parseLedQuery(const String& line) {
|
||||||
if (line == "LED?") {
|
if (line == "LED?") {
|
||||||
for (uint8_t i = 0; i < GAUGE_COUNT; i++) {
|
for (uint8_t i = 0; i < GAUGE_COUNT; i++) {
|
||||||
for (uint8_t j = 0; j < gaugeLedCount[i]; j++) {
|
for (uint8_t j = 0; j < gaugePins[i].ledCount; j++) {
|
||||||
CRGB c = readLed(gaugeLedOffset[i] + j);
|
const CRGB& c = leds[gaugeLedOffset[i] + j];
|
||||||
CMD_PORT.print("LED ");
|
CMD_PORT.print("LED ");
|
||||||
CMD_PORT.print(i);
|
CMD_PORT.print(i);
|
||||||
CMD_PORT.print(' ');
|
CMD_PORT.print(' ');
|
||||||
@@ -919,13 +865,13 @@ bool parseLed(const String& line) {
|
|||||||
char* dash = strchr(idxToken, '-');
|
char* dash = strchr(idxToken, '-');
|
||||||
int idxFirst = atoi(idxToken);
|
int idxFirst = atoi(idxToken);
|
||||||
int idxLast = dash ? atoi(dash + 1) : idxFirst;
|
int idxLast = dash ? atoi(dash + 1) : idxFirst;
|
||||||
if (idxFirst < 0 || idxLast >= gaugeLedCount[id] || idxFirst > idxLast) {
|
if (idxFirst < 0 || idxLast >= gaugePins[id].ledCount || idxFirst > idxLast) {
|
||||||
sendReply("ERR BAD_IDX"); return true;
|
sendReply("ERR BAD_IDX"); return true;
|
||||||
}
|
}
|
||||||
CRGB color(constrain(r, 0, 255), constrain(g, 0, 255), constrain(b, 0, 255));
|
CRGB color(constrain(r, 0, 255), constrain(g, 0, 255), constrain(b, 0, 255));
|
||||||
for (int i = idxFirst; i <= idxLast; i++) {
|
for (int i = idxFirst; i <= idxLast; i++) {
|
||||||
blinkState[gaugeLedOffset[id] + i].active = false;
|
blinkState[gaugeLedOffset[id] + i].active = false;
|
||||||
writeLed(gaugeLedOffset[id] + i, color);
|
leds[gaugeLedOffset[id] + i] = color;
|
||||||
}
|
}
|
||||||
ledsDirty = true;
|
ledsDirty = true;
|
||||||
sendReply("OK");
|
sendReply("OK");
|
||||||
@@ -948,7 +894,7 @@ bool parseBlink(const String& line) {
|
|||||||
char* dash = strchr(idxToken, '-');
|
char* dash = strchr(idxToken, '-');
|
||||||
int idxFirst = atoi(idxToken);
|
int idxFirst = atoi(idxToken);
|
||||||
int idxLast = dash ? atoi(dash + 1) : idxFirst;
|
int idxLast = dash ? atoi(dash + 1) : idxFirst;
|
||||||
if (idxFirst < 0 || idxLast >= gaugeLedCount[id] || idxFirst > idxLast) {
|
if (idxFirst < 0 || idxLast >= gaugePins[id].ledCount || idxFirst > idxLast) {
|
||||||
sendReply("ERR BAD_IDX"); return true;
|
sendReply("ERR BAD_IDX"); return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -969,13 +915,13 @@ bool parseBlink(const String& line) {
|
|||||||
uint8_t globalIdx = gaugeLedOffset[id] + i;
|
uint8_t globalIdx = gaugeLedOffset[id] + i;
|
||||||
BlinkState& bs = blinkState[globalIdx];
|
BlinkState& bs = blinkState[globalIdx];
|
||||||
bs.fx = FX_BLINK;
|
bs.fx = FX_BLINK;
|
||||||
bs.onColor = (count == 7) ? color : readLed(globalIdx);
|
bs.onColor = (count == 7) ? color : leds[globalIdx];
|
||||||
bs.onMs = (uint16_t)onMs;
|
bs.onMs = (uint16_t)onMs;
|
||||||
bs.offMs = (uint16_t)offMs;
|
bs.offMs = (uint16_t)offMs;
|
||||||
bs.currentlyOn = true;
|
bs.currentlyOn = true;
|
||||||
bs.lastMs = nowMs;
|
bs.lastMs = nowMs;
|
||||||
bs.active = true;
|
bs.active = true;
|
||||||
writeLed(globalIdx, bs.onColor);
|
leds[globalIdx] = bs.onColor;
|
||||||
}
|
}
|
||||||
ledsDirty = true;
|
ledsDirty = true;
|
||||||
sendReply("OK");
|
sendReply("OK");
|
||||||
@@ -993,7 +939,7 @@ bool parseBreathe(const String& line) {
|
|||||||
char* dash = strchr(idxToken, '-');
|
char* dash = strchr(idxToken, '-');
|
||||||
int idxFirst = atoi(idxToken);
|
int idxFirst = atoi(idxToken);
|
||||||
int idxLast = dash ? atoi(dash + 1) : idxFirst;
|
int idxLast = dash ? atoi(dash + 1) : idxFirst;
|
||||||
if (idxFirst < 0 || idxLast >= gaugeLedCount[id] || idxFirst > idxLast) {
|
if (idxFirst < 0 || idxLast >= gaugePins[id].ledCount || idxFirst > idxLast) {
|
||||||
sendReply("ERR BAD_IDX"); return true;
|
sendReply("ERR BAD_IDX"); return true;
|
||||||
}
|
}
|
||||||
if (periodMs <= 0) { sendReply("ERR BAD_TIME"); return true; }
|
if (periodMs <= 0) { sendReply("ERR BAD_TIME"); return true; }
|
||||||
@@ -1008,7 +954,7 @@ bool parseBreathe(const String& line) {
|
|||||||
bs.cyclePos = 0;
|
bs.cyclePos = 0;
|
||||||
bs.lastMs = nowMs;
|
bs.lastMs = nowMs;
|
||||||
bs.active = true;
|
bs.active = true;
|
||||||
writeLed(gi, CRGB::Black);
|
leds[gi] = CRGB::Black;
|
||||||
}
|
}
|
||||||
ledsDirty = true;
|
ledsDirty = true;
|
||||||
sendReply("OK");
|
sendReply("OK");
|
||||||
@@ -1026,7 +972,7 @@ bool parseDflash(const String& line) {
|
|||||||
char* dash = strchr(idxToken, '-');
|
char* dash = strchr(idxToken, '-');
|
||||||
int idxFirst = atoi(idxToken);
|
int idxFirst = atoi(idxToken);
|
||||||
int idxLast = dash ? atoi(dash + 1) : idxFirst;
|
int idxLast = dash ? atoi(dash + 1) : idxFirst;
|
||||||
if (idxFirst < 0 || idxLast >= gaugeLedCount[id] || idxFirst > idxLast) {
|
if (idxFirst < 0 || idxLast >= gaugePins[id].ledCount || idxFirst > idxLast) {
|
||||||
sendReply("ERR BAD_IDX"); return true;
|
sendReply("ERR BAD_IDX"); return true;
|
||||||
}
|
}
|
||||||
CRGB color(constrain(r, 0, 255), constrain(g, 0, 255), constrain(b, 0, 255));
|
CRGB color(constrain(r, 0, 255), constrain(g, 0, 255), constrain(b, 0, 255));
|
||||||
@@ -1039,7 +985,7 @@ bool parseDflash(const String& line) {
|
|||||||
bs.dphase = 0;
|
bs.dphase = 0;
|
||||||
bs.lastMs = nowMs;
|
bs.lastMs = nowMs;
|
||||||
bs.active = true;
|
bs.active = true;
|
||||||
writeLed(gi, color); // phase 0 = on
|
leds[gi] = color; // phase 0 = on
|
||||||
}
|
}
|
||||||
ledsDirty = true;
|
ledsDirty = true;
|
||||||
sendReply("OK");
|
sendReply("OK");
|
||||||
@@ -1052,7 +998,7 @@ void updateBlink() {
|
|||||||
bool changed = false;
|
bool changed = false;
|
||||||
|
|
||||||
for (uint8_t i = 0; i < GAUGE_COUNT; i++) {
|
for (uint8_t i = 0; i < GAUGE_COUNT; i++) {
|
||||||
for (uint8_t j = 0; j < gaugeLedCount[i]; j++) {
|
for (uint8_t j = 0; j < gaugePins[i].ledCount; j++) {
|
||||||
uint8_t gi = gaugeLedOffset[i] + j;
|
uint8_t gi = gaugeLedOffset[i] + j;
|
||||||
BlinkState& bs = blinkState[gi];
|
BlinkState& bs = blinkState[gi];
|
||||||
if (!bs.active) continue;
|
if (!bs.active) continue;
|
||||||
@@ -1063,7 +1009,7 @@ void updateBlink() {
|
|||||||
if ((nowMs - bs.lastMs) >= period) {
|
if ((nowMs - bs.lastMs) >= period) {
|
||||||
bs.currentlyOn = !bs.currentlyOn;
|
bs.currentlyOn = !bs.currentlyOn;
|
||||||
bs.lastMs = nowMs;
|
bs.lastMs = nowMs;
|
||||||
writeLed(gi, bs.currentlyOn ? bs.onColor : CRGB::Black);
|
leds[gi] = bs.currentlyOn ? bs.onColor : CRGB::Black;
|
||||||
changed = true;
|
changed = true;
|
||||||
}
|
}
|
||||||
break;
|
break;
|
||||||
@@ -1079,9 +1025,8 @@ void updateBlink() {
|
|||||||
uint8_t bri = (bs.cyclePos < half)
|
uint8_t bri = (bs.cyclePos < half)
|
||||||
? (uint8_t)((uint32_t)bs.cyclePos * 255 / half)
|
? (uint8_t)((uint32_t)bs.cyclePos * 255 / half)
|
||||||
: (uint8_t)((uint32_t)(bs.periodMs - bs.cyclePos) * 255 / half);
|
: (uint8_t)((uint32_t)(bs.periodMs - bs.cyclePos) * 255 / half);
|
||||||
CRGB scaled = bs.onColor;
|
leds[gi] = bs.onColor;
|
||||||
scaled.nscale8(bri ? bri : 1);
|
leds[gi].nscale8(bri ? bri : 1);
|
||||||
writeLed(gi, scaled);
|
|
||||||
changed = true;
|
changed = true;
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
@@ -1090,7 +1035,7 @@ void updateBlink() {
|
|||||||
if ((nowMs - bs.lastMs) >= dur[bs.dphase]) {
|
if ((nowMs - bs.lastMs) >= dur[bs.dphase]) {
|
||||||
bs.lastMs = nowMs;
|
bs.lastMs = nowMs;
|
||||||
bs.dphase = (bs.dphase + 1) & 3;
|
bs.dphase = (bs.dphase + 1) & 3;
|
||||||
writeLed(gi, (bs.dphase == 0 || bs.dphase == 2) ? bs.onColor : CRGB::Black);
|
leds[gi] = (bs.dphase == 0 || bs.dphase == 2) ? bs.onColor : CRGB::Black;
|
||||||
changed = true;
|
changed = true;
|
||||||
}
|
}
|
||||||
break;
|
break;
|
||||||
@@ -1113,7 +1058,6 @@ void processLine(const String& line) {
|
|||||||
if (parseHome(line)) return;
|
if (parseHome(line)) return;
|
||||||
if (parseSweep(line)) return;
|
if (parseSweep(line)) return;
|
||||||
if (parsePosQuery(line)) return;
|
if (parsePosQuery(line)) return;
|
||||||
if (parseCfgQuery(line)) return;
|
|
||||||
if (parseLedQuery(line)) return;
|
if (parseLedQuery(line)) return;
|
||||||
if (parseLed(line)) return;
|
if (parseLed(line)) return;
|
||||||
if (parseBlink(line)) return;
|
if (parseBlink(line)) return;
|
||||||
@@ -1171,11 +1115,10 @@ void setup() {
|
|||||||
// Flatten the per-gauge LED counts into offsets on the shared strip.
|
// Flatten the per-gauge LED counts into offsets on the shared strip.
|
||||||
uint8_t ledOff = 0;
|
uint8_t ledOff = 0;
|
||||||
for (uint8_t i = 0; i < GAUGE_COUNT; i++) {
|
for (uint8_t i = 0; i < GAUGE_COUNT; i++) {
|
||||||
gaugeLedCount[i] = cstrLen(gaugePins[i].ledOrder);
|
|
||||||
gaugeLedOffset[i] = ledOff;
|
gaugeLedOffset[i] = ledOff;
|
||||||
ledOff += gaugeLedCount[i];
|
ledOff += gaugePins[i].ledCount;
|
||||||
}
|
}
|
||||||
FastLED.addLeds<WS2812B, LED_DATA_PIN, RGB>(leds, TOTAL_LEDS);
|
FastLED.addLeds<WS2812B, LED_DATA_PIN, GRB>(leds, TOTAL_LEDS);
|
||||||
FastLED.setBrightness(255);
|
FastLED.setBrightness(255);
|
||||||
FastLED.show();
|
FastLED.show();
|
||||||
|
|
||||||
|
|||||||
62
boot.py
Normal file
62
boot.py
Normal file
@@ -0,0 +1,62 @@
|
|||||||
|
"""
|
||||||
|
boot.py — runs before main.py on every ESP32 boot
|
||||||
|
|
||||||
|
Connects WiFi, runs OTA update, then hands off to main.py.
|
||||||
|
Keep this file as simple as possible — it is never OTA-updated itself
|
||||||
|
(it lives outside the repo folder) so bugs here require USB to fix.
|
||||||
|
"""
|
||||||
|
#import gauge
|
||||||
|
import network
|
||||||
|
import gc
|
||||||
|
import utime
|
||||||
|
import sys
|
||||||
|
|
||||||
|
import ota
|
||||||
|
|
||||||
|
ota.load_config()
|
||||||
|
WIFI_SSID, WIFI_PASSWORD = ota.WIFI_SSID, ota.WIFI_PASSWORD
|
||||||
|
|
||||||
|
def _connect_wifi(timeout_s=20):
|
||||||
|
sta = network.WLAN(network.STA_IF)
|
||||||
|
sta.active(True)
|
||||||
|
sta.config(txpower=15)
|
||||||
|
if sta.isconnected():
|
||||||
|
return True
|
||||||
|
sta.connect(WIFI_SSID, WIFI_PASSWORD)
|
||||||
|
deadline = utime.time() + timeout_s
|
||||||
|
while not sta.isconnected():
|
||||||
|
if utime.time() > deadline:
|
||||||
|
return False
|
||||||
|
utime.sleep_ms(300)
|
||||||
|
return True
|
||||||
|
|
||||||
|
if WIFI_SSID is None:
|
||||||
|
print("[boot] No WiFi credentials — cannot connect, skipping OTA")
|
||||||
|
elif _connect_wifi():
|
||||||
|
ip = network.WLAN(network.STA_IF).ifconfig()[0]
|
||||||
|
print(f"[boot] WiFi connected — {ip}")
|
||||||
|
|
||||||
|
try:
|
||||||
|
ota.update()
|
||||||
|
except Exception as e:
|
||||||
|
print(f"[boot] OTA error: {e} — continuing with existing files")
|
||||||
|
sys.print_exception(e)
|
||||||
|
utime.sleep_ms(5000)
|
||||||
|
ota._fetch_commit_sha = None
|
||||||
|
ota._fetch_manifest = None
|
||||||
|
ota._fetch_dir = None
|
||||||
|
ota._api_get = None
|
||||||
|
ota._download = None
|
||||||
|
ota.urequests = None
|
||||||
|
del ota.urequests
|
||||||
|
del ota
|
||||||
|
gc.collect()
|
||||||
|
del sys.modules["ota"]
|
||||||
|
gc.collect()
|
||||||
|
|
||||||
|
else:
|
||||||
|
print("[boot] WiFi failed — skipping OTA, booting with existing files")
|
||||||
|
|
||||||
|
# main.py runs automatically after boot.py
|
||||||
|
|
||||||
|
|
||||||
File diff suppressed because it is too large
Load Diff
118
gauge.py
118
gauge.py
@@ -28,16 +28,6 @@ import gc
|
|||||||
from umqtt.robust import MQTTClient
|
from umqtt.robust import MQTTClient
|
||||||
from machine import UART
|
from machine import UART
|
||||||
|
|
||||||
# Activate WiFi driver before any heavy heap allocation so it can claim its
|
|
||||||
# contiguous DRAM block before the Python heap fragments the address space.
|
|
||||||
# Only activate if not already running (e.g. boot.py may have started it).
|
|
||||||
gc.collect()
|
|
||||||
_early_wlan = network.WLAN(network.STA_IF)
|
|
||||||
if not _early_wlan.active():
|
|
||||||
_early_wlan.active(True)
|
|
||||||
del _early_wlan
|
|
||||||
gc.collect()
|
|
||||||
|
|
||||||
# ---------------------------------------------------------------------------
|
# ---------------------------------------------------------------------------
|
||||||
# Logging
|
# Logging
|
||||||
# ---------------------------------------------------------------------------
|
# ---------------------------------------------------------------------------
|
||||||
@@ -161,19 +151,27 @@ ARDUINO_TX_PIN = int(_cfg.get("arduino_tx_pin", 17))
|
|||||||
ARDUINO_RX_PIN = int(_cfg.get("arduino_rx_pin", 16))
|
ARDUINO_RX_PIN = int(_cfg.get("arduino_rx_pin", 16))
|
||||||
ARDUINO_BAUD = int(_cfg.get("arduino_baud", 115200))
|
ARDUINO_BAUD = int(_cfg.get("arduino_baud", 115200))
|
||||||
|
|
||||||
_arduino = UART(ARDUINO_UART_ID, baudrate=ARDUINO_BAUD, tx=ARDUINO_TX_PIN, rx=ARDUINO_RX_PIN, timeout=10)
|
_arduino = None
|
||||||
|
|
||||||
|
|
||||||
|
def _ensure_arduino():
|
||||||
|
global _arduino
|
||||||
|
if _arduino is None:
|
||||||
|
_arduino = UART(ARDUINO_UART_ID, baudrate=ARDUINO_BAUD, tx=ARDUINO_TX_PIN, rx=ARDUINO_RX_PIN, timeout=10)
|
||||||
|
return _arduino
|
||||||
|
|
||||||
|
|
||||||
def arduino_send(cmd):
|
def arduino_send(cmd):
|
||||||
"""Send a newline-terminated command to the Arduino."""
|
"""Send a newline-terminated command to the Arduino."""
|
||||||
_arduino.write((cmd + "\n").encode())
|
_ensure_arduino().write((cmd + "\n").encode())
|
||||||
info(f"Arduino → {cmd}")
|
info(f"Arduino → {cmd}")
|
||||||
|
|
||||||
|
|
||||||
def arduino_recv():
|
def arduino_recv():
|
||||||
"""Print any lines waiting in the Arduino RX buffer."""
|
"""Print any lines waiting in the Arduino RX buffer."""
|
||||||
while _arduino.any():
|
uart = _ensure_arduino()
|
||||||
line = _arduino.readline()
|
while uart.any():
|
||||||
|
line = uart.readline()
|
||||||
if line:
|
if line:
|
||||||
print(f"[{_ts()}] ARDU {line.decode().strip()}")
|
print(f"[{_ts()}] ARDU {line.decode().strip()}")
|
||||||
|
|
||||||
@@ -540,11 +538,14 @@ _WIFI_CONNECT_ATTEMPTS = 3
|
|||||||
def _reset_wifi_interface():
|
def _reset_wifi_interface():
|
||||||
global _wifi_sta
|
global _wifi_sta
|
||||||
_wifi_sta = network.WLAN(network.STA_IF)
|
_wifi_sta = network.WLAN(network.STA_IF)
|
||||||
if _wifi_sta.active():
|
if not _wifi_sta.active():
|
||||||
_wifi_sta.active(False)
|
|
||||||
utime.sleep_ms(200)
|
|
||||||
_wifi_sta.active(True)
|
_wifi_sta.active(True)
|
||||||
utime.sleep_ms(500)
|
utime.sleep_ms(500)
|
||||||
|
try:
|
||||||
|
_wifi_sta.disconnect()
|
||||||
|
except Exception:
|
||||||
|
pass
|
||||||
|
utime.sleep_ms(1000)
|
||||||
|
|
||||||
|
|
||||||
def connect_wifi(ssid, password, timeout_s=15, force_reconnect=False):
|
def connect_wifi(ssid, password, timeout_s=15, force_reconnect=False):
|
||||||
@@ -561,6 +562,7 @@ def connect_wifi(ssid, password, timeout_s=15, force_reconnect=False):
|
|||||||
last_error = None
|
last_error = None
|
||||||
for attempt in range(_WIFI_CONNECT_ATTEMPTS):
|
for attempt in range(_WIFI_CONNECT_ATTEMPTS):
|
||||||
info(f"WiFi connecting to '{ssid}' (attempt {attempt + 1}/{_WIFI_CONNECT_ATTEMPTS}) ...")
|
info(f"WiFi connecting to '{ssid}' (attempt {attempt + 1}/{_WIFI_CONNECT_ATTEMPTS}) ...")
|
||||||
|
if not _wifi_sta.isconnected():
|
||||||
_reset_wifi_interface()
|
_reset_wifi_interface()
|
||||||
try:
|
try:
|
||||||
_wifi_sta.connect(ssid, password)
|
_wifi_sta.connect(ssid, password)
|
||||||
@@ -576,7 +578,7 @@ def connect_wifi(ssid, password, timeout_s=15, force_reconnect=False):
|
|||||||
info(f" SSID : {ssid}")
|
info(f" SSID : {ssid}")
|
||||||
info(f" MAC : {mac}")
|
info(f" MAC : {mac}")
|
||||||
info(f" IP : {ip} mask:{mask} gw:{gw} dns:{dns}")
|
info(f" IP : {ip} mask:{mask} gw:{gw} dns:{dns}")
|
||||||
utime.sleep_ms(500)
|
utime.sleep_ms(2000)
|
||||||
return ip
|
return ip
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
last_error = e
|
last_error = e
|
||||||
@@ -601,7 +603,7 @@ def check_wifi():
|
|||||||
|
|
||||||
log_err("WiFi lost connection — attempting reconnect...")
|
log_err("WiFi lost connection — attempting reconnect...")
|
||||||
try:
|
try:
|
||||||
ip = connect_wifi(WIFI_SSID, WIFI_PASSWORD, timeout_s=15, force_reconnect=True)
|
ip = connect_wifi(WIFI_SSID, WIFI_PASSWORD, timeout_s=15)
|
||||||
info(f"WiFi reconnected! IP:{ip}")
|
info(f"WiFi reconnected! IP:{ip}")
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
log_err(f"WiFi reconnect failed: {e}")
|
log_err(f"WiFi reconnect failed: {e}")
|
||||||
@@ -911,10 +913,6 @@ def connect_mqtt():
|
|||||||
except Exception as e:
|
except Exception as e:
|
||||||
last_error = e
|
last_error = e
|
||||||
log_err(f"MQTT connect attempt {attempt + 1} failed: {type(e).__name__}: {e}")
|
log_err(f"MQTT connect attempt {attempt + 1} failed: {type(e).__name__}: {e}")
|
||||||
try:
|
|
||||||
client.sock.close()
|
|
||||||
except Exception:
|
|
||||||
pass
|
|
||||||
gc.collect()
|
gc.collect()
|
||||||
utime.sleep_ms(1000)
|
utime.sleep_ms(1000)
|
||||||
|
|
||||||
@@ -922,27 +920,6 @@ def connect_mqtt():
|
|||||||
raise last_error
|
raise last_error
|
||||||
|
|
||||||
|
|
||||||
_mqtt_check_interval_ms = 30000
|
|
||||||
_last_mqtt_check = 0
|
|
||||||
_discovery_queue = []
|
|
||||||
_discovery_idx = 0
|
|
||||||
_last_discovery_ms = 0
|
|
||||||
_DISCOVERY_INTERVAL_MS = 350
|
|
||||||
|
|
||||||
|
|
||||||
def _compact_discovery_payload(payload):
|
|
||||||
"""Trim optional HA discovery fields when RAM is tight."""
|
|
||||||
compact = dict(payload)
|
|
||||||
|
|
||||||
# Light entities are the largest payloads because they repeat effect metadata.
|
|
||||||
# Keep core functionality, but omit optional effect declarations to reduce heap use.
|
|
||||||
if compact.get("schema") == "json":
|
|
||||||
compact.pop("effect", None)
|
|
||||||
compact.pop("effect_list", None)
|
|
||||||
|
|
||||||
return compact
|
|
||||||
|
|
||||||
|
|
||||||
def check_mqtt():
|
def check_mqtt():
|
||||||
global client_ref, _mqtt_connected, _last_mqtt_check
|
global client_ref, _mqtt_connected, _last_mqtt_check
|
||||||
now = utime.ticks_ms()
|
now = utime.ticks_ms()
|
||||||
@@ -984,10 +961,6 @@ def check_mqtt():
|
|||||||
return True
|
return True
|
||||||
except Exception as e2:
|
except Exception as e2:
|
||||||
log_err(f"MQTT reconnect attempt {attempt + 1} failed: {e2}")
|
log_err(f"MQTT reconnect attempt {attempt + 1} failed: {e2}")
|
||||||
try:
|
|
||||||
client_ref.sock.close()
|
|
||||||
except Exception:
|
|
||||||
pass
|
|
||||||
gc.collect()
|
gc.collect()
|
||||||
utime.sleep_ms(2000)
|
utime.sleep_ms(2000)
|
||||||
|
|
||||||
@@ -995,9 +968,17 @@ def check_mqtt():
|
|||||||
return False
|
return False
|
||||||
|
|
||||||
|
|
||||||
|
_mqtt_check_interval_ms = 30000
|
||||||
|
_last_mqtt_check = 0
|
||||||
|
_discovery_queue = []
|
||||||
|
_discovery_idx = 0
|
||||||
|
_last_discovery_ms = 0
|
||||||
|
_DISCOVERY_INTERVAL_MS = 350
|
||||||
|
|
||||||
|
|
||||||
def _publish_discovery_entity(client, topic, payload, log_msg):
|
def _publish_discovery_entity(client, topic, payload, log_msg):
|
||||||
gc.collect()
|
gc.collect()
|
||||||
client.publish(topic, ujson.dumps(_compact_discovery_payload(payload)), retain=True)
|
client.publish(topic, ujson.dumps(payload), retain=True)
|
||||||
info(log_msg)
|
info(log_msg)
|
||||||
|
|
||||||
|
|
||||||
@@ -1305,12 +1286,42 @@ def apply_motion_defaults():
|
|||||||
send_vfd_state()
|
send_vfd_state()
|
||||||
|
|
||||||
|
|
||||||
|
def _restore_led_states():
|
||||||
|
for i in range(num_gauges):
|
||||||
|
gt = gauge_topics[i]
|
||||||
|
info(f" red={_red_effect[i]} green={_green_effect[i]} status_r={_status_red_effect[i]} status_g={_status_green_effect[i]}")
|
||||||
|
for led_key, led_idx, color, effect_arr, state_topic in [
|
||||||
|
("red", _LED_RED, gauges[i]["ws2812_red"], _red_effect, gt["led_red_state"]),
|
||||||
|
("green", _LED_GREEN, gauges[i]["ws2812_green"], _green_effect, gt["led_green_state"]),
|
||||||
|
("status_red", _LED_STATUS_RED, gauges[i]["ws2812_red"], _status_red_effect, gt["status_red_state"]),
|
||||||
|
("status_green", _LED_STATUS_GREEN, gauges[i]["ws2812_green"], _status_green_effect, gt["status_green_state"]),
|
||||||
|
]:
|
||||||
|
if effect_arr[i]:
|
||||||
|
pub = {"state": "ON", "effect": effect_arr[i]}
|
||||||
|
_publish(state_topic, ujson.dumps(pub), retain=True)
|
||||||
|
if _red_effect[i]:
|
||||||
|
_apply_blink_or_led(i, _LED_RED, gauges[i]["ws2812_red"], _red_effect[i])
|
||||||
|
if _green_effect[i]:
|
||||||
|
_apply_blink_or_led(i, _LED_GREEN, gauges[i]["ws2812_green"], _green_effect[i])
|
||||||
|
if _status_red_effect[i]:
|
||||||
|
_apply_blink_or_led(i, _LED_STATUS_RED, gauges[i]["ws2812_red"], _status_red_effect[i])
|
||||||
|
if _status_green_effect[i]:
|
||||||
|
_apply_blink_or_led(i, _LED_STATUS_GREEN, gauges[i]["ws2812_green"], _status_green_effect[i])
|
||||||
|
|
||||||
|
|
||||||
# ---------------------------------------------------------------------------
|
# ---------------------------------------------------------------------------
|
||||||
# Main
|
# Main
|
||||||
# ---------------------------------------------------------------------------
|
# ---------------------------------------------------------------------------
|
||||||
|
|
||||||
|
|
||||||
def main():
|
def main():
|
||||||
|
gc.collect()
|
||||||
|
_w = network.WLAN(network.STA_IF)
|
||||||
|
if not _w.active():
|
||||||
|
_w.active(True)
|
||||||
|
del _w
|
||||||
|
gc.collect()
|
||||||
|
_ensure_arduino()
|
||||||
gc.collect()
|
gc.collect()
|
||||||
info("=" * 48)
|
info("=" * 48)
|
||||||
info("Gauge MQTT controller starting")
|
info("Gauge MQTT controller starting")
|
||||||
@@ -1318,7 +1329,7 @@ def main():
|
|||||||
info("=" * 48)
|
info("=" * 48)
|
||||||
|
|
||||||
gc.collect()
|
gc.collect()
|
||||||
connect_wifi(WIFI_SSID, WIFI_PASSWORD, force_reconnect=True)
|
connect_wifi(WIFI_SSID, WIFI_PASSWORD)
|
||||||
|
|
||||||
mqtt_attempts = 0
|
mqtt_attempts = 0
|
||||||
while True:
|
while True:
|
||||||
@@ -1331,13 +1342,14 @@ def main():
|
|||||||
if mqtt_attempts % 3 == 0:
|
if mqtt_attempts % 3 == 0:
|
||||||
log_err("WiFi may be stale — forcing reconnect...")
|
log_err("WiFi may be stale — forcing reconnect...")
|
||||||
try:
|
try:
|
||||||
connect_wifi(WIFI_SSID, WIFI_PASSWORD, force_reconnect=True)
|
connect_wifi(WIFI_SSID, WIFI_PASSWORD)
|
||||||
except Exception as we:
|
except Exception as we:
|
||||||
log_err(f"WiFi reconnect failed: {we}")
|
log_err(f"WiFi reconnect failed: {we}")
|
||||||
utime.sleep_ms(5000)
|
utime.sleep_ms(5000)
|
||||||
_subscribe_all(client_ref)
|
_subscribe_all(client_ref)
|
||||||
schedule_discovery()
|
schedule_discovery()
|
||||||
|
|
||||||
|
publish_backlight_states(client_ref)
|
||||||
apply_motion_defaults()
|
apply_motion_defaults()
|
||||||
info("Draining initial retained messages...")
|
info("Draining initial retained messages...")
|
||||||
for _ in range(50):
|
for _ in range(50):
|
||||||
@@ -1350,6 +1362,10 @@ def main():
|
|||||||
gauge_last_rezero[i] = utime.ticks_ms()
|
gauge_last_rezero[i] = utime.ticks_ms()
|
||||||
info("Home command sent")
|
info("Home command sent")
|
||||||
|
|
||||||
|
utime.sleep_ms(100)
|
||||||
|
_restore_led_states()
|
||||||
|
info("LED effects restored")
|
||||||
|
|
||||||
info("Publishing state...")
|
info("Publishing state...")
|
||||||
publish_online(client_ref)
|
publish_online(client_ref)
|
||||||
publish_state(client_ref)
|
publish_state(client_ref)
|
||||||
|
|||||||
Reference in New Issue
Block a user