led_bak.py

import wiringpi
import argparse
import time

parser = argparse.ArgumentParser(description='')
parser.add_argument("--channel", type=int, default=4, help='specify the spi channel')
parser.add_argument("--port", type=int, default=1, help='specify the spi port')
parser.add_argument("--speed", type=int, default=3200000, help='specify the spi speed')
parser.add_argument("--mode", type=int, default=0, help='specify the spi mode')
args = parser.parse_args()

# 1111 1000 F8
# 1100 0000 C0
# sig_1 = 0xf8
# sig_0 = 0xc0

# 1: 1100
# 0: 1000
sig_1 = 0x0C
sig_0 = 0x08


def flatten_arrays(arrays):
    return [element for sublist in arrays for element in sublist]


class WS2812:
    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)]
        wiringpi.wiringPiSPISetupMode(args.channel, args.port, args.speed, args.mode)

        wiringpi.wiringPiSetup()
        wiringpi.pullUpDnControl(6, 1)
        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
        wiringpi.wiringPiSPIDataRW(args.channel, bytes(flatten_arrays(default_tx)))

    def ws2812_light_led(self, red, green, blue, pix_led):

        default_tx = self.ws2812_data
        color = green << 16 | red << 8 | blue
        for i in range(12):
            if color >> (24 - i * 2 - 1) & 1:
                tmp = sig_1 << 4
            else:
                tmp = sig_0 << 4

            default_tx[pix_led][i*2] = tmp

            if color >> (24 - i * 2 - 2) & 1:
                tmp = sig_1
            else:
                tmp = sig_0

            default_tx[pix_led][i*2+1] = tmp

    def ws2812_light_one_led(self, red, green, blue, pix_led):
        self.ws2812_light_led(red, green, blue, pix_led)
        self.ws2812_send_data()

    def ws2812_light_all_led(self, red, green, blue):
        for i in range(self.led_num):
            self.ws2812_light_led(red, green, blue, i)

        self.ws2812_send_data()

    def ws2812_rainbow(self):
        colors = [[0xff, 0, 0], [0, 0xff, 0], [0, 0, 0xff]]
        for i in range(self.led_num):
            cur_color = colors[i % 3]
            self.ws2812_light_led(cur_color[0], cur_color[1], cur_color[2], i)
        self.ws2812_send_data()

    def ws2812_water_lamp(self, red, green, blue, interval_time):
        self.ws2812_shutoff_all()

        for i in range(self.led_num):
            self.ws2812_light_one_led(red, green, blue, i)
            time.sleep(interval_time)

        self.ws2812_shutoff_all()

    def ws2812_shutoff_led(self, n):
        self.ws2812_light_led(0, 0, 0, n)
        self.ws2812_send_data()

    def ws2812_shutoff_all(self):
        for i in range(self.led_num):
            self.ws2812_light_led(0, 0, 0, i)

        self.ws2812_send_data()


if __name__ == '__main__':
    ws2812 = WS2812(10)

    while True:
        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_shutoff_all()
        time.sleep(0.5)
        ws2812.ws2812_water_lamp(0xff, 0, 0, 0.2)
        time.sleep(0.5)