From d352c67192f304774c943a5ef54e879d71a09241 Mon Sep 17 00:00:00 2001
From: Felix Metzner <felixmetzner@outlook.com>
Date: Thu, 18 Apr 2024 20:31:48 +0200
Subject: [PATCH] Minor changes to latex labels for fit components and to
 SystematicsShapePlotter.

---
 .../fit_setups/rdstar_fit_info.py             | 28 +++++++++----------
 .../dedicated_fit_approach/plotting_tools.py  |  4 +--
 2 files changed, 15 insertions(+), 17 deletions(-)

diff --git a/rdstar/offline_analysis/fitting/dedicated_fit_approach/fit_setups/rdstar_fit_info.py b/rdstar/offline_analysis/fitting/dedicated_fit_approach/fit_setups/rdstar_fit_info.py
index 122cf272a..dc36a68d9 100644
--- a/rdstar/offline_analysis/fitting/dedicated_fit_approach/fit_setups/rdstar_fit_info.py
+++ b/rdstar/offline_analysis/fitting/dedicated_fit_approach/fit_setups/rdstar_fit_info.py
@@ -72,9 +72,7 @@ _bbbar_bkg_sig_ids: Tuple[int, ...] = (
     OldMCMatchingIDsQueryInfo.PiLNu.id,
 )
 
