From 6a1f4c5541290fbd76602a78fec083eb46b7dbb6 Mon Sep 17 00:00:00 2001
From: Felix Metzner <felixmetzner@outlook.com>
Date: Mon, 15 Apr 2024 16:47:31 +0200
Subject: [PATCH] More improvements to shape sys plotter.
---
.../dedicated_fit_routine.py | 30 +++++++++++++++----
1 file changed, 25 insertions(+), 5 deletions(-)
diff --git a/rdstar/offline_analysis/fitting/dedicated_fit_approach/dedicated_fit_routine.py b/rdstar/offline_analysis/fitting/dedicated_fit_approach/dedicated_fit_routine.py
index 6b1b5e10a..c4ba31311 100644
--- a/rdstar/offline_analysis/fitting/dedicated_fit_approach/dedicated_fit_routine.py
+++ b/rdstar/offline_analysis/fitting/dedicated_fit_approach/dedicated_fit_routine.py
@@ -1120,6 +1120,7 @@ class RDStarFitEvaluator:
def plot_sig_ff_sys_shape_effects(
self,
fit_status_container: RDStarFitStatusContainer,
+ add_statistical_uncertainties: bool,
) -> Generator[SpecificShapePlotInfoContainer, None, None]:
sys_shape_plotter: SystematicsShapePlotter = SystematicsShapePlotter(
base_output_dir_path=self.base_output_dir_path,
@@ -1139,11 +1140,15 @@ class RDStarFitEvaluator:
plot_subsets=False,
)
- yield from sys_shape_plotter.create_systematics_shape_plots(sys_shape_info=example_sys_shape_info)
+ yield from sys_shape_plotter.create_systematics_shape_plots(
+ sys_shape_info=example_sys_shape_info,
+ add_statistical_uncertainties=add_statistical_uncertainties,
+ )
def plot_norm_ff_sys_shape_effects(
self,
fit_status_container: RDStarFitStatusContainer,
+ add_statistical_uncertainties: bool,
) -> Generator[SpecificShapePlotInfoContainer, None, None]:
sys_shape_plotter: SystematicsShapePlotter = SystematicsShapePlotter(
base_output_dir_path=self.base_output_dir_path,
@@ -1163,11 +1168,15 @@ class RDStarFitEvaluator:
plot_subsets=False,
)
- yield from sys_shape_plotter.create_systematics_shape_plots(sys_shape_info=example_sys_shape_info)
+ yield from sys_shape_plotter.create_systematics_shape_plots(
+ sys_shape_info=example_sys_shape_info,
+ add_statistical_uncertainties=add_statistical_uncertainties,
+ )
def plot_l_id_sys_shape_effects(
self,
fit_status_container: RDStarFitStatusContainer,
+ add_statistical_uncertainties: bool,
) -> Generator[SpecificShapePlotInfoContainer, None, None]:
sys_shape_plotter: SystematicsShapePlotter = SystematicsShapePlotter(
base_output_dir_path=self.base_output_dir_path,
@@ -1187,11 +1196,15 @@ class RDStarFitEvaluator:
plot_subsets=False,
)
- yield from sys_shape_plotter.create_systematics_shape_plots(sys_shape_info=example_sys_shape_info)
+ yield from sys_shape_plotter.create_systematics_shape_plots(
+ sys_shape_info=example_sys_shape_info,
+ add_statistical_uncertainties=add_statistical_uncertainties,
+ )
def plot_tracking_sys_shape_effects(
self,
fit_status_container: RDStarFitStatusContainer,
+ add_statistical_uncertainties: bool,
) -> Generator[SpecificShapePlotInfoContainer, None, None]:
sys_shape_plotter: SystematicsShapePlotter = SystematicsShapePlotter(
base_output_dir_path=self.base_output_dir_path,
@@ -1211,11 +1224,15 @@ class RDStarFitEvaluator:
plot_subsets=False,
)
- yield from sys_shape_plotter.create_systematics_shape_plots(sys_shape_info=example_sys_shape_info)
+ yield from sys_shape_plotter.create_systematics_shape_plots(
+ sys_shape_info=example_sys_shape_info,
+ add_statistical_uncertainties=add_statistical_uncertainties,
+ )
def plot_k_short_sys_shape_effects(
self,
fit_status_container: RDStarFitStatusContainer,
+ add_statistical_uncertainties: bool,
) -> Generator[SpecificShapePlotInfoContainer, None, None]:
sys_shape_plotter: SystematicsShapePlotter = SystematicsShapePlotter(
base_output_dir_path=self.base_output_dir_path,
@@ -1235,7 +1252,10 @@ class RDStarFitEvaluator:
plot_subsets=False,
)
- yield from sys_shape_plotter.create_systematics_shape_plots(sys_shape_info=example_sys_shape_info)
+ yield from sys_shape_plotter.create_systematics_shape_plots(
+ sys_shape_info=example_sys_shape_info,
+ add_statistical_uncertainties=add_statistical_uncertainties,
+ )
def dump_data_for_external_fitter(
self,
--
GitLab