lightbybeat.py

#!/usr/bin/python3
# -*- coding: utf-8 -*-


import pythonlights
import sound
import numpy as np
import math

ctrl = pythonlights.LEDControl()
listener = sound.Listener()

ones = np.ones(3)
peaks = ones * 50000000/256 # Nur ein Schätzwert
longmean = shortmean = ones * 128 # long zum regeln, short zum glätten

sound.INPUT_BLOCK_TIME = 0.1
a = int(250*sound.INPUT_BLOCK_TIME) # Anzahl der Array Element um 250Hz abdecken

t = 0

while True:
    # Dieser Aufruf blockt, bis genug sample da ist.
    spectrum = listener.get_spectrum()

    # blau 0-250 Hz, grün 250-500 Hz, rot 500-1000 Hz
    colors = np.array([np.sum(spectrum[0:a]),np.sum(spectrum[a:a*2]),np.sum(spectrum[a*2:a*4])]) 

    # Normalisiere auf anzeigbare Intensitäten
    colors /= peaks 
    colors = np.clip(colors,0,255)

    # gleitende Mittelwerte
    shortmean = 0.7 * colors + 0.3 * shortmean
    longmean = 0.01 * colors + 0.99 * longmean

    # Korrektur der Maximalwerte um damit die Intesitäten im Mittel um 50% schwanken.
    peaks *= 0.01*longmean/128 + 0.99 

    # Einstellen und senden der berechneten Farbe.
    color = pythonlights.Color(shortmean.astype(int))
    ctrl.set_position(2, color)
    t = (t + 1) % 100
    ctrl.set_gnome(max(5,int(255*math.cos(2.0*t/200*math.pi)**2)))
    ctrl.send()