aboutsummaryrefslogtreecommitdiff
path: root/NeoRuntime/builtin/strandtest/script.py
diff options
context:
space:
mode:
Diffstat (limited to 'NeoRuntime/builtin/strandtest/script.py')
-rw-r--r--NeoRuntime/builtin/strandtest/script.py74
1 files changed, 74 insertions, 0 deletions
diff --git a/NeoRuntime/builtin/strandtest/script.py b/NeoRuntime/builtin/strandtest/script.py
new file mode 100644
index 0000000..50bac1e
--- /dev/null
+++ b/NeoRuntime/builtin/strandtest/script.py
@@ -0,0 +1,74 @@
+from luxcena_neo import NeoBehaviour, utils
+import time
+
+def colorWipe(*color, wait_ms=50):
+ """Wipe color across display a pixel at a time."""
+ for i in range(strip.num_pixels()):
+ strip.set_pixel_color(i, *color)
+ strip.show()
+ time.sleep(wait_ms/1000.0)
+
+def theaterChase(*color, wait_ms=50, iterations=10):
+ """Movie theater light style chaser animation."""
+ for j in range(iterations):
+ for q in range(3):
+ for i in range(0, strip.num_pixels(), 3):
+ strip.set_pixel_color(i+q, *color)
+ strip.show()
+ time.sleep(wait_ms/1000.0)
+ for i in range(0, strip.num_pixels(), 3):
+ strip.set_pixel_color(i+q, 0)
+
+def wheel(pos):
+ """Generate rainbow colors across 0-255 positions."""
+ if pos < 85:
+ return (pos * 3, 255 - pos * 3, 0)
+ elif pos < 170:
+ pos -= 85
+ return (255 - pos * 3, 0, pos * 3)
+ else:
+ pos -= 170
+ return (0, pos * 3, 255 - pos * 3)
+
+def rainbow(wait_ms=20, iterations=1):
+ """Draw rainbow that fades across all pixels at once."""
+ for j in range(256*iterations):
+ for i in range(strip.num_pixels()):
+ strip.set_pixel_color(i, wheel((i+j) & 255))
+ strip.show()
+ time.sleep(wait_ms/1000.0)
+
+def rainbowCycle(wait_ms=20, iterations=5):
+ """Draw rainbow that uniformly distributes itself across all pixels."""
+ for j in range(256*iterations):
+ for i in range(strip.num_pixels()):
+ strip.set_pixel_color(i, wheel(int((i * 256 // strip.num_pixels()) + j) & 255))
+ strip.show()
+ time.sleep(wait_ms/1000.0)
+
+def theaterChaseRainbow(wait_ms=50):
+ """Rainbow movie theater light style chaser animation."""
+ for j in range(256):
+ for q in range(3):
+ for i in range(0, strip.num_pixels(), 3):
+ strip.set_pixel_color(i+q, wheel((i+j) % 255))
+ strip.show()
+ time.sleep(wait_ms/1000.0)
+ for i in range(0, strip.num_pixels(), 3):
+ strip.set_pixel_color(i+q, 0)
+
+class Main(NeoBehaviour):
+
+ def on_start(self):
+ colorWipe(*(255, 0, 0)) # Red wipe
+ colorWipe(*(0, 255, 0)) # Blue wipe
+ colorWipe(*(0, 0, 255)) # Green wipe
+
+ while True:
+ theaterChase(*(127, 127, 127)) # White theater chase
+ theaterChase(*(127, 0, 0)) # Red theater chase
+ theaterChase(*( 0, 0, 127)) # Blue theater chase
+
+ rainbow()
+ rainbowCycle()
+ theaterChaseRainbow()