Update example

Author: mirkokurt

examples/color-your-leds/python/main.py

@@ -6,19 +6,33 @@
 
 ui = WebUI()
 
-def on_set_color(ledid: int, message: dict):
-    print(f"Received set_color message: {message}")
+def on_set_color(id, message: dict):
+    ledid = message.get("led")
+    rgb_color = message.get("color")
+    # Led 1 and 2 are controlled by directly (MPU), while Led 3 and 4 are controlled via Bridge (MCU)
     try:
-        color = message.get("color")
-        Leds.set_led_color(ledid, color)
+        if ledid not in (1, 2, 3, 4):
+            raise ValueError(f"Unknown led '{ledid}'")
+
+        if not rgb_color or not all(k in rgb_color for k in ("r", "g", "b")):
+            raise ValueError("Color must be an object with 'r', 'g', 'b' keys")
+
+        match ledid:
+            case 1:
+                Leds.set_led1_color(rgb_color["r"] != 0, rgb_color["g"] != 0, rgb_color["b"] != 0)
+            case 2:
+                Leds.set_led2_color(rgb_color["r"] != 0, rgb_color["g"] != 0, rgb_color["b"] != 0)
+            case 3:
+                Bridge.call("set_led3_color", rgb_color["r"], rgb_color["g"], rgb_color["b"])
+            case 4:
+                Bridge.call("set_led4_color", rgb_color["r"] != 0, rgb_color["g"] != 0, rgb_color["b"] != 0)
+
     except Exception as e:
         ui.send_message("error", f"LED color set error: {e}")
 
-#Initialize LEDs to off state
+#Initialize LEDs to off state (only 1 and 2 here, 3 and 4 will be set in MCU setup)
 on_set_color(1, {"led": 1, "color": {"r": 0, "g": 0, "b": 0}})
 on_set_color(2, {"led": 2, "color": {"r": 0, "g": 0, "b": 0}})
-on_set_color(3, {"led": 3, "color": {"r": 0, "g": 0, "b": 0}})
-on_set_color(4, {"led": 4, "color": {"r": 0, "g": 0, "b": 0}})
 
 ui.on_message("set_color", on_set_color)
 

examples/color-your-leds/sketch/sketch.ino

@@ -4,33 +4,29 @@
 
 #include <Arduino_RouterBridge.h>
 
-void setRGB(int ledRPin, int ledGPin, int ledBPin, int r, int g, int b) {
-  analogWrite(ledRPin, r);
-  analogWrite(ledGPin, g);
-  analogWrite(ledBPin, b);
+// Led 3 can be controlled via PWM pins
+void set_led3_color(int r, int g, int b) {
+  analogWrite(LED3_R, r);
+  analogWrite(LED3_G, g);
+  analogWrite(LED3_B, b);
 }
 
-void setColor(int ledIndex, int r, int g, int b) {
-  switch (ledIndex) {
-    case 3:
-      setRGB(LED3_R, LED3_G, LED3_B, r, g, b);
-      break;
-    case 4:
-      setRGB(LED4_R, LED4_G, LED4_B, r, g, b);
-      break;
-    default:
-      break;
-  }
+// Led 4 is a simple ON/OFF LED for each color channel, HIGH = OFF, LOW = ON
+void set_led4_color(bool r, bool g, bool b) {
+  digitalWrite(LED_BUILTIN + 3, r ? LOW : HIGH);
+  digitalWrite(LED_BUILTIN + 4, g ? LOW : HIGH);
+  digitalWrite(LED_BUILTIN + 5, b ? LOW : HIGH);
 }
 
 void setup()
 {
-    setColor(3, 0, 0, 0); // Initialize LED 3 to off
-    setColor(4, 0, 0, 0); // Initialize LED 4 to off
+    set_led3_color(0, 0, 0);
+    set_led4_color(false, false, false);
 
     Bridge.begin();
 
-    Bridge.provide("set_led_color", setColor);
+    Bridge.provide("set_led3_color", set_led3_color);
+    Bridge.provide("set_led4_color", set_led4_color);
 }
 
 void loop() {}