Raspberry Pi Pico and GLowBit Rainbow

temp-glob.py

# Raspberry Pi Pico and GLowBit Rainbow
# Combination of sample code from
# GloBit: https://core-electronics.com.au/tutorials/how-to-use-ws2812b-rgb-leds-with-raspberry-pi-pico.html
# Temperature: https://github.com/geerlingguy/baby-safe-temp/blob/master/main.py

import array, utime
from machine import Pin
import rp2

# Get the temperature from the internal RP2040 temperature sensor.
sensor_temp = machine.ADC(4)

# See Raspberry Pi Pico datasheet for the conversion factor.
CONVERSION_FACTOR = 3.3 / (65535)

# Configure the number of WS2812 LEDs.
NUM_LEDS = 13
PIN_NUM = 15
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
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)
    utime.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)
        utime.sleep(wait)
        pixels_show()
    utime.sleep(0.2)


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)

#print("fills")
#for color in COLORS:
#    pixels_fill(color)
#    pixels_show()
#    utime.sleep(0.2)

#print("chases")
#for color in COLORS:
#    color_chase(color, 0.01)

#print("rainbow")
#rainbow_cycle(0)



# Go into a loop.
while True:
    # Get a temperature reading.
    reading = sensor_temp.read_u16() * CONVERSION_FACTOR

    # Convert the temperature into degrees celsius.
    temperature = 27 - (reading - 0.706)/0.001721
    
    pixel = int(temperature) % NUM_LEDS
    
    # If a safe temperature, light the green LED.
    if 20.00 <= temperature <= 22.20:
        pixels_fill(BLACK)
        pixels_set(pixel,GREEN)
        pixels_show()

    # If too hot, light the red LED.
    elif temperature > 22.20:
        pixels_fill(BLACK)
        pixels_set(pixel,RED)
        pixels_show()
    # If too cold, light the blue LED.
    elif temperature < 20.00:
        pixels_fill(BLACK)
        pixels_set(pixel,BLUE)
        pixels_show()
    # If no condition met, we're in an error state, light 'em up!
    else:
        color_chase(WHITE,0.01)

    print(temperature)

    # Sleep for 5 seconds.
    utime.sleep(1)