dataPlotters=[singleMuA,singleMuB,singleEleA,singleEleB,jetA,jetB]
dataEMUPlotters=[singleMuA,singleMuB,singleEleA,singleEleB]
dataJetPlotters=[jetA,jetB]
dataMETPlotters=[metA,metB]
WWLNUJJ = TreePlotter('samples/BulkGravToWWToWlepWhad_narrow_4000.root','tree')
WWLNUJJ.setupFromFile('samples/BulkGravToWWToWlepWhad_narrow_4000.pck')
WWLNUJJ.addCorrectionFactor('xsec','tree')
WWLNUJJ.addCorrectionFactor('genWeight','tree')
WWLNUJJ.addCorrectionFactor('puWeight','tree')
WWLNUJJ.setFillProperties(0,ROOT.kWhite)
WWLNUJJ.setLineProperties(1,ROOT.kRed,3)
WZLNUJJ = TreePlotter('samples/WprimeToWZ_narrow_4000.root','tree')
WZLNUJJ.setupFromFile('samples/WprimeToWZ_narrow_4000.pck')
WZLNUJJ.addCorrectionFactor('3*xsec','tree')
WZLNUJJ.addCorrectionFactor('genWeight','tree')
WZLNUJJ.addCorrectionFactor('puWeight','tree')
WZLNUJJ.setFillProperties(0,ROOT.kWhite)
WZLNUJJ.setLineProperties(1,ROOT.kGreen+3,3)
WHLNUJJ = TreePlotter('samples/WprimeToWhToWlephbb_narrow_4000.root','tree')
WHLNUJJ.setupFromFile('samples/WprimeToWhToWlephbb_narrow_4000.pck')
WHLNUJJ.addCorrectionFactor('xsec','tree')
WHLNUJJ.addCorrectionFactor('genWeight','tree')
WHLNUJJ.addCorrectionFactor('puWeight','tree')
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")
vvPlotters.append(TreePlotter('samples/' + sample + '.root', 'tree'))
vvPlotters[-1].setupFromFile('samples/' + sample + '.pck')
vvPlotters[-1].addCorrectionFactor('xsec', 'tree')
vvPlotters[-1].addCorrectionFactor('genWeight', 'tree')
vvPlotters[-1].addCorrectionFactor('puWeight', 'tree')
vv = MergedPlotter(vvPlotters)
WWLNUJJ = TreePlotter('samples/VBF_RadionToWW_narrow_3000.root', 'tree')
WWLNUJJ.setupFromFile('samples/VBF_RadionToWW_narrow_3000.pck')
WWLNUJJ.addCorrectionFactor('xsec', 'tree')
WWLNUJJ.addCorrectionFactor(100, 'flat')
WWLNUJJ.addCorrectionFactor('genWeight', 'tree')
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")
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")
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")