from machine import Pin, ADC, I2C
from time import sleep
from ssd1306 import SSD1306_I2C

i2c = I2C(0, sda=Pin(0), scl=Pin(1))
display = SSD1306_I2C(128, 64, i2c)

X = [
    [0, 1, 2, 3, 4, 5, 6],      
    [1, 2, 3, 4, 5],            
    [2, 3, 4],                  
    [3]                         
] 

Cores= [[1,0,0], #Vermelho
        [0,1,0], #Verde
        [0,0,1], #Azul
        [1,1,0], #Amarelo
        [1,0,1], #Roxo
        [0,1,1], #Ciano
        [1,1,1], #Branco
        [0,0,0]] #Desligado

Buttons= [Pin(pin, Pin.IN, Pin.PULL_UP) for pin in (28,27,26)]
Leds= [Pin(pin, Pin.OUT) for pin in (5,4,3)]

for x in range(len(X)):
    for y in X[x]:
        display.pixel(x, y, 1)

for cor in Cores:
    for led, x in zip(Leds, cor):
        led.value(cor)
        sleep(1)


display.show()

sleep(2)
display.fill(0)
display.move(X, 0, 2, 1)
display.show()