From 91545d39bbd6677f9a605c8291825c9f74169b00 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Beno=C3=AEt=20Sierro?= Date: Mon, 11 Dec 2023 09:34:13 +0100 Subject: [PATCH] added spectrogram method to Spectrum --- src/scgenerator/spectra.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) 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