ESP32（MicroPython） WS2812 RGB流水灯渐变版

ESP32(MicroPython) RGB流水灯渐变版

本程序实现了渐变颜色的RGB流水灯，速度和亮度可调，分对称和环形两种模式。

主要实现方法是先设定一个可调的基本亮度，渐变开始前第一种颜色亮度为基本亮度的8倍，第二种颜色亮度为0，之后每次渐变第一种颜色的亮度减去基本亮度，第二种颜色的亮度加上基本亮度，经过8个灯后完成渐变。在对称模式下每次加减的亮度为基本亮度的2倍，由于取值不同，对称模式需要增加2个变量记录起始位置而非复用环形模式的变量。

代码如下

#导入Pin模块
from machine import Pin
import time
from machine import PWM
from neopixel import NeoPixel
 
# 五向导航按键，COM引脚接3.3V
key1 = Pin(21, Pin.IN, Pin.PULL_DOWN)
key2 = Pin(19, Pin.IN, Pin.PULL_DOWN)
key3 = Pin(18, Pin.IN, Pin.PULL_DOWN)
key4 = Pin(5, Pin.IN, Pin.PULL_DOWN)
key5 = Pin(17, Pin.IN, Pin.PULL_DOWN)
key6 = Pin(16, Pin.IN, Pin.PULL_DOWN)
 
pin=13
rgb_num=24
rgb_led=NeoPixel(Pin(pin,Pin.OUT),rgb_num)

key_en=1
#按键扫描函数
def key_scan():
    global key_en
    if key_en==1 and (key1.value()==1 or key2.value()==1 or key3.value()==1 or key4.value()==1 or
                      key5.value()==1 or key6.value()==1  ):
        time.sleep_ms(10)
        key_en=0
        if key1.value()==1:
            return 1
        elif key2.value()==1:
            return 2
        elif key3.value()==1:
            return 3
        elif key4.value()==1:
            return 4
        elif key5.value()==1:
            return 5
        elif key6.value()==1:
            return 6
    elif (key1.value()==0 and key2.value()==0 and key3.value()==0 and key4.value()==0 and
          key5.value()==0 and key6.value()==0  ) :
        key_en=1
    return 0

brightness=6
delay=40
mode=1
def key_get(): #获取键值并改变变量的值
    global brightness
    global delay
    global mode
    key=key_scan()
    if key==1 and brightness<30 :
        brightness+=3
    elif key==2 and brightness>3 :
        brightness-=3
    elif key==3 and delay<90 :
        delay+=10
    elif key==4 and delay>10 :
        delay-=10
    elif key==5 and mode<2 :
        mode+=1
    elif key==6 and mode>0 :
        mode-=1     

count=0
count1=0
count2=23
#程序入口
while True:
    key_get()
    if count==23 :
        count=0
    if mode==0 : #关灯
        for i in range(rgb_num):
            rgb_led[i]=(0, 0, 0)
            rgb_led.write()
    if mode==1 : #非对称
        temp=0
        i=count
        count+=1
        r=8*brightness #红色渐变到绿色
        g=0
        while temp<8 :
            r-=brightness
            g+=brightness
            if i>23 :
                i-=24
            rgb_led[i]=(r, g, 0)
            i+=1
            temp+=1
        temp=0
        g=8*brightness #绿色渐变到蓝色
        b=0
        while temp<8 :
            g-=brightness
            b+=brightness
            if i>23 :
                i-=24
            rgb_led[i]=(0, g, b)
            i+=1
            temp+=1
        temp=0
        b=8*brightness #蓝色渐变到红色
        r=0
        while temp<8 :
            b-=brightness
            r+=brightness
            if i>23 :
                i-=24
            rgb_led[i]=(r, 0, b)
            i+=1
            temp+=1
        rgb_led.write()    
        time.sleep_ms(delay)
    if count1==12 :
        count1=0
    if count2==11 :
        count2=23     
    if mode==2 : #对称
        temp=0  #前半部分
        i=count1
        count1+=1
        r=8*brightness #红色渐变到绿色
        g=0
        while temp<4 :
            r-=2*brightness
            g+=2*brightness
            if i>11 :
                i-=12
            rgb_led[i]=(r, g, 0)
            i+=1
            temp+=1
        temp=0
        g=8*brightness #绿色渐变到蓝色
        b=0
        while temp<4 :
            g-=2*brightness
            b+=2*brightness
            if i>11 :
                i-=12
            rgb_led[i]=(0, g, b)
            i+=1
            temp+=1
        temp=0
        b=8*brightness #蓝色渐变到红色
        r=0
        while temp<4 :
            b-=2*brightness
            r+=2*brightness
            if i>11 :
                i-=12
            rgb_led[i]=(r, 0, b)
            i+=1
            temp+=1
        temp=0  #后半部分
        i=count2
        count2-=1
        r=8*brightness #红色渐变到绿色
        g=0
        while temp<4 :
            r-=2*brightness
            g+=2*brightness
            if i<12 :
                i+=12
            rgb_led[i]=(r, g, 0)
            i-=1
            temp+=1
        temp=0
        g=8*brightness #绿色渐变到蓝色
        b=0
        while temp<4 :
            g-=2*brightness
            b+=2*brightness
            if i<12 :
                i+=12
            rgb_led[i]=(0, g, b)
            i-=1
            temp+=1
        temp=0
        b=8*brightness #蓝色渐变到红色
        r=0
        while temp<4 :
            b-=2*brightness
            r+=2*brightness
            if i<12 :
                i+=12
            rgb_led[i]=(r, 0, b)
            i-=1
            temp+=1    
        rgb_led.write()    
        time.sleep_ms(delay)