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.interpolate import CubicSpline
from scipy.fft import fft, ifft, fftfreq from scipy.fft import fft, ifft, fftfreq
class Analyzer: class Analyzer:
""" """
Utility class to study backscattering light in VIRGO 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) (factor_n, coupling_n, factor_d, coupling_d)
) )
)
frequencies = factor_n[0] frequencies = factor_n[0]
@ -284,8 +287,8 @@ class Analyzer:
Cut higher frequencies than cutoff for a given temporal signal Cut higher frequencies than cutoff for a given temporal signal
""" """
sample_spacing = signal[0, 1] - signal[0, 0] 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) frequencies = fftfreq(signal[1].shape[0], sample_spacing)
index_to_remove = where(abs(frequencies) > cutoff) index_to_remove = where(abs(frequencies) > cutoff)
freq_signal[index_to_remove] = 0 freq_signal[index_to_remove] = 0
return array([signal[0], real(ifft(freq_signal))]) return array([signal[0], real(array(ifft(freq_signal)))])