from machine import Pin, ADC
from utime import sleep
import neopixel
import random
num_led=40
led=neopixel.NeoPixel(Pin(5),num_led)
pot33= ADC(Pin(33,Pin.IN))
pot4=ADC(Pin(4,Pin.IN))
gumb=Pin(27, Pin.IN, Pin.PULL_UP)
def nas():
    led.fill([0,0,0])
    sleep(0.2)
    led.fill([0,255,0])
    sleep(0.2)
    led.fill([0,0,0])
    a=random.randint(0,256)
    b=random.randint(0,256)
    c=random.randint(0,256)
    for i in range(num_led-1,-1,-1):
        hit=pot4.read()/10000
        moc=pot33.read()//16
        led[i]=[a,b,c]
        led.write()
        sleep(hit)
        a=(a+random.randint(0,moc))%256
        b=(b+random.randint(0,moc))%256
        c=(c+random.randint(0,moc))%256

        led[i]=[a,b,c]
        led.write()
        if i==0:
            vnos=input("ali hočeš da se lučke premikajo v smeri urinega kazalca DA/NE. ")
            if vnos=="DA":
                ur()
            elif vnos=="NE":
                nas()
    led.fill([0,0,0])

def ur():
    led.fill([0,0,0])
    sleep(0.2)
    led.fill([0,255,0])
    sleep(0.2)
    led.fill([0,0,0])
    sleep(0.2)
    a=random.randint(0,256)
    b=random.randint(0,256)
    c=random.randint(0,256)
    for i in range(num_led):
        moc=pot33.read()//16
        hit=pot4.read()/10000
        led[i]=[a,b,c]
        led.write()
        sleep(hit)
        a=(a+random.randint(0,moc))%256
        b=(b+random.randint(0,moc))%256
        c=(c+random.randint(0,moc))%256

        led[i]=[a,b,c]
        led.write()
        if num_led-1==i:
            vnos=input("ali hočeš da se lučke premikajo v smeri urinega kazalca DA/NE. ")
            if vnos=="DA":
                ur()
            elif vnos=="NE":
                nas()
    led.fill([0,0,0])
while True:
    print("PRITISNI GUMB ZA ZAČETEK IGRE!!!!!!!!")
    sleep(1)
    if gumb.value()==0:
        while True:
            led.fill([255,0,0])
            led.write()
            print(3)
            sleep(0.5)
            led.fill([0,0,0])
            led.write()
            sleep(0.5)
            led.fill([255,0,0])
            led.write()
            print(2)
            sleep(0.5)
            led.fill([0,0,0])
            led.write()
            sleep(0.5)
            led.fill([255,0,0])
            led.write()
            print(1)
            sleep(0.5)
            led.fill([0,0,0])
            led.write()
            print("IGRA SE ZAČENJA!!!!")
            ur()