[tool] spi ws2812

tools/orangepi_spi_ws2812.py

@@ -1,16 +1,31 @@
-import time
-import spidev
 import wiringpi
+import argparse
+import time
+ 
+parser = argparse.ArgumentParser(description='')
+parser.add_argument("--channel", type=int, default=1, help='specify the spi channel')
+parser.add_argument("--port", type=int, default=0, help='specify the spi port')
+parser.add_argument("--speed", type=int, default=6400000, help='specify the spi speed')
+parser.add_argument("--mode", type=int, default=0, help='specify the spi mode')
+args = parser.parse_args()
  
 # 0码 t0H 220ns~380ns
 #（python 中GPIO口无法翻转这么快，python 可能执行一行代码需要800ns）
 # ws2812 波特率如果设置为800kHZ, 1/0.8M=1.25us (0码或者1码所需要的时间)
  
+# 方案1.
 # 使用SPI产生ws2812时序，产生1码和0码，则可以使用传输一个byte 来代表一个0码或者1码，如下：
 # 1111 1000 0xF8  #1码 高电平时间长，低电平时间短
 # 1100 0000 0xC0  #0码 高电平时间短，低电平时间长
 # 则得出SPI 波特率应该设置为800K*8
  
+# 方案2. 见led_bak.py
+# 使用SPI产生ws2812时序，产生1码和0码，则可以使用传输一个半个byte 来代表一个0码或者1码，如下：
+# 1100 0x0c #1码 高电平时间长，低电平时间短
+# 1000 0x08 #0码 高电平时间短，低电平时间长
+# 则得出SPI 波特率应该设置为800K*4
+ 
+ 
 sig_1 = 0xf8
 sig_0 = 0xc0
  
@@ -20,23 +35,22 @@ def flatten_arrays(arrays):
  
  
 class WS2812:
-    def __init__(self, led_num=1):
+    def __init__(self, led_num=10):
         self.led_num = led_num
         self.ws2812_data = [[sig_0 for _ in range(24)] for _ in range(led_num)]
-        spi = spidev.SpiDev()
+        wiringpi.wiringPiSPISetupMode(args.channel, args.port, args.speed, args.mode)
  
         # 设置一下下拉，否则第一盏灯的时序可能不正确
         wiringpi.wiringPiSetup()
         wiringpi.pullUpDnControl(11, 1)
-        spi.open(0, 0)
-        spi.max_speed_hz = 6400000
-        spi.mode = 0
  
-        self.spi = spi
+        print("ws2812_data len", len(self.ws2812_data))
+        print("spi mode: 0x%x" % args.mode)
+        print("max speed: %d Hz (%d KHz)\n" % (args.speed, args.speed / 1000), end='')
  
     def ws2812_send_data(self):
         default_tx = self.ws2812_data
-        self.spi.xfer2(bytes(flatten_arrays(default_tx)))
+        wiringpi.wiringPiSPIDataRW(args.channel, bytes(flatten_arrays(default_tx)))
  
     def ws2812_light_led(self, red, green, blue, pix_led):
         default_tx = self.ws2812_data
@@ -83,16 +97,37 @@ class WS2812:
  
         self.ws2812_send_data()
 
+    def ws2812_gradient_run(self, start_red, start_green, start_blue, end_red, end_green, end_blue, steps, delay):
+	    for step in range(steps):
+	        # 计算当前步的颜色
+	        curr_red = start_red + (end_red - start_red) * step // steps
+	        curr_green = start_green + (end_green - start_green) * step // steps
+	        curr_blue = start_blue + (end_blue - start_blue) * step // steps
+	        
+	        # 清除当前颜色并应用新的渐变色
+	        self.ws2812_shutoff_all()
+	        for i in range(self.led_num):
+	            self.ws2812_light_led(curr_red, curr_green, curr_blue, (i + step) % self.led_num)  # 循环显示渐变色
+	        
+	        self.ws2812_send_data()
+	        time.sleep(delay)
  
 if __name__ == '__main__':
     ws2812 = WS2812(8)
  
     while True:
+        # 示例：从蓝色渐变到红色，再从红色渐变回蓝色，每种颜色渐变10步，每步间隔0.1秒
+        ws2812.ws2812_gradient_run(0, 0, 0xff, 0xff, 0, 0, 10, 0.02)
+        time.sleep(1)  # 等待1秒后开始反向渐变
+        ws2812.ws2812_gradient_run(0xff, 0, 0, 0, 0, 0xff, 10, 0.02)
+        time.sleep(1)  # 再次等待1秒，形成完整的循环
         ws2812.ws2812_light_all_led(0, 0, 0xff)
         time.sleep(0.5)
         ws2812.ws2812_light_all_led(0xff, 0, 0)
         time.sleep(0.5)
         ws2812.ws2812_light_all_led(0, 0xff, 0)
         time.sleep(0.5)
-        # ws2812.ws2812_water_lamp(0xff, 0, 0, 0.2)
+        ws2812.ws2812_shutoff_all()
-        # time.sleep(0.5)
+        time.sleep(0.5)
+        ws2812.ws2812_water_lamp(0xff, 0, 0, 0.2)
+        time.sleep(0.5)