rdmarsh · January 29, 2021 05:24 · Jan 29, 2021
diff --git a/temp-glob.py b/temp-glob.py
@@ -0,0 +1,129 @@
+# 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)