77 lines
1.5 KiB
Python
77 lines
1.5 KiB
Python
import serial
|
|
import time
|
|
import colorsys
|
|
|
|
import sys
|
|
|
|
channels = [
|
|
255, # dimmer
|
|
0, # R
|
|
0, # G
|
|
0, # B
|
|
0, # W
|
|
0, # A
|
|
0, # UV
|
|
0, # Strobe
|
|
0, # function
|
|
0, # function speed
|
|
]
|
|
|
|
start_addr = 1
|
|
|
|
with serial.Serial("/dev/ttyUSB0", 500000) as ser:
|
|
|
|
payload = bytearray(512)
|
|
|
|
FPS = 50
|
|
if len(sys.argv) > 1:
|
|
FPS = int(sys.argv[1])
|
|
|
|
FRAME_TIME = 1 / FPS
|
|
t = 0
|
|
|
|
def sync():
|
|
# wait for sync
|
|
while True:
|
|
b = ser.readline()
|
|
if b.strip() == b"Sync.":
|
|
return
|
|
|
|
sync()
|
|
|
|
print("initial sync.")
|
|
|
|
while True:
|
|
|
|
loop_start = time.time()
|
|
|
|
r, g, b = colorsys.hls_to_rgb(t, 0.5, 1)
|
|
|
|
channels[1] = int(255 * r)
|
|
channels[2] = int(255 * g)
|
|
channels[3] = int(255 * b)
|
|
|
|
payload[(start_addr - 1) : (start_addr - 1 + len(channels))] = channels
|
|
|
|
ser.write(payload)
|
|
ser.flush()
|
|
|
|
response = ser.readline()
|
|
if response.strip() != b"Ack.":
|
|
print(f"received bad response: {response!r}")
|
|
sync()
|
|
continue
|
|
|
|
t += FRAME_TIME
|
|
t %= 1
|
|
|
|
loop_time = time.time() - loop_start
|
|
if loop_time < FRAME_TIME:
|
|
time.sleep(FRAME_TIME - loop_time)
|
|
else:
|
|
print("loop took too long!")
|
|
|
|
print(f"loop time: {1000 * loop_time:0.2f}ms busy, {1000 * (time.time() - loop_start):0.2f}ms total")
|
|
|
|
# print(ser.read_all())
|