ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/dhidas/OSUAnalysis/Tools/scripts/plotWpriOnly.py
Revision: 1.1.1.1 (vendor branch)
Committed: Thu Dec 1 16:28:48 2011 UTC (13 years, 5 months ago) by dhidas
Content type: text/x-python
Branch: dhidas, MAIN
CVS Tags: START, HEAD
Changes since 1.1: +0 -0 lines
Log Message:
osu copy modified

File Contents

# User Rev Content
1 dhidas 1.1 from __future__ import division
2    
3     from tdrStyle import *
4     from math import sqrt
5     from ROOT import *
6    
7     canvases = []
8     scanvases = []
9     counter = 0
10    
11     def plotMttbar():
12     tdrstyle = setTDRStyle();
13     gStyle.SetHatchesSpacing(1.0);
14     lumi = 4000;
15     oldLumi = 4000;
16     scale = lumi / oldLumi;
17     qcdScale = {'default':1.72, 'withMETAndAsymJets': 3.03};
18     # data = TFile.Open("data2.9pb_fullSetOfVars.root");
19    
20     # data = TFile.Open("Run_4000pb_PFElectron_PF2PATJets_PFMET.root");
21     # ttbar = TFile.Open("TTJets_4000pb_PFElectron_PF2PATJets_PFMET.root");
22     # ttbar = TFile.Open("TT_Tune_4000pb_PFElectron_PF2PATJets_PFMET.root");
23     # wjets = TFile.Open("WJetsToLNu_4000pb_PFElectron_PF2PATJets_PFMET.root");
24     # wjets = TFile.Open("WToENu_4000pb_PFElectron_PF2PATJets_PFMET.root");
25     # zjets = TFile.Open("DYJetsToLL_4000pb_PFElectron_PF2PATJets_PFMET.root");
26     # bce1 = TFile.Open("QCD_Pt-20to30_BCtoE_4000pb_PFElectron_PF2PATJets_PFMET.root");
27     # bce2 = TFile.Open("QCD_Pt-30to80_BCtoE_4000pb_PFElectron_PF2PATJets_PFMET.root");
28     # bce3 = TFile.Open("QCD_Pt-80to170_BCtoE_4000pb_PFElectron_PF2PATJets_PFMET.root");
29     # enri1 = TFile.Open("QCD_Pt-20to30_EMEnriched_35.9pb_PFElectron_PF2PATJets_PFMET.root");
30     # enri2 = TFile.Open("QCD_Pt-30to80_EMEnriched_35.9pb_PFElectron_PF2PATJets_PFMET.root");
31     # enri3 = TFile.Open("QCD_Pt-80to170_EMEnriched_35.9pb_PFElectron_PF2PATJets_PFMET.root");
32     # zjets = TFile.Open("DYJets_35.9pb_PFElectron_PF2PATJets_PFMET.root");
33     # Zprime500 = TFile.Open("Zprime_M500GeV_W50GeV_4000pb_PFElectron_PF2PATJets_PFMET.root");
34     # Zprime1500 = TFile.Open("Zprime_M1500GeV_W15GeV_4000pb_PFElectron_PF2PATJets_PFMET.root");
35     # Zprime750 = TFile.Open("Zprime_M750GeV_W7500MeV_35.9pb_PFElectron_PF2PATJets_PFMET.root");
36     # Zprime1000 = TFile.Open("Zprime_M1000GeV_W10GeV_35.9pb_PFElectron_PF2PATJets_PFMET.root");
37     # Zprime2000 = TFile.Open("Zprime_M2000GeV_W20GeV_35.9pb_PFElectron_PF2PATJets_PFMET.root");
38     # Zprime4000 = TFile.Open("Zprime_M4000GeV_W400GeV_4000pb_PFElectron_PF2PATJets_PFMET.root");
39     # Tprime300 = TFile.Open("TprimeToWb_M300_35.9pb_PFElectron_PF2PATJets_PFMET.root");
40     # Tprime350 = TFile.Open("TprimeToWb_M350_35.9pb_PFElectron_PF2PATJets_PFMET.root");
41     WprimeTToTTD_M1000 = TFile.Open("WprimeTToTTD_M1000_4000pb_PFElectron_PF2PATJets_PFMET.root");
42     WprimeTToTTD_M600 = TFile.Open("WprimeTToTTD_M600_4000pb_PFElectron_PF2PATJets_PFMET.root");
43     WprimeTToTTD_M800 = TFile.Open("WprimeTToTTD_M800_4000pb_PFElectron_PF2PATJets_PFMET.root");
44     # WprimeToTTBbar_M1000 = TFile.Open("WprimeToTBbar_M-1000_4000pb_PFElectron_PF2PATJets_PFMET.root");
45    
46     # pj1 = TFile.Open("pj1_36.145pb_PFElectron_PF2PATJets_PFMET.root");
47     # pj2 = TFile.Open("pj2_36.145pb_PFElectron_PF2PATJets_PFMET.root");
48     # pj3 = TFile.Open("pj3_36.145pb_PFElectron_PF2PATJets_PFMET.root");
49     # tW = TFile.Open("tW_36.145pb_PFElectron_PF2PATJets_PFMET.root");
50     # tchan = TFile.Open("tchan_36.145pb_PFElectron_PF2PATJets_PFMET.root");
51     # vqq = TFile.Open("vqq_7.22pb_V4PFCalo.root__fullSetOfVars.root");
52     # Zprime1250 = TFile.Open("/storage/workspace/BristolAnalysisTools/outputfiles/Zprime_M1250GeV_W12500MeV_36.135pb.root");
53     # Zprime1500 = TFile.Open("/storage/workspace/BristolAnalysisTools/outputfiles/Zprime_M1500GeV_W15GeV_36.135pb.root");
54    
55     hists = [];
56     # hists.append('mttbar_QCDEnriched')
57     # hists.append('mttbar_conversions')
58     # hists.append('mttbar_conversions_withMETCut')
59     # hists.append('mttbar_conversions_withMETAndAsymJets')
60     # hists.append('mttbar_conversions_withAsymJetsCut')
61     # hists.append('mttbar_controlRegion')
62     # hists.append("jet12vsjet13");
63     # hists.append("jet12vsjet23");
64     # hists.append("jet13vsjet23");
65     # hists.append("mttbar");
66     # hists.append("tPrimeMass");
67     # hists.append("tPrimepT");
68     # hists.append("tPrimeHT");
69     # hists.append("tPrime_pt");
70     # hists.append("tPrime_px");
71     # hists.append("tPrime_py");
72     # hists.append("tPrime_pz");
73     # hists.append("mLeptonicTopLone");
74     # hists.append("mLepTopLone");
75     # hists.append("mLepTopLoneNM");
76     # hists.append("mHadTopLoneNM");
77     # hists.append("mHadTopLone");
78     # hists.append("chiHadronicTop");
79     # hists.append("chiLeptonicTop");
80     # hists.append("chiGlobal");
81     # hists.append("chiTotal");
82     hists.append("lepTopDeltaR");
83     hists.append("hadTopDeltaR");
84     hists.append("lepTopLep");
85     hists.append("hadTopHad");
86     hists.append("numHadTopMCMatches");
87     hists.append("numLepTopMCMatches");
88     hists.append("numHadTopCorrectID");
89     hists.append("numLepTopCorrectID");
90     # hists.append("mttbar_2ndSolution");
91     # hists.append("mttbar_3rdSolution");
92     # hists.append("mttbar_allSolutions");
93     ##
94     # hists.append("mttbar_withMETCut");
95     # hists.append("mttbar_2ndSolution_withMETCut");
96     # hists.append("mttbar_3rdSolution_withMETCut");
97     # hists.append("mttbar_allSolutions_withMETCut");
98     #
99     # hists.append("mttbar_withMETAndAsymJets");
100     # hists.append("mttbar_2ndSolution_withMETAndAsymJets");
101     # hists.append("mttbar_3rdSolution_withMETAndAsymJets");
102     # hists.append("mttbar_allSolutions_withMETAndAsymJets");
103     #
104     # hists.append("mttbar_withAsymJetsCut");
105     # hists.append("mttbar_2ndSolution_withAsymJetsCut");
106     # hists.append("mttbar_3rdSolution_withAsymJetsCut");
107     # hists.append("mttbar_allSolutions_withAsymJetsCut");
108     #
109     # hists.append("ttbar_pt");
110     # hists.append("ttbar_pt_2ndSolution");
111     # hists.append("ttbar_pt_3rdSolution");
112     # hists.append("ttbar_pt_allSolutions");
113     #
114     # hists.append("ttbar_pt_withMETCut");
115     # hists.append("ttbar_pt_2ndSolution_withMETCut");
116     # hists.append("ttbar_pt_3rdSolution_withMETCut");
117     # hists.append("ttbar_pt_allSolutions_withMETCut");
118     #
119     # hists.append("ttbar_pt_withMETAndAsymJets");
120     # hists.append("ttbar_pt_2ndSolution_withMETAndAsymJets");
121     # hists.append("ttbar_pt_3rdSolution_withMETAndAsymJets");
122     # hists.append("ttbar_pt_allSolutions_withMETAndAsymJets");
123     #
124     # hists.append("ttbar_pt_withAsymJetsCut");
125     # hists.append("ttbar_pt_2ndSolution_withAsymJetsCut");
126     # hists.append("ttbar_pt_3rdSolution_withAsymJetsCut");
127     # # hists.append("ttbar_pt_allSolutions_withAsymJetsCut");
128     # hists.append("angleTops");
129     # hists.append("angleTops_withMETCut");
130     # hists.append("angleTops_withMETAndAsymJets");
131     # hists.append("angleTops_withAsymJetsCut");
132     #
133     # hists.append("leadingJetPt");
134     # hists.append("pt_leadingTop");
135     # hists.append("pt_loneLepTop");
136     # hists.append("pt_lepTop");
137     # hists.append("pt_hadTop");
138     # hists.append("pt_loneHadTop");
139     # hists.append("pt_leadingTop_withMETCut");
140     # hists.append("pt_leadingTop_withMETAndAsymJets");
141     # hists.append("pt_leadingTop_withAsymJetsCut");
142     #
143     # hists.append("pt_NextToLeadingTop");
144     # hists.append("pt_NextToLeadingTop_withMETCut");
145     # hists.append("pt_NextToLeadingTop_withMETAndAsymJets");
146     # hists.append("pt_NextToLeadingTop_withAsymJetsCut");
147     #
148     hists.append("mLeptonicTop");
149     hists.append("mHadronicTop");
150     # hists.append("mAllTop");
151     #
152     # hists.append("ttbar_px");
153     # hists.append("ttbar_py");
154     # hists.append("ttbar_pz");
155     #
156     # hists.append("m3");
157     # hists.append("HT");
158     # hists.append("qkjet_eta");
159     # hists.append("qkjet_rapid");
160     # hists.append("leadjet_rap");
161     # hists.append("lepb_angle");
162     # hists.append("angleHadTopD");
163     hists.append("wpmassplus");
164     hists.append("wpmassminus");
165     # hists.append("MET");
166     # hists.append("METtopangle_ltop");
167     # hists.append("HT3jet_ltop");
168     # hists.append("HT4jet_ltop");
169     # hists.append("mt2jbothchglep4j");
170     # hists.append("mt2jbothchgelec4j");
171     # hists.append("mt2jbothchgmu4j");
172     # hists.append("mt2jbothchglep5j");
173     # hists.append("mt2jbothchgmu5j");
174     # hists.append("mt2jbothchgelec5j");
175     # hists.append("mt2jpluselec4j");
176     # hists.append("mt2jplusmu4j");
177     # hists.append("mt2jpluslep4j");
178     # hists.append("mt2jpluslep5j");
179     # hists.append("mt2jpluselec5j");
180     # hists.append("mt2jplusmu5j");
181     # hists.append("mt2jminuselec4j");
182     # hists.append("mt2jminusmu4j");
183     # hists.append("mt2jminuslep4j");
184     # hists.append("mt2jminuselec5j");
185     # hists.append("mt2jminusmu5j");
186     # hists.append("mt2jminuslep5j");
187     # hists.append("mt4jbothchglep4j");
188     # hists.append("mt4jbothchgelec4j");
189     # hists.append("mt4jbothchgmu4j");
190     # hists.append("mt4jbothchglep5j");
191     # hists.append("mt4jbothchgmu5j");
192     # hists.append("mt4jbothchgelec5j");
193     # hists.append("mt4jpluselec4j");
194     # hists.append("mt4jplusmu4j");
195     # hists.append("mt4jpluslep4j");
196     # hists.append("mt4jpluslep5j");
197     # hists.append("mt4jpluselec5j");
198     # hists.append("mt4jplusmu5j");
199     # hists.append("mt4jminuselec4j");
200     # hists.append("mt4jminusmu4j");
201     # hists.append("mt4jminuslep4j");
202     # hists.append("mt4jminuselec5j");
203     # hists.append("mt4jminusmu5j");
204     # hists.append("mt4jminuslep5j");
205     # hists.append("leadingJetMass");
206     # hists.append("mtW");
207     # hists.append("neutrino_pz");
208     # hists.append("lone_neutrino_pz");
209     suffixes = ["0orMoreBtag",
210     "1orMoreBtag",
211     "2orMoreBtags",
212     "3orMoreBtags",'4orMoreBtags' ]
213     hists = [hist + '_' + suffix for hist in hists for suffix in suffixes]
214    
215     histslep = [];
216     # histslep.append("HT_ltop");
217     # histslep.append("MET_ltop");
218     # histslep.append("METpt_ltop");
219     # histslep.append("mass4jets");
220     # histslep.append("ST_ltop");
221     histslep.append("wpmass");
222     histslep.append("wpnegmass");
223     histslep.append("wpnegmassbad");
224     histslep.append("wpnegmasspos");
225     histslep.append("wpnegmassneg");
226     histslep.append("wpanglmass");
227     histslep.append("deltaPhiTops");
228     histslep.append("deltaphilepjetpos");
229     histslep.append("deltaphilepdjetpos");
230     histslep.append("deltaphilepjetneg");
231     histslep.append("deltaphilepdjetneg");
232     histslep.append("transMasslm1b");
233     suffixes = ["elec", "mu", "lepton"]
234     histslep = [hist + '_' + suffix for hist in histslep for suffix in suffixes]
235    
236     hists += histslep;
237     print hists
238     # hists.append("electron_et");
239     # hists.append("electminus_pt");
240     # hists.append("positron_pt");
241     # hists.append("electron_pt");
242     # hists.append("muon_et");
243     # hists.append("muon_pt");
244    
245     # jetBinned = ["allJets",
246     # "1orMoreJets",
247     # "2orMoreJets",
248     # "3orMoreJets" , "4orMoreJets"]
249     # jetBinnedhists = []
250     # jetBinnedhists.append("QCDest_CombRelIso")
251     # jetBinnedhists.append("QCDest_CombRelIso_1btag")
252     # jetBinnedhists.append("QCDest_CombRelIso_2btag")
253     # jetBinnedhists.append("QCDest_CombRelIso_controlRegion")
254     # jetBinnedhists.append("QCDest_CombRelIso_controlRegion_1btag")
255     # jetBinnedhists.append("QCDest_CombRelIso_controlRegion_2btag")
256     #
257     # jetBinnedhists.append("QCDest_PFIsolation")
258     # jetBinnedhists.append("QCDest_PFIsolation_WithMETCut")
259     # jetBinnedhists.append("QCDest_PFIsolation_WithMETCutAndAsymJetCuts")
260     # jetBinnedhists.append("QCDest_PFIsolation_WithAsymJetCuts")
261    
262     # jetBinnedhists.append("QCDest_PFIsolation_1btag")
263     # jetBinnedhists.append("QCDest_PFIsolation_2btag")
264     # jetBinnedhists.append("QCDest_PFIsolation_controlRegion")
265     # jetBinnedhists.append("QCDest_PFIsolation_controlRegion_1btag")
266     # jetBinnedhists.append("QCDest_PFIsolation_controlRegion_2btag")
267     # jetBinnedhists.append("QCDest_PFIsolation_controlRegion2")
268     # jetBinnedhists.append("QCDest_PFIsolation_controlRegion2_WithMETCut")
269     # jetBinnedhists.append("QCDest_PFIsolation_controlRegion2_WithMETCutAndAsymJetCuts")
270     # jetBinnedhists.append("QCDest_PFIsolation_controlRegion2_WithAsymJetCuts")
271     # jetBinnedhists.append("QCDest_PFIsolation_controlRegion2_1btag")
272     # jetBinnedhists.append("QCDest_PFIsolation_controlRegion2_2btag")
273    
274     # jetBinnedhists = [hist + '_' + suffix for hist in jetBinnedhists for suffix in jetBinned]
275     # hists.extend(jetBinnedhists)
276     gcd = gROOT.cd
277    
278     for histname in hists:
279     currentSelection = 'default'
280     if 'withMETAndAsymJets' in histname:
281     currentSelection = 'withMETAndAsymJets'
282     gcd()
283     print histname
284     # hist_data = data.Get(histname);
285    
286     # hist_data2;
287     # if (histname == "mttbar_rebinned")
288     # hist_data2 = data2.Get(histname);
289     # else
290     # hist_data2 = TH1F(*hist_data);
291     # hist_data.Sumw2();
292     # hist_data2.Sumw2();
293     # hist_ttbar = ttbar.Get(histname);
294     # hist_wjets = wjets.Get(histname);
295     # hist_zjets = zjets.Get(histname);
296    
297     # hist_bce1 = bce1.Get(histname);
298     # hist_bce2 = bce2.Get(histname);
299     # hist_bce3 = bce3.Get(histname);
300     # hist_enri1 = enri1.Get(histname);
301     # hist_enri2 = enri2.Get(histname);
302     # hist_enri3 = enri3.Get(histname);
303     # # hist_pj1 = pj1.Get(histname);
304     # # hist_pj2 = pj2.Get(histname);
305     # hist_pj3 = pj3.Get(histname);
306     # hist_singleTop = tW.Get(histname)
307     # hist_singleTop.Add(tchan.Get(histname))
308    
309     # hist_Zprime500 = Zprime500.Get(histname);
310     # hist_Zprime750 = Zprime750.Get(histname);
311     # hist_Zprime1000 = Zprime1000.Get(histname);
312     # hist_Zprime1500 = Zprime1500.Get(histname);
313     # hist_Zprime2000 = Zprime2000.Get(histname);
314     # hist_Zprime4000 = Zprime4000.Get(histname);
315     # hist_Tprime300 = Tprime300.Get(histname);
316     # hist_Tprime350 = Tprime350.Get(histname);
317     # # hist_Zprime1250 = Zprime1250.Get(histname);
318     hist_wpm1000 = WprimeTToTTD_M1000.Get(histname);
319     hist_wpm600 = WprimeTToTTD_M600.Get(histname);
320     hist_wpm800 = WprimeTToTTD_M800.Get(histname);
321     # hist_wptb1000 = WprimeToTTBbar_M1000.Get(histname);
322    
323     # hist_ttbar.Scale(scale);
324     # hist_wjets.Scale(scale);
325     # hist_zjets.Scale(scale);
326    
327     # hist_bce1.Scale(scale);
328     # hist_bce2.Scale(scale);
329     # hist_bce3.Scale(scale);
330    
331     # hist_enri1.Scale(scale);
332     # hist_enri2.Scale(scale);
333     # hist_enri3.Scale(scale);
334    
335     # hist_pj1.Scale(scale);
336     # hist_pj2.Scale(scale);
337     # hist_pj3.Scale(scale);
338    
339     # scale = scale * 10.0;
340     # hist_Zprime500.Scale(scale);
341     # hist_Zprime750.Scale(scale);
342     # hist_Zprime1000.Scale(scale);
343     # hist_Zprime2000.Scale(scale);
344     # hist_Zprime1500.Scale(scale);
345     # hist_Zprime4000.Scale(scale);
346     # hist_Tprime300.Scale(scale);
347     # hist_Tprime350.Scale(scale);
348     # hist_Zprime1250.Scale(scale);
349     hist_wpm1000.Scale(scale);
350     hist_wpm600.Scale(scale);
351     hist_wpm800.Scale(scale);
352     # hist_wptb1000.Scale(scale);
353    
354     # hist_qcd = hist_bce1.Clone("qcd")#TH1F(*hist_bce1);
355     # hist_qcd.Add(hist_bce2);
356     # hist_qcd.Add(hist_bce3);
357     # hist_qcd.Add(hist_enri1);
358     # hist_qcd.Add(hist_enri2);
359     # hist_qcd.Add(hist_enri3);
360     # hist_qcd.Add(hist_pj1);
361     # hist_qcd.Add(hist_pj2);
362     # hist_qcd.Add(hist_pj3);
363     # hist_qcd.Scale(qcdScale[currentSelection]);
364     # ndata = hist_data.Integral();
365     # ntop = hist_ttbar.Integral();
366     # nwj = hist_wjets.Integral();
367     # nzj = hist_zjets.Integral();
368     # nqcd = hist_qcd.Integral();
369     # sumMC = ntop + nwj + nzj + nqcd;
370     # cout << ndata << " " << sumMC << endl;
371     # hist_wjets.Scale(ndata / sumMC);
372     # hist_ttbar.Scale(ndata / sumMC);
373     # hist_zjets.Scale(ndata / sumMC);
374     # hist_qcd.Scale(ndata / sumMC);
375     # mttbars = ['mttbar_' + suffix for suffix in suffixes]
376     # mttbars2 = ['mttbar_withMETAndAsymJets_' + suffix for suffix in suffixes]
377     # if histname in mttbars or histname in mttbars2:
378     # print "taking QCD shape from DATA"
379     # name = histname.replace('mttbar', 'mttbar_conversions')
380     # hist_qcd = data.Get(name)
381     # if( hist_qcd.Integral() > 0):
382     # hist_qcd.Scale(nqcd/hist_qcd.Integral())
383     # hist_mc = hist_qcd.Clone("all_mc")
384     # hist_mc.Add(hist_ttbar);
385     # hist_mc.Add(hist_zjets);
386     # hist_mc.Add(hist_wjets);
387     # hist_mc.Add(hist_singleTop);
388    
389     # hist_data = hist_ttbar.Clone("test");
390     hist_data = hist_wpm600.Clone("test");
391     rebin = 1;
392     Urange = (0,5000)
393     if ("mttbar" in histname):
394     hist_data.SetXTitle("M_{t#bar{t}}/GeV");
395     hist_data.SetYTitle("Events/(100 GeV)");
396     rebin = 100;
397     Urange = (0, 2000)
398     elif ("tPrimeMass" in histname):
399     hist_data.SetXTitle("T' mass/GeV");
400     hist_data.SetYTitle("Events/(100 GeV)");
401     rebin = 200;
402     Urange = (0, 2000)
403     elif ("wpmass" in histname or "wpnegmass" in histname or "wpanglmass" in histname):
404     hist_data.SetXTitle("W' mass/GeV");
405     hist_data.SetYTitle("Events/(60 GeV)");
406     rebin = 60;
407     Urange = (0, 1500)
408     elif ("jet12" in histname or "jet13" in histname):
409     hist_data.SetXTitle("Jet mass/GeV");
410     hist_data.SetYTitle("Jet mass/GeV)");
411     elif ("tPrimepT" in histname):
412     hist_data.SetXTitle("T' pT/GeV");
413     hist_data.SetYTitle("Events/(8 GeV)");
414     rebin = 8;
415     Urange = (0, 200)
416     elif ("tPrimeHT" in histname):
417     hist_data.SetXTitle("T' HT");
418     hist_data.SetYTitle("Events");
419     rebin = 4;
420     Urange = (0, 1.5)
421     elif ("m3" in histname):
422     hist_data.SetXTitle("M3/GeV");
423     hist_data.SetYTitle("Events/(50 GeV)");
424     rebin = 50;
425     Urange = (0, 1500)
426     elif (histname == "electron_et"):
427     hist_data.SetXTitle("electron E_{T}/GeV");
428     hist_data.SetYTitle("Events/(10 GeV)");
429     rebin = 10;
430     Urange = (0, 300)
431     elif (histname == "electron_pt" or histname == "electminus_pt" or histname == "positron_pt"):
432     hist_data.SetXTitle("electron p_{T}/GeV");
433     hist_data.SetYTitle("Events/(10 GeV)");
434     rebin = 10;
435     Urange = (0, 300)
436     elif (histname == "muon_et"):
437     hist_data.SetXTitle("muon E_{T}/GeV");
438     hist_data.SetYTitle("Events/(10 GeV)");
439     rebin = 10;
440     Urange = (0, 300)
441     elif (histname == "muon_pt"):
442     hist_data.SetXTitle("muon p_{T}/GeV");
443     hist_data.SetYTitle("Events/(10 GeV)");
444     rebin = 10;
445     Urange = (0, 300)
446     elif ("ttbar_pt" in histname):
447     hist_data.SetXTitle("p_{T} of t#bar{t} system/GeV");
448     hist_data.SetYTitle("Events/(10 GeV)");
449     rebin = 10;
450     Urange = (0, 500)
451     elif ("ttbar_px" in histname):
452     hist_data.SetXTitle("p_{x} of t#bar{t} system/GeV");
453     hist_data.SetYTitle("Events/(10 GeV)");
454     rebin = 10;
455     Urange = (0, 500)
456     elif ("ttbar_py" in histname):
457     hist_data.SetXTitle("p_{y} of t#bar{t} system/GeV");
458     hist_data.SetYTitle("Events/(10 GeV)");
459     rebin = 10;
460     Urange = (0, 500)
461     elif ("ttbar_pz" in histname):
462     hist_data.SetXTitle("p_{z} of t#bar{t} system/GeV");
463     hist_data.SetYTitle("Events/(50 GeV)");
464     rebin = 50;
465     Urange = (0, 2000)
466     elif ("METpt_ltop" in histname):
467     hist_data.SetXTitle("MET p_{T}/GeV");
468     hist_data.SetYTitle("Events/(10 GeV)");
469     rebin = 10;
470     Urange = (0, 400)
471     elif ("tPrime_pt" in histname):
472     hist_data.SetXTitle("p_{T} of T' system/GeV");
473     hist_data.SetYTitle("Events/(20 GeV)");
474     rebin = 20;
475     Urange = (0, 200)
476     elif ("tPrime_px" in histname):
477     hist_data.SetXTitle("p_{x} of T' system/GeV");
478     hist_data.SetYTitle("Events/(20 GeV)");
479     rebin = 20;
480     Urange = (0, 200)
481     elif ("tPrime_py" in histname):
482     hist_data.SetXTitle("p_{y} of T' system/GeV");
483     hist_data.SetYTitle("Events/(20 GeV)");
484     rebin = 20;
485     Urange = (0, 200)
486     elif ("tPrime_pz" in histname):
487     hist_data.SetXTitle("p_{z} of T' system/GeV");
488     hist_data.SetYTitle("Events/(100 GeV)");
489     rebin = 200;
490     Urange = (0, 1800)
491     elif ("HT_" in histname):
492     hist_data.SetXTitle("#Sigma p_{T}/GeV");
493     hist_data.SetYTitle("Events/(20 GeV)");
494     rebin = 20;
495     Urange = (0, 2000)
496     elif ("ST" in histname):
497     hist_data.SetXTitle("ST/GeV");
498     hist_data.SetYTitle("Events/(30 GeV)");
499     rebin = 30;
500     Urange = (0, 1000)
501     elif ("HT3jet_ltop" in histname):
502     hist_data.SetXTitle("#Sigma p_{T} of leading 3 jets/GeV");
503     hist_data.SetYTitle("Events/(20 GeV)");
504     rebin = 20;
505     Urange = (0, 1000)
506     elif ("HT4jet_ltop" in histname):
507     hist_data.SetXTitle("#Sigma p_{T} of leading 4 jets/GeV");
508     hist_data.SetYTitle("Events/(50 GeV)");
509     rebin = 50;
510     Urange = (0, 1200)
511     elif (histname == "numberOfJets"):
512     hist_data.SetXTitle("number of jets");
513     hist_data.SetYTitle("Events");
514     elif (histname == "numberOfBJets"):
515     hist_data.SetXTitle("number of b-tagged jets (SSVHE medium)");
516     hist_data.SetYTitle("Events");
517     elif ('MET_' in histname and 'MET_ltop' not in histname):
518     hist_data.SetXTitle("MET/GeV");
519     hist_data.SetYTitle("Events/(10 GeV)");
520     rebin = 2;
521     Urange = (0, 400)
522     elif ('MET_ltop' in histname):
523     hist_data.SetXTitle("MET/GeV");
524     hist_data.SetYTitle("Events/(10 GeV)");
525     rebin = 20;
526     Urange = (0, 400)
527     elif ("leadingJetMass" in histname):
528     hist_data.SetXTitle("leading jet mass/GeV");
529     hist_data.SetYTitle("Events/(5 GeV)");
530     rebin = 5;
531     Urange = (0, 150)
532     elif ("mtW" in histname):
533     hist_data.SetXTitle("transverse W-boson mass/GeV");
534     hist_data.SetYTitle("Events/(10 GeV)");
535     rebin = 10;
536     elif ("both" in histname or "plus" in histname or "minus" in histname):
537     hist_data.SetXTitle("transverse mass/GeV");
538     hist_data.SetYTitle("Events/(100 GeV)");
539     rebin = 100;
540     Urange = (0, 2000)
541     elif ("transMasslm1b" in histname):
542     hist_data.SetXTitle("M_{T} (lep+MET+1b)/GeV");
543     hist_data.SetYTitle("Events/(50 GeV)");
544     rebin = 50;
545     Urange = (0, 700)
546     elif ("electronD0" in histname):
547     hist_data.SetXTitle("electron d_{0} / cm");
548     hist_data.SetYTitle("Events/(0.001 cm)");
549     rebin = 10;
550     elif ("angleTops" in histname or "deltaPhiTops" in histname):
551     hist_data.SetXTitle("angle between top quarks");
552     hist_data.SetYTitle("Events/(0.1 rad)");
553     rebin = 10;
554     elif ( "deltaphilep" in histname):
555     hist_data.SetXTitle("angle between lepton & leading jet");
556     hist_data.SetYTitle("Events/(0.2 rad)");
557     rebin = 20;
558     elif ("lepb_angle" in histname):
559     hist_data.SetXTitle("angle between electron and b jet");
560     hist_data.SetYTitle("Events/(0.1 rad)");
561     rebin = 10;
562     elif ("angleHadTopD" in histname):
563     hist_data.SetXTitle("angle between hadronic top and light jet");
564     hist_data.SetYTitle("Events/(0.2 rad)");
565     rebin = 20;
566     elif ("METtopangle_ltop" in histname):
567     hist_data.SetXTitle("angle between MET and top p_{T}");
568     hist_data.SetYTitle("Events/(0.1 rad)");
569     rebin = 10;
570     Urange = (-2, 2)
571     elif ("qkjet_eta" in histname):
572     hist_data.SetXTitle("#eta of b jet");
573     hist_data.SetYTitle("Events/(0.1 rad)");
574     Urange = (-3, 3)
575     rebin = 10;
576     elif ("_rap" in histname):
577     hist_data.SetXTitle("Jet Rapidity");
578     hist_data.SetYTitle("Events/(0.1 rad)");
579     Urange = (-3, 3)
580     rebin = 20;
581     elif ("neutrino_pz" in histname):
582     hist_data.SetXTitle("neutrino p_{Z} /GeV");
583     hist_data.SetYTitle("Events/(20 GeV)");
584     rebin = 20;
585     Urange = (-500, 500)
586     elif ('mHadTopLone' in histname or 'mHadTopLoneNM' in histname):
587     hist_data.SetXTitle("top mass /GeV");
588     hist_data.SetYTitle("Events/(10 GeV)");
589     rebin = 10;
590     Urange = (0, 400)
591     elif ('mass4jets' in histname):
592     hist_data.SetXTitle("4-jet mass /GeV");
593     hist_data.SetYTitle("Events/(50 GeV)");
594     rebin = 50;
595     Urange = (0, 1500)
596     elif ('mLeptonicTopLone' in histname or 'mLepTopLoneNM' in histname or 'mLepTopLone' in histname):
597     hist_data.SetXTitle("top mass /GeV");
598     hist_data.SetYTitle("Events/(10 GeV)");
599     rebin = 10;
600     Urange = (0, 400)
601     elif ('chiHadronicTop' in histname or 'chiLeptonicTop' in histname):
602     hist_data.SetXTitle("top #chi^{2} /GeV");
603     hist_data.SetYTitle("Events/(1)");
604     rebin = 1;
605     Urange = (0, 30)
606     elif ('chiGlobal' in histname or 'chiTotal' in histname):
607     hist_data.SetXTitle("Event #chi^{2} /GeV");
608     hist_data.SetYTitle("Events/(100)");
609     rebin = 100;
610     elif ('numHadTopMCMatches' in histname or 'numLepTopMCMatches' in histname):
611     hist_data.SetXTitle("Number of MC-Truth Matches");
612     hist_data.SetYTitle("Events");
613     elif ('numHadTopCorrectID' in histname or 'numLepTopCorrectID' in histname):
614     hist_data.SetXTitle("Number of Reco Candidates Matched to MC-Truth Particles");
615     hist_data.SetYTitle("Events");
616     elif ('lepTopDeltaR' in histname or 'hadTopDeltaR' in histname):
617     hist_data.SetXTitle("#Delta R of MC truth top");
618     hist_data.SetYTitle("Events");
619     rebin = 10;
620     elif ('lepTopLep' in histname or 'hadTopHad' in histname):
621     hist_data.SetXTitle("top MC truth identity");
622     hist_data.SetYTitle("Events");
623     Urange = (0, 2)
624     elif ('mHadronicTop' in histname or 'mAllTop' in histname):
625     hist_data.SetXTitle("top mass /GeV");
626     hist_data.SetYTitle("Events/(25 GeV)");
627     rebin = 25;
628     Urange = (0, 700)
629     elif ('mLeptonicTop' in histname):
630     hist_data.SetXTitle("top mass /GeV");
631     hist_data.SetYTitle("Events/(25 GeV)");
632     rebin = 25;
633     Urange = (0, 700)
634     elif ('pt_' in histname and 'Top' in histname):
635     hist_data.SetXTitle("top p_{T} /GeV");
636     hist_data.SetYTitle("Events/(50 GeV)");
637     rebin = 50;
638     Urange = (0, 500)
639     elif ('leadingJetPt' in histname):
640     hist_data.SetXTitle("leading jet p_{T} /GeV");
641     hist_data.SetYTitle("Events/(50 GeV)");
642     rebin = 50;
643     Urange = (0, 500)
644     elif('QCDest_CombRelIso' in histname):
645     hist_data.SetXTitle("relative isolation");
646     hist_data.SetYTitle("Events/(0.1)");
647     rebin = 10;
648     Urange = (0, 2)
649     elif('QCDest_PFIsolation' in histname):
650     hist_data.SetXTitle("ParticleFlow isolation");
651     hist_data.SetYTitle("Events/(0.1)");
652     rebin = 10
653     Urange = (0, 2)
654    
655    
656     hist_data.SetTitleOffset(1.3, "Y");
657     if ('jet1' in histname):
658     hist_data.Rebin2D();
659     # hist_ttbar.Rebin2D();
660     hist_Tprime300.Rebin2D();
661     hist_Tprime350.Rebin2D();
662     hist_data.SetAxisRange(0, 600, "X");
663     hist_data.SetAxisRange(0, 600, "Y");
664     # hist_ttbar.SetAxisRange(0, 600, "X");
665     # hist_ttbar.SetAxisRange(0, 600, "Y");
666     # hist_Tprime300.SetAxisRange(0, 600, "X");
667     # hist_Tprime300.SetAxisRange(0, 600, "Y");
668     # hist_Tprime350.SetAxisRange(0, 600, "X");
669     # hist_Tprime350.SetAxisRange(0, 600, "Y");
670     else:
671     hist_data.Rebin(rebin);
672     # hist_ttbar.Rebin(rebin);
673     # hist_wjets.Rebin(rebin);
674     # hist_zjets.Rebin(rebin);
675     # hist_qcd.Rebin(rebin);
676     # hist_Zprime500.Rebin(rebin);
677     # hist_Zprime1500.Rebin(rebin);
678     # hist_Zprime750.Rebin(rebin);
679     # hist_Zprime1000.Rebin(rebin);
680     # hist_Zprime2000.Rebin(rebin);
681     # hist_Zprime4000.Rebin(rebin);
682     # hist_Tprime300.Rebin(rebin);
683     # hist_Tprime350.Rebin(rebin);
684     # hist_singleTop.Rebin(rebin)
685     hist_wpm1000.Rebin(rebin);
686     hist_wpm600.Rebin(rebin);
687     hist_wpm800.Rebin(rebin);
688     # hist_wptb1000.Rebin(rebin);
689    
690     hist_data.SetAxisRange(Urange[0], Urange[1]);
691     # hist_ttbar.SetAxisRange(Urange[0], Urange[1]);
692     # hist_wjets.SetAxisRange(Urange[0], Urange[1]);
693     # hist_zjets.SetAxisRange(Urange[0], Urange[1]);
694     # hist_qcd.SetAxisRange(Urange[0], Urange[1]);
695     # hist_Zprime500.SetAxisRange(Urange[0], Urange[1]);
696     # hist_Zprime750.SetAxisRange(Urange[0], Urange[1]);
697     # hist_Zprime1000.SetAxisRange(Urange[0], Urange[1]);
698     # hist_Zprime1500.SetAxisRange(Urange[0], Urange[1]);
699     # hist_Zprime2000.SetAxisRange(Urange[0], Urange[1]);
700     # hist_Zprime4000.SetAxisRange(Urange[0], Urange[1]);
701     # hist_Tprime300.SetAxisRange(Urange[0], Urange[1]);
702     # hist_Tprime350.SetAxisRange(Urange[0], Urange[1]);
703     # hist_singleTop.SetAxisRange(Urange[0], Urange[1]);
704     hist_wpm1000.SetAxisRange(Urange[0], Urange[1]);
705     hist_wpm600.SetAxisRange(Urange[0], Urange[1]);
706     hist_wpm800.SetAxisRange(Urange[0], Urange[1]);
707     # hist_wptb1000.SetAxisRange(Urange[0], Urange[1]);
708    
709    
710     hist_data.SetMarkerStyle(8);
711     # hist_data.SetMarkerSize(1.5);
712     hist_data.SetMarkerSize(0.1);
713    
714     # hist_data.SetFillStyle(1001);
715     # hist_data.SetFillColor(kRed);
716     # hist_ttbar.SetFillStyle(1001);
717     # hist_ttbar.SetFillColor(kRed);
718     # hist_wjets.SetFillStyle(1001);
719     # hist_wjets.SetFillColor(4);
720     # hist_zjets.SetFillStyle(1001);
721     # hist_zjets.SetFillColor(8);
722     # hist_qcd.SetFillStyle(1001);
723     # hist_qcd.SetFillColor(kYellow);
724     # hist_qcd.SetFillColor(12);
725     # hist_singleTop.SetFillStyle(1001);
726     # hist_singleTop.SetFillColor(kMagenta)
727     # nbins = hist_qcd.GetXaxis().GetNbins();
728     # binwidth = (hist_qcd.GetXaxis().GetXmax() - hist_qcd.GetXaxis().GetXmin()) / nbins;
729     # for i in range(1, nbins + 1):
730     # yvalue = hist_qcd.GetBinContent(i);
731     # # float xvalue = hist_qcd.GetBinCenter(i);
732     # # float ymin = yvalue - yvalue*0.5;
733     # # float ymax = yvalue + yvalue*0.5;
734     # # float xmin = xvalue - 0.5*binwidth;
735     # # float xmax = xvalue + 0.5*binwidth;
736     # error = yvalue * 0.5;
737     # hist_mc.SetBinError(i, error);
738     # # qcdUncert.SetPointError(i, xmin, xmax, ymin, ymax);
739     #
740     # qcdUncert = TGraphAsymmErrors(hist_mc);
741    
742     linewidth = 6;
743     # hist_Zprime500.SetLineColor(6);
744     # hist_Zprime500.SetLineWidth(linewidth);
745     # hist_Zprime500.SetFillStyle(0);
746     # hist_Zprime500.SetFillColor(kWhite);
747    
748     # hist_Zprime750.SetLineColor(5);
749     # hist_Zprime750.SetLineWidth(linewidth);
750     # hist_Zprime750.SetFillStyle(0);
751     # hist_Zprime750.SetFillColor(kWhite);
752    
753     # hist_Zprime1000.SetLineColor(7);
754     # hist_Zprime1000.SetLineWidth(linewidth);
755     # hist_Zprime1000.SetFillStyle(0);
756     # hist_Zprime1000.SetFillColor(kWhite);
757     # hist_Zprime2000.SetLineColor(42);
758     # hist_Zprime2000.SetLineWidth(linewidth);
759     # hist_Zprime2000.SetFillStyle(0);
760     # hist_Zprime2000.SetFillColor(kWhite);
761     # # hist_Zprime4000.SetLineColor(50);
762     # hist_Zprime4000.SetLineWidth(linewidth);
763     # hist_Zprime4000.SetFillStyle(0);
764     # hist_Zprime4000.SetFillColor(kWhite);
765    
766     # hist_Tprime300.SetLineColor(kOrange + 1);
767     # hist_Tprime300.SetLineWidth(linewidth);
768     # hist_Tprime350.SetLineColor(kGray + 3);
769     # hist_Tprime350.SetLineWidth(linewidth);
770     #
771     # hist_Zprime1250.SetLineColor(kCyan - 5);
772     # hist_Zprime1250.SetLineWidth(linewidth);
773     # hist_Zprime1250.SetFillStyle(0);
774     # hist_Zprime1250.SetFillColor(kWhite);
775     #
776     # hist_Zprime1500.SetLineColor(42);
777     # hist_Zprime1500.SetLineWidth(linewidth);
778     # hist_Zprime1500.SetFillStyle(0);
779     # hist_Zprime1500.SetFillColor(kWhite);
780    
781     hist_wpm1000.SetLineColor(kOrange + 1);
782     hist_wpm1000.SetLineWidth(linewidth);
783     hist_wpm600.SetLineColor(kGray + 3);
784     hist_wpm600.SetLineWidth(linewidth);
785     hist_wpm800.SetLineColor(kCyan);
786     hist_wpm800.SetLineWidth(linewidth);
787     # hist_wptb1000.SetLineColor(kCyan);
788     # hist_wptb1000.SetLineWidth(linewidth);
789    
790     # qcdUncert.SetFillColor(kGray + 3);
791     # qcdUncert.SetFillStyle(3003);
792    
793     leg = TLegend(0.696, 0.35, 0.95, 0.92);
794     leg.SetBorderSize(0);
795     leg.SetLineStyle(0);
796     leg.SetTextFont(42);
797     leg.SetFillStyle(0);
798    
799     # leg.AddEntry(hist_data, "data", "P");
800     # leg.AddEntry(hist_data2, "data(no HLT)", "P");
801     # leg.AddEntry(hist_ttbar, "t#bar{t}", "f");
802     # leg.AddEntry(hist_wjets, "W#rightarrowl#nu", "f");
803     # leg.AddEntry(hist_zjets, "Z/#gamma*#rightarrowl^{+}l^{-}", "f");
804     # leg.AddEntry(hist_qcd, "QCD", "F");
805     # leg.AddEntry(hist_qcd, "QCD/#gamma + jets");
806     # leg.AddEntry(hist_singleTop, "Single-Top")
807     # leg.AddEntry(hist_Zprime500, "Z' 0.5TeV (50 pb)", "L");
808     # leg.AddEntry(hist_Zprime750, "Z' 0.75TeV", "L");
809     # leg.AddEntry(hist_Zprime1000, "Z' 1TeV", "L");
810     # leg.AddEntry(hist_Zprime1500, "Z' 1.5TeV (50 pb)", "L");
811     # leg.AddEntry(hist_Zprime2000, "Z' 2TeV", "L");
812     # leg.AddEntry(hist_Zprime4000, "Z' 4TeV (50 pb)", "L");
813     # leg.AddEntry(hist_Tprime300, "T' 300GeV", "L");
814     # leg.AddEntry(hist_Tprime350, "T' 350GeV", "L");
815     # leg.AddEntry(hist_Zprime1250, "Z' 1.25TeV");
816     leg.AddEntry(hist_wpm600, "W't .6TeV (8 pb)", "L");
817     leg.AddEntry(hist_wpm800, "W't .8TeV (2.2 pb)", "L");
818     leg.AddEntry(hist_wpm1000, "W't 1TeV (0.72 pb)", "L");
819     # leg.AddEntry(hist_wptb1000, "W'-> tb 1TeV (8 pb)", "L");
820    
821    
822     canvases.append(TCanvas("cname" + histname, histname, 1200, 900))
823     canvases[-1].cd().SetRightMargin(0.04);
824     if ('jet1' in histname):
825     hist_data.SetMarkerStyle(8);
826     # hist_ttbar.SetMarkerStyle(21);
827     # hist_ttbar.SetMarkerColor(kRed);
828     hist_Tprime300.SetMarkerColor(kOrange + 1);
829     hist_Tprime300.SetMarkerStyle(22);
830     hist_Tprime350.SetMarkerColor(kGray + 3);
831     hist_Tprime350.SetMarkerStyle(3);
832     else:
833     hs = THStack("MC", "MC");
834     # hs.Add(hist_qcd);
835     # hs.Add(hist_zjets);
836     # hs.Add(hist_wjets);
837     # hs.Add(hist_singleTop);
838     # hs.Add(hist_ttbar);
839     max = 0
840     if hs.GetMaximum() > hist_data.GetMaximum():
841     max = hs.GetMaximum()*1.1
842     else:
843     max = hist_data.GetMaximum()*1.1
844     if hist_wpm600.GetMaximum() > max :
845     max = hist_wpm600.GetMaximum() * 1.1
846    
847     hist_data.GetYaxis().SetRangeUser(0, max);
848     hist_data.SetStats(1);
849     # hist_data.Draw('error');
850     hist_data.Draw();
851     hist_data.SetMarkerStyle(8);
852     # if ('jet1' in histname):
853     # hist_ttbar.Draw("same");
854     # else:
855     # hs.Draw("hist same");
856     # hist_Zprime500.Draw("same");
857     # hist_Zprime750.Draw("same");
858     # hist_Zprime1000.Draw("same");
859     # hist_Zprime2000.Draw("same");
860     # hist_Zprime4000.Draw("same");
861     # hist_Tprime300.Draw("same");
862     # hist_Tprime350.Draw("same");
863     # hist_Zprime1250.Draw("same");
864     # hist_Zprime1500.Draw("same");
865     hist_wpm1000.Draw("same");
866     hist_wpm600.Draw("same");
867     hist_wpm800.Draw("same");
868     # hist_wptb1000.Draw("same");
869     # qcdUncert.Draw("1 same");
870     # hist_data2.Draw("error same");
871     # hist_data.Draw("error same");
872     # hist_data.Draw("same");
873     leg.Draw();
874    
875     text1 = TLatex(3.570061, 23.08044, "CMS Preliminary");
876     text1.SetNDC();
877     text1.SetTextAlign(13);
878     text1.SetX(0.38);
879     text1.SetY(0.928);
880     #text1.SetLineWidth(2);
881     text1.SetTextFont(42);
882     text1.SetTextSizePixels(24);# dflt=28
883     text1.SetTextSize(0.03);
884     text1.Draw();
885    
886     text2 = TLatex(3.570061, 23.08044, "%d pb^{-1} at #sqrt{s} = 7 TeV" % lumi);
887     text2.SetNDC();
888     text2.SetTextAlign(13);
889     text2.SetX(0.38);
890     text2.SetY(0.88);
891     #text2.SetLineWidth(2);
892     text2.SetTextFont(42);
893     text2.SetTextSizePixels(24);# dflt=28
894     text2.SetTextSize(0.03);
895     text2.Draw();
896     canvases[-1].SaveAs('plots/' + histname + '.png')
897    
898     # if (histname == "wpnegmass_elec"):
899     # hist_ttbargood = hist_ttbar.Clone("test");
900     # hist_wprimegood = hist_wpm600.Clone("test");
901     # elif (histname == "wpnegmassbad_elec"):
902     # asymmetryplot(hist_ttbargood, hist_ttbar, 'ttbarasym');
903     # asymmetryplot(hist_wprimegood, hist_wpm600, 'wprimeasym');
904     # asymmetry2plot(hist_ttbargood, hist_ttbar, hist_wprimegood, hist_wpm600, 'wpttcombined');
905     # plusminuscmp(hist_wpm600, hist_wprimegood, "wpnegmassgoodbad", 0);
906     # hist_wpm600.Add(hist_ttbar);
907     # hist_wprimegood.Add(hist_ttbargood);
908     # plusminuscmp(hist_wpm600, hist_wprimegood, "wpnegmtotgoodbad", 0);
909     # elif (histname == "wpnegmasspos"):
910     # hist_wprimepos = hist_wpm600.Clone("test");
911     # hist_ttbarbad = hist_ttbar.Clone("test");
912     # elif (histname == "wpnegmassneg"):
913     # plusminuscmp(hist_wprimepos, hist_wpm600, 'wpmassplusminus', 1);
914    
915    
916     # canvases[-1].SaveAs('plots/' + histname + '.root')
917     # canvases[-1].SaveAs('/storage/results/' + histname + '.png')
918    
919     # cu_hist_data = getCumulativePlot(hist_data, "data");
920     # cu_hist_ttbar = getCumulativePlot(hist_ttbar, "ttbar");
921     # cu_hist_wjets = getCumulativePlot(hist_wjets, "wjets");
922     # cu_hist_zjets = getCumulativePlot(hist_zjets, "zjets");
923     # cu_hist_qcd = getCumulativePlot(hist_qcd, "qcd");
924     # cu_hist_singleTop = getCumulativePlot(hist_singleTop, "singleTop");
925     # cu_hist_wpm1000 = getCumulativePlot(hist_wpm1000, "Wprimet1000");
926     # cu_hist_wpm600 = getCumulativePlot(hist_wpm600, "Wprimet600");
927     # cu_hist_wpm800 = getCumulativePlot(hist_wpm800, "Wprimet800");
928     # cu_hist_wptb1000 = getCumulativePlot(hist_wptb1000, "Wprimetb1000");
929     # cu_hist_Zprime500 = getCumulativePlot(hist_Zprime500, "Zprime500");
930     # cu_hist_Zprime750 = getCumulativePlot(hist_Zprime750, "Zprime750");
931     # cu_hist_Zprime1000 = getCumulativePlot(hist_Zprime1000, "Zprime1000");
932     # cu_hist_Zprime2000 = getCumulativePlot(hist_Zprime2000, "Zprime2000");
933     # cu_hist_Zprime4000 = getCumulativePlot(hist_Zprime4000, "Zprime4000");
934     # cu_hist_Tprime300 = getCumulativePlot(hist_Tprime300, "Tprime300");
935     # cu_hist_Tprime350 = getCumulativePlot(hist_Tprime350, "Tprime350");
936     ## cu_hist_Zprime1250 = getCumulativePlot(hist_Zprime1250, "Zprime1250");
937     ## cu_hist_Zprime1500 = getCumulativePlot(hist_Zprime1500, "Zprime1500");
938     # cu_hist_data.SetYTitle("Integrated Events/(50 GeV)");
939     ##
940     #
941     # cu_hist_data.SetAxisRange(Urange[0], Urange[1]);
942     # cu_hist_ttbar.SetAxisRange(Urange[0], Urange[1]);
943     # cu_hist_wjets.SetAxisRange(Urange[0], Urange[1]);
944     # cu_hist_zjets.SetAxisRange(Urange[0], Urange[1]);
945     # cu_hist_qcd.SetAxisRange(Urange[0], Urange[1]);
946     # cu_hist_singleTop.SetAxisRange(Urange[0], Urange[1]);
947     # cu_hist_wpm1000.SetAxisRange(Urange[0], Urange[1]);
948     # cu_hist_wpm600.SetAxisRange(Urange[0], Urange[1]);
949     # cu_hist_wpm800.SetAxisRange(Urange[0], Urange[1]);
950     # cu_hist_wptb1000.SetAxisRange(Urange[0], Urange[1]);
951    
952     # cu_hs = THStack("cu_MC", "cu_MC");
953     # cu_hs.Add(cu_hist_qcd);
954     # cu_hs.Add(cu_hist_zjets);
955     # cu_hs.Add(cu_hist_wjets);
956     # cu_hs.Add(cu_hist_ttbar);
957    
958     # scanvases.append(TCanvas("cu_cname" + histname, histname + "(cu)", 1200, 900))
959     # scanvases[-1].cd().SetRightMargin(0.04);
960     # cu_hist_data.Draw("error");
961     # cu_hs.Draw("hist same");
962     # cu_hist_wpm1000.Draw("same");
963     # cu_hist_wpm600.Draw("same");
964     # cu_hist_wpm800.Draw("same");
965     # cu_hist_wptb1000.Draw("same");
966     # cu_hist_Zprime500.Draw("same");
967     # cu_hist_Zprime750.Draw("same");
968     # cu_hist_Zprime1000.Draw("same");
969     # cu_hist_Zprime2000.Draw("same");
970     # cu_hist_Zprime4000.Draw("same");
971     # cu_hist_Tprime300.Draw("same");
972     # cu_hist_Tprime350.Draw("same");
973     ### cu_hist_Zprime1250.Draw("same");
974     ### cu_hist_Zprime1500.Draw("same");
975     ## # cu_hist_data2.Draw("error same");
976     # cu_hist_data.Draw("error same");
977     # leg.Draw();
978     ##
979     # text1.Draw();
980     ##
981     # text2.Draw();
982     # scanvases[-1].SaveAs('/storage/results/' + histname + '_integrated.png')
983     # scanvases[-1].SaveAs('plots/' + histname + '_integrated.png')
984    
985    
986    
987     # tdrGrid: Turns the grid lines on (true) or off (false)
988    
989     #def tdrGrid(bool gridOn):
990     # tdrStyle.SetPadGridX(gridOn);
991     # tdrStyle.SetPadGridY(gridOn);
992     #
993    
994     # fixOverlay: Redraws the axis
995    
996     def setCorrErrs(hist_good, hist_bad, hist_diff):
997     nbins = hist_diff.GetXaxis().GetNbins();
998     for i in range(1, nbins):
999     goodval = hist_good.GetBinContent(i);
1000     badval = hist_bad.GetBinContent(i);
1001     errsum = goodval + badval;
1002     if errsum > 0:
1003     errprod = goodval * badval;
1004     # Assume 90% correlation
1005     tot = 2.0 * errprod * (errsum + 3.6 * sqrt(errprod));
1006     tot = sqrt(tot) / (errsum * errsum);
1007     hist_diff.SetBinError(i, tot);
1008    
1009    
1010    
1011     def setUncorrErrs(hist_good, hist_bad, hist_diff):
1012     nbins = hist_diff.GetXaxis().GetNbins();
1013     for i in range(1, nbins):
1014     goodval = hist_good.GetBinContent(i);
1015     badval = hist_bad.GetBinContent(i);
1016     err = goodval + badval;
1017     err = err * err * err;
1018     if err > 0:
1019     err = 4.0 * sqrt((goodval * badval) / err);
1020     hist_diff.SetBinError(i, err);
1021    
1022    
1023     def asymmetryplot(hist_good, hist_bad, histname):
1024     hist_total = hist_good.Clone("test");
1025     hist_total.Sumw2();
1026     hist_total.Add(hist_bad);
1027     hist_diff = hist_good.Clone("test");
1028     hist_diff.Sumw2();
1029     hist_diff.Add(hist_bad, -1);
1030     hist_total.Scale(0.5);
1031     hist_diff.Divide(hist_total);
1032     setCorrErrs(hist_good, hist_bad, hist_diff);
1033     hist_diff.SetXTitle("W' mass/GeV");
1034     hist_diff.SetYTitle("Asymmetry");
1035     hist_diff.SetAxisRange(200, 1100);
1036     hist_diff.SetLineWidth(2);
1037     # hist_diff.SetFillStyle(1001);
1038     hist_diff.SetFillStyle(3001);
1039     hist_diff.SetFillColor(kRed);
1040     canvases[-1].SetGridy();
1041     hist_diff.GetYaxis().SetRangeUser(-1, 1);
1042     hist_diff.Draw("hist E1");
1043     canvases[-1].SaveAs('plots/' + histname + '.png')
1044    
1045    
1046     def asymmetry2plot(ttbar_good, ttbar_bad, wp_good, wp_bad, histname):
1047     hist_total = ttbar_good.Clone("test");
1048     hist_total.Sumw2();
1049     hist_total.Add(ttbar_bad);
1050     hist_total.Add(wp_good);
1051     hist_total.Add(wp_bad);
1052     hist_total.Scale(0.5);
1053     hist_diff = ttbar_good.Clone("test");
1054     hist_diff.Sumw2();
1055     hist_diff.Add(wp_good);
1056     hist_diff.Add(ttbar_bad, -1);
1057     hist_diff.Add(wp_bad, -1);
1058     hist_diff.Divide(hist_total);
1059     # nbins = hist_diff.GetXaxis().GetNbins();
1060     hist_good = ttbar_good.Clone("test");
1061     hist_good.Add(wp_good);
1062     hist_bad = ttbar_bad.Clone("test");
1063     hist_bad.Add(wp_bad);
1064     setCorrErrs(hist_good, hist_bad, hist_diff);
1065     hist_diff.SetXTitle("W' mass/GeV");
1066     hist_diff.SetYTitle("Asymmetry");
1067     hist_diff.SetAxisRange(200, 1100);
1068     hist_diff.SetLineWidth(2);
1069     # hist_diff.SetFillStyle(1001);
1070     hist_diff.SetFillStyle(3001);
1071     hist_diff.SetFillColor(kRed);
1072     canvases[-1].SetGridy();
1073     hist_diff.GetYaxis().SetRangeUser(-1, 1);
1074     hist_diff.Draw("hist E1");
1075     canvases[-1].SaveAs('plots/' + histname + '.png')
1076    
1077    
1078    
1079     def plusminuscmp(hist_pos, hist_neg, hist_name, sign):
1080     hist_neg.SetXTitle("Reco W' mass/GeV");
1081     hist_neg.SetYTitle("Events/(60 GeV)");
1082     hist_pos.SetLineColor(kGray + 3);
1083     hist_neg.SetLineColor(kRed);
1084     linewidth = 6;
1085     hist_pos.SetLineWidth(linewidth);
1086     hist_neg.SetLineWidth(linewidth);
1087     leg = TLegend(0.696, 0.35, 0.95, 0.92);
1088     leg.SetBorderSize(0);
1089     leg.SetLineStyle(0);
1090     leg.SetTextFont(42);
1091     leg.SetFillStyle(0);
1092     if sign == 1:
1093     leg.AddEntry(hist_pos, "W't+ .6TeV (8 pb)", "L");
1094     leg.AddEntry(hist_neg, "W't- .6TeV (8 pb)", "L");
1095     else:
1096     leg.AddEntry(hist_pos, "W't bad .6TeV (8 pb)", "L");
1097     leg.AddEntry(hist_neg, "W't good .6TeV (8 pb)", "L");
1098     hist_pos.SetStats(1);
1099     hist_neg.Draw();
1100     hist_pos.Draw("same");
1101     leg.Draw();
1102     canvases[-1].SaveAs('plots/' + hist_name + '.png')
1103    
1104    
1105     def fixOverlay():
1106     gPad.RedrawAxis();
1107    
1108    
1109     def getCumulativePlot(initial, type):
1110     global counter
1111     counter = counter + 1;
1112     name = initial.GetName()
1113     name = "cu_" + name + "_" + type + str(counter);
1114     title = initial.GetTitle()
1115     title = "cu_" + title + "_" + type;
1116     xaxis = initial.GetXaxis().GetTitle();
1117     yaxis = initial.GetYaxis().GetTitle();
1118     nBins = initial.GetNbinsX();
1119     cu = TH1F(name, title, nBins, initial.GetXaxis().GetXmin(), initial.GetXaxis().GetXmax());
1120     for bin in range(1,nBins+1):
1121     cu.SetBinContent(bin, initial.Integral(bin, nBins));
1122    
1123     cu.SetFillStyle(initial.GetFillStyle());
1124     cu.SetFillColor(initial.GetFillColor());
1125     cu.SetLineColor(initial.GetLineColor());
1126     cu.SetMarkerSize(initial.GetMarkerSize());
1127     cu.SetMarkerStyle(initial.GetMarkerStyle());
1128     cu.SetMarkerColor(initial.GetMarkerColor());
1129     cu.SetLineWidth(initial.GetLineWidth());
1130     cu.GetXaxis().SetTitle(xaxis);
1131     cu.GetYaxis().SetTitle(yaxis);
1132     return cu;
1133    
1134     if __name__ == "__main__":
1135     gROOT.SetBatch(True)
1136     gROOT.ProcessLine('gErrorIgnoreLevel = 1001;')
1137     plotMttbar()
1138     # print "press enter to quit"
1139     # a = raw_input()