Add circle pattern

2025-11-24 22:07:39 +13:00
parent a17380c250
commit fe2717f51d
2 changed files with 211 additions and 2 deletions
--- a/test/patterns/circle.py
+++ b/test/patterns/circle.py
@@ -0,0 +1,132 @@
+#!/usr/bin/env python3
+import uasyncio as asyncio
+import utime
+from machine import WDT
+from settings import Settings
+from patterns import Patterns
+
+
+async def main():
+    s = Settings()
+    pin = s.get("led_pin", 10)
+    num = s.get("num_leds", 30)
+
+    p = Patterns(pin=pin, num_leds=num)
+    wdt = WDT(timeout=10000)
+    
+    # Test 1: Basic circle (n1=50, n2=100, n3=200, n4=0)
+    print("Test 1: Basic circle (n1=50, n2=100, n3=200, n4=0)")
+    p.set_param("br", 255)
+    p.set_param("n1", 50)   # Head moves 50 LEDs/second
+    p.set_param("n2", 100)  # Max length 100 LEDs
+    p.set_param("n3", 200)  # Tail moves 200 LEDs/second
+    p.set_param("n4", 0)    # Min length 0 LEDs
+    p.set_param("cl", [(255, 0, 0)])  # Red
+    p.select("circle")
+    task = asyncio.create_task(p.run())
+    start = utime.ticks_ms()
+    # Run for 5 seconds to see full cycle
+    while utime.ticks_diff(utime.ticks_ms(), start) < 5000:
+        wdt.feed()
+        await asyncio.sleep_ms(10)
+    await p.stop()
+    await task
+    
+    # Test 2: Slow growth, fast shrink (n1=20, n2=50, n3=100, n4=0)
+    print("Test 2: Slow growth, fast shrink (n1=20, n2=50, n3=100, n4=0)")
+    p.stopped = False
+    p.set_param("n1", 20)   # Head moves 20 LEDs/second (slow)
+    p.set_param("n2", 50)   # Max length 50 LEDs
+    p.set_param("n3", 100)  # Tail moves 100 LEDs/second (fast)
+    p.set_param("n4", 0)    # Min length 0 LEDs
+    p.set_param("cl", [(0, 255, 0)])  # Green
+    p.select("circle")
+    task = asyncio.create_task(p.run())
+    start = utime.ticks_ms()
+    while utime.ticks_diff(utime.ticks_ms(), start) < 5000:
+        wdt.feed()
+        await asyncio.sleep_ms(10)
+    await p.stop()
+    await task
+    
+    # Test 3: Fast growth, slow shrink (n1=100, n2=30, n3=20, n4=0)
+    print("Test 3: Fast growth, slow shrink (n1=100, n2=30, n3=20, n4=0)")
+    p.stopped = False
+    p.set_param("n1", 100)  # Head moves 100 LEDs/second (fast)
+    p.set_param("n2", 30)   # Max length 30 LEDs
+    p.set_param("n3", 20)   # Tail moves 20 LEDs/second (slow)
+    p.set_param("n4", 0)    # Min length 0 LEDs
+    p.set_param("cl", [(0, 0, 255)])  # Blue
+    p.select("circle")
+    task = asyncio.create_task(p.run())
+    start = utime.ticks_ms()
+    while utime.ticks_diff(utime.ticks_ms(), start) < 5000:
+        wdt.feed()
+        await asyncio.sleep_ms(10)
+    await p.stop()
+    await task
+    
+    # Test 4: With minimum length (n1=50, n2=40, n3=100, n4=10)
+    print("Test 4: With minimum length (n1=50, n2=40, n3=100, n4=10)")
+    p.stopped = False
+    p.set_param("n1", 50)   # Head moves 50 LEDs/second
+    p.set_param("n2", 40)   # Max length 40 LEDs
+    p.set_param("n3", 100)  # Tail moves 100 LEDs/second
+    p.set_param("n4", 10)  # Min length 10 LEDs (never fully disappears)
+    p.set_param("cl", [(255, 255, 0)])  # Yellow
+    p.select("circle")
+    task = asyncio.create_task(p.run())
+    start = utime.ticks_ms()
+    while utime.ticks_diff(utime.ticks_ms(), start) < 5000:
+        wdt.feed()
+        await asyncio.sleep_ms(10)
+    await p.stop()
+    await task
+    
+    # Test 5: Very fast (n1=200, n2=20, n3=200, n4=0)
+    print("Test 5: Very fast (n1=200, n2=20, n3=200, n4=0)")
+    p.stopped = False
+    p.set_param("n1", 200)  # Head moves 200 LEDs/second (very fast)
+    p.set_param("n2", 20)   # Max length 20 LEDs
+    p.set_param("n3", 200)  # Tail moves 200 LEDs/second (very fast)
+    p.set_param("n4", 0)    # Min length 0 LEDs
+    p.set_param("cl", [(255, 0, 255)])  # Magenta
+    p.select("circle")
+    task = asyncio.create_task(p.run())
+    start = utime.ticks_ms()
+    while utime.ticks_diff(utime.ticks_ms(), start) < 3000:
+        wdt.feed()
+        await asyncio.sleep_ms(10)
+    await p.stop()
+    await task
+    
+    # Test 6: Very slow (n1=10, n2=25, n3=10, n4=0)
+    print("Test 6: Very slow (n1=10, n2=25, n3=10, n4=0)")
+    p.stopped = False
+    p.set_param("n1", 10)   # Head moves 10 LEDs/second (very slow)
+    p.set_param("n2", 25)  # Max length 25 LEDs
+    p.set_param("n3", 10)   # Tail moves 10 LEDs/second (very slow)
+    p.set_param("n4", 0)    # Min length 0 LEDs
+    p.set_param("cl", [(0, 255, 255)])  # Cyan
+    p.select("circle")
+    task = asyncio.create_task(p.run())
+    start = utime.ticks_ms()
+    while utime.ticks_diff(utime.ticks_ms(), start) < 5000:
+        wdt.feed()
+        await asyncio.sleep_ms(10)
+    await p.stop()
+    await task
+    
+    # Cleanup
+    print("Test complete, turning off")
+    p.stopped = False
+    p.select("off")
+    task = asyncio.create_task(p.run())
+    await asyncio.sleep_ms(100)
+    await p.stop()
+    await task
+
+
+if __name__ == "__main__":
+    asyncio.run(main())
+