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())