adding plotting script that just shows and overlays the final aggregated efficiencies

Author: jprendi

NERDAnalysis2026/compare_TOT_eff.py

@@ -0,0 +1,113 @@
+import ROOT
+import os
+import argparse
+
+ROOT.gROOT.SetBatch(True)
+ROOT.gErrorIgnoreLevel = ROOT.kError
+
+parser = argparse.ArgumentParser(description="Overlay efficiency graphs from two aggregated ROOT files.")
+parser.add_argument('--f1', required=True, help='First input ROOT file (e.g., aggregated_NGT.root)')
+parser.add_argument('--l1', required=True, help='Legend label for first file (e.g., "NGT")')
+parser.add_argument('--f2', required=True, help='Second input ROOT file (e.g., aggregated_HTS.root)')
+parser.add_argument('--l2', required=True, help='Legend label for second file (e.g., "HTS TestStand")')
+parser.add_argument('--outdir', default='plots_comparison', help='Directory to save overlaid plots')
+args = parser.parse_args()
+
+# List of expected variables based on your first script
+VARIABLES = ["eta", "ptEB", "ptEE", "phiEB", "phiEE"]
+
+def get_variable_from_name(gname):
+    """Extracts the variable name from the graph name for the X-axis label."""
+    for var in VARIABLES:
+        if f"_{var}_" in gname:
+            return var
+    return "Variable"
+
+def main():
+    os.makedirs(args.outdir, exist_ok=True)
+
+    file1 = ROOT.TFile.Open(args.f1)
+    file2 = ROOT.TFile.Open(args.f2)
+
+    if not file1 or file1.IsZombie() or not file2 or file2.IsZombie():
+        raise SystemExit("Error: Could not open one or both input ROOT files.")
+
+    keys = file1.GetListOfKeys()
+    #graph_names = [k.GetName() for k in keys if k.ReadObj().InheritsFrom("TGraphAsymmErrors")]
+    graph_names = [k.GetName() for k in keys if k.ReadObj().InheritsFrom("TGraphAsymmErrors") and "Total" in k.GetName()]
+
+    if not graph_names:
+        print(f"No TGraphAsymmErrors found in {args.f1}.")
+        return
+
+    for gname in graph_names:
+        g1 = file1.Get(gname)
+        g2 = file2.Get(gname)
+
+        if not g1 or not g2:
+            continue
+
+        c = ROOT.TCanvas(f"c_{gname}", "", 800, 600)
+        c.SetRightMargin(0.05)
+        c.SetLeftMargin(0.12)
+        c.SetBottomMargin(0.12)
+        c.SetGridy()
+
+        # Styling
+        g1.SetLineColor(ROOT.kBlue + 1)
+        g1.SetMarkerColor(ROOT.kBlue + 1)
+        g1.SetMarkerStyle(20)
+        g1.SetLineWidth(2)
+
+        g2.SetLineColor(ROOT.kRed + 1)
+        g2.SetMarkerColor(ROOT.kRed + 1)
+        g2.SetMarkerStyle(21)
+        g2.SetLineWidth(2)
+
+        # Determine Y-axis range including error bars
+        y_min, y_max = 1.0, 0.0
+        for g in [g1, g2]:
+            for i in range(g.GetN()):
+                y = g.GetY()[i]
+                y_err_low = g.GetErrorYlow(i)
+                y_err_high = g.GetErrorYhigh(i)
+                
+                if y > 0: # Ignore empty bins
+                    if (y - y_err_low) < y_min: y_min = y - y_err_low
+                    if (y + y_err_high) > y_max: y_max = y + y_err_high
+        
+        # Add slight padding to min/max
+        g1.SetMaximum(min(1.05, y_max * 1.05))
+        g1.SetMinimum(max(0.0, y_min * 0.9))
+        
+        # Clean up title based on Script 1's 'eff_' prefix
+        clean_title = gname.replace("eff_", "Efficiency: ").replace("_", " ")
+        g1.SetTitle(clean_title)
+        
+        # Dynamic axis labels
+        g1.GetYaxis().SetTitle("Efficiency")
+        g1.GetXaxis().SetTitle(get_variable_from_name(gname))
+
+        g1.Draw("AP")
+        g2.Draw("P SAME")
+
+        # Adjusted legend to avoid overlapping data in the bottom left
+        leg = ROOT.TLegend(0.60, 0.20, 0.90, 0.35)
+        leg.SetBorderSize(0)
+        leg.SetFillStyle(0)
+        leg.SetTextSize(0.035)
+        leg.AddEntry(g1, args.l1, "lp")
+        leg.AddEntry(g2, args.l2, "lp")
+        leg.Draw()
+
+        out_path = os.path.join(args.outdir, f"compare_{gname}.png")
+        c.SaveAs(out_path)
+        c.SaveAs(out_path.replace(".png", ".pdf"))
+        
+        # Cleanup memory for the canvas
+        c.Close()
+
+    print(f"Comparison plots saved to {args.outdir}/")
+
+if __name__ == "__main__":
+    main()