ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/dhidas/OSUAnalysis/Tools/scripts/makePlots.py
Revision: 1.1
Committed: Thu Dec 1 16:28:48 2011 UTC (13 years, 5 months ago) by dhidas
Content type: text/x-python
Branch point for: dhidas, MAIN
Log Message:
Initial revision

File Contents

# User Rev Content
1 dhidas 1.1 from HistGetter import *
2     from HistPlotter import *
3     from time import sleep
4     from ROOT import *
5    
6     if __name__ == '__main__':
7     gROOT.SetBatch(True)
8     gROOT.ProcessLine('gErrorIgnoreLevel = 1001;')
9    
10     files = {'data':"/storage/results/QCD_Estimate/Preapproval3/data_36.145pb_PFElectron_PF2PATJets_PFMET.root",
11     'ttbar' : "/storage/results/QCD_Estimate/Preapproval3/ttjet_36.145pb_PFElectron_PF2PATJets_PFMET.root",
12     'wjets' : "/storage/results/QCD_Estimate/Preapproval3/wj_36.145pb_PFElectron_PF2PATJets_PFMET.root",
13     'zjets' : "/storage/results/QCD_Estimate/Preapproval3/zj_36.145pb_PFElectron_PF2PATJets_PFMET.root",
14     'bce1' : "/storage/results/QCD_Estimate/Preapproval3/bce1_36.145pb_PFElectron_PF2PATJets_PFMET.root",
15     'bce2' : "/storage/results/QCD_Estimate/Preapproval3/bce2_36.145pb_PFElectron_PF2PATJets_PFMET.root",
16     'bce3' : "/storage/results/QCD_Estimate/Preapproval3/bce3_36.145pb_PFElectron_PF2PATJets_PFMET.root",
17     'enri1' : "/storage/results/QCD_Estimate/Preapproval3/enri1_36.145pb_PFElectron_PF2PATJets_PFMET.root",
18     'enri2' : "/storage/results/QCD_Estimate/Preapproval3/enri2_36.145pb_PFElectron_PF2PATJets_PFMET.root",
19     'enri3' : "/storage/results/QCD_Estimate/Preapproval3/enri3_36.145pb_PFElectron_PF2PATJets_PFMET.root",
20     'pj1' : "/storage/results/QCD_Estimate/Preapproval3/pj1_36.145pb_PFElectron_PF2PATJets_PFMET.root",
21     'pj2' : "/storage/results/QCD_Estimate/Preapproval3/pj2_36.145pb_PFElectron_PF2PATJets_PFMET.root",
22     'pj3' : "/storage/results/QCD_Estimate/Preapproval3/pj3_36.145pb_PFElectron_PF2PATJets_PFMET.root"}
23    
24     histsNames = ['mttbar_conversions_allBtags',
25     'mttbar_conversions_withMETAndAsymJets_allBtags',
26     'mttbar_antiIsolated_withMETAndAsymJets_allBtags',
27     'mttbar_antiIsolated_allBtags',
28     'mttbar_controlRegion_withMETAndAsymJets_allBtags',
29     'mttbar_controlRegion_allBtags']
30    
31     # HistGetter.samplefiles = files
32     # HG = HistGetter()
33     setStyle()
34     hists = getHistsFromFiles( histsNames, files )
35     hists = addSampleSum( hists )
36     hists = applyDefaultStylesAndColors( hists )
37    
38     hists = rebin( hists, 50, 'mttbar*' )
39     hists = setXRange( hists, ( 300, 1800 ), 'mttbar*' )
40    
41     hists = setXTitle( hists, 'm(t#bar{t})/GeV', 'mttbar*' )
42     hists = setYTitle( hists, 'Events/(50 GeV)', 'mttbar*' )
43    
44     hists = makeDetailedMCStack( hists )
45    
46     normConv = hists['data']['mttbar_conversions_withMETAndAsymJets_allBtags']
47     normContr = hists['data']['mttbar_controlRegion_withMETAndAsymJets_allBtags']
48     normAnti = hists['data']['mttbar_antiIsolated_withMETAndAsymJets_allBtags']
49    
50     normConv.SetYTitle("a.u/50GeV");
51    
52     normConv.Sumw2()
53     normContr.Sumw2()
54     normAnti.Sumw2()
55    
56     normConv.Scale(1/normConv.Integral())
57     normContr.Scale(1/normContr.Integral())
58     normAnti.Scale(1/normAnti.Integral())
59    
60     diff = normConv.Clone()
61     diff.Divide(normAnti)
62     line = TLine(300, 1, 1800, 1)
63     line.SetLineColor(1)
64     line.SetLineWidth(4)
65    
66    
67     c = TCanvas("cname", 'cname', 1920, 1080)
68     diff.SetYTitle("conversions/non-iso electrons /50GeV");
69     diff.Draw('error')
70     line.Draw('same')
71     c.SaveAs('shapeRatio_conversions_NonIsolatedElectrons.png')
72    
73     diff = normConv.Clone()
74     diff.Divide(normContr)
75    
76     c = TCanvas("cname2", 'cname2', 1920, 1080)
77     diff.SetYTitle("conversions/fake electrons /50GeV");
78     diff.Draw('error')
79     line.Draw('same')
80     c.SaveAs('shapeRatio_conversions_fakeElectrons.png')
81    
82     c = TCanvas("cname3", 'cname3', 1920, 1080)
83    
84     normConv.SetFillColor(kYellow)
85     normConv.SetFillStyle( 1001 )
86    
87     normContr.SetFillColor(kAzure - 2)
88     normContr.SetFillStyle( 3005 )
89    
90     normAnti.SetFillColor(kRed + 1)
91     normAnti.SetFillStyle( 3004 )
92     normConv.GetYaxis().SetRangeUser(0, 0.2);
93     normConv.Draw('hist')
94     normContr.Draw('hist same')
95     normAnti.Draw('hist same')
96    
97     leg = TLegend(0.696, 0.35, 0.94, 0.92);
98     leg.SetBorderSize(0);
99     leg.SetLineStyle(0);
100     leg.SetTextFont(42);
101     leg.SetFillStyle(0);
102    
103     leg.AddEntry(normConv, "conversions", "f");
104     leg.AddEntry(normContr, "fake electrons", "f");
105     leg.AddEntry(normAnti, "non isolated electrons", "f");
106     leg.Draw()
107    
108     c.SaveAs('shape_comparison.png')
109    
110     c = TCanvas("cname4", 'cname4', 1920, 1080)
111     histname = 'mttbar_antiIsolated_withMETAndAsymJets_allBtags'
112     hist = hists['allMCDetailed'][histname]
113     hists['enri1'][histname].GetYaxis().SetRangeUser(0, hist.GetMaximum()*1.4);
114     hists['enri1'][histname].Draw('hist')
115     hist.Draw('hist same')
116    
117     leg = TLegend(0.696, 0.35, 0.94, 0.92);
118     leg.SetBorderSize(0);
119     leg.SetLineStyle(0);
120     leg.SetTextFont(42);
121     leg.SetFillStyle(0);
122    
123     leg.AddEntry(hists['enri1'][histname], "QCD EMEnr. 20to30", "f");
124     leg.AddEntry(hists['enri2'][histname], "QCD EMEnr. 30to80", "f");
125     leg.AddEntry(hists['enri3'][histname], "QCD EMEnr. 80to170", "f");
126     leg.AddEntry(hists['bce1'][histname], "QCD BCToE. 20to30", "f");
127     leg.AddEntry(hists['bce2'][histname], "QCD BCToE. 30to80", "f");
128     leg.AddEntry(hists['bce3'][histname], "QCD BCToE. 80to170", "f");
129     leg.AddEntry(hists['pj1'][histname], "QCD #gamma + jets 40to100", "f");
130     leg.AddEntry(hists['pj2'][histname], "QCD #gamma + jets 100to200", "f");
131     leg.AddEntry(hists['pj3'][histname], "QCD #gamma + jets 200toINF", "f");
132     leg.Draw()
133    
134     c.SaveAs('detailed_MC_NonIsoElectrons.png')
135    
136