diff --git a/pyleds/program/promise/christmas_lights.py b/pyleds/program/promise/christmas_lights.py new file mode 100644 index 0000000..af021ae --- /dev/null +++ b/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)