Jimmy
74b4b495f9
Add a broad set of new pattern modules and matching pattern smoke scripts so the new effects can be validated directly on-device.
40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
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
|