brmlab/ledbar
4
0
Fork 0
mirror of https://github.com/brmlab/ledbar.git synced 2025-06-09 13:24:01 +02:00
Code Issues Projects Releases Packages Wiki Activity

Update host_python/equalizer.py

Browse source 
fix range of autocorrelation
This commit is contained in:
pborky 2012-12-22 23:16:20 +01:00
parent 88fcfc09a8
commit a6e28d711b
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
host_python/equalizer.py
View file

@ -157,8 +157,8 @@ def loop( stream ):
#TODO: dynamic attenuation based on average power of input signal over timespan, threshold to cut off noise
x = np.concatenate(history*history_diminish)*window/ATTENUATION
# compute power spectral desity using autocarelate approach
psd = np.abs(np.fft.fft(np.correlate(x,x,'full')))[...,np.newaxis]
# estimate power spectral desity using autocorelate approach
psd = np.abs(np.fft.fft(np.correlate(x,x,'same')))[...,np.newaxis]
# frequencies
freqs = np.fft.fftfreq(psd.shape[0],1./RATE)[...,np.newaxis]
# frequency band vector _orthogonal_ to freqs
@ -175,7 +175,7 @@ def loop( stream ):
))
for pixel in xrange(EPIXELS):
c = get_color(energy[pixel])
c = get_color(energy[pixel]) # consider using energy**0.5 instead
if SYMMETRIC == 1:
l.set_pixel(PIXELS / 2 + pixel, c[0], c[1], c[2])
l.set_pixel(PIXELS / 2 - (pixel + 1), c[0], c[1], c[2])