kopia lustrzana https://github.com/blaz-r/pico_python_ws2812b
79 wiersze
3.0 KiB
Python
79 wiersze
3.0 KiB
Python
import array, time
|
|
from machine import Pin
|
|
import rp2
|
|
|
|
@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()
|
|
|
|
#delay here is the reset time. You need a pause to reset the LED strip back to the initial LED
|
|
#however, if you have quite a bit of processing to do before the next time you update the strip
|
|
#you could put in delay=0 (or a lower delay)
|
|
class ws2812b:
|
|
def __init__(self, num_leds, state_machine, pin, delay=0.001):
|
|
self.pixels = array.array("I", [0 for _ in range(num_leds)])
|
|
self.sm = rp2.StateMachine(state_machine, ws2812, freq=8000000, sideset_base=Pin(pin))
|
|
self.sm.active(1)
|
|
self.num_leds = num_leds
|
|
self.delay = delay
|
|
self.brightnessvalue = 255
|
|
|
|
# Set the overal value to adjust brightness when updating leds
|
|
def brightness(self, brightness = None):
|
|
if brightness == None:
|
|
return self.brightnessvalue
|
|
else:
|
|
if (brightness < 1):
|
|
brightness = 1
|
|
if (brightness > 255):
|
|
brightness = 255
|
|
self.brightnessvalue = brightness
|
|
|
|
# Create a gradient with two RGB colors between "pixel1" and "pixel2" (inclusive)
|
|
def set_pixel_line_gradient(self, pixel1, pixel2, left_red, left_green, left_blue, right_red, right_green, right_blue):
|
|
if pixel2 - pixel1 == 0: return
|
|
|
|
right_pixel = max(pixel1, pixel2)
|
|
left_pixel = min(pixel1, pixel2)
|
|
|
|
for i in range(right_pixel - left_pixel + 1):
|
|
fraction = i / (right_pixel - left_pixel)
|
|
red = round((right_red - left_red) * fraction + left_red)
|
|
green = round((right_green - left_green) * fraction + left_green)
|
|
blue = round((right_blue - left_blue) * fraction + left_blue)
|
|
|
|
self.set_pixel(left_pixel + i, red, green, blue)
|
|
|
|
# Set an array of pixels starting from "pixel1" to "pixel2" to the desired color.
|
|
def set_pixel_line(self, pixel1, pixel2, red, green, blue):
|
|
for i in range(pixel1, pixel2+1):
|
|
self.set_pixel(i, red, green, blue)
|
|
|
|
def set_pixel(self, pixel_num, red, green, blue):
|
|
# Adjust color values with brightnesslevel
|
|
blue = round(blue * (self.brightness() / 255))
|
|
red = round(red * (self.brightness() / 255))
|
|
green = round(green * (self.brightness() / 255))
|
|
|
|
self.pixels[pixel_num] = blue | red << 8 | green << 16
|
|
|
|
def show(self):
|
|
for i in range(self.num_leds):
|
|
self.sm.put(self.pixels[i],8)
|
|
time.sleep(self.delay)
|
|
|
|
def fill(self, red, green, blue):
|
|
for i in range(self.num_leds):
|
|
self.set_pixel(i, red, green, blue)
|
|
time.sleep(self.delay)
|