updating work in progress ana_ABCD

uhh-cms · Jul 10, 2024 · 770cfcc · 770cfcc
1 parent 30fd941
commit 770cfcc
Showing 1 changed file with 34 additions and 28 deletions.
diff --git a/hbt/config/hist_hooks.py b/hbt/config/hist_hooks.py
@@ -1,6 +1,31 @@
 import order as od
 
 
+def create_regions(id, regions, mc_hist, data_hist, cat_name, cat_parent):
+    if "os" in cat_name:
+        if "non_iso" in cat_name:
+            regions[cat_parent]["A"] = {}
+            regions[cat_parent]["A"]["mc"] = mc_hist.counts()[id]
+            regions[cat_parent]["A"]["data"] = data_hist.counts()[id]
+            print("...created region A for category " + cat_parent)
+        else:
+            regions[cat_parent]["B"] = {}
+            regions[cat_parent]["B"]["mc"] = mc_hist.counts()[id]
+            regions[cat_parent]["B"]["data"] = data_hist.counts()[id]
+            print("...created region B for category " + cat_parent)
+    elif "ss" in cat_name:
+        if "non_iso" in cat_name:
+            regions[cat_parent]["C"] = {}
+            regions[cat_parent]["C"]["mc"] = mc_hist.counts()[id]
+            regions[cat_parent]["C"]["data"] = data_hist.counts()[id]
+            print("...created region C for category " + cat_parent)
+        else:
+            regions[cat_parent]["D"] = {}
+            regions[cat_parent]["D"]["mc"] = mc_hist.counts()[id]
+            regions[cat_parent]["D"]["data"] = data_hist.counts()[id]
+            print("...created region D for category " + cat_parent)
+
+
 def add_hooks(cfg: od.Config):
 
     def ana_ABCD(task, hists):
@@ -30,34 +55,15 @@ def ana_ABCD(task, hists):
                 # define ABCD regions
                 if cat_parent not in regions:
                     regions[cat_parent] = {}
-                elif "os" in cat_name:
-                    if "non_iso" in cat_name:
-                        regions[cat_parent]["A"] = {}
-                        regions[cat_parent]["A"]["mc"] = mc_hist.counts()[id]
-                        regions[cat_parent]["A"]["data"] = data_hist.counts()[id]
-                        print("...created region A for category " + cat_parent)
-                    else:
-                        regions[cat_parent]["B"] = {}
-                        regions[cat_parent]["B"]["mc"] = mc_hist.counts()[id]
-                        regions[cat_parent]["B"]["data"] = data_hist.counts()[id]
-                        print("...created region B for category " + cat_parent)
-                elif "ss" in cat_name:
-                    if "non_iso" in cat_name:
-                        regions[cat_parent]["C"] = {}
-                        regions[cat_parent]["C"]["mc"] = mc_hist.counts()[id]
-                        regions[cat_parent]["C"]["data"] = data_hist.counts()[id]
-                        print("...created region C for category " + cat_parent)
-                    else:
-                        regions[cat_parent]["D"] = {}
-                        regions[cat_parent]["D"]["mc"] = mc_hist.counts()[id]
-                        regions[cat_parent]["D"]["data"] = data_hist.counts()[id]
-                        print("...created region D for category " + cat_parent)
-
-                from IPython import embed
-                embed()
-                quit()
-                # calculate rate factor
-                #C_rate = regions[cat_parent]
+                    create_regions(id, regions, mc_hist, data_hist, cat_name, cat_parent)
+                else:
+                    create_regions(id, regions, mc_hist, data_hist, cat_name, cat_parent)
+
+            from IPython import embed
+            embed()
+            quit()
+            # calculate rate factor
+            #C_rate = regions[cat_parent]
 
             hists[cfg.processes.n.qcd] = mc_hist * 0.5