Added rgbw_racers.py written by Julian
This commit is contained in:
Thomas Kolb
parent
8ae13d77e9
commit
f6cfbb12f3
|
|
@ -0,0 +1,123 @@
|
|||
#!/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
|
||||
Loading…
Reference in New Issue