from machine import Pin, PWM
from time import sleep

D7 = Pin(23,Pin.OUT)
D6 = Pin(22,Pin.OUT)
D5 = Pin(21,Pin.OUT)
D4 = Pin(19,Pin.OUT)
D3 = Pin(18,Pin.OUT)
D2 = Pin(5,Pin.OUT)
D1 = Pin(17,Pin.OUT)
D0 = Pin(16,Pin.OUT)
E = Pin(4,Pin.OUT)
Rs = Pin(0,Pin.OUT)
R = PWM(Pin(25),freq=50)
G = PWM(Pin(26),freq=50)
B = PWM(Pin(27),freq=50)
push = Pin(12,Pin.IN,Pin.PULL_UP)

def peligro(x,y,z,a):
    D7.value(int(x[0]))
    D6.value(int(x[1]))
    D5.value(int(x[2]))
    D4.value(int(x[3]))
    D3.value(int(x[4]))
    D2.value(int(x[5]))
    D1.value(int(x[6]))
    D0.value(int(x[7]))
    Rs.value(int(x[8]))
    E.value(1)
    sleep(0.1)
    E.value(0)
    R.duty(y)
    G.duty(z)
    B.duty(a)

lista_1 = ["001100000", "001100000", "001100000", "001110000", "000101000", "000011000", "000001100", "000000010"]
lista_A = ["010000011", "010011001", "010101001", "010011111", "000000010"]
lista_P = ["010100001", "010100101", "010001011", "010000111", "010000011", "010101011", "010000111", "010010011", "010011111", "010011101", "000000010"]
lista_S = ["010100111", "010010011", "010001111" ,"010000011", "000000010"]

for n in lista_1:
    peligro(n,0,0,0)

n=0
i=0

while(1):
    if push.value()==0:
        sleep(0.2)
        n=n+1
        if n > 2:
            n=0
    while push.value()==0:
            sleep(0.02)
            
    if n==0: 
        for i in lista_A:
            peligro(i,1023,0,0)
            sleep(0.5)

    elif n==1:
        for i in lista_P:
            peligro(i,1023,800,0)
            sleep(0.5)

    else:
        for i in lista_S:
            peligro(i,0,1023,0)
            sleep(0.5)