ATLAS Offline Software
Loading...
Searching...
No Matches
plotIsoValidation.py
Go to the documentation of this file.
1# Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
2import os, sys, re, ROOT, argparse
3from math import fabs
4import logging
5
6
7def DrawTLatex(x, y, text, size=18, font=43, align=11):
8 tex = ROOT.TLatex()
9 tex.SetTextAlign(align)
10 tex.SetTextSize(size)
11 tex.SetTextFont(font)
12 tex.SetNDC()
13 tex.DrawLatex(x, y, text)
14
15
16
17def saveHisto(plot_options, histo):
18 if not histo.GetEntries(): return False
19
20 Canvas_Name = "Isolation_%s_%s"%("Corr" if options.useCorrectedCones else "Orig" ,
21 histo.GetName())
22 c1 = ROOT.TCanvas(Canvas_Name, "canvas", 800 , 600)
23
24 c1.cd()
25 c1.SetTopMargin(0.13)
26 if histo.GetDimension() == 2:
27 c1.SetRightMargin(0.15)
28 c1.SetLeftMargin(0.12)
29 if histo.GetDimension() == 1:
30 histo.SetLineColor(ROOT.kRed)
31 histo.SetLineWidth(2)
32
33 histo.Draw()
34 else: histo.Draw("colz")
35 DrawTLatex(c1.GetLeftMargin(), 0.92,"%s (%s)" %(histo.GetTitle(), "corrected" if plot_options.useCorrectedCones else "vanilla") )
36
37 c1.SaveAs("%s/%s.pdf"%(options.outDir, Canvas_Name))
38 c1.SaveAs("%s/AllIsolationPlots_%s.pdf"%(options.outDir, "Corr" if options.useCorrectedCones else "Orig" ))
39
40 return True
41if __name__ == '__main__':
42 parser = argparse.ArgumentParser(description='This script generates validation plots of the isolation correction tool. For more help type \"python plotIsoValidation.py -h\"',
43 prog='plotIsoValidation',
44 formatter_class=argparse.ArgumentDefaultsHelpFormatter)
45
46 parser.add_argument("--inFile", help="Where is the input generated by the IsolationCorrection test alg 'IsolationSelection/testIsoCloseByCorr_jobOptions.py'",
47 default="IsoCloseByTestFile.root")
48 parser.add_argument("--outDir", help = "Output directory of the final plots", default="Plots/")
49 parser.add_argument("--useCorrectedCones", help="", action ='store_true', default=False)
50 options = parser.parse_args()
51
52 if not os.path.exists(options.inFile):
53 logging.error("There is no file %s"%(options.inFile))
54 exit(1)
55
56 inFile = ROOT.TFile.Open(options.inFile, "READ")
57 if not inFile or not inFile.IsOpen():
58 logging.error("ROOT file %s is invalid please check whether the test job processed successfully"%(options.inFile))
59 exit(1)
60
61 anaTree = inFile.Get("IsoCorrTest")
62 if not anaTree:
63 logging.error(" Where is my Testing tree in file %s"%(options.inFile))
64 exit(1)
65
68 Elec_Track_Iso = ROOT.TH1D("elec_track_iso", "track isolation", 55, -5, 50)
69 Elec_Track_Iso.GetXaxis().SetTitle(" p_{T}^{TrackIsolation} (electrons) [GeV]")
70 Elec_Calo_Iso = ROOT.TH1D("elec_calo_iso", "calorimeter isolation", 55, -5, 50)
71 Elec_Calo_Iso.GetXaxis().SetTitle(" E_{T}^{CaloIsolation} (electrons) [GeV]")
72
73 Muon_Track_Iso = ROOT.TH1D("muon_track_iso", "track isolation", 55, -5, 50)
74 Muon_Track_Iso.GetXaxis().SetTitle(" p_{T}^{TrackIsolation} (muons) [GeV]")
75 Muon_Calo_Iso = ROOT.TH1D("muon_calo_iso", "calorimeter isolation", 55, -5, 50)
76 Muon_Calo_Iso.GetXaxis().SetTitle(" E_{T}^{CaloIsolation} (muons) [GeV]")
77
78
79 Elec_RelTrack_Iso = ROOT.TH1D("elec_rel_track_iso", "track isolation", 52, -0.04, 1)
80 Elec_RelTrack_Iso.GetXaxis().SetTitle("p_{T}^{TrackIsolation} / p_{T} (electrons)")
81 Elec_RelCalo_Iso = ROOT.TH1D("elec_rel_calo_iso", "track isolation", 52, -0.04, 1)
82 Elec_RelCalo_Iso.GetXaxis().SetTitle("E_{T}^{CaloIsolation} / p_{T} (electrons)")
83
84 Muon_RelTrack_Iso = ROOT.TH1D("muon_rel_track_iso", "track isolation", 52, -0.04, 1)
85 Muon_RelTrack_Iso.GetXaxis().SetTitle("p_{T}^{TrackIsolation} / p_{T} (muons)")
86 Muon_RelCalo_Iso = ROOT.TH1D("muon_rel_calo_iso", "track isolation", 52, -0.04, 1)
87 Muon_RelCalo_Iso.GetXaxis().SetTitle("E_{T}^{CaloIsolation} / p_{T} (muons)")
88
89
90 Elec_Track_polution_dR = ROOT.TH2D("elec_track_polution_dR", "track_isolaiton against dR",50, 0, 0.5, 100, -1.5, 1.5 )
91 Elec_Track_polution_dR.GetXaxis().SetTitle("#Delta R (e, #it{l})")
92 Elec_Track_polution_dR.GetYaxis().SetTitle("p_{T}^{TrackIsolation}(electrons) / p_{T}( Associated close-by track)")
93
94 Elec_Topo_polution_dR = ROOT.TH2D("elec_calo_polution_dR", "calorimeter isolation against dR",50, 0, 0.5, 100, -1.5, 1.5)
95 Elec_Topo_polution_dR.GetXaxis().SetTitle("#Delta R (e, #it{l})")
96 Elec_Topo_polution_dR.GetYaxis().SetTitle("E_{T}^{CaloIsolation} (electrons) / E_{T}( Associated close-by cluster)")
97
98 Muon_Track_polution_dR = ROOT.TH2D("muon_track_polution_dR", "track_isolaiton against dR", 50, 0, 0.5, 100, -1.5, 1.5)
99 Muon_Track_polution_dR.GetXaxis().SetTitle("#Delta R (#mu, #it{l})")
100 Muon_Track_polution_dR.GetYaxis().SetTitle("p_{T}^{TrackIsolation}(muons) / p_{T}( Associated close-by track)")
101
102 Muon_Topo_polution_dR = ROOT.TH2D("muon_calo_polution_dR", "calorimeter isolation against dR", 50, 0, 0.5, 100, -1.5, 1.5)
103 Muon_Topo_polution_dR.GetXaxis().SetTitle("#Delta R (#mu, #it{l})")
104 Muon_Topo_polution_dR.GetYaxis().SetTitle("E_{T}^{CaloIsolation} (muons) / E_{T}( Associated close-by cluster)")
105
106 N_Isolated_Muon = ROOT.TH1D("N_muons", "Isolated #mu", 6, 0, 6);
107 N_Isolated_Muon.GetXaxis().SetTitle("N(#mu) - passing iso WP" );
108
109 N_Isolated_Elec = ROOT.TH1D("N_elec", "Elec #mu", 6, 0, 6);
110 N_Isolated_Elec.GetXaxis().SetTitle("N(e) - passing iso WP");
111
112 AllHistos = [Elec_Track_Iso,
113 Elec_Calo_Iso ,
114 Muon_Track_Iso,
115 Muon_Calo_Iso,
116 Elec_RelTrack_Iso,
117 Muon_RelTrack_Iso,
118 Elec_RelCalo_Iso,
119 Muon_RelCalo_Iso,
120 Elec_Track_polution_dR,
121 Elec_Topo_polution_dR,
122 Muon_Track_polution_dR,
123 Muon_Topo_polution_dR ,
124 N_Isolated_Muon,
125 N_Isolated_Elec,
126 ]
127 treeReader = ROOT.TTreeReader("IsoCorrTest", inFile)
128 branch_names = [ B.GetName() for B in anaTree.GetListOfBranches() ]
129
130 mu_trk_cone_names = [b [ b.find("Corr" if options.useCorrectedCones else "Orig") + 5 :] for b in branch_names if b.find("Muons_%s_pt"%("Corr" if options.useCorrectedCones else "Orig")) != -1]
131 mu_calo_cone_names = [ b [ b.find("Corr" if options.useCorrectedCones else "Orig") + 5 :] for b in branch_names if b.find("Muons_%s_topo"%("Corr" if options.useCorrectedCones else "Orig")) != -1 or
132 b.find("Muons_%s_neflow"%("Corr" if options.useCorrectedCones else "Orig")) != -1
133 ]
134
135 el_trk_cone_names = [ b [ b.find("Corr" if options.useCorrectedCones else "Orig") + 5 :] for b in branch_names if b.find("Electrons_%s_pt"%("Corr" if options.useCorrectedCones else "Orig")) != -1]
136 el_calo_cone_names = [ b [ b.find("Corr" if options.useCorrectedCones else "Orig") + 5 :] for b in branch_names if b.find("Electrons_%s_topo"%("Corr" if options.useCorrectedCones else "Orig")) != -1 or
137 b.find("Electrons_%s_neflow"%("Corr" if options.useCorrectedCones else "Orig")) != -1
138 ]
139
140 if len (el_calo_cone_names) > 0:
141 for H in [H for H in AllHistos if H.GetName().find("elec") != -1 and H.GetName().find("calo") != -1]: H.SetTitle(el_calo_cone_names[0])
142 if len (el_trk_cone_names) > 0:
143 for H in [H for H in AllHistos if H.GetName().find("elec") != -1 and H.GetName().find("track") != -1]: H.SetTitle(el_trk_cone_names[0])
144 if len (mu_calo_cone_names) > 0:
145 for H in [H for H in AllHistos if H.GetName().find("muon") != -1 and H.GetName().find("calo") != -1]: H.SetTitle(mu_calo_cone_names[0])
146 if len (mu_trk_cone_names) > 0:
147 for H in [H for H in AllHistos if H.GetName().find("muon") != -1 and H.GetName().find("track") != -1]: H.SetTitle(mu_trk_cone_names[0])
148
149
150 mu_pt = ROOT.TTreeReaderArray(float)(treeReader, "Muons_pt")
151 mu_eta = ROOT.TTreeReaderArray(float)(treeReader, "Muons_eta")
152 mu_phi = ROOT.TTreeReaderArray(float)(treeReader, "Muons_phi")
153 mu_trk = ROOT.TTreeReaderArray(float)(treeReader, "Muons_trackPt")
154
155 mu_clu = ROOT.TTreeReaderArray(float)(treeReader, "Muons_clusterEt")
156 mu_clu_eta = ROOT.TTreeReaderArray(float)(treeReader, "Muons_clusterEta")
157 mu_clu_phi = ROOT.TTreeReaderArray(float)(treeReader, "Muons_clusterPhi")
158
159
160 mu_trk_iso = ROOT.TTreeReaderArray(float)(treeReader, "Muons_%s_%s"%("Corr" if options.useCorrectedCones else "Orig", mu_trk_cone_names[0])) if len(mu_trk_cone_names) > 0 else None
161 mu_calo_iso = ROOT.TTreeReaderArray(float)(treeReader, "Muons_%s_%s"%("Corr" if options.useCorrectedCones else "Orig",mu_calo_cone_names[0])) if len (mu_calo_cone_names) > 0 else None
162
163 el_pt = ROOT.TTreeReaderArray(float)(treeReader, "Electrons_pt")
164 el_eta = ROOT.TTreeReaderArray(float)(treeReader, "Electrons_eta")
165 el_phi = ROOT.TTreeReaderArray(float)(treeReader, "Electrons_phi")
166 el_trk = ROOT.TTreeReaderArray(float)(treeReader, "Electrons_trackPt")
167 el_clu = ROOT.TTreeReaderArray(float)(treeReader, "Electrons_clusterEt")
168 el_clu_eta = ROOT.TTreeReaderArray(float)(treeReader, "Electrons_clusterEta")
169 el_clu_phi = ROOT.TTreeReaderArray(float)(treeReader, "Electrons_clusterPhi")
170
171 el_trk_iso = ROOT.TTreeReaderArray(float)(treeReader, "Electrons_%s_%s"%("Corr" if options.useCorrectedCones else "Orig", el_trk_cone_names[0])) if len(el_trk_cone_names) > 0 else None
172 el_calo_iso = ROOT.TTreeReaderArray(float)(treeReader, "Electrons_%s_%s"%("Corr" if options.useCorrectedCones else "Orig",el_calo_cone_names[0])) if len(el_calo_cone_names) > 0 else None
173
174 el_pass_iso = ROOT.TTreeReaderArray(bool)(treeReader, "Muons_%sPassIso"%("Corr" if options.useCorrectedCones else "Orig"))
175 mu_pass_iso = ROOT.TTreeReaderArray(bool)(treeReader, "Electrons_%sPassIso"%("Corr" if options.useCorrectedCones else "Orig"))
176
177
178 while treeReader.Next():
179 muon_t_vec = []
180 elec_t_vec = []
181
182 elec_t_calo_vec = []
183 muon_t_calo_vec = []
184
185 N_Isolated_Elec.Fill(len([i for i in range(el_pass_iso.GetSize()) if el_pass_iso[i] == True]))
186 N_Isolated_Muon.Fill(len([i for i in range(mu_pass_iso.GetSize()) if mu_pass_iso[i] == True]))
187
188 for m in range(mu_pt.GetSize()):
189 mu = ROOT.TLorentzVector()
190 mu.SetPtEtaPhiM( mu_pt[m], mu_eta[m], mu_phi[m], 0)
191 muon_t_vec += [mu]
192 calo_mu = ROOT.TLorentzVector()
193 calo_mu.SetPtEtaPhiM( mu_clu[m], mu_clu_eta[m], mu_clu_phi[m], 0)
194 muon_t_calo_vec += [calo_mu]
195
196 for e in range(el_pt.GetSize()):
197 el = ROOT.TLorentzVector()
198 el.SetPtEtaPhiM( el_pt[e], el_eta[e], el_phi[e], 0)
199 elec_t_vec += [el]
200 calo_el = ROOT.TLorentzVector()
201 calo_el.SetPtEtaPhiM( el_clu[e], el_clu_eta[e], el_clu_phi[e], 0)
202 elec_t_calo_vec += [calo_el]
203
204 for m, mu in enumerate(muon_t_vec):
205 if mu_trk_iso:
206 Muon_Track_Iso.Fill(mu_trk_iso[m] / 1.e3)
207 Muon_RelTrack_Iso.Fill( mu_trk_iso[m]/ mu.Pt())
208 if mu_calo_iso:
209 Muon_Calo_Iso.Fill(mu_calo_iso[m] / 1.e3)
210 Muon_RelTrack_Iso.Fill(mu_calo_iso[m] / mu.Pt())
211
212 for m1 in range(m):
213 if mu_trk_iso: Muon_Track_polution_dR.Fill( mu.DeltaR(muon_t_vec[m1]), mu_trk_iso[m] / mu_trk[m1] )
214 if mu_calo_iso: Muon_Topo_polution_dR.Fill( mu.DeltaR(muon_t_calo_vec[m1]), mu_calo_iso[m] / (mu_clu[m1] if mu_clu[m1]!= 0. else 1.) )
215
216 for e, el in enumerate(elec_t_vec):
217 if mu_trk_iso: Muon_Track_polution_dR.Fill( mu.DeltaR(el), mu_trk_iso[m] / el_trk[e] )
218 if mu_calo_iso: Muon_Topo_polution_dR.Fill( mu.DeltaR(el), mu_calo_iso[m] / el.E() )
219
220
221 for e, el in enumerate(elec_t_vec):
222 if el_trk_iso:
223 Elec_Track_Iso.Fill(el_trk_iso[e] / 1.e3)
224 Elec_RelTrack_Iso.Fill( el_trk_iso[e]/ el.Pt())
225 for e1 in range(e):
226 if el_trk_iso: Elec_Track_polution_dR.Fill( el.DeltaR(elec_t_vec[e1]), el_trk_iso[e] / el_trk[e1] )
227 if el_calo_iso: Elec_Topo_polution_dR.Fill( el.DeltaR(elec_t_vec[e1]), el_calo_iso[e] / elec_t_vec[e1].E() )
228
229 for m, mu in enumerate(muon_t_vec):
230 if el_trk_iso: Elec_Track_polution_dR.Fill( mu.DeltaR(el), el_trk_iso[e] / mu_trk[m])
231 if el_calo_iso: Elec_Topo_polution_dR.Fill( mu.DeltaR(muon_t_calo_vec[m]), el_calo_iso[e] / (mu_clu[m] if mu_clu[m]!= 0. else 1.))
232
233
234 # do this here, since before it destroys the argparse
235 ROOT.gROOT.Macro("rootlogon.C")
236 ROOT.gROOT.SetStyle("ATLAS")
237 ROOT.gROOT.SetBatch(1)
238 ROOT.gStyle.SetPalette(ROOT.kViridis)
239 if not os.path.exists(options.outDir):
240 logging.info("Create directory %s"%(options.outDir))
241 os.system("mkdir -p %s"%(options.outDir))
242
243 dummy_canvas = ROOT.TCanvas("dummy", "dummy", 800, 600)
244 dummy_canvas.SaveAs("%s/AllIsolationPlots_%s.pdf["%(options.outDir,
245 "Corr" if options.useCorrectedCones else "Orig" ) )
246
247 saved_once = False
248 for H in AllHistos: saved_once = saveHisto(options, H) or saved_once
249
250 if saved_once: dummy_canvas.SaveAs("%s/AllIsolationPlots_%s.pdf]"%(options.outDir,
251 "Corr" if options.useCorrectedCones else "Orig" ) )
252 else: os.system("rm %s/AllIsolationPlots_%s.pdf"%(options.outDir,
253 "Corr" if options.useCorrectedCones else "Orig" ) )
254
find
std::string find(const std::string &s)
return a remapped string
Definition hcg.cxx:138
plotIsoValidation.DrawTLatex
DrawTLatex(x, y, text, size=18, font=43, align=11)
Definition plotIsoValidation.py:7
plotIsoValidation.saveHisto
saveHisto(plot_options, histo)
Definition plotIsoValidation.py:17
Generated on for ATLAS Offline Software by doxygen 1.14.0