Adding example code for bf uncertainties.

rdstar/offline_analysis/fitting/dedicated_fit_approach/dedicated_fit_routine.py

@@ -1168,3 +1168,104 @@ class RDStarFitEvaluator:
         fit_data_frame: pd.DataFrame = rdstar_fit_instance.internal_mc_sample_df
         fit_data_frame.to_hdf(path_or_buf=tmp_fit_data_file_path, key="fitting")
         os.rename(src=tmp_fit_data_file_path, dst=fit_data_file_path)
+
+
+# TODO: BF variation code snippet
+__new_code_for_bf_sys: str = """
+def __d_meson_decay_bf_uncertainties_example_code():
+    df_mc["sig_d_decay_abs"] = df_mc["sig_d_decay"].apply(lambda x: tuple([abs(x_) for x_ in x]) if isinstance(x, tuple) else x)
+
+    df_mc["D_BF_variation_nominal"] = 1
+    df_mc["D_BF_variation_0_up"  ] = 1
+    df_mc["D_BF_variation_0_down"] = 1
+    df_mc["D_BF_variation_1_up"  ] = 1
+    df_mc["D_BF_variation_1_down"] = 1
+    df_mc["D_BF_variation_2_up"  ] = 1
+    df_mc["D_BF_variation_2_down"] = 1
+    df_mc["D_BF_variation_3_up"  ] = 1
+    df_mc["D_BF_variation_3_down"] = 1
+    df_mc["D_BF_variation_4_up"  ] = 1
+    df_mc["D_BF_variation_4_down"] = 1
+    df_mc["D_BF_variation_5_up"  ] = 1
+    df_mc["D_BF_variation_5_down"] = 1
+    df_mc["D_BF_variation_6_up"  ] = 1
+    df_mc["D_BF_variation_6_down"] = 1
+    df_mc["D_BF_variation_7_up"  ] = 1
+    df_mc["D_BF_variation_7_down"] = 1
+    df_mc["D_BF_variation_8_up"  ] = 1
+    df_mc["D_BF_variation_8_down"] = 1
+    df_mc["D_BF_variation_9_up"  ] = 1
+    df_mc["D_BF_variation_9_down"] = 1
+
+    def vary_D_BR(df_mc, i, D_BR, selection, debug=False):
+        df_mc.loc[selection, f"D_BF_variation_{i}_up"] = 1 / D_BR.n * (D_BR.n + D_BR.s)
+        df_mc.loc[selection, f"D_BF_variation_{i}_down"] = 1 / D_BR.n * (D_BR.n - D_BR.s)
+        if debug:
+            print(
+                df_mc.loc[selection, ["__weight_overall__", f"D_BF_variation_{i}_up", f"D_BF_variation_{i}_down"]
+                         ].applymap(lambda x: f"{x:4f}").head(n=3)
+            )
+        return df_mc
+
+
+    # (321, 211, 999, 999, 999, 999, 999)      57534  # 0
+    # (213, 321, 999, 999, 999, 999, 999)      29801  # 1
+    # (20213, 321, 999, 999, 999, 999, 999)    22938  # 2
+    # (321, 211, 211, 999, 999, 999, 999)      10147  # 3
+    # (323, 211, 999, 999, 999, 999, 999)       9623  # 4
+    # (321, 211, 211, 211, 999, 999, 999)       7294  # 5
+    # (321, 321, 999, 999, 999, 999, 999)       5238  # 6
+    # (10311, 211, 999, 999, 999, 999, 999)     4928  # 7
+    # (313, 113, 999, 999, 999, 999, 999)       4793  # 8
+    # (313, 111, 999, 999, 999, 999, 999)       4746  # 9
+
+    df_mc = vary_D_BR(df_mc, 0,
+        ufloat(0.0395, 0.03/100),
+        df_mc["sig_d_decay_abs"] == (321, 211, 999, 999, 999, 999, 999)
+    )
+
+    df_mc = vary_D_BR(df_mc, 1,
+        ufloat(0.113, 0.7/100),
+        df_mc["sig_d_decay_abs"] == (213, 321, 999, 999, 999, 999, 999)
+    )
+
+    df_mc = vary_D_BR(df_mc, 2,
+        ufloat(0.0750, 0.32/100),
+        df_mc["sig_d_decay_abs"] == (20213, 321, 999, 999, 999, 999, 999)
+    )
+
+    df_mc = vary_D_BR(df_mc, 3,
+        ufloat(0.0938 - 0.0104 - 0.0125, 0.16/100),
+        df_mc["sig_d_decay_abs"] == (321, 211, 211, 999, 999, 999, 999)
+    )
+
+    df_mc = vary_D_BR(df_mc, 4,
+        ufloat(0.0164 + 0.0231, (0.37e-3**2 + (0.3/100)**2)**0.5),
+        df_mc["sig_d_decay_abs"] == (323, 211, 999, 999, 999, 999, 999)
+    )
+
+    df_mc = vary_D_BR(df_mc, 5,
+        ufloat(0.0181, 0.07/100),
+        df_mc["sig_d_decay_abs"] == (321, 211, 211, 211, 999, 999, 999)
+    )
+
+    df_mc = vary_D_BR(df_mc, 6,
+        ufloat(0.00408, 0.06e-3),
+        df_mc["sig_d_decay_abs"] == (321, 321, 999, 999, 999, 999, 999)
+    )
+
+    df_mc = vary_D_BR(df_mc, 7,
+        ufloat(0.0104, 0.12/100),
+        df_mc["sig_d_decay_abs"] == (10311, 211, 999, 999, 999, 999, 999)
+    )
+
+    df_mc = vary_D_BR(df_mc, 8,
+        ufloat(0.0023, 0.05/100),
+        df_mc["sig_d_decay_abs"] == (313, 113, 999, 999, 999, 999, 999)
+    )
+
+    df_mc = vary_D_BR(df_mc, 9,
+        ufloat(0.0195, 0.24/100),
+        df_mc["sig_d_decay_abs"] == (313, 111, 999, 999, 999, 999, 999)
+    )
+"""