From 975d517a391937f7401c56a471252aa614ce1b37 Mon Sep 17 00:00:00 2001
From: Felix Metzner <felixmetzner@outlook.com>
Date: Thu, 18 Apr 2024 21:21:05 +0200
Subject: [PATCH] Adding variation number to plot titles.
---
.../fitting/dedicated_fit_approach/plotting_tools.py | 8 ++++++--
1 file changed, 6 insertions(+), 2 deletions(-)
diff --git a/rdstar/offline_analysis/fitting/dedicated_fit_approach/plotting_tools.py b/rdstar/offline_analysis/fitting/dedicated_fit_approach/plotting_tools.py
index 69985e778..d2941023f 100644
--- a/rdstar/offline_analysis/fitting/dedicated_fit_approach/plotting_tools.py
+++ b/rdstar/offline_analysis/fitting/dedicated_fit_approach/plotting_tools.py
@@ -524,9 +524,11 @@ class SystematicsShapePlotter:
sys_shape_info: ShapePlotInfoContainer,
reco_ch_info: RecoChannelInfo,
) -> Generator[SpecificShapePlotInfoContainer, None, None]:
- for subset_id in range(sys_shape_info.number_of_eigendirections):
+ n_eigendirs: int = sys_shape_info.number_of_eigendirections
+ for subset_id in range(n_eigendirs):
yield self.plot_systematics_shape_overview_plot_for(
subset_id=subset_id,
+ number_of_subsets=n_eigendirs,
sys_shape_info=sys_shape_info,
reco_ch_info=reco_ch_info,
)
@@ -554,6 +556,7 @@ class SystematicsShapePlotter:
def plot_systematics_shape_overview_plot_for(
self,
subset_id: int,
+ number_of_subsets: int,
sys_shape_info: ShapePlotInfoContainer,
reco_ch_info: RecoChannelInfo,
) -> SpecificShapePlotInfoContainer:
@@ -635,7 +638,8 @@ class SystematicsShapePlotter:
shape_plot_info=_shape_plot_info,
)
- title_str: str = sys_shape_info.latex_str + " Sys. Shape Effect in " + reco_ch_info.latex_label
+ var_str: str = "" if number_of_subsets == 1 else f"({subset_id})"
+ title_str: str = sys_shape_info.latex_str + f" Sys. Shape Effect{var_str} in " + reco_ch_info.latex_label
fig.suptitle(title_str, fontsize=18)
_shape_plot_infos.append(_shape_plot_info)
--
GitLab