example_240_leds.py

# -*- coding: utf-8 -*-

"""
This example shows some animations on 4 meters of NeoPixels strip with 240 LEDs.
"""


import pyb
import math

from ws2812 import WS2812


def color_gen(i=0):
    while True:
        i += 1
        red = (1 + math.sin(i * 0.1)) * 127 + 1
        green = (1 + math.sin(i * 0.1324)) * 127 + 1
        blue = (1 + math.sin(i * 0.1654)) * 127 + 1

        total = red + green + blue
        red = int(red / total * 255)
        green = int(green / total * 255)
        blue = int(blue / total * 255)

        yield red, green, blue

colors = color_gen()


def animation_1(led_count):
    data = [(0, 0, 0) for i in range(led_count)]
    step = 0
    while True:
        data[step % led_count] = next(colors)
        yield data
        step += 1


def animation_2(led_count, offset=3, length=1):
    data = [(0, 0, 0) for i in range(led_count)]
    offsets = range(0, led_count, offset)
    step = 0
    while True:
        pos = step % led_count
        rgb = next(colors)
        for off in offsets:
            data[pos - off] = rgb
            data[pos - off - length] = (0, 0, 0)
        yield data
        step += 1


def animation_3(led_count, offset=10):
    data = [(0, 0, 0) for i in range(led_count)]
    offsets = range(0, led_count, offset)
    step = 0
    while True:
        pos = step % led_count
        rgb = next(colors)
        for off in offsets:
            data[pos - off] = rgb
        yield data
        step += 1


stripe = WS2812(spi_bus=1, led_count=240, intensity=0.05)

anim_2 = animation_2(stripe.led_count)
anim_3 = animation_3(stripe.led_count)
anim_4 = animation_2(stripe.led_count, 15, 5)

while True:
    anim_1 = animation_1(stripe.led_count)
    for i in range(240):
        stripe.show(next(anim_1))

    for i in range(120):
        stripe.show(next(anim_2))
        pyb.delay(50)

    for i in range(240):
        stripe.show(next(anim_3))

    for i in range(240):
        stripe.show(next(anim_4))