from machine import Pin, ADC
from time import sleep
import neopixel
import random
 
nl = 50
led = neopixel.NeoPixel(Pin(18), nl)
speed = ADC(Pin(32))
btn = Pin(2, Pin.IN, Pin.PULL_UP)

sez = []

for i in range(50):
    sez.append(i)

sez.reverse()

def izklop(i):
    led[i-1] = [0, 0, 0]

def off():
    for i in range(nl):
        led[i] = [0, 0, 0]
    led.write()

def v_levo():
    r = random.randrange(0, 256)
    g = random.randrange(0, 256)
    b = random.randrange(0, 256)
    clr = [r, g, b]
    for i in range(nl):
        led[i] = clr
        led.write()
        sleep(max(speed.read()/10000, 0.1))
    r += 50
    g += 50
    b += 50
    if r > 255:
        r = r - 255
    if g > 255:
        g = g - 255
    if b > 255:
        b = b - 255
    for i in range(5):
        led.fill([0,0,0])
        led.write()
        sleep(0.4)
        led.fill([r, g, b])
        led.write()
        sleep(0.4)

def v_desno(sez):
    clr = [random.randrange(0,256), random.randrange(0,256), random.randrange(0,256)]
    for i in sez:
        led[i] = clr
        led.write()
        sleep(max(speed.read()/10000, 0.1))
    
    

while True:
    off()
    v_levo()
    while True:
        led.fill([0,0,0])
        levo_ali_desno = input("napiši l za levo animacijo in d za desno animacijo")
        if levo_ali_desno == "l":
            v_levo()
        elif levo_ali_desno == "d":
            v_desno(sez)
        led.fill([0,0,0])