feat: Autorebin option in make_template script (#133)

* fixing pt range of fixed WP histos

* Updated unfo for JECs in Summer23BPix campaing


* feat: add autorebin option and new cross section entry for TBbartoLplusNuBbar process

* fix: correct formatting

* fix: correct formatting in cross section entry for TBbartoLplusNuBbar process

---------

Co-authored-by: Agostino De Iorio <[email protected]>

Author: adeiorio

scripts/make_template.py

@@ -1,6 +1,10 @@
 from coffea.util import load
+import numpy as np
 import uproot, sys, os, argparse, hist
 from BTVNanoCommissioning.helpers.xs_scaler import collate, scaleSumW
+from BTVNanoCommissioning.utils.plot_utils import (
+    rebin_hist,
+)
 import numpy as np
 
 parser = argparse.ArgumentParser(description="Make templates from coffea files")
@@ -44,6 +48,11 @@
     default=None,
     help="Specify mergemap as dict, '{merge1:[dataset1,dataset2]...}' Also works with the json file with dict",
 )
+parser.add_argument(
+    "--autorebin",
+    default=None,
+    help="Rebin the plotting variables, input `int` or `list`. int: merge N bins. list of number: rebin edges(non-uniform bin is possible)",
+)
 
 
 def create_template(inputs, variable, mergemap, axis, lumi, output):
@@ -54,7 +63,29 @@ def create_template(inputs, variable, mergemap, axis, lumi, output):
     hall = merge_all["all"][variable][axis]
 
     fout = uproot.recreate(f"{variable}_{output}")
+    subtract = False
+    if "osss" in axis.keys():
+        if "sum" in str(axis["osss"]):
+            subtract = True
+            axis.pop("osss")
     for data in collated.keys():
+        if subtract:
+            collated[data][variable] = (
+                collated[data][variable][{"osss": 0}]
+                + collated[data][variable][{"osss": 1}] * -1
+            )
+        if arg.autorebin is not None:
+            if arg.autorebin.isdigit():
+                rebin = int(arg.autorebin)
+            else:
+                rebin = np.array([float(i) for i in arg.autorebin.split(",")])
+            fout[data.replace("-", "_")] = rebin_hist(
+                collated[data][variable][axis],
+                collated[data][variable].axes[-1].name,
+                rebin,
+            )
+        else:
+            fout[data.replace("-", "_")] = collated[data][variable][axis]
         if variable not in collated[data].keys():
             raise f"{variable} not in {data}"
         elif np.sum(collated[data][variable].values()) == 0:
@@ -90,6 +121,7 @@ def create_template(inputs, variable, mergemap, axis, lumi, output):
             for d in inputs[f].keys():
                 flists.append(d)
         mergemap = {d: d for d in flists}
+
     axis = json.loads(arg.axis)
     # modify axis to hist methods
     for a in axis:

src/BTVNanoCommissioning/helpers/xsection.py

@@ -152,6 +152,12 @@
         "energy": "13.6",
         "comment": "from TOP-22-012 with PDG 2023 value",
     },
+    {
+        "process_name": "TBbartoLplusNuBbar-s-channel-4FS_TuneCP5_13p6TeV_amcatnlo-pythia8",
+        "cross_section": "2.278",
+        "energy": "13.6",
+        "comment": "from XSDB",
+    },
     {
         "process_name": "TTTo2L2Nu_CP5_13p6TeV_powheg-pythia8",
         "cross_section": "96.9",