Updates for fakes with u_T^\mu bins and other fixes

scripts/analysisTools/tests/testPlots1D.py

@@ -47,6 +47,7 @@ def plotDistribution1D(
     ratioPadYaxisTitle="Data/pred::0.9,1.1",
     scaleToUnitArea=False,
     noRatioPanel=False,
+    scaleProcs={},
 ):
 
     createPlotDirAndCopyPhp(outfolder_dataMC)
@@ -71,13 +72,17 @@ def plotDistribution1D(
             hmc[d].SetLineColor(ROOT.kBlack)
             hmc[d].SetMarkerSize(0)
             hmc[d].SetMarkerStyle(0)
+            if d in scaleProcs.keys():
+                hmc[d].Scale(scaleProcs[d])
             stackIntegral += hmc[d].Integral()
 
     if scaleToUnitArea:
         hdata.Scale(1.0 / hdata.Integral())
 
     stack_1D = ROOT.THStack("stack_1D", "signal and backgrounds")
-    hmcSortedKeys = sorted(hmc.keys(), key=lambda x: hmc[x].Integral())
+    hmcSortedKeys = sorted(
+        [x for x in hmc.keys() if x in datasets], key=lambda x: hmc[x].Integral()
+    )
     for i in hmcSortedKeys:
         if scaleToUnitArea:
             hmc[i].Scale(1.0 / stackIntegral)

scripts/analysisTools/w_mass_13TeV/makeEfficiencyCorrection.py

@@ -58,7 +58,7 @@
         "--corrvar",
         type=str,
         default="run",
-        choices=["run", "phi"],
+        choices=["run", "phi", "utAngleSign"],
         help="Variable to be used for the correction",
     )
     parser.add_argument(
@@ -162,6 +162,11 @@
             f"{round(corrEdges[i], 2)} < #phi^{{#mu}} < {round(corrEdges[i+1], 2)}"
             for i in range(nCorrBins)
         ]
+    elif args.corrvar == "utAngleSign":
+        legEntries = [
+            "#it{u}_{T}^{#mu} < 0",
+            "#it{u}_{T}^{#mu} > 0",
+        ]
     else:
         legEntries = [f"{corrvar} bin {i}" for i in range(nCorrBins)]
 
@@ -175,7 +180,7 @@
         sign = "minus" if args.charge < 0 else "plus"
         chargePostfix = f"_{sign}"
 
-    # create a TH2 and convrt into boost again to save it
+    # create a TH2 and convert into boost again to save it
     nx = heffiroot["Data"][0].GetNbinsX()
     xmin = heffiroot["Data"][0].GetXaxis().GetBinLowEdge(1)
     xmax = heffiroot["Data"][0].GetXaxis().GetBinLowEdge(1 + nx)
@@ -209,9 +214,20 @@
                 f"{corrvar}-eta ID = {i} {j}: {round(corr,3)} +/- {round(corr_unc,3)}"
             )
 
-    title_var = "Run" if corrvar == "run" else "#phi" if corrvar == "phi" else "CorrVar"
-
-    effRange = "0.85,1.05" if corrvar == "run" else "0.80,1.1"
+    titles = {
+        "run": "Run",
+        "phi": "#phi^{#mu}",
+        "utAngleSign": "sign(#it{u}_{T}^{#mu})",
+    }
+    title_var = "CorrVar"
+    if corrvar in titles.keys():
+        title_var = titles[corrvar]
+
+    effrange = "0.80,1.1"
+    if corrvar == "run":
+        effRange = "0.85,1.05"
+    elif corrvar == "utAngleSign":
+        effRange = "0.8,1.2"
 
     for d in datasets:
         drawNTH1(