"""
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)