pyleds/program/promise/christmas_lights.py

@@ -0,0 +1,38 @@
+from pyleds.lib.Program import Program
+from rpi_ws281x import Color
+import time
+
+def name():
+    return 'Christmas Lights'
+
+class Christmas(Program):
+    def run(self, args: [] = None):
+        # Configuration
+        wait_ms = 100  # Speed of the animation
+
+        # Define classic festive colors
+        RED = Color(255, 0, 0)
+        GREEN = Color(0, 255, 0)
+        WARM_WHITE = Color(200, 180, 60) # A golden-ish warm white
+
+        colors = [RED, GREEN, WARM_WHITE]
+
+        # This determines how many pixels of the same color are next to each other
+        group_size = 2
+
+        offset = 0
+        while self.get_loop().status():
+            num_pixels = self._lm.count_pixels()
+
+            for i in range(num_pixels):
+                # Calculate which color index to use based on pixel position and current offset
+                # This creates the "moving" effect
+                color_index = ((i + offset) // group_size) % len(colors)
+                self._lm.set_pixel_color(i, colors[color_index])
+
+            self._lm.show()
+
+            # Increment offset to move the lights
+            offset = (offset + 1) % (len(colors) * group_size)
+
+            time.sleep(wait_ms / 1000.0)