outdir = '/afs/cern.ch/user/n/nawoods/www/ZZPlots/firstICHEPPlots'
# tpVersionHash = 'v2.0-13-g36fc26c' #'v2.0-11-gafcf7cc' #'v1.1-4-ga295b14-extended'
#
# TP = TPFunctionManager(tpVersionHash)
z1eMCWeight = 'e1EffScaleFactor * e2EffScaleFactor * e1EffScaleFactor * e2EffScaleFactor'
z2eMCWeight = 'e3EffScaleFactor * e4EffScaleFactor'
#z1mMCWeight = 'm1EffScaleFactor * m1TrkRecoEffScaleFactor * m2EffScaleFactor * m2TrkRecoEffScaleFactor'
#z2mMCWeight = 'm3EffScaleFactor * m3TrkRecoEffScaleFactor * m4EffScaleFactor * m4TrkRecoEffScaleFactor'
z1mMCWeight = 'm1EffScaleFactor * m2EffScaleFactor'
z2mMCWeight = 'm3EffScaleFactor * m4EffScaleFactor'
puWt = WeightStringMaker('puWeight')
fPUScale = root_open(os.path.join(os.environ['zza'], 'ZZAnalyzer',
'data', 'pileupReweighting',
'pileup_MC_80x_271036-276811_69200.root'))
puSFHist = fPUScale.puweight
puSFStr = puWt.makeWeightStringFromHist(puSFHist, 'nTruePU')
mcWeight = {
'eeee' : '(genWeight*{0}*{1}*{2})'.format(puSFStr, z1eMCWeight, z2eMCWeight),
'eemm' : '(genWeight*{0}*{1}*{2})'.format(puSFStr, z1eMCWeight, z1mMCWeight),
'mmmm' : '(genWeight*{0}*{1}*{2})'.format(puSFStr, z1mMCWeight, z2mMCWeight),
}
mcWeight['zz'] = [mcWeight['eeee'], mcWeight['eemm'], mcWeight['mmmm']]
units = {
from rootpy.io import root_open
from rootpy.plotting import Hist
import os
# lots of prep things
tpVersionHash = 'v2.0-13-g36fc26c' #'v2.0-11-gafcf7cc' #'v1.1-4-ga295b14-extended'
TP = TPFunctionManager(tpVersionHash)
z1eMCWeight = '*'.join(TP.getTPString('e%d'%ne, 'TightID')+'*'+TP.getTPString('e%d'%ne, 'IsoTight') for ne in range(1,3))
z2eMCWeight = '*'.join(TP.getTPString('e%d'%ne, 'TightID')+'*'+TP.getTPString('e%d'%ne, 'IsoTight') for ne in range(3,5))
z1mMCWeight = '*'.join(TP.getTPString('m%d'%nm, 'TightID')+'*'+TP.getTPString('m%d'%nm, 'IsoTight') for nm in range(1,3))
z2mMCWeight = '*'.join(TP.getTPString('m%d'%nm, 'TightID')+'*'+TP.getTPString('m%d'%nm, 'IsoTight') for nm in range(3,5))
wts = WeightStringMaker('puWeight')
fPUScale = root_open(os.environ['zza']+'/ZZAnalyzer/data/pileupReweighting/PUScaleFactors_29Feb2016.root')
puScaleFactorHist = fPUScale.Get("puScaleFactor")
puScaleFactorStr = wts.makeWeightStringFromHist(puScaleFactorHist, 'nTruePU')
mcWeight = {
'eeee' : '(GenWeight*{0}*{1}*{2})'.format(puScaleFactorStr, z1eMCWeight, z2eMCWeight),
'eemm' : '(GenWeight*{0}*{1}*{2})'.format(puScaleFactorStr, z1eMCWeight, z1mMCWeight),
'mmmm' : '(GenWeight*{0}*{1}*{2})'.format(puScaleFactorStr, z1mMCWeight, z2mMCWeight),
}
mcWeight['zz'] = [mcWeight['eeee'], mcWeight['eemm'], mcWeight['mmmm']]
bkg = BkgManager('01mar2016')
exit(0)
tpVersionHash = 'v2.0-13-g36fc26c' #'v1.1-4-ga295b14-extended' #v1.1-1-g4cbf52a_v2'
TP = TPFunctionManager(tpVersionHash)
scales=['', 'down', 'up']
fPUScale = root_open(os.path.join(os.environ['zza'],'ZZAnalyzer','data/pileupReweighting/PUScaleFactors_29Feb2016.root'))
puScaleFactorHist = {
'' : fPUScale.Get("puScaleFactor"),
'up' : fPUScale.Get("puScaleFactor_ScaleUp"),
'down' : fPUScale.Get("puScaleFactor_ScaleDown"),
}
wts = WeightStringMaker('puWeight')
puScaleFactorStr = {s:wts.makeWeightStringFromHist(h, 'nTruePU') for s,h in puScaleFactorHist.iteritems()}
bkg = BkgManager('01mar2016')
cr3PScale = {
ch : bkg.fullString3P1F(ch) for ch in ['eeee','eemm','mmmm']
}
cr3PScale['zz'] = [cr3PScale[c] for c in ['eeee','eemm','mmmm']]
cr2PScale = {
ch : bkg.fullString2P2F(ch) for ch in ['eeee','eemm','mmmm']
}
