light_maymays/microcontroller/ctrl.py

54 lines
1.1 KiB
Python

import serial
import time
import colorsys
channels = [
192, # pan
0, # tilt
134, # dimmer
255, # R
0x88, # G
0, # B
0, # W
1, # movement speed
0, # RST
]
start_addr = 10
with serial.Serial("/dev/ttyUSB0", 115200) as ser:
payload = bytearray(512)
FPS = 20
FRAME_TIME = 1 / FPS
t = 0
while True:
loop_start = time.time()
r, g, b = colorsys.hls_to_rgb(t, 0.5, 1)
channels[3] = int(255 * r)
channels[4] = int(255 * g)
channels[5] = int(255 * b)
payload[(start_addr - 1) : (start_addr - 1 + len(channels))] = channels
ser.write(payload)
ser.flush()
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())