# Example using PIO to drive a set of WS2812 LEDs.

import array, time
from machine import Pin
import rp2

# Configure the number of WS2812 LEDs.
NUM_LEDS = 16
PIN_NUM = 6
brightness = 50

@rp2.asm_pio(sideset_init=rp2.PIO.OUT_LOW, out_shiftdir=rp2.PIO.SHIFT_LEFT, autopull=True, pull_thresh=24)
def ws2812():
    T1 = 2
    T2 = 5
    T3 = 3
    wrap_target()
    label("bitloop")
    out(x, 1)               .side(0)    [T3 - 1]
    jmp(not_x, "do_zero")   .side(1)    [T1 - 1]
    jmp("bitloop")          .side(1)    [T2 - 1]
    label("do_zero")
    nop()                   .side(0)    [T2 - 1]
    wrap()


# Create the StateMachine with the ws2812 program, outputting on pin
sm = rp2.StateMachine(0, ws2812, freq=8_000_000, sideset_base=Pin(PIN_NUM))

# Start the StateMachine, it will wait for data on its FIFO.
sm.active(1)

# Display a pattern on the LEDs via an array of LED RGB values.
ar = array.array("I", [0 for _ in range(NUM_LEDS)])

##########################################################################
def pixels_show():
    dimmer_ar = array.array("I", [0 for _ in range(NUM_LEDS)])
    for i,c in enumerate(ar):
        r = int(((c >> 8) & 0xFF) * brightness)
        g = int(((c >> 16) & 0xFF) * brightness)
        b = int((c & 0xFF) * brightness)
        dimmer_ar[i] = (g<<16) + (r<<8) + b
    sm.put(dimmer_ar, 8)
    time.sleep_ms(10)

def pixels_set(i, color):
    ar[i] = (color[1]<<16) + (color[0]<<8) + color[2]

def pixels_fill(color):
    for i in range(len(ar)):
        pixels_set(i, color)

def color_chase(color, wait):
    for i in range(NUM_LEDS):
        pixels_set(i, color)
        time.sleep(wait)
        pixels_show()
    time.sleep(0.2)
 
def wheel(pos):
    # Input a value 0 to 255 to get a color value.
    # The colours are a transition r - g - b - back to r.
    if pos < 0 or pos > 255:
        return (0, 0, 0)
    if pos < 85:
        return (255 - pos * 3, pos * 3, 0)
    if pos < 170:
        pos -= 85
        return (0, 255 - pos * 3, pos * 3)
    pos -= 170
    return (pos * 3, 0, 255 - pos * 3)
 
 
def rainbow_cycle(wait):
    for j in range(255):
        for i in range(NUM_LEDS):
            rc_index = (i * 250 // NUM_LEDS) + j
            pixels_set(i, wheel(rc_index & 255))
        pixels_show()
        time.sleep(wait)

vig = (132,135,146)
azure = (0,128,255)
red = (255,0,0)
orange = (255,125,0)
yellow = (255,255,0)
spring_green = (125,255,0)
green = (0,255,0)
turquoise = (0,125,255)
cyan = (0,255,255)
ocean = (0,125,255)
blue = (0,0,255)
violet = (125,0,255)
magenta = (255,0,255)
raspberry = (255,0,125)

colours = [
vig,
red,
azure,
orange,
yellow,
spring_green,
turquoise,
cyan,
ocean,
green,
blue,
violet,
magenta,
raspberry] 



# print("fills")
# while True:
#     for color in colours:       
#         pixels_fill(color)
#         pixels_show()
#         time.sleep(0.3)

# print("chases")
# while True:
#     for color in colours:       
#         color_chase(color, 0.2)


#rainbow
# while True:
#     print("rainbow")
#     rainbow_cycle(0.7)


#solo round 

# for i in range(17):
#     pixels_set(i, cyan)
#     pixels_show()
#     pixels_set(i, (0,0,0))
#     time.sleep(1)


#solo round multi colour

# for col in colours:
#     for i in range(16):
#         pixels_set(i, col)
#         pixels_show()
#         pixels_set(i, (0,0,0))
#         time.sleep(0.2)


# solo round multi colour

for col in colours:
    for i in range(16):
        pixels_set(i, col)
        pixels_show()
        pixels_set(i, (0,0,0))
        time.sleep(0.5)


# # two colours on opposite time

# for col in colours:
#     for i in range(0,16):
#         pixels_set(i, col)
#         pixels_set(-i, col)
#         pixels_show()
#         pixels_set(i, (0,0,0))
#         pixels_set(-i, (0,0,0))
#         time.sleep(0.2)



#same in whil loop

# end = 16
# start = 0
# while start < end:
#     pixels_set(start, (254,2,222))
#     pixels_set(start-1, (25,254,22))
#     pixels_show()
#     #pixels_set(i, (0,0,0))
#     time.sleep(0.3)
    
#     #pixels_set(i, (0,0,0))
#     time.sleep(0.3)
#     start+=1