From 1f60aaa65c17f9510b8912bbf5b6753a2823cb6e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Beno=C3=AEt=20Sierro?= Date: Tue, 12 Dec 2023 12:27:52 +0100 Subject: [PATCH] fft shift spectrogram --- src/scgenerator/spectra.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/scgenerator/spectra.py b/src/scgenerator/spectra.py index ef9888d..78581a6 100644 --- 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