dedebenui/scgenerator
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

reverted some noise changes

Browse Source
This commit is contained in:
Benoît Sierro
2024-01-11 16:25:46 +01:00
parent 7c7e7b7533
commit 9b2907b1cb
2 changed files with 82 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

54
examples/noise_study.py
View File

@@ -5,34 +5,54 @@ import scgenerator as sc
def main():
nperseg = 513
nf = 513
nseg = 240
ntot = (nperseg - 1) * (nseg - 1)
ntot = (nf - 1) * (nseg - 1)
fs = 44100
nf = 45611
f = np.linspace(0, fs // 2, nf)
nf_full = 45611
f = np.linspace(0, fs // 2, nf_full)
spec = 1 / (f + 1)
spec += np.exp(-(((f - (0.2 * fs)) / (0.5 * fs)) ** 2))
plt.plot(f, spec)
plt.xscale("log")
plt.yscale("log")
plt.show()
spec += np.exp(-(((f - (0.1 * fs)) / (0.5 * fs)) ** 2))
spec += np.exp(-(((f - (0.45 * fs)) / (0.02 * fs)) ** 2))
# spec = np.ones_like(f) * 1e-5
# spec += np.exp(-(((f - 15000) / 300) ** 2))
noise = sc.noise.NoiseMeasurement(f, spec)
t, y = noise.time_series(nf=nperseg, nseg=nseg)
# tf, yf = noise.time_series(nperseg=(ntot // 2) + 1, nseg=1)
t, y = noise.time_series(nf=nf, nseg=nseg)
tf, yf = noise.time_series(nf=ntot // 2 + 1)
print(y.std(), yf.std())
plt.plot(t, y, ls="", marker=".")
plt.plot(tf, yf, ls="", marker=".")
plt.show()
newnoise = noise.from_time_series(y, t[1] - t[0], nperseg=(nperseg - 1) * 2)
newnoise = noise.from_time_series(y, t[1] - t[0], nf=nf)
newnoise_full = noise.from_time_series(yf, tf[1] - tf[0], nperseg=(nf - 1) * 2)
print(newnoise.psd[1:-1].var())
print(newnoise_full.psd[1:-1].var())
fig, ax = plt.subplots()
ax.plot(*noise.plottable())
ax.plot(*newnoise.plottable())
ax.plot(*newnoise_full.plottable(discard_nyquist=True))
ax.set_xscale("log")
axin = plt.gca().inset_axes(
[0.1, 0.1, 0.3, 0.3],
xlim=(1.5e4, fs / 1.9),
ylim=(-3.1, 2.4),
yticklabels=[],
)
# axin.set_xscale("log")
axin.plot(*newnoise.plottable())
axin.plot(*newnoise_full.plottable())
axin.plot(*noise.plottable())
ax.indicate_inset_zoom(axin)
axin.tick_params(labelbottom=False)
plt.plot(*noise.plottable())
f, sxx = noise.sample_spectrum(nperseg)
plt.plot(f, 10 * np.log10(sxx))
plt.plot(*newnoise.plottable())
plt.xscale("log")
plt.show()