2 Commits

Author SHA1 Message Date
Eerik Abel
f2e97ed465 Strips program 2026-01-02 16:25:23 +02:00
545ff966c8 add generated christmas lights (#9)
Reviewed-on: #9
2025-12-21 14:54:56 +00:00
3 changed files with 52 additions and 35 deletions

View File

@@ -0,0 +1,14 @@
from lib.Program import Program
def name():
return '2-color strips'
class Strips(Program):
def run(self, args = None) -> None:
for i in range(self._lm.count_pixels()):
if i // 5 % 2:
color = (255, 255, 255)
else:
color = args['color']
self._lm.set_pixel_color(i, (color[0] << 16) | (color[1] << 8) | color[2])
self._lm.show()

View File

@@ -0,0 +1,38 @@
from lib.Program import Program
from rpi_ws281x import Color
import time
def name():
return 'Christmas Lights'
class ChristmasLights(Program):
def run(self, args: [] = None):
# Configuration
wait_ms = 0 # 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)
self._lm.get_tempo().wait()
time.sleep(wait_ms / 1000.0)

View File

@@ -1,35 +0,0 @@
from lib.Program import Program
from rpi_ws281x import Color
import time
def name():
return 'Vikermasetsus'
def wheel(pos):
"""Generate rainbow colors across 0-255 positions."""
if pos < 85:
return Color(pos * 3, 255 - pos * 3, 0)
elif pos < 170:
pos -= 85
return Color(255 - pos * 3, 0, pos * 3)
else:
pos -= 170
return Color(0, pos * 3, 255 - pos * 3)
class Vikermasetsus(Program):
def run(self, args: [] = None):
wait_ms = 20
iterations = 5
while self.get_loop().status():
"""Draw rainbow that uniformly distributes itself across all pixels."""
for j in range(256 * iterations):
if not self.get_loop().status():
break
for i in range(self._lm.count_pixels()):
self._lm.set_pixel_color(i, wheel(
(int(i * 2560 / self._lm.count_pixels()) + j) & 255))
self._lm.show()
time.sleep(wait_ms / 1000.0)