155 lines
6.0 KiB
Python
Executable File
155 lines
6.0 KiB
Python
Executable File
#!/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()
|
|
|
|
|
|
print("{:>10} {:>8} {:>8}".format("frame", "AVG", "MAX"))
|
|
# Hauptschleife
|
|
f = 0 # frame counter
|
|
clear_leds = []
|
|
while True:
|
|
for strip, led in clear_leds:
|
|
# Clear leds
|
|
s.set_color(strip, led, 0,0,0,0)
|
|
clear_leds = []
|
|
|
|
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']]))
|
|
clear_leds.append((strip, led))
|
|
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']]))
|
|
clear_leds.append((strip, led))
|
|
|
|
# Add tail of speed-depending length and logarithmicly decaying brightness
|
|
tail = r['speed'] * 10
|
|
if tail > 0:
|
|
tail_range = range(1, int(tail))
|
|
else:
|
|
tail_range = range(-1, int(tail), -1)
|
|
for i in tail_range:
|
|
if not 0 <= int(r['loc'])-i < MAXLOC:
|
|
continue
|
|
strip, led = loc2sl(int(r['loc'])-i)
|
|
color = tuple([int(c / (abs(i)*(abs(tail)/2)) ) for c in r['color'][:3]]+[0]) # blackout white
|
|
s.add_color(strip, led, *color)
|
|
clear_leds.append((strip, led))
|
|
# last tail segment is interpolated
|
|
if tail_range:
|
|
if not 0 <= int(r['loc'])-i < MAXLOC:
|
|
continue
|
|
last_loc = r['loc']-i
|
|
strip, led = loc2sl(math.ceil(last_loc))
|
|
s.add_color(strip, led, *tuple([int(c*(last_loc-int(last_loc))) for c in color]))
|
|
clear_leds.append((strip, led))
|
|
|
|
# Random speed variation, limited to [-2, 2] interval
|
|
r['speed'] = max(-2, min(2, r['speed'] + (random.random()-0.5)*0.01))
|
|
|
|
# Use speed to define W-intensity ("whiteness")
|
|
r['color'][3] = min(int(abs(r['speed'])*128), 255)
|
|
|
|
if f % 600 == 0:
|
|
print("{:>10} {:>8.2f} {:>8.2f}".format(
|
|
f,
|
|
sum([abs(r['speed']) for r in racers])/len(racers),
|
|
max([abs(r['speed']) for r in racers])))
|
|
|
|
s.commit()
|
|
|
|
# Warte bis zum nächsten Frame
|
|
time.sleep(interval)
|
|
f += 1
|