import ev3_dc as ev3
import keyboard
import matplotlib.pyplot as plt
from thread_task import Periodic

MAC = '00:16:53:50:8A:86'  # Cambia esto por la dirección MAC de tu ladrillo EV3

exit = False
keyboard.add_hotkey('q', lambda: globals().update(exit=True))

redList = []  # Lista para almacenar el color rojo
greenList = []  # Lista para almacenar el color verde
blueList = []  # Lista para almacenar el color azul

with ev3.EV3(protocol=ev3.BLUETOOTH, host=MAC) as robot:
    # Conectar el sensor de color en el Puerto 1
    sensor = ev3.Color(ev3.PORT_1, ev3_obj=robot)

    def read_sensor():
        rgb = sensor.rgb_raw
        redList.append(rgb.red)
        greenList.append(rgb.green)
        blueList.append(rgb.blue)
        print(f"Luz RGB: R{rgb.red}, G{rgb.green}, B{rgb.blue}")

    print("Presiona 'q' para interrumpir el programa")
    sensor_task = Periodic(0.5, read_sensor)
    sensor_task.start()
    while not exit:
        plt.clf()
        plt.title('Lecturas del Sensor de Color RGB')
        plt.plot(redList, 'r-')
        plt.plot(greenList, 'g-')
        plt.plot(blueList, 'b-')
        plt.show(block=False)
        plt.draw()
        plt.pause(1)
    sensor_task.stop()
    print("Programa terminado")