More improvements to shape sys plotter.

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,