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

15
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
@ -128,7 +129,7 @@ class Analyzer:
]
)
self.movement[1] -= mean(self.movement[1])
self.movement = self.movement[:,:-500000]
self.movement = self.movement[:, :-500000]
def psd_signal(self, signal, fft_length=10):
"""
@ -257,8 +258,10 @@ class Analyzer:
]
)
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 = (
self.interpolate(
(factor_n, coupling_n, factor_d, coupling_d)
)
)
frequencies = factor_n[0]
@ -283,9 +286,9 @@ 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])
sample_spacing = signal[0, 1] - signal[0, 0]
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)))])