1 |
#!/usr/bin/env python
|
2 |
###################################################
|
3 |
# This python script ...
|
4 |
####################################################
|
5 |
|
6 |
import os,sys,string,re,math
|
7 |
from array import array
|
8 |
|
9 |
try:
|
10 |
from ROOT import *
|
11 |
except:
|
12 |
print "\nCannot load PYROOT, make sure you have setup ROOT in the path"
|
13 |
print "and pyroot library is also defined in the variable PYTHONPATH, try:\n"
|
14 |
if (os.getenv("PYTHONPATH")):
|
15 |
print " setenv PYTHONPATH ${PYTHONPATH}:$ROOTSYS/lib\n"
|
16 |
else:
|
17 |
print " setenv PYTHONPATH $ROOTSYS/lib\n"
|
18 |
#sys.exit()
|
19 |
|
20 |
#import Inspector
|
21 |
#import Style
|
22 |
|
23 |
gROOT.SetStyle("Plain")
|
24 |
gStyle.SetOptStat(0)
|
25 |
DEBUG = False
|
26 |
|
27 |
# style
|
28 |
#thestyle = Style.Style()
|
29 |
HasCMSStyle = False
|
30 |
style = None
|
31 |
if os.path.isfile('tdrstyle.C'):
|
32 |
gROOT.ProcessLine('.L tdrstyle.C')
|
33 |
ROOT.setTDRStyle()
|
34 |
print "Found tdrstyle.C file, using this style."
|
35 |
HasCMSStyle = True
|
36 |
if os.path.isfile('CMSTopStyle.cc'):
|
37 |
gROOT.ProcessLine('.L CMSTopStyle.cc+')
|
38 |
style = CMSTopStyle()
|
39 |
print "Found CMSTopStyle.cc file, use TOP style if requested in xml file."
|
40 |
if not HasCMSStyle:
|
41 |
print "No CMSTopStyle.cc"
|
42 |
|
43 |
#FOpen = ['results_ttbar.root',
|
44 |
# 'results_STtch.root',
|
45 |
# 'results_STtch_bar.root',
|
46 |
# 'results_STsch.root',
|
47 |
# 'results_STsch_bar.root',
|
48 |
# 'results_STtWch.root']
|
49 |
#
|
50 |
########################################
|
51 |
##def Disc(ofile, name):
|
52 |
#######################################
|
53 |
#for ofile in FOpen:
|
54 |
# tmpname = ofile.replace('results_','')
|
55 |
# name = tmpname.replace('.root','')
|
56 |
#
|
57 |
# infile = TFile(ofile)
|
58 |
# histo1 = infile.Get("/MET/Stlep_"+name)
|
59 |
# histo2 = infile.Get("/MET/Ht_"+name)
|
60 |
#
|
61 |
# scale1 = 1/histo1.Integral()
|
62 |
# scale2 = 1/histo2.Integral()
|
63 |
#
|
64 |
# histo1.Scale(scale1)
|
65 |
# histo2.Scale(scale2)
|
66 |
#
|
67 |
# #totalhisto = TH1F(histo1*histo2)
|
68 |
#
|
69 |
# #if ofile[1]:
|
70 |
# # c1.cd()
|
71 |
# # histo1.Draw("Hist")
|
72 |
# #else:
|
73 |
# # c1.cd()
|
74 |
# # histo1.Draw("Hist same")
|
75 |
# #if ofile[len(FOpen)]: c1.SaveAs("test.png")
|
76 |
|
77 |
def legend():
|
78 |
leg = TLegend(0.75,0.65,0.93,0.93)
|
79 |
leg.SetMargin(0.12)
|
80 |
leg.SetTextSize(0.035)
|
81 |
leg.SetFillColor(0)
|
82 |
leg.SetFillStyle(0)
|
83 |
leg.SetBorderSize(0)
|
84 |
leg.AddEntry(httbar_1, style.TtbarText, "l" )
|
85 |
leg.AddEntry(hSTsch_bar_1,style.ST_t_sText, "l" )
|
86 |
leg.AddEntry(hSTtWch_bar_1,style.ST_tWText, "l" )
|
87 |
leg.Draw()
|
88 |
return leg
|
89 |
|
90 |
|
91 |
# Open files
|
92 |
fttbar = TFile("/eos/uscms/store/user/algomez/FourTop/resultsTreeAnalyzer/results09/results_ttbar.root")
|
93 |
httbar_1 = fttbar.Get("/MET/Stjet_ttbar")
|
94 |
httbar_1.Scale(1/httbar_1.Integral())
|
95 |
httbar_2 = fttbar.Get("/MET/Ht_ttbar")
|
96 |
httbar_2.Scale(1/httbar_2.Integral())
|
97 |
httbar_3 = fttbar.Get("/MET/Stlep_ttbar")
|
98 |
httbar_3.Scale(1/httbar_3.Integral())
|
99 |
httbar_4 = fttbar.Get("/jets/Njets_cut1_ttbar")
|
100 |
httbar_4.Scale(1/httbar_4.Integral())
|
101 |
httbar_5 = fttbar.Get("/jets/Nbjets_CSVL_ttbar")
|
102 |
httbar_5.Scale(1/httbar_5.Integral())
|
103 |
#httbar_6 = fttbar.Get("/jets/Nbjets_CSVL_ttbar")
|
104 |
#httbar_6.Scale(1/httbar_6.Integral())
|
105 |
|
106 |
fSTtch = TFile("/eos/uscms/store/user/algomez/FourTop/resultsTreeAnalyzer/results09/results_STtch.root")
|
107 |
hSTtch_1 = fSTtch.Get("/MET/Stjet_STtch")
|
108 |
hSTtch_2 = fSTtch.Get("/MET/Ht_STtch")
|
109 |
hSTtch_3 = fSTtch.Get("/MET/Stlep_STtch")
|
110 |
hSTtch_4 = fSTtch.Get("/jets/Njets_cut1_STtch")
|
111 |
hSTtch_5 = fSTtch.Get("/jets/Nbjets_CSVL_STtch")
|
112 |
|
113 |
fSTtch_bar = TFile("/eos/uscms/store/user/algomez/FourTop/resultsTreeAnalyzer/results09/results_STtch_bar.root")
|
114 |
hSTtch_bar_1 = fSTtch_bar.Get("/MET/Stjet_STtch_bar")
|
115 |
hSTtch_bar_1.Add(hSTtch_1)
|
116 |
hSTtch_bar_2 = fSTtch_bar.Get("/MET/Ht_STtch_bar")
|
117 |
hSTtch_bar_2.Add(hSTtch_2)
|
118 |
hSTtch_bar_3 = fSTtch_bar.Get("/MET/Stlep_STtch_bar")
|
119 |
hSTtch_bar_3.Add(hSTtch_3)
|
120 |
hSTtch_bar_4 = fSTtch_bar.Get("/jets/Njets_cut1_STtch_bar")
|
121 |
hSTtch_bar_4.Add(hSTtch_4)
|
122 |
hSTtch_bar_5 = fSTtch_bar.Get("/jets/Nbjets_CSVL_STtch_bar")
|
123 |
hSTtch_bar_5.Add(hSTtch_5)
|
124 |
|
125 |
fSTsch = TFile("/eos/uscms/store/user/algomez/FourTop/resultsTreeAnalyzer/results09/results_STsch.root")
|
126 |
hSTsch_1 = fSTsch.Get("/MET/Stjet_STsch")
|
127 |
hSTsch_1.Add(hSTtch_bar_1)
|
128 |
hSTsch_2 = fSTsch.Get("/MET/Ht_STsch")
|
129 |
hSTsch_2.Add(hSTtch_bar_2)
|
130 |
hSTsch_3 = fSTsch.Get("/MET/Stlep_STsch")
|
131 |
hSTsch_3.Add(hSTtch_bar_3)
|
132 |
hSTsch_4 = fSTsch.Get("/jets/Njets_cut1_STsch")
|
133 |
hSTsch_4.Add(hSTtch_bar_4)
|
134 |
hSTsch_5 = fSTsch.Get("/jets/Nbjets_CSVL_STsch")
|
135 |
hSTsch_5.Add(hSTtch_bar_5)
|
136 |
|
137 |
fSTsch_bar = TFile("/eos/uscms/store/user/algomez/FourTop/resultsTreeAnalyzer/results09/results_STsch_bar.root")
|
138 |
hSTsch_bar_1 = fSTsch_bar.Get("/MET/Stjet_STsch_bar")
|
139 |
hSTsch_bar_1.Add(hSTsch_1)
|
140 |
hSTsch_bar_1.Sumw2()
|
141 |
hSTsch_bar_1.Scale(1/hSTsch_bar_1.Integral())
|
142 |
hSTsch_bar_2 = fSTsch_bar.Get("/MET/Ht_STsch_bar")
|
143 |
hSTsch_bar_2.Add(hSTsch_2)
|
144 |
hSTsch_bar_2.Sumw2()
|
145 |
hSTsch_bar_2.Scale(1/hSTsch_bar_2.Integral())
|
146 |
hSTsch_bar_3 = fSTsch_bar.Get("/MET/Stlep_STsch_bar")
|
147 |
hSTsch_bar_3.Add(hSTsch_3)
|
148 |
hSTsch_bar_3.Sumw2()
|
149 |
hSTsch_bar_3.Scale(1/hSTsch_bar_3.Integral())
|
150 |
hSTsch_bar_4 = fSTsch_bar.Get("/jets/Njets_cut1_STsch_bar")
|
151 |
hSTsch_bar_4.Add(hSTsch_4)
|
152 |
hSTsch_bar_4.Sumw2()
|
153 |
hSTsch_bar_4.Scale(1/hSTsch_bar_4.Integral())
|
154 |
hSTsch_bar_5 = fSTsch_bar.Get("/jets/Nbjets_CSVL_STsch_bar")
|
155 |
hSTsch_bar_5.Add(hSTsch_5)
|
156 |
hSTsch_bar_5.Sumw2()
|
157 |
hSTsch_bar_5.Scale(1/hSTsch_bar_5.Integral())
|
158 |
|
159 |
fSTtWch = TFile("/eos/uscms/store/user/algomez/FourTop/resultsTreeAnalyzer/results09/results_STtWch.root")
|
160 |
hSTtWch_1 = fSTtWch.Get("/MET/Stjet_STtWch")
|
161 |
hSTtWch_2 = fSTtWch.Get("/MET/Ht_STtWch")
|
162 |
hSTtWch_3 = fSTtWch.Get("/MET/Stlep_STtWch")
|
163 |
hSTtWch_4 = fSTtWch.Get("/jets/Njets_cut1_STtWch")
|
164 |
hSTtWch_5 = fSTtWch.Get("/jets/Nbjets_CSVL_STtWch")
|
165 |
|
166 |
fSTtWch_bar = TFile("/eos/uscms/store/user/algomez/FourTop/resultsTreeAnalyzer/results09/results_STtWch_bar.root")
|
167 |
hSTtWch_bar_1 = fSTtWch_bar.Get("/MET/Stjet_STtWch_bar")
|
168 |
hSTtWch_bar_1.Add(hSTtWch_1)
|
169 |
hSTtWch_bar_1.Sumw2()
|
170 |
hSTtWch_bar_1.Scale(1/hSTtWch_bar_1.Integral())
|
171 |
hSTtWch_bar_2 = fSTtWch_bar.Get("/MET/Ht_STtWch_bar")
|
172 |
hSTtWch_bar_2.Add(hSTtWch_2)
|
173 |
hSTtWch_bar_2.Sumw2()
|
174 |
hSTtWch_bar_2.Scale(1/hSTtWch_bar_2.Integral())
|
175 |
hSTtWch_bar_3 = fSTtWch_bar.Get("/MET/Stlep_STtWch_bar")
|
176 |
hSTtWch_bar_3.Add(hSTtWch_3)
|
177 |
hSTtWch_bar_3.Sumw2()
|
178 |
hSTtWch_bar_3.Scale(1/hSTtWch_bar_3.Integral())
|
179 |
hSTtWch_bar_4 = fSTtWch_bar.Get("/jets/Njets_cut1_STtWch_bar")
|
180 |
hSTtWch_bar_4.Add(hSTtWch_4)
|
181 |
hSTtWch_bar_4.Sumw2()
|
182 |
hSTtWch_bar_4.Scale(1/hSTtWch_bar_4.Integral())
|
183 |
hSTtWch_bar_5 = fSTtWch_bar.Get("/jets/Nbjets_CSVL_STtWch_bar")
|
184 |
hSTtWch_bar_5.Add(hSTtWch_5)
|
185 |
hSTtWch_bar_5.Sumw2()
|
186 |
hSTtWch_bar_5.Scale(1/hSTtWch_bar_5.Integral())
|
187 |
|
188 |
|
189 |
can1 = TCanvas('can1', '', 800,600)
|
190 |
can1.cd()
|
191 |
can1.SetLogy()
|
192 |
|
193 |
httbar_1.SetLineColor(style.TtbarColor)
|
194 |
httbar_1.SetMaximum(1)
|
195 |
httbar_1.SetYTitle('1/N dN/dx')
|
196 |
httbar_1.Draw("Hist")
|
197 |
|
198 |
hSTsch_bar_1.SetLineColor(style.ST_t_sColor)
|
199 |
hSTsch_bar_1.SetLineWidth(1)
|
200 |
hSTsch_bar_1.Draw("Hist same")
|
201 |
|
202 |
hSTtWch_bar_1.SetLineColor(style.ST_tWColor)
|
203 |
hSTtWch_bar_1.SetLineWidth(2)
|
204 |
hSTtWch_bar_1.Draw("Hist same")
|
205 |
|
206 |
leg = legend()
|
207 |
can1.SaveAs("diff_Stjet_Top.png")
|
208 |
|
209 |
can2 = TCanvas('can2', '', 800,600)
|
210 |
can2.cd()
|
211 |
can2.SetLogy()
|
212 |
|
213 |
httbar_2.SetLineColor(style.TtbarColor)
|
214 |
httbar_2.SetMaximum(1)
|
215 |
httbar_2.SetYTitle('1/N dN/dx')
|
216 |
httbar_2.Draw("Hist")
|
217 |
|
218 |
hSTsch_bar_2.SetLineColor(style.ST_t_sColor)
|
219 |
hSTsch_bar_2.SetLineWidth(1)
|
220 |
hSTsch_bar_2.Draw("Hist same")
|
221 |
|
222 |
hSTtWch_bar_2.SetLineColor(style.ST_tWColor)
|
223 |
hSTtWch_bar_2.SetLineWidth(2)
|
224 |
hSTtWch_bar_2.Draw("Hist same")
|
225 |
|
226 |
leg = legend()
|
227 |
can2.SaveAs("diff_Ht_Top.png")
|
228 |
|
229 |
can3 = TCanvas('can3', '', 800,600)
|
230 |
can3.cd()
|
231 |
can3.SetLogy()
|
232 |
|
233 |
httbar_3.SetLineColor(style.TtbarColor)
|
234 |
httbar_3.SetMaximum(1)
|
235 |
httbar_3.SetYTitle('1/N dN/dx')
|
236 |
httbar_3.Draw("Hist")
|
237 |
|
238 |
hSTsch_bar_3.SetLineColor(style.ST_t_sColor)
|
239 |
hSTsch_bar_3.SetLineWidth(1)
|
240 |
hSTsch_bar_3.Draw("Hist same")
|
241 |
|
242 |
hSTtWch_bar_3.SetLineColor(style.ST_tWColor)
|
243 |
hSTtWch_bar_3.SetLineWidth(2)
|
244 |
hSTtWch_bar_3.Draw("Hist same")
|
245 |
|
246 |
leg = legend()
|
247 |
can3.SaveAs("diff_Stlep_Top.png")
|
248 |
|
249 |
can4 = TCanvas('can4', '', 800,600)
|
250 |
can4.cd()
|
251 |
can4.SetLogy()
|
252 |
|
253 |
httbar_4.SetLineColor(style.TtbarColor)
|
254 |
httbar_4.SetMaximum(1)
|
255 |
httbar_4.SetYTitle('1/N dN/dx')
|
256 |
httbar_4.Draw("Hist")
|
257 |
|
258 |
hSTsch_bar_4.SetLineColor(style.ST_t_sColor)
|
259 |
hSTsch_bar_4.SetLineWidth(1)
|
260 |
hSTsch_bar_4.Draw("Hist same")
|
261 |
|
262 |
hSTtWch_bar_4.SetLineColor(style.ST_tWColor)
|
263 |
hSTtWch_bar_4.SetLineWidth(2)
|
264 |
hSTtWch_bar_4.Draw("Hist same")
|
265 |
leg = legend()
|
266 |
can4.SaveAs("diff_Njets_Top.png")
|
267 |
|
268 |
can5 = TCanvas('can5', '', 800,600)
|
269 |
can5.cd()
|
270 |
can5.SetLogy()
|
271 |
|
272 |
httbar_5.SetLineColor(style.TtbarColor)
|
273 |
httbar_5.SetMaximum(1)
|
274 |
httbar_5.SetYTitle('1/N dN/dx')
|
275 |
httbar_5.Draw("Hist")
|
276 |
|
277 |
hSTsch_bar_5.SetLineColor(style.ST_t_sColor)
|
278 |
hSTsch_bar_5.SetLineWidth(1)
|
279 |
hSTsch_bar_5.Draw("Hist same")
|
280 |
|
281 |
hSTtWch_bar_5.SetLineColor(style.ST_tWColor)
|
282 |
hSTtWch_bar_5.SetLineWidth(2)
|
283 |
hSTtWch_bar_5.Draw("Hist same")
|
284 |
|
285 |
leg = legend()
|
286 |
can5.SaveAs("diff_Nbjets_CSVL_Top.png")
|