diff --git a/src/scgenerator/spectra.py b/src/scgenerator/spectra.py index b517738..f14cf1d 100644 --- a/src/scgenerator/spectra.py +++ b/src/scgenerator/spectra.py @@ -5,7 +5,7 @@ import warnings from dataclasses import dataclass from functools import cached_property from pathlib import Path -from typing import Callable, Generic, Iterator, Sequence, TypeVar, overload +from typing import Callable, Generic, Iterable, Iterator, Sequence, TypeVar, overload import numpy as np @@ -161,6 +161,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) + freq_int = afreq_int freq_amp = afreq_amp