kopia lustrzana https://github.com/pimoroni/pimoroni-pico
63 wiersze
2.2 KiB
Python
63 wiersze
2.2 KiB
Python
import plasma
|
|
from plasma import plasma_stick
|
|
from random import uniform
|
|
|
|
"""
|
|
Snow in a bottle! Always winter, never Christmas!
|
|
Adjust SNOW_INTENSITY for more snow.
|
|
"""
|
|
|
|
# Set how many LEDs you have
|
|
NUM_LEDS = 50
|
|
|
|
# How much snow? [bigger number = more snowflakes]
|
|
SNOW_INTENSITY = 0.0002
|
|
|
|
# Change RGB colours here (RGB colour picker: https://g.co/kgs/k2Egjk )
|
|
BACKGROUND_COLOUR = [30, 50, 50] # dim blue
|
|
SNOW_COLOUR = [240, 255, 255] # bluish white
|
|
|
|
# how quickly current colour changes to target colour [1 - 255]
|
|
FADE_UP_SPEED = 255 # abrupt change for a snowflake
|
|
FADE_DOWN_SPEED = 1
|
|
|
|
|
|
def display_current():
|
|
# paint our current LED colours to the strip
|
|
for i in range(NUM_LEDS):
|
|
led_strip.set_rgb(i, current_leds[i][0], current_leds[i][1], current_leds[i][2])
|
|
|
|
|
|
def move_to_target():
|
|
# nudge our current colours closer to the target colours
|
|
for i in range(NUM_LEDS):
|
|
for c in range(3): # 3 times, for R, G & B channels
|
|
if current_leds[i][c] < target_leds[i][c]:
|
|
current_leds[i][c] = min(current_leds[i][c] + FADE_UP_SPEED, target_leds[i][c]) # increase current, up to a maximum of target
|
|
elif current_leds[i][c] > target_leds[i][c]:
|
|
current_leds[i][c] = max(current_leds[i][c] - FADE_DOWN_SPEED, target_leds[i][c]) # reduce current, down to a minimum of target
|
|
|
|
|
|
# Create a list of [r, g, b] values that will hold current LED colours, for display
|
|
current_leds = [[0] * 3 for i in range(NUM_LEDS)]
|
|
# Create a list of [r, g, b] values that will hold target LED colours, to move towards
|
|
target_leds = [[0] * 3 for i in range(NUM_LEDS)]
|
|
|
|
# set up the WS2812 / NeoPixel™ LEDs
|
|
led_strip = plasma.WS2812(NUM_LEDS, 0, 0, plasma_stick.DAT, color_order=plasma.COLOR_ORDER_RGB)
|
|
|
|
# start updating the LED strip
|
|
led_strip.start()
|
|
|
|
while True:
|
|
for i in range(NUM_LEDS):
|
|
# randomly add snow
|
|
if SNOW_INTENSITY > uniform(0, 1):
|
|
# set a target to start a snowflake
|
|
target_leds[i] = SNOW_COLOUR
|
|
# slowly reset snowflake to background
|
|
if current_leds[i] == target_leds[i]:
|
|
target_leds[i] = BACKGROUND_COLOUR
|
|
move_to_target() # nudge our current colours closer to the target colours
|
|
display_current() # display current colours to strip
|