ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/nowaf/StudyXClean/rootScripts/MakeMetStudy.py
Revision: 1.1
Committed: Thu Feb 25 06:29:24 2010 UTC (15 years, 2 months ago) by nowak
Content type: text/x-python
Branch: MAIN
Log Message:
more stuff

File Contents

# Content
1 import ROOT
2 import sys
3 sys.path.append( '/afs/naf.desy.de/user/n/nowaf/CMSSW_3_3_6/src/LeptonSel/LeptonSel/rootScripts' )
4 import definitions as Def
5
6 def main(argv):
7 #import sys
8
9 basepath = ''
10 top = ''
11
12 try:
13 options, arguments = getopt.getopt(argv, "hp:t:", ["help","path=","topology="])
14
15 except getopt.GetoptError:
16 usage()
17 sys.exit(2)
18
19 for opt, arg in options:
20 if opt in ("-h", "--help"):
21 usage()
22 sys.exit(2)
23 elif opt in ("-p", "--path"):
24 basepath = arg
25 elif opt in ("-t", "--topology"):
26 top = arg
27
28 if basepath == '':
29 print "Error! Need path to .root file to continue."
30 sys.exit(2)
31 elif top == '':
32 print "Warning! no topology specified. Using 'all' "
33 top = 'all'
34
35 return basepath, top
36
37 def usage():
38
39 print "Script to plot all XC study plots for the di lepton analysis"
40 print "Options and arguments:"
41 print "-h, --help\t\t: print this help message and exit"
42 print "-p, --path\t\t: relative path to the .root file"
43
44 pass
45
46 if __name__ == "__main__":
47
48 import getopt
49
50 basepath, top = main(sys.argv[1:])
51
52
53 Def.SetGlobalStyles( "emr" )
54 ROOT.gStyle.SetPadRightMargin( 0.1 )
55
56 saves = {}
57
58 fin = Def.ReadInRootFile( basepath )
59 fin.cd()
60
61 c_mets = ROOT.TCanvas( "c_mets", "mets" ,1,71,600,600 )
62 mets = fin.Get( "h_newMet_vs_oldMet_" + top )
63 print type( fin )
64 print type( mets )
65 mets.SetXTitle( "new MET [GeV]" )
66 mets.SetYTitle( "old MET [GeV]" )
67 mets.SetTitle( "" )
68 mets.Draw( "COLZ" )
69 saves["mets"] = mets
70
71 c_jeNumbers = ROOT.TCanvas( "c_jeNumbers", "jeNumbers" ,1,71,600,300 )
72 c_jeNumbers.Divide(2,1)
73 c_jeNumbers.cd(1)
74 je_new = fin.Get( "h_j_vs_e_newMETg_" + top )
75 je_new.SetXTitle( "number of jets altered" )
76 je_new.SetYTitle( "number of electrons dropped" )
77 je_new.SetTitle( "newMET > oldMET" )
78 je_new.Draw( "COLZ" )
79 c_jeNumbers.cd(2)
80 je_old = fin.Get( "h_j_vs_e_oldMETg_" + top )
81 je_old.SetXTitle( "number of jets altered" )
82 je_old.SetYTitle( "number of electrons dropped" )
83 je_old.SetTitle( "newMET < oldMET" )
84 je_old.Draw( "COLZ" )
85 saves["je_new"] = je_new
86 saves["je_old"] = je_old
87
88 c_jeNumbers_2 = ROOT.TCanvas( "c_jeNumbers_2", "jeNumbers_2" ,1,71,600,300 )
89 c_jeNumbers_2.Divide(2,1)
90 c_jeNumbers_2.cd(1)
91 je_new_2 = fin.Get( "h_dj_vs_ie_newMETg_" + top )
92 je_new_2.SetXTitle( "number of jets dropped" )
93 je_new_2.SetYTitle( "number of electrons isolated" )
94 je_new_2.SetTitle( "newMET > oldMET" )
95 je_new_2.Draw( "COLZ" )
96 c_jeNumbers_2.cd(2)
97 je_old_2 = fin.Get( "h_dj_vs_ie_oldMETg_" + top )
98 je_old_2.SetXTitle( "number of jets dropped" )
99 je_old_2.SetYTitle( "number of electrons isolated" )
100 je_old_2.SetTitle( "newMET < oldMET" )
101 je_old_2.Draw( "COLZ" )
102 saves["je_new_2"] = je_new_2
103 saves["je_old_2"] = je_old_2
104
105 c_jeNumbers_3 = ROOT.TCanvas( "c_jeNumbers_3", "jeNumbers_3" ,1,71,600,300 )
106 c_jeNumbers_3.Divide(2,1)
107 c_jeNumbers_3.cd(1)
108 je_new_3 = fin.Get( "h_aj_vs_ie_newMETg_" + top )
109 je_new_3.SetXTitle( "number of jets altered" )
110 je_new_3.SetYTitle( "number of electrons isolated" )
111 je_new_3.SetTitle( "newMET > oldMET" )
112 je_new_3.Draw( "COLZ" )
113 c_jeNumbers_3.cd(2)
114 je_old_3 = fin.Get( "h_aj_vs_ie_oldMETg_" + top )
115 je_old_3.SetXTitle( "number of jets altered" )
116 je_old_3.SetYTitle( "number of electrons isolated" )
117 je_old_3.SetTitle( "newMET < oldMET" )
118 je_old_3.Draw( "COLZ" )
119 saves["je_new_3"] = je_new_3
120 saves["je_old_3"] = je_old_3
121
122 c_jeNumbers_Et = ROOT.TCanvas( "c_jeNumbers_Et", "jeNumbers_Et" ,1,71,600,600 )
123 c_jeNumbers_Et.Divide(2,2)
124 c_jeNumbers_Et.cd(1)
125 je_m_new_Et = fin.Get( "h_m_anewj_vs_aoldj_newMETg_" + top )
126 je_m_new_Et.SetXTitle( "new Jet E_{t} (matched) [GeV]" )
127 je_m_new_Et.SetYTitle( "old Jet E_{t} (matched) [GeV]" )
128 je_m_new_Et.SetTitle( "newMET > oldMET" )
129 je_m_new_Et.Draw( "COLZ" )
130 c_jeNumbers_Et.cd(2)
131 je_m_old_Et = fin.Get( "h_m_anewj_vs_aoldj_oldMETg_" + top )
132 je_m_old_Et.SetXTitle( "new Jet E_{t} (matched) [GeV]" )
133 je_m_old_Et.SetYTitle( "old Jet E_{t} (matched) [GeV]" )
134 je_m_old_Et.SetTitle( "newMET < oldMET" )
135 je_m_old_Et.Draw( "COLZ" )
136 saves["je_m_new_Et"] = je_m_new_Et
137 saves["je_m_old_Et"] = je_m_old_Et
138 c_jeNumbers_Et.cd(3)
139 je_nm_new_Et = fin.Get( "h_nm_anewj_vs_aoldj_newMETg_" + top )
140 je_nm_new_Et.SetXTitle( "new Jet E_{t} (not matched) [GeV]" )
141 je_nm_new_Et.SetYTitle( "old Jet E_{t} (not matched) [GeV]" )
142 je_nm_new_Et.SetTitle( "newMET > oldMET" )
143 je_nm_new_Et.Draw( "COLZ" )
144 c_jeNumbers_Et.cd(4)
145 je_nm_old_Et = fin.Get( "h_nm_anewj_vs_aoldj_oldMETg_" + top )
146 je_nm_old_Et.SetXTitle( "new Jet E_{t} (not matched) [GeV]" )
147 je_nm_old_Et.SetYTitle( "old Jet E_{t} (not matched) [GeV]" )
148 je_nm_old_Et.SetTitle( "newMET < oldMET" )
149 je_nm_old_Et.Draw( "COLZ" )
150 saves["je_nm_new_Et"] = je_nm_new_Et
151 saves["je_nm_old_Et"] = je_nm_old_Et
152
153 c_jeNumbers_Diff = ROOT.TCanvas( "c_jeNumbers_Diff", "jeNumbers_Diff" ,1,71,600,600 )
154 c_jeNumbers_Diff.Divide(2,2)
155 c_jeNumbers_Diff.cd(1)
156 je_m_new_Diff = fin.Get( "h_m_Danewj_vs_Daoldj_newMETg_" + top )
157 je_m_new_Diff.SetXTitle( "Et(CORR) - Et(RAW) new jet (matched) [GeV]" )
158 je_m_new_Diff.SetYTitle( "Et(CORR) - Et(RAW) old jet (matched) [GeV]" )
159 je_m_new_Diff.SetTitle( "newMET > oldMET" )
160 je_m_new_Diff.Draw( "COLZ" )
161 c_jeNumbers_Diff.cd(2)
162 je_m_old_Diff = fin.Get( "h_m_Danewj_vs_Daoldj_oldMETg_" + top )
163 je_m_old_Diff.SetXTitle( "Et(CORR) - Et(RAW) new jet (matched) [GeV]" )
164 je_m_old_Diff.SetYTitle( "Et(CORR) - Et(RAW) new jet (matched) [GeV]" )
165 je_m_old_Diff.SetTitle( "newMET < oldMET" )
166 je_m_old_Diff.Draw( "COLZ" )
167 saves["je_m_new_Diff"] = je_m_new_Diff
168 saves["je_m_old_Diff"] = je_m_old_Diff
169 c_jeNumbers_Diff.cd(3)
170 je_nm_new_Diff = fin.Get( "h_nm_Danewj_vs_Daoldj_newMETg_" + top )
171 je_nm_new_Diff.SetXTitle( "Et(CORR) - Et(RAW) new jet (not matched) [GeV]" )
172 je_nm_new_Diff.SetYTitle( "Et(CORR) - Et(RAW) old jet (not matched) [GeV]" )
173 je_nm_new_Diff.SetTitle( "newMET > oldMET" )
174 je_nm_new_Diff.Draw( "COLZ" )
175 c_jeNumbers_Diff.cd(4)
176 je_nm_old_Diff = fin.Get( "h_nm_Danewj_vs_Daoldj_oldMETg_" + top )
177 je_nm_old_Diff.SetXTitle( "Et(CORR) - Et(RAW) new jet (not matched) [GeV]" )
178 je_nm_old_Diff.SetYTitle( "Et(CORR) - Et(RAW) new jet (not matched) [GeV]" )
179 je_nm_old_Diff.SetTitle( "newMET < oldMET" )
180 je_nm_old_Diff.Draw( "COLZ" )
181 saves["je_nm_new_Diff"] = je_nm_new_Diff
182 saves["je_nm_old_Diff"] = je_nm_old_Diff
183
184
185 ### number of matched/notmatched altered/dropped/not changed jets/elecs
186 hist_list_j_new = [ fin.Get( "h_j_nm_newMetg_" + top ), fin.Get( "h_j_m_newMetg_" + top ),\
187 fin.Get( "h_aj_nm_newMetg_" + top ), fin.Get( "h_aj_m_newMetg_" + top ), \
188 fin.Get( "h_dj_nm_newMetg_" + top ), fin.Get( "h_dj_m_newMetg_" + top ) ]
189
190 color_list_j = [50,8,12,9,28,30]
191 case_list_j = ["nm, not altered", "m, notaltered", "nm, altered", "m altered", "nm dropped", "m, dropped"]
192
193 c_je = ROOT.TCanvas( "c_je", "je" ,1,71,600,600 )
194 c_je.Divide(2,2)
195
196 c_je.cd(1)
197 Def.DrawSmall( c_je,\
198 hist_list_j_new, color_list_j, case_list_j, \
199 "number of jets", "", "new MET > old MET",\
200 True, 0, False,\
201 "ur", "je1", saves)
202
203 hist_list_j_old = [ fin.Get( "h_j_nm_oldMetg_" + top ), fin.Get( "h_j_m_oldMetg_" + top ),\
204 fin.Get( "h_aj_nm_oldMetg_" + top ), fin.Get( "h_aj_m_oldMetg_" + top ), \
205 fin.Get( "h_dj_nm_oldMetg_" + top ), fin.Get( "h_dj_m_oldMetg_" + top ) ]
206
207 c_je.cd(2)
208 Def.DrawSmall( c_je,\
209 hist_list_j_old, color_list_j, case_list_j, \
210 "number of jets", "", "new MET < old MET",\
211 True, 0, False,\
212 "ur", "je2", saves)
213
214 hist_list_e_new = [ fin.Get( "h_e_nm_newMetg_" + top ), fin.Get( "h_e_m_newMetg_" + top ),\
215 fin.Get( "h_de_nm_newMetg_" + top ), fin.Get( "h_de_m_newMetg_" + top ) ]
216
217 color_list_e = [50,8,28,30]
218 case_list_e = ["nm ", "m ","nm dropped", "m, dropped"]
219
220 c_je.cd(3)
221 Def.DrawSmall( c_je,\
222 hist_list_e_new, color_list_e, case_list_e, \
223 "number of electrons", "", "new MET > old MET",\
224 True, 0, False,\
225 "ur", "je3", saves)
226
227 hist_list_e_old = [ fin.Get( "h_e_nm_oldMetg_" + top ), fin.Get( "h_e_m_oldMetg_" + top ),\
228 fin.Get( "h_de_nm_oldMetg_" + top ), fin.Get( "h_de_m_oldMetg_" + top ) ]
229
230 c_je.cd(4)
231 Def.DrawSmall( c_je,\
232 hist_list_e_old, color_list_e, case_list_e, \
233 "number of electrons", "", "new MET < old MET",\
234 True, 0, False,\
235 "ur", "je4", saves)
236
237 ### Pt of matched/notmatched altered/dropped/not changed jets/elecs
238 hist_list_jPt_new = [ fin.Get( "h_jPt_nm_newMetg_" + top ), fin.Get( "h_jPt_m_newMetg_" + top ),\
239 fin.Get( "h_ajPt_nm_newMetg_" + top ), fin.Get( "h_ajPt_m_newMetg_" + top ), \
240 fin.Get( "h_djPt_nm_newMetg_" + top ), fin.Get( "h_djPt_m_newMetg_" + top ) ]
241
242 #color_list_j = [50,8,12,9,28,30]
243 #case_list_j = ["nm, not altered", "m, notaltered", "nm, altered", "m altered", "nm dropped", "m, dropped"]
244
245 c_jePt = ROOT.TCanvas( "c_jePt", "jePt" ,1,71,600,600 )
246 c_jePt.Divide(2,2)
247
248 c_jePt.cd(1)
249 Def.DrawSmall( c_jePt,\
250 hist_list_jPt_new, color_list_j, case_list_j, \
251 "P_{t} (jets) [GeV]", "", "new MET > old MET",\
252 True, 0, False,\
253 "ur", "jePt1", saves)
254
255 hist_list_jPt_old = [ fin.Get( "h_jPt_nm_oldMetg_" + top ), fin.Get( "h_jPt_m_oldMetg_" + top ),\
256 fin.Get( "h_ajPt_nm_oldMetg_" + top ), fin.Get( "h_ajPt_m_oldMetg_" + top ), \
257 fin.Get( "h_djPt_nm_oldMetg_" + top ), fin.Get( "h_djPt_m_oldMetg_" + top ) ]
258
259 c_jePt.cd(2)
260 Def.DrawSmall( c_jePt,\
261 hist_list_jPt_old, color_list_j, case_list_j, \
262 "P_{t} (jets) [GeV]", "", "new MET < old MET",\
263 True, 0, False,\
264 "ur", "jePt2", saves)
265
266 hist_list_ePt_new = [ fin.Get( "h_ePt_nm_newMetg_" + top ), fin.Get( "h_ePt_m_newMetg_" + top ),\
267 fin.Get( "h_dePt_nm_newMetg_" + top ), fin.Get( "h_dePt_m_newMetg_" + top ) ]
268
269 #color_list_e = [50,8,28,30]
270 #case_list_e = ["nm ", "m ","nm dropped", "m, dropped"]
271
272 c_jePt.cd(3)
273 Def.DrawSmall( c_jePt,\
274 hist_list_ePt_new, color_list_e, case_list_e, \
275 "P_{t} (electrons) [GeV]", "", "new MET > old MET",\
276 True, 0, False,\
277 "ur", "jePt3", saves)
278
279 hist_list_ePt_old = [ fin.Get( "h_ePt_nm_oldMetg_" + top ), fin.Get( "h_ePt_m_oldMetg_" + top ),\
280 fin.Get( "h_dePt_nm_oldMetg_" + top ), fin.Get( "h_dePt_m_oldMetg_" + top ) ]
281
282 c_jePt.cd(4)
283 Def.DrawSmall( c_jePt,\
284 hist_list_ePt_old, color_list_e, case_list_e, \
285 "P_{t} (electrons) [GeV]", "", "new MET < old MET",\
286 True, 0, False,\
287 "ur", "jePt4", saves)
288
289
290 c_jcorr = ROOT.TCanvas( "c_jcorr", "jcorr" ,1,71,600,300 )
291 c_jcorr.Divide(2,1)
292
293 hist_list_jraw = [ fin.Get( "h_DRawj_nm_newMetg_" + top ), fin.Get( "h_DRawj_m_newMetg_" + top ),\
294 fin.Get( "h_DRawj_nm_oldMetg_" + top ), fin.Get( "h_DRawj_m_oldMetg_" + top ) ]
295
296 case_list_corr = [ "nm, new > old", "m, new > old", "nm, new < old", "m, new < old" ]
297
298 c_jcorr.cd(1)
299 Def.DrawSmall( c_jcorr,\
300 hist_list_jraw, color_list_e, case_list_corr, \
301 "j_{new} (RAW) - j_{old} (RAW)", "", "",\
302 False, 0, False,\
303 "ur", "jePtRaw", saves)
304
305 hist_list_jcorr = [ fin.Get( "h_DCorrj_nm_newMetg_" + top ), fin.Get( "h_DCorrj_m_newMetg_" + top ),\
306 fin.Get( "h_DCorrj_nm_oldMetg_" + top ), fin.Get( "h_DCorrj_m_oldMetg_" + top ) ]
307
308 c_jcorr.cd(2)
309 Def.DrawSmall( c_jcorr,\
310 hist_list_jcorr, color_list_e, case_list_corr, \
311 "#Delta j_{new} - #Delta j_{old}", "", "",\
312 False, 0, False,\
313 "ur", "jePtCorr", saves)
314
315 c_acc = ROOT.TCanvas( "c_acc", "acc" ,1,71,600,600 )
316 c_acc.Divide(2,2)
317
318 hist_list_eta = [ fin.Get( "h_e_m_eta_newMetg_" + top ), fin.Get( "h_e_m_eta_oldMetg_" + top ),\
319 fin.Get( "h_e_m_eta_sameMet_" + top )]
320
321 color_list_acc = [50,8,13]
322 case_list_acc = [ "new > old", "new < old", "new = old" ]
323
324 c_acc.cd(1)
325 Def.DrawSmall( c_acc,\
326 hist_list_eta, color_list_acc, case_list_acc, \
327 "#eta (electrons)", "", "",\
328 False, 0, False,\
329 "ur", "eeta", saves)
330
331 hist_list_number = [ fin.Get( "h_ie_newMetg_" + top ), fin.Get( "h_ie_oldMetg_" + top ),\
332 fin.Get( "h_ie_sameMet_" + top )]
333
334 c_acc.cd(2)
335 Def.DrawSmall( c_acc,\
336 hist_list_number, color_list_acc, case_list_acc, \
337 "number of isolated electrons", "", "",\
338 False, 0, False,\
339 "ur", "ie", saves)
340
341 hist_list_gen = [ fin.Get( "h_gene_acc_newMetg_" + top ), fin.Get( "h_gene_acc_oldMetg_" + top ),\
342 fin.Get( "h_gene_acc_sameMet_" + top )]
343
344 c_acc.cd(3)
345 Def.DrawSmall( c_acc,\
346 hist_list_gen, color_list_acc, case_list_acc, \
347 "number of gen electrons in acceptance", "", "",\
348 False, 0, False,\
349 "ur", "gene", saves)
350
351 #c_acc.cd(4)
352
353 c_geneie = ROOT.TCanvas( "c_gene_ie", "gene_ie" ,1,71,600,600 )
354 c_geneie.Divide(2,2)
355
356 c_geneie.cd(1)
357 gene_vs_ie = fin.Get( "h_gene_vs_ie_newMETg_" + top )
358 gene_vs_ie.SetXTitle( "number of gen electrons in acceptance" )
359 gene_vs_ie.SetYTitle( "number of isolated electrons" )
360 gene_vs_ie.SetTitle( "newMET > oldMET" )
361 gene_vs_ie.Draw( "COLZ" )
362 saves["gene_vs_ie"] = gene_vs_ie
363
364 c_geneie.cd(2)
365 gene_vs_ie_old = fin.Get( "h_gene_vs_ie_oldMETg_" + top )
366 gene_vs_ie_old.SetXTitle( "number of gen electrons in acceptance" )
367 gene_vs_ie_old.SetYTitle( "number of isolated electrons" )
368 gene_vs_ie_old.SetTitle( "newMET < oldMET" )
369 gene_vs_ie_old.Draw( "COLZ" )
370 saves["gene_vs_ie_old"] = gene_vs_ie_old
371
372 c_geneie.cd(3)
373 gene_vs_ie_same = fin.Get( "h_gene_vs_ie_sameMET_" + top )
374 gene_vs_ie_same.SetXTitle( "number of gen electrons in acceptance" )
375 gene_vs_ie_same.SetYTitle( "number of isolated electrons" )
376 gene_vs_ie_same.SetTitle( "newMET = oldMET" )
377 gene_vs_ie_same.Draw( "COLZ" )
378 saves["gene_vs_ie_same"] = gene_vs_ie_same
379
380
381 Def.DontQuit()