dedebenui/scgenerator
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fft shift spectrogram

Browse Source
This commit is contained in:
Benoît Sierro
2023-12-12 12:27:52 +01:00
parent f33fbc4127
commit 1f60aaa65c
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/scgenerator/spectra.py
View File

@@ -162,7 +162,9 @@ class Spectrum(np.ndarray):
return pulse.g12(self, axis)[..., self.wl_order] return pulse.g12(self, axis)[..., self.wl_order]
def spectrogram(self, delays: Iterable[float], gate_width: float = 2e-13) -> np.ndarray: 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_int = afreq_int
freq_amp = afreq_amp freq_amp = afreq_amp