notwa/dsp
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

update 42

Browse source
This commit is contained in:
Connor Olding 2018-03-03 04:04:22 -08:00
parent 3968fbddc4
commit 23f87a1385
1 changed files with 16 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
lib/smoothfft.py
View file

@ -40,13 +40,11 @@ def smoothfft2(xs, ys, bw=1, precision=512, compensate=True):
return xs2, ys2 return xs2, ys2
def smoothfft3(ys, bw=1, precision=512): def smoothfft_setup(size, precision=512, bw=1/6):
"""performs log-lin smoothing on magnitude data""" dotme = np.zeros((size, precision))
size = len(ys)
xs = np.arange(0, 1, 1/size)
xs = np.arange(0, 1, 1/size)
xs2 = np.logspace(-np.log2(size), 0, precision, base=2) xs2 = np.logspace(-np.log2(size), 0, precision, base=2)
ys2 = np.zeros(precision)
comp = np.zeros(precision) comp = np.zeros(precision)
log2_xs2 = np.log2(xs2) log2_xs2 = np.log2(xs2)
@ -54,6 +52,17 @@ def smoothfft3(ys, bw=1, precision=512):
dist = np.abs(log2_xs2 - np.log2(x + 1e-35)) / bw dist = np.abs(log2_xs2 - np.log2(x + 1e-35)) / bw
window = np.exp(-dist**2 * 4) # gaussian (untruncated) window = np.exp(-dist**2 * 4) # gaussian (untruncated)
comp += window comp += window
ys2 += ys[i] * window dotme[i] = window
return xs2, ys2 / comp dotme /= comp
return xs2, dotme
def smoothfft3(ys, bw=1, precision=512, srate=None):
"""performs log-lin smoothing on magnitude data"""
xs2, dotme = smoothfft_setup(len(ys), precision, bw)
if srate is None:
return xs2, ys @ dotme
else:
return xs2 * (srate / 2), ys @ dotme