linarphy/backscattering-analyzer
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix lsp server error by putting array

Browse source
This commit is contained in:
linarphy 2024-05-17 10:49:21 +02:00
parent 9b34f5b6b9
commit d1e649d61e
No known key found for this signature in database
GPG key ID: E61920135EFF2295
1 changed files with 9 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
src/backscattering_analyzer/analyzer.py
View file

@ -15,6 +15,7 @@ from scipy.signal import welch as psd
from scipy.interpolate import CubicSpline
from scipy.fft import fft, ifft, fftfreq
class Analyzer:
"""
Utility class to study backscattering light in VIRGO
@ -257,9 +258,11 @@ class Analyzer:
]
)
factor_n, coupling_n, factor_d, coupling_d = self.interpolate(
factor_n, coupling_n, factor_d, coupling_d = (
self.interpolate(
(factor_n, coupling_n, factor_d, coupling_d)
)
)
frequencies = factor_n[0]
@ -284,8 +287,8 @@ class Analyzer:
Cut higher frequencies than cutoff for a given temporal signal
"""
sample_spacing = signal[0, 1] - signal[0, 0]
freq_signal = fft(signal[1])
freq_signal = array(fft(signal[1]))
frequencies = fftfreq(signal[1].shape[0], sample_spacing)
index_to_remove = where(abs(frequencies) > cutoff)
freq_signal[index_to_remove] = 0
return array([signal[0], real(ifft(freq_signal))])
return array([signal[0], real(array(ifft(freq_signal)))])