import utime
from machine import Pin
from neopixel import NeoPixel

class WS2812:
    """WS2812（可编程 RGB LED）"""

    def __init__(self, pin=16, led_beads=1):
        """
        :param pin: 引脚编号，RP2350ZERO板载WS2812使用引脚16
        :param led_beads: LED灯珠数
        """
        if not isinstance(led_beads, int):
            raise TypeError("led_beads数据类型应为整数")

        if not led_beads >= 1:
            raise ValueError("led_beads应大于等于1")

        self.led_beads = led_beads

        # 尝试初始化LED
        try:
            self.led = NeoPixel(Pin(pin, Pin.OUT), self.led_beads)
        except Exception as exception:
            raise RuntimeError(f"初始化LED发生异常，{str(exception)}") from exception

    def set_colors(self, colors):
        """
        设置灯珠颜色
        :param colors: 包含每颗灯珠RGB的颜色列表
        """
        if not isinstance(colors, list):
            raise TypeError("colors数据类型应为列表")

        if len(colors) != self.led_beads:
            raise ValueError("colors列表长度应与LED灯珠数一致")

        for bead_idx, bead_color in enumerate(colors):
            if not isinstance(bead_color, tuple):
                raise TypeError("灯珠颜色数据类型应为元组")

            if len(bead_color) != 3:
                raise ValueError("灯珠颜色长度应为3")

            for channel_idx, channel_value in enumerate(bead_color):
                if not isinstance(channel_value, int):
                    raise TypeError("通道数据数据类型应为整数")

                if not 0 <= channel_value <= 255:
                    raise ValueError("通道数据应在0～255范围内")

        for bead_idx, bead_color in enumerate(colors):
            # 将RGB转为GRB并设置灯珠颜色
            self.led[bead_idx] = (bead_color[1], bead_color[0], bead_color[2])

        # LED刷新
        self.led.write()

    def iridesce(self, hue_offset=20):
        """通过HSV颜色空间生成渐变炫彩效果
        :param hue_offset: 色相偏移角度
        """
        if not isinstance(hue_offset, int):
            raise TypeError("hue_offset数据类型应为整数")

        if not 0 <= hue_offset <= 60:
            raise ValueError("hue_offset应在0～60范围内")

        # 初始化HSV颜色空间参数，其中hue为色相，saturation为饱和度，value为亮度
        hue, saturation, value = 0, 255, 128

        while True:
            # 初始化颜色列表
            colors = []
            
            for bead_idx in range(self.led_beads):
                # 色相
                hue = (hue + bead_idx * hue_offset) % 360
                # 色域
                region = hue // 60
                # 色域内相对位置
                remainder = (hue % 60) * 255 // 60
                # RGB颜色通道中的最小值分量
                primary = (value * (255 - saturation)) // 255
                # RGB颜色通道中的过渡分量
                tertiary = (value * remainder) // 255

                # 根据色域将HSV转为RGB
                if region == 0:
                    color = (value, tertiary, primary)  # 由红色渐变为黄色
                elif region == 1:
                    color = (value - tertiary, value, primary)  # 由黄色渐变为绿色
                elif region == 2:
                    color = (primary, value, tertiary)  # 由绿色渐变为青色
                elif region == 3:
                    color = (primary, value - tertiary, value)  # 由青色渐变为蓝色
                elif region == 4:
                    color = (tertiary, primary, value)  # 由蓝色渐变为洋红色
                else:
                    color = (value, primary, value - tertiary)  # 由洋红色渐变为红色
                
                colors.append(color)

            self.set_colors(colors)

            # 根据色域动态调整色相步长
            hue = hue + [1, 3, 3, 3, 3, 1][region]

            utime.sleep_ms(20)


"""
使用示例

RP2350ZERP 使用板载WS2812，其占用引脚16

import utime
from utils.led import WS2812

rgb = WS2812()

while True:
    rgb.iridesce()

    utime.sleep_ms(20)
"""