from machine import Pin, SPI
import epaper_babasss as epaper2in9

# Configuración de pines para SPI y e-paper
cs = Pin(5)
dc = Pin(17)
rst = Pin(16)
busy = Pin(4)
sck = Pin(18)
mosi = Pin(23)
miso = Pin(14)
spi = SPI(baudrate=2000000, polarity=0, phase=0, sck=sck, mosi=mosi, miso=miso)

# Inicializar la pantalla e-paper
e = epaper2in9.EPD(spi, cs, dc, rst, busy)
e.init()

# Dimensiones de la pantalla
w = 128
h = 296

def fill_screen_with_color(color):
    if color not in [0x00, 0xFF]:
        raise ValueError("Color debe ser 0x00 (negro), 0xFF (blanco) o 0x00 (rojo)")

    print(f"# Llenando la pantalla con el color {'negro' if color == 0x00 else 'blanco' if color == 0xFF else 'rojo'}")
    
    # Limpiar el buffer de la pantalla con el color especificado
    if color == 0xFF:  # Blanco
        e.Clear(0xFF, 0xFF)
    elif color == 0x00:  # Negro
        e.Clear(0x00, 0x00)
    else:  # Rojo
        e.Clear(0xFF, 0x00)  # Blanco en el buffer negro, rojo en el buffer rojo
    e.display()

# Llenar la pantalla con rojo
fill_screen_with_color(0x00)  # Ajustar el valor si el rojo tiene un código específico en tu biblioteca

# Descomenta la siguiente línea para llenar la pantalla con blanco
# fill_screen_with_color(0xFF)