-_continuum_bkg_sig_ids: Tuple[int, ...] = (
-    OldMCMatchingIDsQueryInfo.CBKG.id,
-)
+_continuum_bkg_sig_ids: Tuple[int, ...] = (OldMCMatchingIDsQueryInfo.CBKG.id,)
 
 dstst_l_in_BpD_bkg_sig_ids: Tuple[int, ...] = RDStarFitSignalIDwithRecoModeInfo.get_modified_sig_id(
     sig_ids=_dstst_l_bkg_sig_ids,
@@ -254,7 +252,7 @@ component_Dss_in_cB: ComponentInfo = ComponentInfo(
 component_Dss_in_nB: ComponentInfo = ComponentInfo(
     name="DSS_in_nB",
     comp_index=51,
-    latex_label="_" + LatexDecayProcesses.to_dstst_l_nu,
+    latex_label=LatexDecayProcesses.to_dstst_l_nu,
     sig_ids=dstst_l_in_BzD_bkg_sig_ids + dstst_l_in_BzDst_bkg_sig_ids,
     color=KITColors.kit_red,
 )
@@ -269,21 +267,21 @@ component_Dss_in_BpD: ComponentInfo = ComponentInfo(
 component_Dss_in_BzD: ComponentInfo = ComponentInfo(
     name="DSS_in_BzD",
     comp_index=51,
-    latex_label="_" + LatexDecayProcesses.to_dstst_l_nu,
+    latex_label=LatexDecayProcesses.to_dstst_l_nu,
     sig_ids=dstst_l_in_BzD_bkg_sig_ids,
     color=KITColors.kit_red,
 )
 component_Dss_in_BpDst: ComponentInfo = ComponentInfo(
     name="DSS_in_BpDst",
     comp_index=52,
-    latex_label="_" + LatexDecayProcesses.to_dstst_l_nu,
+    latex_label=LatexDecayProcesses.to_dstst_l_nu,
     sig_ids=dstst_l_in_BpDst_bkg_sig_ids,
     color=KITColors.kit_red,
 )
 component_Dss_in_BzDst: ComponentInfo = ComponentInfo(
     name="DSS_in_BzDst",
     comp_index=53,
-    latex_label="_" + LatexDecayProcesses.to_dstst_l_nu,
+    latex_label=LatexDecayProcesses.to_dstst_l_nu,
     sig_ids=dstst_l_in_BzDst_bkg_sig_ids,
     color=KITColors.kit_red,
 )
@@ -301,7 +299,7 @@ component_bbbar_in_cB_bkg: ComponentInfo = ComponentInfo(
 component_bbbar_in_nB_bkg: ComponentInfo = ComponentInfo(
     name="BBbarBKG_in_nB",
     comp_index=61,
-    latex_label="_" + LatexDecayProcesses.to_b_bbar_bkg,
+    latex_label=LatexDecayProcesses.to_b_bbar_bkg,
     sig_ids=bbbar_in_BzD_bkg_sig_ids + bbbar_in_BzDst_bkg_sig_ids,
     color=KITColors.grey,
 )
@@ -316,21 +314,21 @@ component_bbbar_in_BpD_bkg: ComponentInfo = ComponentInfo(
 component_bbbar_in_BzD_bkg: ComponentInfo = ComponentInfo(
     name="BBbarBKG_in_BzD",
     comp_index=61,
-    latex_label="_" + LatexDecayProcesses.to_b_bbar_bkg,
+    latex_label=LatexDecayProcesses.to_b_bbar_bkg,
     sig_ids=bbbar_in_BzD_bkg_sig_ids,
     color=KITColors.grey,
 )
 component_bbbar_in_BpDst_bkg: ComponentInfo = ComponentInfo(
     name="BBbarBKG_in_BpDst",
     comp_index=62,
-    latex_label="_" + LatexDecayProcesses.to_b_bbar_bkg,
+    latex_label=LatexDecayProcesses.to_b_bbar_bkg,
     sig_ids=bbbar_in_BpDst_bkg_sig_ids,
     color=KITColors.grey,
 )
 component_bbbar_in_BzDst_bkg: ComponentInfo = ComponentInfo(
     name="BBbarBKG_in_BzDst",
     comp_index=63,
-    latex_label="_" + LatexDecayProcesses.to_b_bbar_bkg,
+    latex_label=LatexDecayProcesses.to_b_bbar_bkg,
     sig_ids=bbbar_in_BzDst_bkg_sig_ids,
     color=KITColors.grey,
 )
@@ -348,7 +346,7 @@ component_continuum_in_cB_bkg: ComponentInfo = ComponentInfo(
 component_continuum_in_nB_bkg: ComponentInfo = ComponentInfo(
     name="CBKG_in_nB",
     comp_index=71,
-    latex_label="_" + LatexDecayProcesses.to_continuum_txt,
+    latex_label=LatexDecayProcesses.to_continuum_txt,
     sig_ids=continuum_in_BzD_bkg_sig_ids + continuum_in_BzDst_bkg_sig_ids,
     color=KITColors.light_grey,
 )
@@ -363,21 +361,21 @@ component_continuum_in_BpD_bkg: ComponentInfo = ComponentInfo(
 component_continuum_in_BzD_bkg: ComponentInfo = ComponentInfo(
     name="CBKG_in_BzD",
     comp_index=71,
-    latex_label="_" + LatexDecayProcesses.to_continuum_txt,
+    latex_label=LatexDecayProcesses.to_continuum_txt,
     sig_ids=continuum_in_BzD_bkg_sig_ids,
     color=KITColors.light_grey,
 )
 component_continuum_in_BpDst_bkg: ComponentInfo = ComponentInfo(
     name="CBKG_in_BpDst",
     comp_index=72,
-    latex_label="_" + LatexDecayProcesses.to_continuum_txt,
+    latex_label=LatexDecayProcesses.to_continuum_txt,
     sig_ids=continuum_in_BpDst_bkg_sig_ids,
     color=KITColors.light_grey,
 )
 component_continuum_in_BzDst_bkg: ComponentInfo = ComponentInfo(
     name="CBKG_in_BzDst",
     comp_index=73,
-    latex_label="_" + LatexDecayProcesses.to_continuum_txt,
+    latex_label=LatexDecayProcesses.to_continuum_txt,
     sig_ids=continuum_in_BzDst_bkg_sig_ids,
     color=KITColors.light_grey,
 )
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 f31f51100..ffa89bfeb 100644
--- a/rdstar/offline_analysis/fitting/dedicated_fit_approach/plotting_tools.py
+++ b/rdstar/offline_analysis/fitting/dedicated_fit_approach/plotting_tools.py
@@ -566,7 +566,7 @@ class SystematicsShapePlotter:
         )
 
         if self.fig_size is None:
-            fig_size: Tuple[float, float] = (8.0, n_plot_rows * 2.0)
+            fig_size: Tuple[float, float] = (12.0, n_plot_rows * 3.0)
         else:
             fig_size = self.fig_size
 
@@ -574,7 +574,7 @@ class SystematicsShapePlotter:
             nrows=n_plot_rows,
             ncols=4,
             figsize=fig_size,
-            dpi=200,
+            dpi=300,
             sharex="all",
             sharey="all",
         )  # type: FigureType, np.ndarray
-- 
GitLab