fft shift spectrogram

2023-12-12 12:27:52 +01:00
parent f33fbc4127
commit 1f60aaa65c
1 changed files with 3 additions and 1 deletions
--- a/src/scgenerator/spectra.py
+++ b/src/scgenerator/spectra.py
@@ -162,7 +162,9 @@ class Spectrum(np.ndarray):
        return pulse.g12(self, axis)[..., self.wl_order]

    def spectrogram(self, delays: Iterable[float], gate_width: float = 2e-13) -> np.ndarray:
-        return pulse.spectrogram(self.t, delays, self.time_amp, gate_width)
+        return np.fft.fftshift(
+            pulse.spectrogram(self.t, delays, self.time_amp, gate_width), axes=-1
+        )

    freq_int = afreq_int
    freq_amp = afreq_amp