feat(patterns): add expanded animation pack with smoke tests
Add a broad set of new pattern modules and matching pattern smoke scripts so the new effects can be validated directly on-device.
This commit is contained in:
39
src/patterns/plasma.py
Normal file
39
src/patterns/plasma.py
Normal file
@@ -0,0 +1,39 @@
|
||||
import utime
|
||||
|
||||
|
||||
class Plasma:
|
||||
def __init__(self, driver):
|
||||
self.driver = driver
|
||||
|
||||
def _wheel(self, pos):
|
||||
if pos < 85:
|
||||
return (pos * 3, 255 - pos * 3, 0)
|
||||
if pos < 170:
|
||||
pos -= 85
|
||||
return (255 - pos * 3, 0, pos * 3)
|
||||
pos -= 170
|
||||
return (0, pos * 3, 255 - pos * 3)
|
||||
|
||||
def run(self, preset):
|
||||
scale = max(1, int(preset.n1) if int(preset.n1) > 0 else 6)
|
||||
speed = max(1, int(preset.n2) if int(preset.n2) > 0 else 2)
|
||||
contrast = max(1, int(preset.n3) if int(preset.n3) > 0 else 2)
|
||||
t = self.driver.step % 256
|
||||
last = utime.ticks_ms()
|
||||
while True:
|
||||
d = max(1, int(preset.d))
|
||||
now = utime.ticks_ms()
|
||||
if utime.ticks_diff(now, last) >= d:
|
||||
for i in range(self.driver.num_leds):
|
||||
v = ((i * scale + t) & 255)
|
||||
v2 = (((i * scale // max(1, contrast)) - (t * 2)) & 255)
|
||||
c = self._wheel((v + v2) & 255)
|
||||
self.driver.n[i] = self.driver.apply_brightness(c, preset.b)
|
||||
self.driver.n.write()
|
||||
t = (t + speed) % 256
|
||||
self.driver.step = t
|
||||
last = utime.ticks_add(last, d)
|
||||
if not preset.a:
|
||||
yield
|
||||
return
|
||||
yield
|
||||
Reference in New Issue
Block a user