初始化提交

boards/default_src/micropython/origin/build/lib/ws2812.py

@@ -0,0 +1,84 @@
+"""
+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)