def plot_concentrations(
res: xr.Dataset,
ax: Axis,
center_λ: float | None,
linlog: bool = False,
linthresh: float = 1,
linscale: float = 1,
main_irf_nr: int = 0,
cycler: Cycler | None = PlotStyle().cycler,
title: str = "Concentrations",
) -> None:
"""Plot traces on the given axis ``ax``.
Parameters
----------
res: xr.Dataset
Result dataset from a pyglotaran optimization.
ax: Axis
Axis to plot the traces on
center_λ: float | None
Center wavelength (λ in nm)
linlog: bool
Whether to use 'symlog' scale or not. Defaults to False.
linthresh: float
A single float which defines the range (-x, x), within which the plot is linear.
This avoids having the plot go to infinity around zero. Defaults to 1.
linscale: float
This allows the linear range (-linthresh to linthresh) to be stretched
relative to the logarithmic range.
Its value is the number of decades to use for each half of the linear range.
For example, when linscale == 1.0 (the default), the space used for the
positive and negative halves of the linear range will be equal to one
decade in the logarithmic range. Defaults to 1.
main_irf_nr: int
Index of the main ``irf`` component when using an ``irf``
parametrized with multiple peaks. Defaults to 0.
cycler : Cycler | None
Plot style cycler to use. Defaults to PlotStyle().data_cycler_solid.
title: str
Title used for the plot axis. Defaults to "Concentrations".
See Also
--------
get_shifted_traces
"""
add_cycler_if_not_none(ax, cycler)
traces = get_shifted_traces(res, center_λ, main_irf_nr)
if "spectral" in traces.coords:
traces.sel(spectral=center_λ, method="nearest").plot.line(x="time",
ax=ax)
else:
traces.plot.line(x="time", ax=ax)
ax.set_title(title)
if linlog:
ax.set_xscale("symlog", linthresh=linthresh, linscale=linscale)
def plot_residual(
res: xr.Dataset,
ax: Axis,
linlog: bool = False,
linthresh: float = 1,
show_data: bool = False,
cycler: Cycler | None = PlotStyle().cycler,
) -> None:
"""Plot data or residual on a 2D contour plot.
Parameters
----------
res : xr.Dataset
Result dataset
ax : Axis
Axis to plot on.
linlog : bool
Whether to use 'symlog' scale or not. Defaults to False.
linthresh : float
A single float which defines the range (-x, x), within which the plot is linear.
This avoids having the plot go to infinity around zero. Defaults to 1.
show_data : bool
Whether to show the data or the residual. Defaults to False.
cycler : Cycler | None
Plot style cycler to use. Defaults to PlotStyle().cycler.
"""
add_cycler_if_not_none(ax, cycler)
data = res.data if show_data else res.residual
title = "dataset" if show_data else "residual"
shape = np.array(data.shape)
dims = data.coords.dims
# Handle different dimensionality of data
if min(shape) == 1:
data.plot.line(x=dims[shape.argmax()], ax=ax)
elif min(shape) < 5:
data.plot(x="time", ax=ax)
else:
data.plot(x="time", ax=ax, add_colorbar=False)
if linlog:
ax.set_xscale("symlog", linthresh=linthresh)
ax.set_title(title)
def plot_lsv_residual(
res: xr.Dataset,
ax: Axis,
indices: Sequence[int] = range(2),
linlog: bool = False,
linthresh: float = 1,
cycler: Cycler | None = PlotStyle().cycler,
show_legend: bool = True,
) -> None:
"""Plot left singular vectors (time) of the residual matrix.
Parameters
----------
res : xr.Dataset
Result dataset
ax : Axis
Axis to plot on.
indices : Sequence[int]
Indices of the singular vector to plot. Defaults to range(4).
linlog : bool
Whether to use 'symlog' scale or not. Defaults to False.
linthresh : float
A single float which defines the range (-x, x), within which the plot is linear.
This avoids having the plot go to infinity around zero. Defaults to 1.
cycler : Cycler | None
Plot style cycler to use. Defaults to PlotStyle().cycler.
show_legend: bool
Whether or not to show the legend. Defaults to True.
"""
add_cycler_if_not_none(ax, cycler)
if "weighted_residual_left_singular_vectors" in res:
rLSV = res.weighted_residual_left_singular_vectors
else:
rLSV = res.residual_left_singular_vectors
_plot_svd_vetors(rLSV, indices, "left_singular_value_index", ax,
show_legend)
ax.set_title("res. LSV")
if linlog:
ax.set_xscale("symlog", linthresh=linthresh)
def plot_data_and_fits(
result: ResultLike,
wavelength: float,
axis: Axis,
center_λ: float | None = None,
main_irf_nr: int = 0,
linlog: bool = False,
linthresh: float = 1,
divide_by_scale: bool = True,
per_axis_legend: bool = False,
y_label: str = "a.u.",
cycler: Cycler | None = PlotStyle().data_cycler_solid,
) -> None:
"""Plot data and fits for a given ``wavelength`` on a given ``axis``.
If the wavelength isn't part of a dataset, that dataset will be skipped.
Parameters
----------
result : ResultLike
Data structure which can be converted to a mapping.
wavelength : float
Wavelength to plot data and fits for.
axis: Axis
Axis to plot the data and fits on.
center_λ: float | None
Center wavelength (λ in nm)
main_irf_nr : int
Index of the main ``irf`` component when using an ``irf``
parametrized with multiple peaks. Defaults to 0.
linlog : bool
Whether to use 'symlog' scale or not. Defaults to False.
linthresh : float
A single float which defines the range (-x, x), within which the plot is linear.
This avoids having the plot go to infinity around zero. Defaults to 1.
divide_by_scale : bool
Whether or not to divide the data by the dataset scale used for optimization.
Defaults to True.
per_axis_legend: bool
Whether to use a legend per plot or for the whole figure. Defaults to False.
y_label: str
Label used for the y-axis of each subplot.
cycler : Cycler | None
Plot style cycler to use. Defaults to PlotStyle().data_cycler_solid.
See Also
--------
plot_fit_overview
"""
result_map = result_dataset_mapping(result)
add_cycler_if_not_none(axis, cycler)
for dataset_name in result_map.keys():
spectral_coords = result_map[dataset_name].coords["spectral"].values
if spectral_coords.min() <= wavelength <= spectral_coords.max():
result_data = result_map[dataset_name].sel(spectral=[wavelength],
method="nearest")
scale = extract_dataset_scale(result_data, divide_by_scale)
irf_loc = extract_irf_location(result_data, center_λ, main_irf_nr)
result_data = result_data.assign_coords(
time=result_data.coords["time"] - irf_loc)
(result_data.data / scale).plot(x="time",
ax=axis,
label=f"{dataset_name}_data")
(result_data.fitted_data / scale).plot(x="time",
ax=axis,
label=f"{dataset_name}_fit")
else:
[next(axis._get_lines.prop_cycler) for _ in range(2)]
if linlog:
axis.set_xscale("symlog", linthresh=linthresh)
axis.set_ylabel(y_label)
if per_axis_legend is True:
axis.legend()