17313c254b
* Moved all module settings to a new config.py file * Completely overhauled visualize.py and added a new radiate effect that colours the radiative beats according the beat frequency. * Improved some constants like the decay constant to be parametric so that they scale to any led strip size * Added temporal dithering to Beat.update_pixels() so that it now supports fractional speed values. Being limited to integral values was starting to become a problem. * Overhauled and simplified the LED module. * When updating pixels, the LED module no longer sends UDP packets for pixels that have not changed. This optimization reduces the packet load significantly and should allow for higher refresh rates. * Renamed lookup_table.npy to gamm_table.npy to better reflect that the table is used for gamma correction of the LED strip
94 lines
2.9 KiB
Python
94 lines
2.9 KiB
Python
from __future__ import print_function
|
|
import time
|
|
import socket
|
|
import numpy as np
|
|
import config
|
|
|
|
_sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
|
_gamma = np.load('gamma_table.npy')
|
|
_prev_pixels = np.tile(0, (config.N_PIXELS, 3))
|
|
|
|
pixels = np.tile(0, (config.N_PIXELS, 3))
|
|
"""Array containing the pixel values for the LED strip"""
|
|
|
|
|
|
def update():
|
|
global pixels, _prev_pixels
|
|
pixels = np.clip(pixels, 0, 255)
|
|
m = ''
|
|
for i in range(config.N_PIXELS):
|
|
# Ignore pixels if they haven't changed (saves bandwidth)
|
|
if np.array_equal(pixels[i], _prev_pixels[i]):
|
|
continue
|
|
r = _gamma[pixels[i][0]] if config.GAMMA_CORRECTION else pixels[i][0]
|
|
g = _gamma[pixels[i][1]] if config.GAMMA_CORRECTION else pixels[i][1]
|
|
b = _gamma[pixels[i][2]] if config.GAMMA_CORRECTION else pixels[i][2]
|
|
m += chr(i) + chr(r) + chr(g) + chr(b)
|
|
_prev_pixels = pixels
|
|
_sock.sendto(m, (config.UDP_IP, config.UDP_PORT))
|
|
|
|
|
|
|
|
# def set_all(R, G, B):
|
|
# for i in range(config.N_PIXELS):
|
|
# set_pixel(i, R, G, B)
|
|
# update_pixels()
|
|
|
|
|
|
# def autocolor(x, speed=1.0):
|
|
# dt = 2.0 * np.pi / config.N_PIXELS
|
|
# t = time.time() * speed
|
|
# def r(t): return (np.sin(t + 0.0) + 1.0) * 1.0 / 2.0
|
|
# def g(t): return (np.sin(t + (2.0 / 3.0) * np.pi) + 1.0) * 1.0 / 2.0
|
|
# def b(t): return (np.sin(t + (4.0 / 3.0) * np.pi) + 1.0) * 1.0 / 2.0
|
|
# for n in range(config.N_PIXELS):
|
|
# set_pixel(N=n,
|
|
# R=r(n * dt + t) * x[n],
|
|
# G=g(n * dt + t) * x[n],
|
|
# B=b(n * dt + t) * x[n],
|
|
# gamma_correction=True)
|
|
# update_pixels()
|
|
|
|
|
|
# def set_pixel(N, R, G, B, gamma_correction=True):
|
|
# global _m
|
|
# r = int(min(max(R, 0), 255))
|
|
# g = int(min(max(G, 0), 255))
|
|
# b = int(min(max(B, 0), 255))
|
|
# if gamma_correction:
|
|
# r = _gamma_table[r]
|
|
# g = _gamma_table[g]
|
|
# b = _gamma_table[b]
|
|
# if _m is None:
|
|
# _m = chr(N) + chr(r) + chr(g) + chr(b)
|
|
# else:
|
|
# _m += chr(N) + chr(r) + chr(g) + chr(b)
|
|
|
|
|
|
# def update_pixels():
|
|
# global _m
|
|
# _sock.sendto(_m, (config.UDP_IP, config.UDP_PORT))
|
|
# _m = None
|
|
|
|
|
|
# def rainbow(brightness=255.0, speed=1.0, fps=10):
|
|
# offset = 132
|
|
# dt = 2.0 * np.pi / config.N_PIXELS
|
|
# def r(t): return (np.sin(t + 0.0) + 1.0) * brightness / 2.0 + offset
|
|
# def g(t): return (np.sin(t + (2.0 / 3.0) * np.pi) + 1.0) * brightness / 2.0 + offset
|
|
# def b(t): return (np.sin(t + (4.0 / 3.0) * np.pi) + 1.0) * brightness / 2.0 + offset
|
|
# while True:
|
|
# t = time.time() * speed
|
|
# for n in range(config.N_PIXELS):
|
|
# T = t + n * dt
|
|
# set_pixel(N=n, R=r(T), G=g(T), B=b(T))
|
|
# update_pixels()
|
|
# time.sleep(1.0 / fps)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
while True:
|
|
update()
|
|
#set_all(0, 0, 0)
|
|
# rainbow(speed=0.025, fps=40, brightness=0)
|