added spectrogram method to Spectrum

This commit is contained in:
Benoît Sierro
2023-12-11 09:34:13 +01:00
parent a9aef82240
commit 91545d39bb

View File

@@ -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