import picozero
from time import sleep
from picozero import LED, Button, RGBLED, DistanceSensor

led = LED(2)
button=Button(1)
rgb = RGBLED(red=3, green=28, blue=27)
ds = DistanceSensor(echo=20, trigger=21, max_distance = 4)
distanceSensorOn = False

def OneA():
    for i in range(10):
        print(i+1)
        led.on()
        sleep(1)
        led.off()
        sleep(1)

def OneB():
    for i in range(20):
        time = .10 *(i)
        print(time)
        led.on()
        sleep(2-time)
        print(2-time)
        led.off()
        sleep(2-time)
    

def OneC():
    def divisible_conditional(num):
        for i in range(100):
            if i % num == 0:
                print("LED on")
                led.on()
                sleep(1)
            elif i % num != 0:
                print("LED off")
                led.off()
                sleep(1)
    number = int(input("Enter a number 1-100: "))
    divisible_conditional(number)


def TwoA():
    while True:
        if button.is_pressed:
            led.on()
            print("Button is pressed")
        else:
            led.off()
            print("Button is not pressed")
        sleep(0.1)
    

def TwoB():
    count = 0
    while True:
        if button.is_pressed:
            count += 1
            print("Button is pressed")
            print(f"# of button presses:  + {count}")
            sleep(0.5)
        

def TwoC():
    while True:
        if button.is_pressed:
            print("Button is pressed")
            led.toggle()
            sleep(.5)
        

def ThreeA():
    rgb.color = (255, 0, 0)  # full red
    sleep(1)
    rgb.color = (0, 255, 0)  # full green
    sleep(1)
    rgb.color = (0, 0, 255)  # cyan
    sleep(1)
    rgb.color = (0, 0, 0)  # off
    sleep(1)


def ThreeB():
    count = 0
    colors = [(0,0,0), (255, 0, 0), (0, 255, 0), (0, 0 , 255)]
    while True:
        rgb.color = colors[count]
        if count == 0:
            print("LED is off")
        elif count == 1:
            print("LED is red")
        elif count == 2:
            print("LED is green")
        elif count == 3:
            print("LED is blue")
        if button.is_pressed:
            print("Button is pressed")
            count += 1
        if count > 3:
            count = 0
        sleep(.5)

while True:
    if button.is_pressed:
        print("Button is pressed")
        distanceSensorOn = True
        sleep(1)

    while distanceSensorOn:
        print(ds.distance)
        if ds.distance >= 4:
            rgb.color = (0, 255, 0)
        elif ds.distance >= 2 and ds.distance < 3:
            rgb.color = (255, 255, 0)
        elif ds.distance < 2:
            rgb.color = (255, 0, 0)
        sleep(0.1)
        if button.is_pressed:
            print("Button is pressed")
            rgb.color = (0,0,0)
            distanceSensorOn = False
            sleep(1)
    


#OneA()
#OneB()
#OneC()
#TwoA()
#TwoB()
#TwoC()
#ThreeA()
#ThreeB()