def plot_das(res: xr.Dataset,
ax: Axis,
title: str = "DAS",
cycler: Cycler | None = PlotStyle().cycler) -> None:
"""Plot DAS (Decay Associated Spectra) on ``ax``.
Parameters
----------
res : xr.Dataset
Result dataset
ax : Axis
Axis to plot on.
title : str
Title of the plot. Defaults to "DAS".
cycler : Cycler | None
Plot style cycler to use. Defaults to PlotStyle().cycler.
"""
add_cycler_if_not_none(ax, cycler)
keys = [
v for v in res.data_vars
if v.startswith(("decay_associated_spectra", "species_spectra"))
]
for key in keys:
das = res[key]
das.plot.line(x="spectral", ax=ax)
ax.set_title(title)
ax.get_legend().remove()
def plot_norm_sas(res: xr.Dataset,
ax: Axis,
title: str = "norm SAS",
cycler: Cycler | None = PlotStyle().cycler) -> None:
"""Plot normalized SAS (Species Associated Spectra) on ``ax``.
Parameters
----------
res : xr.Dataset
Result dataset
ax : Axis
Axis to plot on.
title : str
Title of the plot. Defaults to "norm SAS".
cycler : Cycler | None
Plot style cycler to use. Defaults to PlotStyle().cycler.
"""
add_cycler_if_not_none(ax, cycler)
keys = [
v for v in res.data_vars
if v.startswith(("species_associated_spectra", "species_spectra"))
]
for key in keys:
sas = res[key]
# sas = res.species_associated_spectra
(sas / np.abs(sas).max(dim="spectral")).plot.line(x="spectral", ax=ax)
ax.set_title(title)
ax.get_legend().remove()