| 1 |
from __future__ import division
|
| 2 |
|
| 3 |
from tdrStyle import *
|
| 4 |
from ROOT import *
|
| 5 |
|
| 6 |
canvases = []
|
| 7 |
scanvases = []
|
| 8 |
|
| 9 |
def plotMttbar():
|
| 10 |
tdrstyle = setTDRStyle();
|
| 11 |
gStyle.SetHatchesSpacing(1.0);
|
| 12 |
lumi = 36.145;#3.06 + 1.37 + 2.53 + 0.26;
|
| 13 |
oldLumi = 36.145;
|
| 14 |
scale = lumi / oldLumi;
|
| 15 |
qcdScale = {'default':0.78, 'withMETAndAsymJets': 1.58};
|
| 16 |
# data = TFile.Open("data2.9pb_fullSetOfVars.root");
|
| 17 |
data = TFile.Open("/storage/results/outputfiles/Preapproval2/data_36.145pb_PFElectron_PF2PATJets_PFMET.root");
|
| 18 |
ttbar = TFile.Open("/storage/results/outputfiles/Preapproval2/ttjet_36.145pb_PFElectron_PF2PATJets_PFMET.root");
|
| 19 |
wjets = TFile.Open("/storage/results/outputfiles/Preapproval2/wj_36.145pb_PFElectron_PF2PATJets_PFMET.root");
|
| 20 |
zjets = TFile.Open("/storage/results/outputfiles/Preapproval2/zj_36.145pb_PFElectron_PF2PATJets_PFMET.root");
|
| 21 |
bce1 = TFile.Open("/storage/results/outputfiles/Preapproval2/bce1_36.145pb_PFElectron_PF2PATJets_PFMET.root");
|
| 22 |
bce2 = TFile.Open("/storage/results/outputfiles/Preapproval2/bce2_36.145pb_PFElectron_PF2PATJets_PFMET.root");
|
| 23 |
bce3 = TFile.Open("/storage/results/outputfiles/Preapproval2/bce3_36.145pb_PFElectron_PF2PATJets_PFMET.root");
|
| 24 |
enri1 = TFile.Open("/storage/results/outputfiles/Preapproval2/enri1_36.145pb_PFElectron_PF2PATJets_PFMET.root");
|
| 25 |
enri2 = TFile.Open("/storage/results/outputfiles/Preapproval2/enri2_36.145pb_PFElectron_PF2PATJets_PFMET.root");
|
| 26 |
enri3 = TFile.Open("/storage/results/outputfiles/Preapproval2/enri3_36.145pb_PFElectron_PF2PATJets_PFMET.root");
|
| 27 |
pj1 = TFile.Open("/storage/results/outputfiles/Preapproval2/pj1_36.145pb_PFElectron_PF2PATJets_PFMET.root");
|
| 28 |
pj2 = TFile.Open("/storage/results/outputfiles/Preapproval2/pj2_36.145pb_PFElectron_PF2PATJets_PFMET.root");
|
| 29 |
pj3 = TFile.Open("/storage/results/outputfiles/Preapproval2/pj3_36.145pb_PFElectron_PF2PATJets_PFMET.root");
|
| 30 |
tW = TFile.Open("/storage/results/outputfiles/Preapproval2/tW_36.145pb_PFElectron_PF2PATJets_PFMET.root");
|
| 31 |
tchan = TFile.Open("/storage/results/outputfiles/Preapproval2/tchan_36.145pb_PFElectron_PF2PATJets_PFMET.root");
|
| 32 |
# vqq = TFile.Open("vqq_7.22pb_V4PFCalo.root__fullSetOfVars.root");
|
| 33 |
# Zprime500 = TFile.Open("/storage/workspace/BristolAnalysisTools/outputfiles/Zprime_M500GeV_W5GeV_36.145pb.root");
|
| 34 |
# Zprime750 = TFile.Open("/storage/workspace/BristolAnalysisTools/outputfiles/Zprime_M750GeV_W7500MeV_36.145pb.root");
|
| 35 |
# Zprime1000 = TFile.Open("/storage/workspace/BristolAnalysisTools/outputfiles/Zprime_M1TeV_W10GeV_36.145pb.root");
|
| 36 |
# Zprime1250 = TFile.Open("/storage/workspace/BristolAnalysisTools/outputfiles/Zprime_M1250GeV_W12500MeV_36.145pb.root");
|
| 37 |
# Zprime1500 = TFile.Open("/storage/workspace/BristolAnalysisTools/outputfiles/Zprime_M1500GeV_W15GeV_36.145pb.root");
|
| 38 |
|
| 39 |
hists = [];
|
| 40 |
# hists.append('mttbar_QCDEnriched')
|
| 41 |
hists.append('mttbar_conversions')
|
| 42 |
# hists.append('mttbar_conversions_withMETCut')
|
| 43 |
hists.append('mttbar_conversions_withMETAndAsymJets')
|
| 44 |
# hists.append('mttbar_conversions_withAsymJetsCut')
|
| 45 |
# hists.append('mttbar_controlRegion')
|
| 46 |
# hists.append("mttbar");
|
| 47 |
# hists.append("mttbar_2ndSolution");
|
| 48 |
# hists.append("mttbar_3rdSolution");
|
| 49 |
# hists.append("mttbar_allSolutions");
|
| 50 |
|
| 51 |
# hists.append("mttbar_withMETCut");
|
| 52 |
# hists.append("mttbar_2ndSolution_withMETCut");
|
| 53 |
# hists.append("mttbar_3rdSolution_withMETCut");
|
| 54 |
# hists.append("mttbar_allSolutions_withMETCut");
|
| 55 |
|
| 56 |
# hists.append("mttbar_withMETAndAsymJets");
|
| 57 |
# hists.append("mttbar_2ndSolution_withMETAndAsymJets");
|
| 58 |
# hists.append("mttbar_3rdSolution_withMETAndAsymJets");
|
| 59 |
# hists.append("mttbar_allSolutions_withMETAndAsymJets");
|
| 60 |
|
| 61 |
# hists.append("mttbar_withAsymJetsCut");
|
| 62 |
# hists.append("mttbar_2ndSolution_withAsymJetsCut");
|
| 63 |
# hists.append("mttbar_3rdSolution_withAsymJetsCut");
|
| 64 |
# hists.append("mttbar_allSolutions_withAsymJetsCut");
|
| 65 |
|
| 66 |
# hists.append("ttbar_pt");
|
| 67 |
# hists.append("ttbar_pt_2ndSolution");
|
| 68 |
# hists.append("ttbar_pt_3rdSolution");
|
| 69 |
# hists.append("ttbar_pt_allSolutions");
|
| 70 |
#
|
| 71 |
# hists.append("ttbar_pt_withMETCut");
|
| 72 |
# hists.append("ttbar_pt_2ndSolution_withMETCut");
|
| 73 |
# hists.append("ttbar_pt_3rdSolution_withMETCut");
|
| 74 |
# hists.append("ttbar_pt_allSolutions_withMETCut");
|
| 75 |
#
|
| 76 |
# hists.append("ttbar_pt_withMETAndAsymJets");
|
| 77 |
# hists.append("ttbar_pt_2ndSolution_withMETAndAsymJets");
|
| 78 |
# hists.append("ttbar_pt_3rdSolution_withMETAndAsymJets");
|
| 79 |
# hists.append("ttbar_pt_allSolutions_withMETAndAsymJets");
|
| 80 |
#
|
| 81 |
# hists.append("ttbar_pt_withAsymJetsCut");
|
| 82 |
# hists.append("ttbar_pt_2ndSolution_withAsymJetsCut");
|
| 83 |
# hists.append("ttbar_pt_3rdSolution_withAsymJetsCut");
|
| 84 |
# hists.append("ttbar_pt_allSolutions_withAsymJetsCut");
|
| 85 |
# hists.append("angleTops");
|
| 86 |
# hists.append("angleTops_withMETCut");
|
| 87 |
# hists.append("angleTops_withMETAndAsymJets");
|
| 88 |
# hists.append("angleTops_withAsymJetsCut");
|
| 89 |
#
|
| 90 |
# hists.append("pt_leadingTop");
|
| 91 |
# hists.append("pt_leadingTop_withMETCut");
|
| 92 |
# hists.append("pt_leadingTop_withMETAndAsymJets");
|
| 93 |
# hists.append("pt_leadingTop_withAsymJetsCut");
|
| 94 |
#
|
| 95 |
# hists.append("pt_NextToLeadingTop");
|
| 96 |
# hists.append("pt_NextToLeadingTop_withMETCut");
|
| 97 |
# hists.append("pt_NextToLeadingTop_withMETAndAsymJets");
|
| 98 |
# hists.append("pt_NextToLeadingTop_withAsymJetsCut");
|
| 99 |
#
|
| 100 |
# hists.append("mLeptonicTop");
|
| 101 |
# hists.append("mHadronicTop");
|
| 102 |
# hists.append("mAllTop");
|
| 103 |
#
|
| 104 |
# hists.append("ttbar_px");
|
| 105 |
# hists.append("ttbar_py");
|
| 106 |
# hists.append("ttbar_pz");
|
| 107 |
#
|
| 108 |
# hists.append("m3");
|
| 109 |
# hists.append("HT");
|
| 110 |
# hists.append("MET");
|
| 111 |
# hists.append("leadingJetMass");
|
| 112 |
# hists.append("mtW");
|
| 113 |
# hists.append("neutrino_pz");
|
| 114 |
suffixes = ["allBtags",
|
| 115 |
"1orMoreBtag",
|
| 116 |
"2orMoreBtags",
|
| 117 |
"3orMoreBtags",'4orMoreBtags' ]
|
| 118 |
hists = [hist + '_' + suffix for hist in hists for suffix in suffixes]
|
| 119 |
|
| 120 |
jetBinned = ["allJets",
|
| 121 |
"1orMoreJets",
|
| 122 |
"2orMoreJets",
|
| 123 |
"3orMoreJets" , "4orMoreJets"]
|
| 124 |
jetBinnedhists = []
|
| 125 |
# jetBinnedhists.append('diElectronMass')
|
| 126 |
# jetBinnedhists.append("QCDest_CombRelIso")
|
| 127 |
# jetBinnedhists.append("QCDest_CombRelIso_1btag")
|
| 128 |
# jetBinnedhists.append("QCDest_CombRelIso_2btag")
|
| 129 |
# jetBinnedhists.append("QCDest_CombRelIso_controlRegion")
|
| 130 |
# jetBinnedhists.append("QCDest_CombRelIso_controlRegion_1btag")
|
| 131 |
# jetBinnedhists.append("QCDest_CombRelIso_controlRegion_2btag")
|
| 132 |
#
|
| 133 |
# jetBinnedhists.append("QCDest_PFIsolation")
|
| 134 |
# jetBinnedhists.append("QCDest_PFIsolation_WithMETCut")
|
| 135 |
# jetBinnedhists.append("QCDest_PFIsolation_WithMETCutAndAsymJetCuts")
|
| 136 |
# jetBinnedhists.append("QCDest_PFIsolation_WithAsymJetCuts")
|
| 137 |
#
|
| 138 |
# jetBinnedhists.append("QCDest_PFIsolation_1btag")
|
| 139 |
# jetBinnedhists.append("QCDest_PFIsolation_2btag")
|
| 140 |
# jetBinnedhists.append("QCDest_PFIsolation_controlRegion")
|
| 141 |
# jetBinnedhists.append("QCDest_PFIsolation_controlRegion_1btag")
|
| 142 |
# jetBinnedhists.append("QCDest_PFIsolation_controlRegion_2btag")
|
| 143 |
# jetBinnedhists.append("QCDest_PFIsolation_controlRegion2")
|
| 144 |
# jetBinnedhists.append("QCDest_PFIsolation_controlRegion2_WithMETCut")
|
| 145 |
jetBinnedhists.append("QCDest_PFIsolation_controlRegion2_WithMETCutAndAsymJetCuts")
|
| 146 |
# jetBinnedhists.append("QCDest_PFIsolation_controlRegion2_WithAsymJetCuts")
|
| 147 |
# jetBinnedhists.append("QCDest_PFIsolation_controlRegion2_1btag")
|
| 148 |
# jetBinnedhists.append("QCDest_PFIsolation_controlRegion2_2btag")
|
| 149 |
|
| 150 |
jetBinnedhists = [hist + '_' + suffix for hist in jetBinnedhists for suffix in jetBinned]
|
| 151 |
hists.extend(jetBinnedhists)
|
| 152 |
gcd = gROOT.cd
|
| 153 |
|
| 154 |
for histname in hists:
|
| 155 |
currentSelection = 'default'
|
| 156 |
if 'withMETAndAsymJets' in histname:
|
| 157 |
currentSelection = 'withMETAndAsymJets'
|
| 158 |
gcd()
|
| 159 |
print histname
|
| 160 |
hist_data = data.Get(histname);
|
| 161 |
|
| 162 |
# hist_data2;
|
| 163 |
# if (histname == "mttbar_rebinned")
|
| 164 |
# hist_data2 = data2.Get(histname);
|
| 165 |
# else
|
| 166 |
# hist_data2 = TH1F(*hist_data);
|
| 167 |
# hist_data.Sumw2();
|
| 168 |
# hist_data2.Sumw2();
|
| 169 |
hist_ttbar = ttbar.Get(histname);
|
| 170 |
hist_wjets = wjets.Get(histname);
|
| 171 |
hist_zjets = zjets.Get(histname);
|
| 172 |
|
| 173 |
hist_bce1 = bce1.Get(histname);
|
| 174 |
hist_bce2 = bce2.Get(histname);
|
| 175 |
hist_bce3 = bce3.Get(histname);
|
| 176 |
hist_enri1 = enri1.Get(histname);
|
| 177 |
hist_enri2 = enri2.Get(histname);
|
| 178 |
hist_enri3 = enri3.Get(histname);
|
| 179 |
hist_pj1 = pj1.Get(histname);
|
| 180 |
hist_pj2 = pj2.Get(histname);
|
| 181 |
hist_pj3 = pj3.Get(histname);
|
| 182 |
hist_singleTop = tW.Get(histname)
|
| 183 |
hist_singleTop.Add(tchan.Get(histname))
|
| 184 |
|
| 185 |
# hist_Zprime500 = Zprime500.Get(histname);
|
| 186 |
# hist_Zprime750 = Zprime750.Get(histname);
|
| 187 |
# hist_Zprime1000 = Zprime1000.Get(histname);
|
| 188 |
# hist_Zprime1250 = Zprime1250.Get(histname);
|
| 189 |
# hist_Zprime1500 = Zprime1500.Get(histname);
|
| 190 |
|
| 191 |
hist_ttbar.Scale(scale);
|
| 192 |
hist_wjets.Scale(scale);
|
| 193 |
hist_zjets.Scale(scale);
|
| 194 |
|
| 195 |
hist_bce1.Scale(scale);
|
| 196 |
hist_bce2.Scale(scale);
|
| 197 |
hist_bce3.Scale(scale);
|
| 198 |
|
| 199 |
hist_enri1.Scale(scale);
|
| 200 |
hist_enri2.Scale(scale);
|
| 201 |
hist_enri3.Scale(scale);
|
| 202 |
|
| 203 |
hist_pj1.Scale(scale);
|
| 204 |
hist_pj2.Scale(scale);
|
| 205 |
hist_pj3.Scale(scale);
|
| 206 |
|
| 207 |
# hist_Zprime500.Scale(scale);
|
| 208 |
# hist_Zprime750.Scale(scale);
|
| 209 |
# hist_Zprime1000.Scale(scale);
|
| 210 |
# hist_Zprime1250.Scale(scale);
|
| 211 |
# hist_Zprime1500.Scale(scale);
|
| 212 |
|
| 213 |
hist_qcd = hist_bce1.Clone("qcd")#TH1F(*hist_bce1);
|
| 214 |
hist_qcd.Add(hist_bce2);
|
| 215 |
hist_qcd.Add(hist_bce3);
|
| 216 |
hist_qcd.Add(hist_enri1);
|
| 217 |
hist_qcd.Add(hist_enri2);
|
| 218 |
hist_qcd.Add(hist_enri3);
|
| 219 |
hist_qcd.Add(hist_pj1);
|
| 220 |
hist_qcd.Add(hist_pj2);
|
| 221 |
hist_qcd.Add(hist_pj3);
|
| 222 |
hist_qcd.Scale(qcdScale[currentSelection]);
|
| 223 |
# ndata = hist_data.Integral();
|
| 224 |
# ntop = hist_ttbar.Integral();
|
| 225 |
# nwj = hist_wjets.Integral();
|
| 226 |
# nzj = hist_zjets.Integral();
|
| 227 |
nqcd = hist_qcd.Integral();
|
| 228 |
# sumMC = ntop + nwj + nzj + nqcd;
|
| 229 |
# cout << ndata << " " << sumMC << endl;
|
| 230 |
# hist_wjets.Scale(ndata / sumMC);
|
| 231 |
# hist_ttbar.Scale(ndata / sumMC);
|
| 232 |
# hist_zjets.Scale(ndata / sumMC);
|
| 233 |
# hist_qcd.Scale(ndata / sumMC);
|
| 234 |
mttbars = ['mttbar_' + suffix for suffix in suffixes]
|
| 235 |
mttbars2 = ['mttbar_withMETAndAsymJets_' + suffix for suffix in suffixes]
|
| 236 |
if histname in mttbars or histname in mttbars2:
|
| 237 |
print "taking QCD shape from DATA"
|
| 238 |
name = histname.replace('mttbar', 'mttbar_conversions')
|
| 239 |
hist_qcd = data.Get(name)
|
| 240 |
if( hist_qcd.Integral() > 0):
|
| 241 |
hist_qcd.Scale(nqcd/hist_qcd.Integral())
|
| 242 |
hist_mc = hist_qcd.Clone("all_mc")
|
| 243 |
hist_mc.Add(hist_ttbar);
|
| 244 |
hist_mc.Add(hist_zjets);
|
| 245 |
hist_mc.Add(hist_wjets);
|
| 246 |
hist_mc.Add(hist_singleTop);
|
| 247 |
|
| 248 |
rebin = 1;
|
| 249 |
Urange = (0,5000)
|
| 250 |
if ("mttbar" in histname):
|
| 251 |
hist_data.SetXTitle("m(t#bar{t})/GeV");
|
| 252 |
hist_data.SetYTitle("Events/(50 GeV)");
|
| 253 |
rebin = 50;
|
| 254 |
Urange = (300, 1800)
|
| 255 |
elif ("m3" in histname):
|
| 256 |
hist_data.SetXTitle("M3/GeV");
|
| 257 |
hist_data.SetYTitle("Events/(50 GeV)");
|
| 258 |
rebin = 50;
|
| 259 |
Urange = (0, 1500)
|
| 260 |
elif (histname == "electron_et"):
|
| 261 |
hist_data.SetXTitle("electron p_{T}/GeV");
|
| 262 |
hist_data.SetYTitle("Events/(5 GeV)");
|
| 263 |
rebin = 5;
|
| 264 |
elif ("ttbar_pt" in histname):
|
| 265 |
hist_data.SetXTitle("p_{T} of t#bar{t} system/GeV");
|
| 266 |
hist_data.SetYTitle("Events/(10 GeV)");
|
| 267 |
rebin = 10;
|
| 268 |
Urange = (0, 500)
|
| 269 |
elif ("ttbar_px" in histname):
|
| 270 |
hist_data.SetXTitle("p_{x} of t#bar{t} system/GeV");
|
| 271 |
hist_data.SetYTitle("Events/(10 GeV)");
|
| 272 |
rebin = 10;
|
| 273 |
Urange = (0, 500)
|
| 274 |
elif ("ttbar_py" in histname):
|
| 275 |
hist_data.SetXTitle("p_{y} of t#bar{t} system/GeV");
|
| 276 |
hist_data.SetYTitle("Events/(10 GeV)");
|
| 277 |
rebin = 10;
|
| 278 |
Urange = (0, 500)
|
| 279 |
elif ("ttbar_pz" in histname):
|
| 280 |
hist_data.SetXTitle("p_{z} of t#bar{t} system/GeV");
|
| 281 |
hist_data.SetYTitle("Events/(50 GeV)");
|
| 282 |
rebin = 50;
|
| 283 |
Urange = (0, 2000)
|
| 284 |
elif ("HT" in histname):
|
| 285 |
hist_data.SetXTitle("#Sigma p_{T}/GeV");
|
| 286 |
hist_data.SetYTitle("Events/(50 GeV)");
|
| 287 |
rebin = 50;
|
| 288 |
Urange = (0, 2000)
|
| 289 |
elif (histname == "numberOfJets"):
|
| 290 |
hist_data.SetXTitle("number of jets");
|
| 291 |
hist_data.SetYTitle("Events");
|
| 292 |
elif (histname == "numberOfBJets"):
|
| 293 |
hist_data.SetXTitle("number of b-tagged jets (SSVHE medium)");
|
| 294 |
hist_data.SetYTitle("Events");
|
| 295 |
elif ('MET_' in histname):
|
| 296 |
hist_data.SetXTitle("MET/GeV");
|
| 297 |
hist_data.SetYTitle("Events/(10 GeV)");
|
| 298 |
rebin = 2;
|
| 299 |
Urange = (0, 300)
|
| 300 |
elif ("leadingJetMass" in histname):
|
| 301 |
hist_data.SetXTitle("leading jet mass/GeV");
|
| 302 |
hist_data.SetYTitle("Events/(5 GeV)");
|
| 303 |
rebin = 5;
|
| 304 |
Urange = (0, 150)
|
| 305 |
elif ("mtW" in histname):
|
| 306 |
hist_data.SetXTitle("transverse W-boson mass/GeV");
|
| 307 |
hist_data.SetYTitle("Events/(10 GeV)");
|
| 308 |
rebin = 10;
|
| 309 |
elif ("electronD0" in histname):
|
| 310 |
hist_data.SetXTitle("electron d_{0} / cm");
|
| 311 |
hist_data.SetYTitle("Events/(0.001 cm)");
|
| 312 |
rebin = 10;
|
| 313 |
elif ("angleTops" in histname):
|
| 314 |
hist_data.SetXTitle("angle between top quarks");
|
| 315 |
hist_data.SetYTitle("Events/(0.1 rad)");
|
| 316 |
rebin = 20;
|
| 317 |
elif ("neutrino_pz" in histname):
|
| 318 |
hist_data.SetXTitle("neutrino p_{Z} /GeV");
|
| 319 |
hist_data.SetYTitle("Events/(10 GeV)");
|
| 320 |
rebin = 10;
|
| 321 |
Urange = (-500, 500)
|
| 322 |
elif ('mHadronicTop' in histname or 'mLeptonicTop' in histname or 'mAllTop' in histname):
|
| 323 |
hist_data.SetXTitle("top mass /GeV");
|
| 324 |
hist_data.SetYTitle("Events/(20 GeV)");
|
| 325 |
rebin = 20;
|
| 326 |
elif ('pt_leadingTop' in histname or 'pt_NextToLeadingTop' in histname):
|
| 327 |
hist_data.SetXTitle("top p_{T} /GeV");
|
| 328 |
hist_data.SetYTitle("Events/(20 GeV)");
|
| 329 |
rebin = 20;
|
| 330 |
elif('QCDest_CombRelIso' in histname):
|
| 331 |
hist_data.SetXTitle("relative isolation");
|
| 332 |
hist_data.SetYTitle("Events/(0.1)");
|
| 333 |
rebin = 10;
|
| 334 |
Urange = (0, 2)
|
| 335 |
elif('QCDest_PFIsolation' in histname):
|
| 336 |
hist_data.SetXTitle("Relative isolation");
|
| 337 |
hist_data.SetYTitle("Events/(0.1)");
|
| 338 |
rebin = 10
|
| 339 |
Urange = (0, 2)
|
| 340 |
elif('diElectron' in histname):
|
| 341 |
hist_data.SetXTitle("m(ee)");
|
| 342 |
hist_data.SetYTitle("Events/(1 GeV)");
|
| 343 |
rebin = 5
|
| 344 |
Urange = (0, 200)
|
| 345 |
|
| 346 |
|
| 347 |
hist_data.Rebin(rebin);
|
| 348 |
hist_ttbar.Rebin(rebin);
|
| 349 |
hist_wjets.Rebin(rebin);
|
| 350 |
hist_zjets.Rebin(rebin);
|
| 351 |
hist_qcd.Rebin(rebin);
|
| 352 |
hist_singleTop.Rebin(rebin)
|
| 353 |
|
| 354 |
hist_data.SetAxisRange(Urange[0], Urange[1]);
|
| 355 |
hist_ttbar.SetAxisRange(Urange[0], Urange[1]);
|
| 356 |
hist_wjets.SetAxisRange(Urange[0], Urange[1]);
|
| 357 |
hist_zjets.SetAxisRange(Urange[0], Urange[1]);
|
| 358 |
hist_qcd.SetAxisRange(Urange[0], Urange[1]);
|
| 359 |
hist_singleTop.SetAxisRange(Urange[0], Urange[1]);
|
| 360 |
|
| 361 |
hist_data.SetMarkerStyle(8);
|
| 362 |
|
| 363 |
hist_ttbar.SetFillStyle(1001);
|
| 364 |
hist_ttbar.SetFillColor(kRed + 1);
|
| 365 |
hist_wjets.SetFillStyle(1001);
|
| 366 |
hist_wjets.SetFillColor(kGreen - 3);
|
| 367 |
hist_zjets.SetFillStyle(1001);
|
| 368 |
hist_zjets.SetFillColor(kAzure - 2);
|
| 369 |
hist_qcd.SetFillStyle(1001);
|
| 370 |
hist_qcd.SetFillColor(kYellow);
|
| 371 |
hist_singleTop.SetFillStyle(1001);
|
| 372 |
hist_singleTop.SetFillColor(kMagenta)
|
| 373 |
|
| 374 |
# hist_Zprime500.SetLineColor(kTeal - 9);
|
| 375 |
# hist_Zprime500.SetFillStyle(0);
|
| 376 |
# hist_Zprime500.SetFillColor(kWhite);
|
| 377 |
#
|
| 378 |
# hist_Zprime750.SetLineColor(kBlue - 6);
|
| 379 |
# hist_Zprime750.SetFillStyle(0);
|
| 380 |
# hist_Zprime750.SetFillColor(kWhite);
|
| 381 |
#
|
| 382 |
# hist_Zprime1000.SetLineColor(28);
|
| 383 |
# hist_Zprime1000.SetFillStyle(0);
|
| 384 |
# hist_Zprime1000.SetFillColor(kWhite);
|
| 385 |
#
|
| 386 |
# hist_Zprime1250.SetLineColor(kCyan - 5);
|
| 387 |
# hist_Zprime1250.SetFillStyle(0);
|
| 388 |
# hist_Zprime1250.SetFillColor(kWhite);
|
| 389 |
#
|
| 390 |
# hist_Zprime1500.SetLineColor(kOrange + 1);
|
| 391 |
# hist_Zprime1500.SetFillStyle(0);
|
| 392 |
# hist_Zprime1500.SetFillColor(kWhite);
|
| 393 |
|
| 394 |
# qcdUncert.SetFillColor(kGray + 3);
|
| 395 |
# qcdUncert.SetFillStyle(3003);
|
| 396 |
|
| 397 |
leg = TLegend(0.696, 0.35, 0.94, 0.92);
|
| 398 |
leg.SetBorderSize(0);
|
| 399 |
leg.SetLineStyle(0);
|
| 400 |
leg.SetTextFont(42);
|
| 401 |
leg.SetFillStyle(0);
|
| 402 |
|
| 403 |
leg.AddEntry(hist_data, "data", "P");
|
| 404 |
# leg.AddEntry(hist_data2, "data(no HLT)", "P");
|
| 405 |
leg.AddEntry(hist_ttbar, "t#bar{t}", "f");
|
| 406 |
leg.AddEntry(hist_wjets, "W#rightarrowl#nu", "f");
|
| 407 |
leg.AddEntry(hist_zjets, "Z/#gamma*#rightarrowl^{+}l^{-}", "f");
|
| 408 |
leg.AddEntry(hist_qcd, "QCD/#gamma + jets");
|
| 409 |
leg.AddEntry(hist_singleTop, "Single-Top")
|
| 410 |
# leg.AddEntry(hist_Zprime500, "Z' 0.5TeV (50pb)");
|
| 411 |
# leg.AddEntry(hist_Zprime750, "Z' 0.75TeV (50pb)");
|
| 412 |
# leg.AddEntry(hist_Zprime1000, "Z' 1TeV (50pb)");
|
| 413 |
# leg.AddEntry(hist_Zprime1250, "Z' 1.25TeV (50pb)");
|
| 414 |
# leg.AddEntry(hist_Zprime1500, "Z' 1.5TeV (50pb)");
|
| 415 |
|
| 416 |
|
| 417 |
canvases.append(TCanvas("cname" + histname, histname, 1200, 900))
|
| 418 |
canvases[-1].cd().SetRightMargin(0.04);
|
| 419 |
hs = THStack("MC", "MC");
|
| 420 |
hs.Add(hist_qcd);
|
| 421 |
hs.Add(hist_zjets);
|
| 422 |
hs.Add(hist_wjets);
|
| 423 |
hs.Add(hist_singleTop);
|
| 424 |
hs.Add(hist_ttbar);
|
| 425 |
max = 0
|
| 426 |
if hs.GetMaximum() > hist_data.GetMaximum():
|
| 427 |
max = hs.GetMaximum()*1.4
|
| 428 |
else:
|
| 429 |
max = hist_data.GetMaximum()*1.4
|
| 430 |
|
| 431 |
hist_data.GetYaxis().SetRangeUser(0, max);
|
| 432 |
hist_data.Draw('error');
|
| 433 |
hs.Draw("hist same");
|
| 434 |
# hist_Zprime500.Draw("same");
|
| 435 |
# hist_Zprime750.Draw("same");
|
| 436 |
# hist_Zprime1000.Draw("same");
|
| 437 |
# hist_Zprime1250.Draw("same");
|
| 438 |
# hist_Zprime1500.Draw("same");
|
| 439 |
# qcdUncert.Draw("1 same");
|
| 440 |
# hist_data2.Draw("error same");
|
| 441 |
hist_data.Draw("error same");
|
| 442 |
leg.Draw();
|
| 443 |
text1 = add_cms_label(lumi, '4orMoreJets')
|
| 444 |
# text1 = TLatex(3.570061, 23.08044, "CMS Preliminary");
|
| 445 |
# text1.SetNDC();
|
| 446 |
# text1.SetTextAlign(13);
|
| 447 |
# text1.SetX(0.38);
|
| 448 |
# text1.SetY(0.928);
|
| 449 |
# #text1.SetLineWidth(2);
|
| 450 |
# text1.SetTextFont(42);
|
| 451 |
# text1.SetTextSizePixels(24);# dflt=28
|
| 452 |
text1.Draw();
|
| 453 |
#
|
| 454 |
# text2 = TLatex(3.570061, 23.08044, "~%.1f pb^{-1} at #sqrt{s} = 7 TeV" % lumi);
|
| 455 |
# text2.SetNDC();
|
| 456 |
# text2.SetTextAlign(13);
|
| 457 |
# text2.SetX(0.38);
|
| 458 |
# text2.SetY(0.88);
|
| 459 |
# #text2.SetLineWidth(2);
|
| 460 |
# text2.SetTextFont(42);
|
| 461 |
# text2.SetTextSizePixels(24);# dflt=28
|
| 462 |
# text2.Draw();
|
| 463 |
canvases[-1].SaveAs('/storage/results/' + histname + '.png')
|
| 464 |
|
| 465 |
cu_hist_data = getCumulativePlot(hist_data, "data");
|
| 466 |
cu_hist_ttbar = getCumulativePlot(hist_ttbar, "ttbar");
|
| 467 |
cu_hist_wjets = getCumulativePlot(hist_wjets, "wjets");
|
| 468 |
cu_hist_zjets = getCumulativePlot(hist_zjets, "zjets");
|
| 469 |
cu_hist_qcd = getCumulativePlot(hist_qcd, "qcd");
|
| 470 |
cu_hist_singleTop = getCumulativePlot(hist_singleTop, "singleTop");
|
| 471 |
## cu_hist_Zprime500 = getCumulativePlot(hist_Zprime500, "Zprime500");
|
| 472 |
## cu_hist_Zprime750 = getCumulativePlot(hist_Zprime750, "Zprime750");
|
| 473 |
## cu_hist_Zprime1000 = getCumulativePlot(hist_Zprime1000, "Zprime1000");
|
| 474 |
## cu_hist_Zprime1250 = getCumulativePlot(hist_Zprime1250, "Zprime1250");
|
| 475 |
## cu_hist_Zprime1500 = getCumulativePlot(hist_Zprime1500, "Zprime1500");
|
| 476 |
cu_hist_data.SetYTitle("Integrated Events/(50 GeV)");
|
| 477 |
##
|
| 478 |
#
|
| 479 |
cu_hist_data.SetAxisRange(Urange[0], Urange[1]);
|
| 480 |
cu_hist_ttbar.SetAxisRange(Urange[0], Urange[1]);
|
| 481 |
cu_hist_wjets.SetAxisRange(Urange[0], Urange[1]);
|
| 482 |
cu_hist_zjets.SetAxisRange(Urange[0], Urange[1]);
|
| 483 |
cu_hist_qcd.SetAxisRange(Urange[0], Urange[1]);
|
| 484 |
cu_hist_singleTop.SetAxisRange(Urange[0], Urange[1]);
|
| 485 |
|
| 486 |
cu_hs = THStack("cu_MC", "cu_MC");
|
| 487 |
cu_hs.Add(cu_hist_qcd);
|
| 488 |
cu_hs.Add(cu_hist_zjets);
|
| 489 |
cu_hs.Add(cu_hist_wjets);
|
| 490 |
cu_hs.Add(cu_hist_ttbar);
|
| 491 |
|
| 492 |
scanvases.append(TCanvas("cu_cname" + histname, histname + "(cu)", 1200, 900))
|
| 493 |
scanvases[-1].cd().SetRightMargin(0.04);
|
| 494 |
cu_hist_data.Draw("error");
|
| 495 |
cu_hs.Draw("hist same");
|
| 496 |
### cu_hist_Zprime500.Draw("same");
|
| 497 |
### cu_hist_Zprime750.Draw("same");
|
| 498 |
### cu_hist_Zprime1000.Draw("same");
|
| 499 |
### cu_hist_Zprime1250.Draw("same");
|
| 500 |
### cu_hist_Zprime1500.Draw("same");
|
| 501 |
## # cu_hist_data2.Draw("error same");
|
| 502 |
cu_hist_data.Draw("error same");
|
| 503 |
leg.Draw();
|
| 504 |
##
|
| 505 |
text1.Draw();
|
| 506 |
##
|
| 507 |
# text2.Draw();
|
| 508 |
scanvases[-1].SaveAs('/storage/results/' + histname + '_integrated.png')
|
| 509 |
|
| 510 |
|
| 511 |
|
| 512 |
# tdrGrid: Turns the grid lines on (true) or off (false)
|
| 513 |
|
| 514 |
#def tdrGrid(bool gridOn):
|
| 515 |
# tdrStyle.SetPadGridX(gridOn);
|
| 516 |
# tdrStyle.SetPadGridY(gridOn);
|
| 517 |
#
|
| 518 |
|
| 519 |
# fixOverlay: Redraws the axis
|
| 520 |
|
| 521 |
def fixOverlay():
|
| 522 |
gPad.RedrawAxis();
|
| 523 |
|
| 524 |
|
| 525 |
def getCumulativePlot(initial, type):
|
| 526 |
name = initial.GetName()
|
| 527 |
name = "cu_" + name + "_" + type;
|
| 528 |
title = initial.GetTitle()
|
| 529 |
title = "cu_" + title + "_" + type;
|
| 530 |
xaxis = initial.GetXaxis().GetTitle();
|
| 531 |
yaxis = initial.GetYaxis().GetTitle();
|
| 532 |
nBins = initial.GetNbinsX();
|
| 533 |
cu = TH1F(name, title, nBins, initial.GetXaxis().GetXmin(), initial.GetXaxis().GetXmax());
|
| 534 |
for bin in range(1,nBins+1):
|
| 535 |
cu.SetBinContent(bin, initial.Integral(bin, nBins));
|
| 536 |
|
| 537 |
cu.SetFillStyle(initial.GetFillStyle());
|
| 538 |
cu.SetFillColor(initial.GetFillColor());
|
| 539 |
cu.SetLineColor(initial.GetLineColor());
|
| 540 |
cu.SetMarkerSize(initial.GetMarkerSize());
|
| 541 |
cu.SetMarkerStyle(initial.GetMarkerStyle());
|
| 542 |
cu.SetMarkerColor(initial.GetMarkerColor());
|
| 543 |
cu.SetLineWidth(initial.GetLineWidth());
|
| 544 |
cu.GetXaxis().SetTitle(xaxis);
|
| 545 |
cu.GetYaxis().SetTitle(yaxis);
|
| 546 |
return cu;
|
| 547 |
|
| 548 |
def add_cms_label( lumi, njet = "" ):
|
| 549 |
jetBinsLatex = {'0jet':'0 jet', 'allJets':'#geq 0 jets', '1jet':'1 jet', '1orMoreJets':'#geq 1 jet',
|
| 550 |
'2jets':'2 jets', '2orMoreJets':'#geq 2 jets', '3jets':'3 jets', '3orMoreJets':'#geq 3 jets',
|
| 551 |
'4orMoreJets':'#geq 4 jets'}
|
| 552 |
|
| 553 |
mytext = TPaveText( 0.3, 0.8, 0.6, 0.93, "NDC" );
|
| 554 |
mytext.AddText( "CMS Preliminary" );
|
| 555 |
mytext.AddText( "%.1f pb^{-1} at #sqrt{s} = 7 TeV" % lumi );
|
| 556 |
if njet != "":
|
| 557 |
mytext.AddText( "e+jets, %s" % jetBinsLatex[njet] )
|
| 558 |
|
| 559 |
mytext.SetFillStyle( 0 );
|
| 560 |
mytext.SetBorderSize( 0 );
|
| 561 |
mytext.SetTextFont( 42 );
|
| 562 |
mytext.SetTextAlign( 13 );
|
| 563 |
return mytext
|
| 564 |
|
| 565 |
if __name__ == "__main__":
|
| 566 |
gROOT.SetBatch(True)
|
| 567 |
gROOT.ProcessLine('gErrorIgnoreLevel = 1001;')
|
| 568 |
plotMttbar()
|
| 569 |
# print "press enter to quit"
|
| 570 |
# a = raw_input()
|
