85 lines
2.2 KiB
Python
85 lines
2.2 KiB
Python
"""
|
|
WS2812 RGB
|
|
|
|
Micropython library for the WS2812 NeoPixel-RGB
|
|
=======================================================
|
|
|
|
#Preliminary composition 20240110
|
|
|
|
dahanzimin From the Mixly Team
|
|
"""
|
|
from time import sleep
|
|
from machine import bitstream
|
|
|
|
class NeoPixel:
|
|
def __init__(self, pin, n, bpp=3, timing=1, ORDER=(1, 0, 2, 3), default=None, multiplex=False, leds=0):
|
|
self.pin = pin
|
|
self.bpp = bpp
|
|
self.leds = leds
|
|
self.rgbs = n-leds
|
|
self.ORDER = ORDER
|
|
self.multiplex = multiplex
|
|
self.rgb_buf = bytearray(self.rgbs * bpp)
|
|
self.led_buf = bytearray(self.leds * bpp)
|
|
self.timing = (((350, 850, 800, 400) if timing else (800, 1700, 1600, 900)) if isinstance(timing, int) else timing)
|
|
if not self.multiplex: self.pin.init(self.pin.OUT,value=default)
|
|
self.write()
|
|
|
|
def __len__(self):
|
|
return self.rgbs
|
|
|
|
def __setitem__(self, n, v):
|
|
for i in range(self.bpp):
|
|
self.rgb_buf[n * self.bpp + self.ORDER[i]] = v[i]
|
|
|
|
def __getitem__(self, n):
|
|
return tuple(self.rgb_buf[n* self.bpp + self.ORDER[i]] for i in range(self.bpp))
|
|
|
|
def led_set(self, n, v):
|
|
for i in range(self.bpp):
|
|
self.led_buf[n * self.bpp + self.ORDER[i]] = v[i]
|
|
|
|
def led_get(self, n):
|
|
return tuple(self.led_buf[n * self.bpp + self.ORDER[i]] for i in range(self.bpp))
|
|
|
|
def fill(self, v):
|
|
for i in range(self.bpp):
|
|
j = self.ORDER[i]
|
|
while j < self.rgbs * self.bpp:
|
|
self.rgb_buf[j] = v[i]
|
|
j += self.bpp
|
|
|
|
def write(self):
|
|
if self.multiplex: self.pin.init(self.pin.OUT)
|
|
bitstream(self.pin, 0, self.timing, self.rgb_buf + self.led_buf)
|
|
if self.multiplex: self.pin.init(self.pin.IN)
|
|
|
|
def color_chase(self,R, G, B, wait):
|
|
for i in range(self.rgbs):
|
|
self.__setitem__(i,(R, G, B))
|
|
self.write()
|
|
sleep(wait/1000)
|
|
|
|
def rainbow_cycle(self, wait, clear=True):
|
|
for j in range(255):
|
|
for i in range(self.rgbs):
|
|
rc_index = (i * 256 // self.rgbs) + j
|
|
self.__setitem__(i,self.wheel(rc_index & 255))
|
|
self.write()
|
|
sleep(wait / 1000 / 256)
|
|
if clear:
|
|
self.fill((0, 0, 0))
|
|
self.write()
|
|
|
|
def wheel(self,pos):
|
|
if pos < 0 or pos > 255:
|
|
return (0, 0, 0)
|
|
elif pos < 85:
|
|
return (pos * 3, 255 - pos * 3, 0)
|
|
elif pos < 170:
|
|
pos -= 85
|
|
return (255 - pos * 3, 0, pos * 3)
|
|
else:
|
|
pos -= 170
|
|
return (0, pos * 3, 255 - pos * 3)
|