sk6812-client/rgbw_racers.py

#!/usr/bin/env python
# encoding: utf-8

# Copyright (c) 2022 Julian Hammer
# This file is licensed under the terms of the MIT license. See COPYING for
# details.

# Importieren von sk6812_multistrip.py, welches das Netzwerkprotokoll für die
# LED-Leiste abstrahiert.
import sk6812_multistrip as sk6812

import time    # für sleep()
import sys     # zum Lesen der IP-Adresse aus den Kommandozeilenargumenten

import math
import random
import colorsys

# Konstanten
NLED   = 150     # Anzahl der LEDs pro logischen Strip
NSTRIP = 2       # Anzahl der logischen Strips

# Objekt initialisieren
s = sk6812.SK6812(sys.argv[1], 2703)

# Wartezeit zwischen den Animationsschritten. Achtung, die LED-Leiste arbeitet
# immer mit 60 FPS! Dies ist insbesondere bei Verwendung von fade_color()
# relevant.
interval = 1.0/60

last_frame = [(0,0,0,0)] * NLED * NSTRIP

def hsv2rgb(h,s,v):
    return tuple(round(i * 255) for i in colorsys.hsv_to_rgb(h,s,v))

def random_color():
    return list(hsv2rgb(random.random(), 1, 1)) + [random.randint(0,255)]

def loc2sl(loc):
    # translates 1d-coordinate (loc) to
    # (strip, led) with:
    # (0,0) ... (0, 149) (1, 149) ... (1, 0)
    strip = loc // NLED
    led = loc % NLED
    if strip == 1:
        led = NLED - led
    return strip, led

MAXLOC = NLED*NSTRIP

racers = [
    dict(loc=random.randint(0,299), color=random_color(), speed=(random.random()-0.5)*0.2),
    dict(loc=random.randint(0,299), color=random_color(), speed=(random.random()-0.5)*0.2),
    dict(loc=random.randint(0,299), color=random_color(), speed=(random.random()-0.5)*0.2),
    dict(loc=random.randint(0,299), color=random_color(), speed=(random.random()-0.5)*0.2),
    dict(loc=random.randint(0,299), color=random_color(), speed=(random.random()-0.5)*0.2),
    dict(loc=random.randint(0,299), color=random_color(), speed=(random.random()-0.5)*0.2),
    dict(loc=random.randint(0,299), color=random_color(), speed=(random.random()-0.5)*0.2),
    dict(loc=random.randint(0,299), color=random_color(), speed=(random.random()-0.5)*0.2),
    dict(loc=random.randint(0,299), color=random_color(), speed=(random.random()-0.5)*0.2),
    dict(loc=random.randint(0,299), color=random_color(), speed=(random.random()-0.5)*0.2),
    dict(loc=random.randint(0,299), color=random_color(), speed=(random.random()-0.5)*0.3),
    dict(loc=random.randint(0,299), color=random_color(), speed=(random.random()-0.5)*0.3),
    dict(loc=random.randint(0,299), color=random_color(), speed=(random.random()-0.5)*0.3),
    dict(loc=random.randint(0,299), color=random_color(), speed=(random.random()-0.5)*0.3),
    dict(loc=random.randint(0,299), color=random_color(), speed=(random.random()-0.5)*0.3),
    dict(loc=random.randint(0,299), color=random_color(), speed=(random.random()-0.5)*0.3),
    dict(loc=random.randint(0,299), color=random_color(), speed=(random.random()-0.5)*0.3),
    dict(loc=random.randint(0,299), color=random_color(), speed=(random.random()-0.5)*0.3),
    dict(loc=random.randint(0,299), color=random_color(), speed=(random.random()-0.5)*0.5),
    dict(loc=random.randint(0,299), color=random_color(), speed=(random.random()-0.5)*0.5),
    dict(loc=random.randint(0,299), color=random_color(), speed=(random.random()-0.5)*0.5),
    dict(loc=random.randint(0,299), color=random_color(), speed=(random.random()-0.5)*0.5),
    dict(loc=random.randint(0,299), color=random_color(), speed=(random.random()-0.5)*2),
    dict(loc=random.randint(0,299), color=random_color(), speed=(random.random()-0.5)*2),
]

# Clear all leds at start
for loc in range(MAXLOC):
    strip, led = loc2sl(loc)
    s.set_color(strip, led, 0,0,0,0)
    if loc % 150 == 0:
        s.commit()
s.commit()

# Hauptschleife
f = 0  # frame counter
while True:
    for r in racers:
        # Clear leds
        strip, led = loc2sl(int(r['loc']))
        s.set_color(strip, led, 0,0,0,0)
        strip, led = loc2sl(math.ceil(r['loc']))
        s.set_color(strip, led, 0,0,0,0)
        
    for r in racers:
        # Update location
        r['loc'] += r['speed']
        # Reverse at ends (0 and MAXLOC-1)
        if r['loc'] >= MAXLOC:
            r['loc'] = MAXLOC - (r['loc'] % MAXLOC)
            r['speed'] *= -1
        if r['loc'] < 0:
            r['loc'] *= -1
            r['speed'] *= -1

        # Interpolate color linearly between two leds
        strip, led = loc2sl(int(r['loc']))
        s.add_color(strip, led, *tuple([int(c*(math.ceil(r['loc'])-r['loc'])) for c in r['color']]))
        strip, led = loc2sl(math.ceil(r['loc']))
        s.add_color(strip, led, *tuple([int(c*(r['loc']-int(r['loc']))) for c in r['color']]))

        # Random speed variation
        r['speed'] += (random.random()-0.5)*0.01

        # Use speed to define W-intensity ("whiteness")
        r['color'][3] = min(int(abs(r['speed'])*200), 255)

    #print(max([abs(r['speed']) for r in racers]), min([abs(r['speed']) for r in racers]))
    s.commit()
    # Warte bis zum nächsten Frame
    time.sleep(interval)
    f += 1