from modules import ws2812

_ai_led_matrix_total = 36
_ai_led_matrix = ws2812(24,_ai_led_matrix_total)
_ai_brightness = 10

import time

def rgba_to_rgb_conversion(red, green, blue):
    if _ai_brightness <= 255:
        alpha = _ai_brightness / 255
    elif _ai_brightness > 255:
        alpha = 255 / 255
    elif _ai_brightness < 0:
        alpha = 0 / 255
    final_red = int((1 - alpha) * 0 + alpha * red)
    final_green = int((1 - alpha) * 0 + alpha * green)
    final_blue = int((1 - alpha) * 0 + alpha * blue)
    final = (final_red, final_green, final_blue)
    return final

def _ai_led_draw_rectangle(x, y, w, h, rgb):
    for i in range(x, x+ w, 1):
        for j in range(y, y+h, 1):
            _a = _ai_led_matrix.set_led((((i*6)//6)-1)*6+(j-1),(rgb))

while True:
    for b in range(0, 50, 1):
        _ai_brightness = b
        _ai_led_draw_rectangle(1,1,6,6,rgba_to_rgb_conversion(255, 0, 0))
        _ai_led_matrix.display()
        time.sleep_ms(1)
    for b in range(50, 0, -1):
        _ai_brightness = b
        _ai_led_draw_rectangle(1,1,6,6,rgba_to_rgb_conversion(255, 0, 0))
        _ai_led_matrix.display()
        time.sleep_ms(1)