# ws2812_test_v05 tlfong01 2021mar06hkt2114

# Configuration
# Piromoni Tiny2040
# Acer Intel Core i5 PC on Chinese Win10
# MicroPython (Raspberry Pi Pico)
# Thonny 3.3.3

# References
#   (1) Raspberry Pi Pico and NeoPixel Example in MicroPython - PiBits
#       http://www.pibits.net/code/raspberry-pi-pico-and-neopixel-example-in-micropython.php#codesyntax_1
#   (2) MicroPython: WS2812B Addressable RGB LEDs with ESP3/ESP8266 - RandomNerdTutorials
#       https://randomnerdtutorials.com/micropython-ws2812b-addressable-rgb-leds-
#       neopixel-esp32-esp8266/
#   (3) NeoPixel dithering with Pico - Ben Everard, RpiBlog 2021jan22
#       https://www.raspberrypi.org/blog/neopixel-dithering-with-pico/
#   (4) How to use WS2812B RGB LEDs with Raspberry Pi Pico - Michael, Core Electronics 2021feb08
#       https://core-electronics.com.au/tutorials/how-to-use-ws2812b-rgb-leds-with-raspberry-pi-pico.html
#   (5) Pico Examples: pico-examples/pio/ws2812/ - Rpi, GitHub
#       https://github.com/raspberrypi/pico-examples/tree/master/pio/ws2812

import array, time
from machine import Pin
import rp2
 
# Configure the number of WS2812 LEDs, pins and brightness.
NUM_LEDS = 7
PIN_NUM = 16
brightness = 0.1
  
@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(16).
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)
 
BLACK = (0, 0, 0)
RED = (255, 0, 0)
YELLOW = (255, 150, 0)
GREEN = (0, 255, 0)
CYAN = (0, 255, 255)
BLUE = (0, 0, 255)
PURPLE = (180, 0, 255)
WHITE = (255, 255, 255)
COLORS = (BLACK, RED, YELLOW, GREEN, CYAN, BLUE, PURPLE, WHITE)

# *** Main ***

print('Begin ws2812_test_v05(), ...')
for color in COLORS:
    pixels_fill(color)
    pixels_show()
    time.sleep(0.5)
print('End   ws2812_test_v05()')    

# *** End ***

# *** Begin sample output, ... tlfong01  2021mar06hkt2120 ***
'''
>>> %Run -c $EDITOR_CONTENT
Begin ws2812_test_v05(), ...
End   ws2812_test_v05()
>>> 
'''
# *** End sample output. ***