131 lines
3.3 KiB
Python
131 lines
3.3 KiB
Python
"""
|
|
On-device test for the Point pattern using mpremote.
|
|
|
|
Usage (from pico/ dir or project root with adjusted paths):
|
|
|
|
mpremote connect <device> cp src/*.py :
|
|
mpremote connect <device> cp src/patterns/*.py :patterns
|
|
mpremote connect <device> cp lib/*.py :
|
|
mpremote connect <device> cp test/test_point.py :
|
|
mpremote connect <device> run test_point.py
|
|
|
|
This script:
|
|
- Instantiates Presets
|
|
- Creates a few in-memory 'point' presets with different ranges/colors
|
|
- Selects each one so you can visually confirm the segments
|
|
"""
|
|
|
|
from presets import Presets, Preset
|
|
|
|
|
|
def make_point_preset(name, colors, n_values, brightness=255):
|
|
"""
|
|
Helper to build a Preset for the 'point' pattern.
|
|
|
|
colors: list of up to 4 (r,g,b) tuples
|
|
n_values: list/tuple of 8 ints [n1..n8]
|
|
"""
|
|
# Pad or trim colors to 4 entries
|
|
cs = list(colors)[:4]
|
|
while len(cs) < 4:
|
|
cs.append((0, 0, 0))
|
|
|
|
n1, n2, n3, n4, n5, n6, n7, n8 = n_values
|
|
data = {
|
|
"p": "point",
|
|
"c": cs,
|
|
"b": brightness,
|
|
"n1": n1,
|
|
"n2": n2,
|
|
"n3": n3,
|
|
"n4": n4,
|
|
"n5": n5,
|
|
"n6": n6,
|
|
"n7": n7,
|
|
"n8": n8,
|
|
# 'a' is not used by point; it's static
|
|
}
|
|
return name, Preset(data)
|
|
|
|
|
|
def show_and_wait(presets, name, preset_obj, wait_ms):
|
|
"""Select a static 'point' preset and hold it for wait_ms."""
|
|
presets.presets[name] = preset_obj
|
|
presets.select(name)
|
|
# Point draws immediately in run(), then just yields; one tick is enough.
|
|
presets.tick()
|
|
|
|
import utime
|
|
|
|
start = utime.ticks_ms()
|
|
while utime.ticks_diff(utime.ticks_ms(), start) < wait_ms:
|
|
# Keep ticking in case other logic ever depends on it
|
|
presets.tick()
|
|
|
|
|
|
def main():
|
|
presets = Presets()
|
|
presets.load()
|
|
|
|
num_leds = presets.strip_length(0)
|
|
if num_leds <= 0:
|
|
print("No strips; aborting point test.")
|
|
return
|
|
|
|
print("Starting point pattern test...")
|
|
|
|
quarter = num_leds // 4
|
|
half = num_leds // 2
|
|
|
|
point_presets = []
|
|
|
|
# 1. Single band: first quarter, red
|
|
point_presets.append(
|
|
make_point_preset(
|
|
"point_red_q1",
|
|
colors=[(255, 0, 0)],
|
|
n_values=[0, quarter - 1, 0, -1, 0, -1, 0, -1],
|
|
)
|
|
)
|
|
|
|
# 2. Two bands: red first half, green second half
|
|
point_presets.append(
|
|
make_point_preset(
|
|
"point_red_green_halves",
|
|
colors=[(255, 0, 0), (0, 255, 0)],
|
|
n_values=[0, half - 1, half, num_leds - 1, 0, -1, 0, -1],
|
|
)
|
|
)
|
|
|
|
# 3. Three bands: R, G, B quarters
|
|
point_presets.append(
|
|
make_point_preset(
|
|
"point_rgb_quarters",
|
|
colors=[(255, 0, 0), (0, 255, 0), (0, 0, 255)],
|
|
n_values=[
|
|
0,
|
|
quarter - 1, # red
|
|
quarter,
|
|
2 * quarter - 1, # green
|
|
2 * quarter,
|
|
3 * quarter - 1, # blue
|
|
0,
|
|
-1,
|
|
],
|
|
)
|
|
)
|
|
|
|
# Show each for ~4 seconds
|
|
for name, preset_obj in point_presets:
|
|
print("Showing point preset:", name)
|
|
show_and_wait(presets, name, preset_obj, wait_ms=4000)
|
|
|
|
print("Point pattern test finished. Turning off LEDs.")
|
|
presets.select("off")
|
|
presets.tick()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|
|
|