QCDPlotters = getPlotters('QCD_HT',False)
QCD = MergedPlotter(QCDPlotters)
DATAPlotters = getPlotters('SingleMuon_,SingleElectron_,MET_',True)
data=MergedPlotter(DATAPlotters)
SigPlotters = getPlotters('BulkGravToWWToWlepWhad_narrow_1400',False)
sig = MergedPlotter(SigPlotters)
#Fill properties
VJets.setFillProperties(1001,ROOT.kAzure-9)
BKG.setFillProperties(1001,ROOT.kAzure-9)
ttVV.setFillProperties(1001,ROOT.kSpring-5)
W.setFillProperties(1001,ROOT.kSpring-5)
QCD.setFillProperties(1001,ROOT.kGray)
#Stack for lnu+J
lnujjStack = StackPlotter()
lnujjStack.addPlotter(QCD,"QCD","QCD multijet","background")
#lnujjStack.addPlotter(VJets,"WJets","V+Jets","background")
lnujjStack.addPlotter(BKG,"Bkg","V+Jets","background")
#lnujjStack.addPlotter(ttVV,"top","top/VV","background")
lnujjStack.addPlotter(W,"W","W","background")
lnujjStack.addPlotter(data,"data_obs","Data","data")
#dataPlotters=[singleMu,singleEle,jetHT]
dataPlotters=[jetHT]
data = MergedPlotter(dataPlotters)
#WWZ = TreePlotter('samples/WprimeToWZToWhadZhad_narrow_2000.pck','tree')
#WWZ.setupFromFile('samples/WprimeToWZToWhadZhad_narrow_2000.pck')
#WWZ.setFillProperties(0,ROOT.kWhite)
#WWZ.setLineProperties(1,ROOT.kOrange+12,3)
#RSGWWLNuQQ..addCorrectionFactor('xsec',0.001,0.0,'lnN')
#Stack
lnujStack = StackPlotter()
lnujStack.addPlotter(WJets,"W+jets","W+Jets","background")
lnujStack.addPlotter(TTJets,"TTJets","t#bar{t}+jets","background")
#vvStack.addPlotter(RSGWWLNuQQ,"RSG2000","RSGWW #rightarrow l#nu QQ","signal")
lnujStack.addPlotter(data,"data_obs","Data","data")
jjStack = StackPlotter()
#lnujStack.addPlotter(WJets,"W+jets","W+Jets","background")
jjStack.addPlotter(QCD,"QCD","QCD","background")
#vvStack.addPlotter(RSGWWLNuQQ,"RSG2000","RSGWW #rightarrow l#nu QQ","signal")
jjStack.addPlotter(data,"data_obs","Data","data")
ST.setFillProperties(1001, ROOT.kSpring + 2)
VV.setFillProperties(1001, ROOT.kOrange - 2)
#VH.setFillProperties(1001,ROOT.kRed-9)
#TopVVVH.setFillProperties(1001,ROOT.kSpring+5)
nonRes.setFillProperties(
1001, ROOT.TColor.GetColor("#A5D2FF")) #ROOT.kBlue+1)#ROOT.kAzure-9)
res.setFillProperties(
1001, ROOT.TColor.GetColor("#60B037")) #ROOT.kGreen+1)#ROOT.kSpring-5)
#resW.setFillProperties(1001,ROOT.TColor.GetColor("#37B04D"))#ROOT.kGreen+1)#ROOT.kSpring-5)
#resTop.setFillProperties(1001,ROOT.TColor.GetColor("#C3D631"))#ROOT.kGreen+1)#ROOT.kSpring-5)
#Sig.setFillProperties(1001,ROOT.kRed+1)
## Stack for lnu+J
lnujjStack = StackPlotter()
if LIKETEMPLATES:
lnujjStack.addPlotter(nonRes, "nonRes", "W+jets", "background")
lnujjStack.addPlotter(res, "res", "W+V/t", "background")
#lnujjStack.addPlotter(resW,"resW","W peak","background")
#lnujjStack.addPlotter(resTop,"resTop","top peak","background")
else:
lnujjStack.addPlotter(QCD, "QCD", "QCD multijet", "background")
#lnujjStack.addPlotter(VH,"VH","VH","background")
lnujjStack.addPlotter(VV, "VV", "VV", "background")
lnujjStack.addPlotter(VJets, "WJets", "V+Jets", "background")
#lnujjStack.addPlotter(Top,"top","top","background")
lnujjStack.addPlotter(ST, "ST", "single top", "background")
lnujjStack.addPlotter(TT, "TT", "t#bar{t}", "background")
#lnujjStack.addPlotter(TopVVVH,"topVVVH","top, VV, VH","background")
#change the CMS_lumi variables (see CMS_lumi.py)
lumi_13TeV = "35.9 fb^{-1}"
lumi_sqrtS = "13 TeV" # used with iPeriod = 0, e.g. for simulation-only plots (default is an empty string)
iPeriod = 4
iPos = 11
#zjPlotters=[]
BKGPlotters = getPlotters(
'DYJetsToLL_M50_HT,WJetsToLNu_HT,TT_pow,T_tWch,TBar_tWch', False)
BKG = MergedPlotter(BKGPlotters)
VVPlotters = getPlotters('WZTo2L2Q,WZTo1L1Nu2Q', False)
VV = MergedPlotter(VVPlotters)
DATAPlotters = getPlotters('SingleMuon_,SingleElectron_', True)
data = MergedPlotter(DATAPlotters)
SigPlotters = getPlotters('WprimeToWZToWhadZlep_narrow_1400', False)
sig = MergedPlotter(SigPlotters)
#Fill properties
BKG.setFillProperties(1001, ROOT.kAzure - 9)
VV.setFillProperties(1001, ROOT.kSpring - 5)
#Stack for lnu+J
lljjStack = StackPlotter()
lljjStack.addPlotter(BKG, "Bkg", "V+Jets", "background")
lljjStack.addPlotter(VV, "VV", "VV", "background")
lljjStack.addPlotter(sig, "sig", "X #rightarrow WZ", "signal")
lljjStack.addPlotter(data, "data_obs", "Data", "data")
#Fill properties
WJets.setFillProperties(1001,ROOT.kAzure-9)
tt.setFillProperties(1001,ROOT.kGreen-5)
ZJets.setFillProperties(1001,ROOT.kAzure+5)
GJets.setFillProperties(1001,ROOT.kYellow)
QCD.setFillProperties(1001,ROOT.kGray)
dataEMU = MergedPlotter(dataEMUPlotters)
dataJet = MergedPlotter(dataJetPlotters)
#Stack for lnu+J
lnujjStack = StackPlotter()
lnujjStack.addPlotter(QCD,"QCD","QCD multijet","background")
#lnujjStack.addPlotter(GJets,"GJets","#gamma+Jets","background")
lnujjStack.addPlotter(WJets,"WJets","W+Jets","background")
lnujjStack.addPlotter(tt,"tt","t#bar{t}","background")
lnujjStack.addPlotter(WWLNUJJ,"X1","X #rightarrow WW","signal")
lnujjStack.addPlotter(WZLNUJJ,"X2","X #rightarrow WZ","signal")
lnujjStack.addPlotter(WHLNUJJ,"X3","X #rightarrow WH","signal")
lnujjStack.addPlotter(dataEMU,"data_obs","Data","data")
#Stack for ll+J
lljjStack = StackPlotter()
lljjStack.addPlotter(tt,"tt","t#bar{t}","background")
lljjStack.addPlotter(ZJets,"ZJets","Z+Jets","background")
lljjStack.addPlotter(dataEMU,"data_obs","Data","data")
'((HLT_ISOELE_50ns==1||(HLT_ELE==1))&&(!(HLT_ISOMU_50ns==1||(HLT_MU==1))))',
0.0, 'lnN')
jetHT = TreePlotter('samples/JetHT_Run2015B.root', 'tree')
#jetHT.addCorrectionFactor('hlt','((HLT_JJ==1||HLT_HT==1)&&(!(HLT_ISOMU_50ns==1||(HLT_MU==1)||(HLT_ISOELE_50ns==1||(HLT_ELE==1)))))',0.0,'lnN')
#dataPlotters=[singleMu,singleEle,jetHT]
dataPlotters = [jetHT]
data = MergedPlotter(dataPlotters)
#WWZ = TreePlotter('samples/WprimeToWZToWhadZhad_narrow_2000.pck','tree')
#WWZ.setupFromFile('samples/WprimeToWZToWhadZhad_narrow_2000.pck')
#WWZ.setFillProperties(0,ROOT.kWhite)
#WWZ.setLineProperties(1,ROOT.kOrange+12,3)
#RSGWWLNuQQ..addCorrectionFactor('xsec',0.001,0.0,'lnN')
#Stack
lnujStack = StackPlotter()
lnujStack.addPlotter(WJets, "W+jets", "W+Jets", "background")
lnujStack.addPlotter(TTJets, "TTJets", "t#bar{t}+jets", "background")
#vvStack.addPlotter(RSGWWLNuQQ,"RSG2000","RSGWW #rightarrow l#nu QQ","signal")
lnujStack.addPlotter(data, "data_obs", "Data", "data")
jjStack = StackPlotter()
#lnujStack.addPlotter(WJets,"W+jets","W+Jets","background")
jjStack.addPlotter(QCD, "QCD", "QCD", "background")
#vvStack.addPlotter(RSGWWLNuQQ,"RSG2000","RSGWW #rightarrow l#nu QQ","signal")
jjStack.addPlotter(data, "data_obs", "Data", "data")
STnonRes.setFillProperties(1001, ROOT.TColor.GetColor("#98ADDB"))
VV.setFillProperties(1001, ROOT.kOrange - 2)
#VH.setFillProperties(1001,ROOT.kRed-9)
#TopVVVH.setFillProperties(1001,ROOT.kSpring+5)
nonRes.setFillProperties(
1001, ROOT.TColor.GetColor("#A5D2FF")) #ROOT.kBlue+1)#ROOT.kAzure-9)
res.setFillProperties(
1001, ROOT.TColor.GetColor("#60B037")) #ROOT.kGreen+1)#ROOT.kSpring-5)
#resW.setFillProperties(1001,ROOT.TColor.GetColor("#37B04D"))#ROOT.kGreen+1)#ROOT.kSpring-5)
#resTop.setFillProperties(1001,ROOT.TColor.GetColor("#C3D631"))#ROOT.kGreen+1)#ROOT.kSpring-5)
#Sig.setFillProperties(1001,ROOT.kRed+1)
## Stack for lnu+J
lnujjStack = StackPlotter()
if BKGDSPLITMETHOD == "b2":
lnujjStack.addPlotter(nonRes, "nonRes", "W+jets", "background")
lnujjStack.addPlotter(res, "res", "W+V/t", "background")
#lnujjStack.addPlotter(resW,"resW","W peak","background")
#lnujjStack.addPlotter(resTop,"resTop","top peak","background")
elif BKGDSPLITMETHOD == "b1":
lnujjStack.addPlotter(QCD, "QCD", "QCD multijet", "background")
#lnujjStack.addPlotter(VH,"VH","VH","background")
lnujjStack.addPlotter(VV, "VV", "VV", "background")
lnujjStack.addPlotter(VJets, "WJets", "V+jets", "background")
#lnujjStack.addPlotter(Top,"top","top","background")
lnujjStack.addPlotter(ST, "ST", "single top", "background")
lnujjStack.addPlotter(TT, "TT", "t#bar{t}", "background")
#lnujjStack.addPlotter(TopVVVH,"topVVVH","top, VV, VH","background")
import ROOT
from CMGTools.VVResonances.plotting.TreePlotter import TreePlotter
from CMGTools.VVResonances.plotting.StackPlotter import StackPlotter
ROOT.gROOT.ProcessLine('.x tdrstyle.C')
w = TreePlotter('samples/W.root','tree')
w.addCorrectionFactor('xsec','xsec',0.0,'lnN')
w.addCorrectionFactor('mc','l_mcPt>0',0.0,'lnN')
w.setFillProperties(0,ROOT.kWhite)
w.setLineProperties(1,ROOT.kOrange+10,3)
qcd = TreePlotter('samples/W.root','tree')
qcd.addCorrectionFactor('xsec','xsec',0.0,'lnN')
qcd.addCorrectionFactor('mc','l_mcPt==0',0.0,'lnN')
qcd.setFillProperties(3003,ROOT.kBlue)
qcd.setLineProperties(1,ROOT.kBlue,3)
#Stack
lStack = StackPlotter()
lStack.addPlotter(qcd,"QCD","Non prompt","background")
lStack.addPlotter(w,"Prompt_leptons","Prompt","signal")
from CMGTools.VVResonances.plotting.TreePlotter import TreePlotter
from CMGTools.VVResonances.plotting.MergedPlotter import MergedPlotter
from CMGTools.VVResonances.plotting.StackPlotter import StackPlotter
#create the W+jets plotters
wjPlotters = []
for sample in [
'WJetsToLNu_HT100to200', 'WJetsToLNu_HT200to400',
'WJetsToLNu_HT400to600', 'WJetsToLNu_HT600toInf'
]:
wjPlotters.append(TreePlotter('samples/' + sample + '.root', 'tree'))
wjPlotters[-1].setupFromFile('samples/' + sample + '.pck')
wjPlotters[-1].addCorrectionFactor('xsec', 'xsec', 0.0, 'lnN')
WJets = MergedPlotter(wjPlotters)
WJets.setFillProperties(1001, ROOT.kAzure - 9)
RSGWWLNuQQ = TreePlotter('samples/RSGravToWWToLNQQ_2000.root', 'tree')
RSGWWLNuQQ.setupFromFile('samples/RSGravToWWToLNQQ_2000.pck')
RSGWWLNuQQ.setFillProperties(0, ROOT.kWhite)
RSGWWLNuQQ.setLineProperties(1, ROOT.kOrange + 10, 3)
#RSGWWLNuQQ..addCorrectionFactor('xsec',0.001,0.0,'lnN')
#Stack
vvStack = StackPlotter()
vvStack.addPlotter(WJets, "W+jets", "W+Jets", "background")
vvStack.addPlotter(RSGWWLNuQQ, "RSG2000", "RSGWW #rightarrow l#nu QQ",
"signal")
WWLNUJJ.addCorrectionFactor('puWeight', 'tree')
WWLNUJJ.setFillProperties(0, ROOT.kWhite)
WWLNUJJ.setLineProperties(1, ROOT.kRed, 3)
#Fill properties
WJets.setFillProperties(1001, ROOT.kAzure - 9)
vv.setFillProperties(1001, ROOT.kOrange)
ttO.setFillProperties(1001, ROOT.kSpring - 5)
ttM.setFillProperties(1001, ROOT.kTeal - 1)
#ZJets.setFillProperties(1001,ROOT.kAzure+5)
#GJets.setFillProperties(1001,ROOT.kYellow)
QCD.setFillProperties(1001, ROOT.kGray)
data = MergedPlotter(dataPlotters)
#Stack for lnu+J
lnujjStack = StackPlotter()
#lnujjStack.addPlotter(QCD,"QCD","QCD multijet","background")
lnujjStack.addPlotter(ttO, "tt", "t#bar{t} (other)", "background")
lnujjStack.addPlotter(WJets, "WJets", "W+Jets", "background")
lnujjStack.addPlotter(ttM, "tt", "t#bar{t} (W)", "background")
lnujjStack.addPlotter(vv, "VV", "SM WV", "background")
#lnujjStack.addPlotter(WWLNUJJ,"X1","VBF X #rightarrow WW","signal")
#lnujjStack.addPlotter(WZLNUJJ,"X2","X #rightarrow WZ","signal")
#lnujjStack.addPlotter(WHLNUJJ,"X3","X #rightarrow WH","signal")
lnujjStack.addPlotter(data, "data_obs", "Data", "data")
jjStack = StackPlotter()
jjStack.addPlotter(QCD, "QCD", "QCD multijet", "background")
jjStack.addPlotter(data, "data_obs", "Data", "data")
from CMGTools.VVResonances.plotting.StackPlotter import StackPlotter
#create the W+jets plotters
wjPlotters=[]
for sample in ['WJetsToLNu_HT100to200','WJetsToLNu_HT200to400','WJetsToLNu_HT400to600','WJetsToLNu_HT600toInf']:
wjPlotters.append(TreePlotter('samples/'+sample+'.root','tree'))
wjPlotters[-1].setupFromFile('samples/'+sample+'.pck')
wjPlotters[-1].addCorrectionFactor('xsec','xsec',0.0,'lnN')
wjPlotters[-1].addCorrectionFactor('genWeight','genWeight',0.0,'lnN')
WJets = MergedPlotter(wjPlotters)
WJets.setFillProperties(1001,ROOT.kAzure-9)
RSGWWLNuQQ = TreePlotter('samples/RSGravToWWToLNQQ_kMpl01_2500.root','tree')
RSGWWLNuQQ.setupFromFile('samples/RSGravToWWToLNQQ_kMpl01_2500.pck')
RSGWWLNuQQ.setFillProperties(0,ROOT.kWhite)
RSGWWLNuQQ.setLineProperties(1,ROOT.kOrange+10,3)
RSGWWLNuQQ.addCorrectionFactor('xsec',0.001,0.0,'lnN')
#Stack
vvStack = StackPlotter()
vvStack.addPlotter(WJets,"W+jets","W+Jets","background")
vvStack.addPlotter(RSGWWLNuQQ,"RSG2000","RSGWW #rightarrow l#nu QQ","signal")
dataEMUPlotters = [singleMuA, singleMuB, singleEleA, singleEleB]
dataJetPlotters = [jetA, jetB]
dataMETPlotters = [metA, metB]
#Fill properties
WJets.setFillProperties(1001, ROOT.kAzure - 9)
tt.setFillProperties(1001, ROOT.kGreen - 5)
ZJets.setFillProperties(1001, ROOT.kAzure + 5)
GJets.setFillProperties(1001, ROOT.kYellow)
QCD.setFillProperties(1001, ROOT.kGray)
dataEMU = MergedPlotter(dataEMUPlotters)
dataJet = MergedPlotter(dataJetPlotters)
#Stack for lnu+J
lnujjStack = StackPlotter()
lnujjStack.addPlotter(QCD, "QCD", "QCD multijet", "background")
lnujjStack.addPlotter(GJets, "GJets", "#gamma+Jets", "background")
lnujjStack.addPlotter(tt, "tt", "t#bar{t}", "background")
lnujjStack.addPlotter(WJets, "WJets", "W+Jets", "background")
lnujjStack.addPlotter(dataEMU, "data_obs", "Data", "data")
#Stack for ll+J
lljjStack = StackPlotter()
lljjStack.addPlotter(tt, "tt", "t#bar{t}", "background")
lljjStack.addPlotter(ZJets, "ZJets", "Z+Jets", "background")
lljjStack.addPlotter(dataEMU, "data_obs", "Data", "data")
jjStack = StackPlotter()
jjStack.addPlotter(QCD, "QCD", "QCD multijet", "background")
