import   pyb
import   mathfrom   ws2812 import WS2812ring   = WS2812(spi_bus=1, led_count=8, intensity=0.1)def   data_generator(led_count):data = [(0, 0, 0) for i in   range(led_count)]step = 0while True:red = int((1 + math.sin(step *   0.1324)) * 127)green = int((1 + math.sin(step *   0.1654)) * 127)blue = int((1 + math.sin(step * 0.1))   * 127)data[step % led_count] = (red, green,   blue)yield datastep += 1for   data in data_generator(ring.led_count):ring.show(data)pyb.delay(100)

import   gc
import   pybclass   WS2812:"""Driver for WS2812 RGB LEDs. May be used   for controlling single LED or chainof LEDs.Example of use:chain = WS2812(spi_bus=1,   led_count=4)data = [(255, 0, 0),    # red(0, 255, 0),    # green(0, 0, 255),    # blue(85, 85, 85),   # white]chain.show(data)Version: 1.0"""buf_bytes = (0x11, 0x13, 0x31, 0x33)def __init__(self, spi_bus=1,   led_count=1, intensity=1):"""Params:* spi_bus = SPI bus ID (1 or 2)* led_count = count of LEDs* intensity = light intensity (float   up to 1)"""self.led_count = led_countself.intensity = intensity# prepare SPI data buffer (4 bytes   for each color)self.buf_length = self.led_count * 3   * 4self.buf = bytearray(self.buf_length)# SPI initself.spi = pyb.SPI(spi_bus,   pyb.SPI.MASTER, baudrate=3200000, polarity=0, phase=1)# turn LEDs offself.show([])def show(self, data):"""Show RGB data on LEDs. Expected data   = [(R, G, B), ...] where R, G and Bare intensities of colors in range   from 0 to 255. One RGB tuple for eachLED. Count of tuples may be less than   count of connected LEDs."""self.fill_buf(data)self.send_buf()def send_buf(self):"""Send buffer over SPI."""self.spi.send(self.buf)gc.collect()def update_buf(self, data, start=0):"""Fill a part of the buffer with RGB   data.Order of colors in buffer is changed   from RGB to GRB because WS2812 LEDhas GRB order of colors. Each color   is represented by 4 bytes in buffer(1 byte for each 2 bits).Returns the index of the first unfilled   LEDNote: If you find this function ugly,   it's because speed optimisationsbeated purity of code."""buf = self.bufbuf_bytes = self.buf_bytesintensity = self.intensitymask = 0x03index = start * 12for red, green, blue in data:red = int(red * intensity)green = int(green * intensity)blue = int(blue * intensity)buf[index] = buf_bytes[green   >> 6 & mask]buf[index+1] = buf_bytes[green   >> 4 & mask]buf[index+2] = buf_bytes[green   >> 2 & mask]buf[index+3] = buf_bytes[green   & mask]buf[index+4] = buf_bytes[red   >> 6 & mask]buf[index+5] = buf_bytes[red   >> 4 & mask]buf[index+6] = buf_bytes[red   >> 2 & mask]buf[index+7] = buf_bytes[red   & mask]buf[index+8] = buf_bytes[blue   >> 6 & mask]buf[index+9] = buf_bytes[blue   >> 4 & mask]buf[index+10] = buf_bytes[blue   >> 2 & mask]buf[index+11] = buf_bytes[blue   & mask]index += 12return index // 12def fill_buf(self, data):"""Fill buffer with RGB data.All LEDs after the data are turned   off."""end = self.update_buf(data)# turn off the rest of the LEDsbuf = self.bufoff = self.buf_bytes[0]for index in range(end * 12,   self.buf_length):buf[index] = offindex += 1