139 lines
4.3 KiB
Python
139 lines
4.3 KiB
Python
#!/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 n_chase (n1=5, n2=5, n3=1, n4=1)
|
|
print("Test 1: Basic n_chase (n1=5, n2=5, n3=1, n4=1)")
|
|
p.set_param("br", 255)
|
|
p.set_param("dl", 200)
|
|
p.set_param("n1", 5) # 5 LEDs color0
|
|
p.set_param("n2", 5) # 5 LEDs color1
|
|
p.set_param("n3", 1) # Move 1 forward on even steps
|
|
p.set_param("n4", 1) # Move 1 forward on odd steps
|
|
p.set_param("cl", [(255, 0, 0), (0, 255, 0)]) # Red and Green
|
|
p.select("n_chase")
|
|
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 2: Forward and backward (n3=2, n4=-1)
|
|
print("Test 2: Forward and backward (n3=2, n4=-1)")
|
|
p.stopped = False
|
|
p.set_param("n1", 3)
|
|
p.set_param("n2", 3)
|
|
p.set_param("n3", 2) # Move 2 forward on even steps
|
|
p.set_param("n4", -1) # Move 1 backward on odd steps
|
|
p.set_param("dl", 150)
|
|
p.set_param("cl", [(0, 0, 255), (255, 255, 0)]) # Blue and Yellow
|
|
p.select("n_chase")
|
|
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 3: Large segments (n1=10, n2=5)
|
|
print("Test 3: Large segments (n1=10, n2=5, n3=3, n4=3)")
|
|
p.stopped = False
|
|
p.set_param("n1", 10) # 10 LEDs color0
|
|
p.set_param("n2", 5) # 5 LEDs color1
|
|
p.set_param("n3", 3) # Move 3 forward
|
|
p.set_param("n4", 3) # Move 3 forward
|
|
p.set_param("dl", 200)
|
|
p.set_param("cl", [(255, 128, 0), (128, 0, 255)]) # Orange and Purple
|
|
p.select("n_chase")
|
|
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 4: Fast movement (n3=5, n4=5)
|
|
print("Test 4: Fast movement (n3=5, n4=5)")
|
|
p.stopped = False
|
|
p.set_param("n1", 4)
|
|
p.set_param("n2", 4)
|
|
p.set_param("n3", 5) # Move 5 forward
|
|
p.set_param("n4", 5) # Move 5 forward
|
|
p.set_param("dl", 100)
|
|
p.set_param("cl", [(255, 0, 255), (0, 255, 255)]) # Magenta and Cyan
|
|
p.select("n_chase")
|
|
task = asyncio.create_task(p.run())
|
|
start = utime.ticks_ms()
|
|
while utime.ticks_diff(utime.ticks_ms(), start) < 2000:
|
|
wdt.feed()
|
|
await asyncio.sleep_ms(10)
|
|
await p.stop()
|
|
await task
|
|
|
|
# Test 5: Backward movement (n3=-2, n4=-2)
|
|
print("Test 5: Backward movement (n3=-2, n4=-2)")
|
|
p.stopped = False
|
|
p.set_param("n1", 6)
|
|
p.set_param("n2", 4)
|
|
p.set_param("n3", -2) # Move 2 backward
|
|
p.set_param("n4", -2) # Move 2 backward
|
|
p.set_param("dl", 200)
|
|
p.set_param("cl", [(255, 255, 255), (0, 0, 0)]) # White and Black
|
|
p.select("n_chase")
|
|
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: Alternating forward/backward (n3=3, n4=-2)
|
|
print("Test 6: Alternating forward/backward (n3=3, n4=-2)")
|
|
p.stopped = False
|
|
p.set_param("n1", 5)
|
|
p.set_param("n2", 5)
|
|
p.set_param("n3", 3) # Move 3 forward on even steps
|
|
p.set_param("n4", -2) # Move 2 backward on odd steps
|
|
p.set_param("dl", 250)
|
|
p.set_param("cl", [(255, 0, 0), (0, 255, 0)]) # Red and Green
|
|
p.select("n_chase")
|
|
task = asyncio.create_task(p.run())
|
|
start = utime.ticks_ms()
|
|
while utime.ticks_diff(utime.ticks_ms(), start) < 4000:
|
|
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())
|
|
|
|
|