def getLooseLeptonString(nMu, nE):
return looseMuIDString(
ptCut=10) + "==" + str(nMu) + "&&" + looseEleIDString(
ptCut=10, absEtaCut=2.5) + "==" + str(nE)
def getLooseLeptonString(nMu, nE): return looseMuIDString(ptCut=10) + "==" + str(nMu) + "&&" + looseEleIDString(ptCut=10, absEtaCut=2.5) + "==" + str(nE)
def _estimate(self, region, channel, setup):
printHeader("DD TTZ prediction for '%s' channel %s" %(self.name, channel))
#Sum of all channels for 'all'
if channel=='all':
return sum( [ self.cachedEstimate(region, c, channel, setup) for c in ['MuMu', 'EE', 'EMu'] ] )
else:
#Data driven for EE, EMu and MuMu.
preSelection = setup.preselection('MC', channel=channel)
#check lumi consistency
assert abs(1.-setup.lumi[channel]/setup.sample['Data'][channel]['lumi'])<0.01, "Lumi specified in setup %f does not match lumi in data sample %f in channel %s"%(setup.lumi[channel], setup.sample['Data'][channel]['lumi'], channel)
selection_MC_2l = "&&".join([region.cutString(setup.sys['selectionModifier']), preSelection['cut']])
weight = preSelection['weightStr']
yield_MC_2l = setup.lumi[channel]/1000.*u_float(getYieldFromChain(setup.sample['TTZ'][channel]['chain'], cutString = selection_MC_2l, weight=weight, returnError = True) )
if setup.verbose: print "yield_MC_2l: %s"%yield_MC_2l
muonSelection_loosePt = looseMuIDString(ptCut=10)
electronSelection_loosePt = looseEleIDString(ptCut=10)
#mu_mu_mu
MuMuMuSelection = "nGoodMuons>=2" + '&&' + muonSelection_loosePt + "==3"
if setup.parameters['useTriggers']: MuMuMuSelection += '&&HLT_3mu'
#e_e_e
EEESelection = "nGoodElectrons>=2" + '&&' + electronSelection_loosePt + "==3"
if setup.parameters['useTriggers']: EEESelection += '&&HLT_3e'
#e_e_mu
EEMuSelection = "(nGoodMuons+nGoodElectrons)>=2" + "&&" + electronSelection_loosePt + "==2&&" + muonSelection_loosePt + "==1"
if setup.parameters['useTriggers']: EEMuSelection += '&&HLT_2e1mu'
#mu_mu_e
MuMuESelection = "(nGoodMuons+nGoodElectrons)>=2" + "&&" + electronSelection_loosePt + "==1&&" + muonSelection_loosePt + "==2"
if setup.parameters['useTriggers']: MuMuESelection += '&&HLT_2mu1e'
MC_hadronSelection = setup.selection('MC', hadronicSelection = True,
**setup.defaultParameters(update={'nJets': self.nJets, 'nBTags':self.nMediumBTags, 'metMin': 0., 'metSigMin':0., 'dPhiJetMet':0. })
)['cut']
data_hadronSelection = setup.selection('Data', hadronicSelection = True,
**setup.defaultParameters(update={'nJets': self.nJets, 'nBTags':self.nMediumBTags, 'metMin': 0., 'metSigMin':0., 'dPhiJetMet':0. })
)['cut']
#loose bjet selection added here
if self.nLooseBTags[0]>=0:
MC_hadronSelection += '&&Sum$(Jet_pt>30&&abs(Jet_eta)<2.4&&Jet_id&&Jet_btagCSV>0.605)>='+str(self.nLooseBTags[0])
data_hadronSelection += '&&Sum$(Jet_pt>30&&abs(Jet_eta)<2.4&&Jet_id&&Jet_btagCSV>0.605)>='+str(self.nLooseBTags[0])
if self.nLooseBTags[1]>=0:
MC_hadronSelection += '&&Sum$(Jet_pt>30&&abs(Jet_eta)<2.4&&Jet_id&&Jet_btagCSV>0.605)<='+str(self.nLooseBTags[1])
data_hadronSelection += '&&Sum$(Jet_pt>30&&abs(Jet_eta)<2.4&&Jet_id&&Jet_btagCSV>0.605)<='+str(self.nLooseBTags[1])
MC_MuMuMu = "&&".join([
MC_hadronSelection,
MuMuMuSelection,
"abs(mlmZ_mass-91.2)<10"
])
MC_EEE = "&&".join([
MC_hadronSelection,
EEESelection,
"abs(mlmZ_mass-91.2)<10"
])
MC_EEMu = "&&".join([
MC_hadronSelection,
EEMuSelection,
"abs(mlmZ_mass-91.2)<10"
])
MC_MuMuE = "&&".join([
MC_hadronSelection,
MuMuESelection,
"abs(mlmZ_mass-91.2)<10"
])
MC_3l = "(("+MC_MuMuMu+")||("+MC_EEE+")||("+MC_EEMu+")||("+MC_MuMuE+"))"
data_MuMuMu = "&&".join([
data_hadronSelection,
MuMuMuSelection,
"abs(mlmZ_mass-91.2)<10"
])
data_EEE = "&&".join([
data_hadronSelection,
EEESelection,
"abs(mlmZ_mass-91.2)<10"
])
data_EEMu = "&&".join([
data_hadronSelection,
EEMuSelection,
"abs(mlmZ_mass-91.2)<10"
])
data_MuMuE = "&&".join([
data_hadronSelection,
MuMuESelection,
"abs(mlmZ_mass-91.2)<10"
])
######yield_MC_3l computed for ALL channels but lumi changes slightly here depending on channel
yield_MC_3l = setup.lumi[channel]/1000.*u_float( getYieldFromChain(setup.sample['TTZ'][channel]['chain'], cutString = MC_3l, weight=weight, returnError = True))
if setup.verbose: print "yield_MC_looseSelection_3l: %s"%yield_MC_3l
yield_data_MuMuMu = u_float( getYieldFromChain(setup.sample['Data']['MuMu']['chain'], cutString = data_MuMuMu, weight=weight, returnError = True))
if setup.verbose: print "yield_data_looseSelection_MuMuMu: %s"%yield_data_MuMuMu
yield_data_EEE = u_float( getYieldFromChain(setup.sample['Data']['EE']['chain'], cutString = data_EEE, weight=weight, returnError = True))
if setup.verbose: print "yield_data_looseSelection_EEE: %s"%yield_data_EEE
yield_data_EMu = u_float( getYieldFromChain(setup.sample['Data']['EMu']['chain'], cutString = "(("+data_MuMuE+')||('+data_EEMu+'))', weight=weight, returnError = True))
if setup.verbose: print "yield_data_looseSelection_EMu: %s"%yield_data_EMu
yield_data_3l = yield_data_MuMuMu+yield_data_EEE+yield_data_EMu
if setup.verbose: print "yield_data_3l: %s"%yield_data_3l
#electroweak subtraction
print "\n Substracting electroweak backgrounds from data: \n"
yield_other = u_float(0., 0.)
for s in ['TTJets' , 'DY', 'other']:
yield_other+= setup.lumi[channel]/1000.* u_float(getYieldFromChain(setup.sample[s][channel]['chain'], cutString = MC_3l, weight=weight, returnError=True))
if setup.verbose: print "yield_looseSelection_other %s added, now: %s"%(s, yield_other)
normRegYield = yield_data_3l - yield_other
if normRegYield.val<0: print "\n !!!Warning!!! \n Negative normalization region yield data: (%s), MC: (%s) \n"%(yield_data_3l, yield_other)
print "normRegYield", normRegYield
print "\n Control Region predictys ", normRegYield, " TTZ events in data; ", yield_MC_3l, " TTZ events in MC. Ratio ---> ", (normRegYield/yield_MC_3l)
print "DD-TTZ ---> ", (normRegYield/yield_MC_3l)*yield_MC_2l
return (normRegYield/yield_MC_3l)*yield_MC_2l
)
args = argParser.parse_args()
# Logging
import StopsDilepton.tools.logger as logger
logger = logger.get_logger(args.logLevel, logFile = None )
import RootTools.core.logger as logger_rt
logger_rt = logger_rt.get_logger(args.logLevel, logFile = None )
#make samples
#from StopsDilepton.samples.cmgTuples_Fall15_mAODv2_25ns_2l_postProcessed_dilep import *
#from StopsDilepton.samples.cmgTuples_Data25ns_mAODv2_postProcessed import *
if args.mode=="doubleMu":
leptonSelectionString = "&&".join([looseMuIDString()+"==2", looseEleIDString()+"==0"])
trigger = "HLT_mumuIso"
elif args.mode=="doubleEle":
leptonSelectionString = "&&".join([looseMuIDString()+"==0", looseEleIDString()+"==2"])
trigger = "HLT_ee_DZ"
elif args.mode=="muEle":
leptonSelectionString = "&&".join([looseMuIDString()+"==1", looseEleIDString()+"==1"])
trigger = "HLT_mue"
else:
raise ValueError( "Mode %s not known"%args.mode )
# Extra requirements on data
#dataFilterCut = "(Flag_HBHENoiseIsoFilter&&Flag_HBHENoiseFilter&&Flag_CSCTightHaloFilter&&Flag_goodVertices&&Flag_eeBadScFilter&&Flag_EcalDeadCellTriggerPrimitiveFilter)"
filterCut = "(Flag_HBHENoiseIsoFilter&&Flag_HBHENoiseFilter&&Flag_CSCTightHaloFilter&&Flag_goodVertices&&Flag_eeBadScFilter&&Flag_EcalDeadCellTriggerPrimitiveFilter)"
import StopsDilepton.tools.user as user
presel_met = 'met_pt>' + str(metcut)
presel_njet = 'nGoodJets' + njetscut[0]
presel_nMedBjet = 'nBTags' + nMedBjetscut[0]
presel_nLooseBjet = 'Sum$(Jet_pt>30&&abs(Jet_eta)<2.4&&Jet_id&&Jet_btagCSV>' + str(
LooseBtagcoeff) + ')' + nLooseBjetscut[0]
presel_metsig = 'met_pt/sqrt(Sum$(Jet_pt*(Jet_pt>30&&abs(Jet_eta)<2.4&&Jet_id)))>' + str(
metsignifcut)
presel_dPhi = 'cos(met_phi-Jet_phi[0])<cos(' + str(
dphicut) + ')&&cos(met_phi-Jet_phi[1])<cos(' + str(dphicut) + ')'
data = [DoubleMuon_Run2015D, DoubleEG_Run2015D, MuonEG_Run2015D]
presel_flavour_MuMuMu = 'nGoodMuons>=2&&HLT_3mu&&' + looseMuIDString(
10) + '==3'
presel_flavour_EEE = 'nGoodElectrons>=2&&HLT_3e&&' + looseEleIDString(
10) + '==3'
presel_flavour_MuMuE = '(nGoodElectrons+nGoodMuons)>=2&&HLT_2mu1e&&' + looseEleIDString(
10) + '==1&&' + looseMuIDString(10) + '==2'
presel_flavour_EEMu = '(nGoodElectrons+nGoodMuons)>=2&&HLT_2e1mu&&' + looseEleIDString(
10) + '==2&&' + looseMuIDString(10) + '==1'
luminosity = data[0]["lumi"]
datacut = "(Flag_HBHENoiseFilter&&Flag_goodVertices&&Flag_CSCTightHaloFilter&&Flag_eeBadScFilter&&weight>0)"
preselection = presel_njet + '&&' + presel_nLooseBjet + '&&' + presel_nMedBjet + '&&' + presel_met + '&&' + presel_metsig + '&&' + presel_dPhi
backgrounds = [
DY_HT_LO, TTJets_Lep, TTZ, singleTop, diBoson, triBoson, TTXNoZ,
WJetsToLNu_HT
]
args = argParser.parse_args()
# Logging
import StopsDilepton.tools.logger as logger
logger = logger.get_logger(args.logLevel, logFile=None)
import RootTools.core.logger as logger_rt
logger_rt = logger_rt.get_logger(args.logLevel, logFile=None)
#make samples
#from StopsDilepton.samples.cmgTuples_Fall15_mAODv2_25ns_2l_postProcessed_dilep import *
#from StopsDilepton.samples.cmgTuples_Data25ns_mAODv2_postProcessed import *
if args.mode == "doubleMu":
leptonSelectionString = "&&".join(
[looseMuIDString() + "==2",
looseEleIDString() + "==0"])
trigger = "HLT_mumuIso"
elif args.mode == "doubleEle":
leptonSelectionString = "&&".join(
[looseMuIDString() + "==0",
looseEleIDString() + "==2"])
trigger = "HLT_ee_DZ"
elif args.mode == "muEle":
leptonSelectionString = "&&".join(
[looseMuIDString() + "==1",
looseEleIDString() + "==1"])
trigger = "HLT_mue"
else:
raise ValueError("Mode %s not known" % args.mode)
# Extra requirements on data
def _estimate(self, region, channel, setup):
printHeader("DD TTZ prediction for '%s' channel %s" %
(self.name, channel))
#Sum of all channels for 'all'
if channel == 'all':
return sum([
self.cachedEstimate(region, c, channel, setup)
for c in ['MuMu', 'EE', 'EMu']
])
else:
#Data driven for EE, EMu and MuMu.
preSelection = setup.preselection('MC', channel=channel)
#check lumi consistency
assert abs(
1. -
setup.lumi[channel] / setup.sample['Data'][channel]['lumi']
) < 0.01, "Lumi specified in setup %f does not match lumi in data sample %f in channel %s" % (
setup.lumi[channel], setup.sample['Data'][channel]['lumi'],
channel)
selection_MC_2l = "&&".join([
region.cutString(setup.sys['selectionModifier']),
preSelection['cut']
])
weight = preSelection['weightStr']
yield_MC_2l = setup.lumi[channel] / 1000. * u_float(
getYieldFromChain(setup.sample['TTZ'][channel]['chain'],
cutString=selection_MC_2l,
weight=weight,
returnError=True))
if setup.verbose: print "yield_MC_2l: %s" % yield_MC_2l
muonSelection_loosePt = looseMuIDString(ptCut=10)
electronSelection_loosePt = looseEleIDString(ptCut=10)
#mu_mu_mu
MuMuMuSelection = "nGoodMuons>=2" + '&&' + muonSelection_loosePt + "==3"
if setup.parameters['useTriggers']: MuMuMuSelection += '&&HLT_3mu'
#e_e_e
EEESelection = "nGoodElectrons>=2" + '&&' + electronSelection_loosePt + "==3"
if setup.parameters['useTriggers']: EEESelection += '&&HLT_3e'
#e_e_mu
EEMuSelection = "(nGoodMuons+nGoodElectrons)>=2" + "&&" + electronSelection_loosePt + "==2&&" + muonSelection_loosePt + "==1"
if setup.parameters['useTriggers']: EEMuSelection += '&&HLT_2e1mu'
#mu_mu_e
MuMuESelection = "(nGoodMuons+nGoodElectrons)>=2" + "&&" + electronSelection_loosePt + "==1&&" + muonSelection_loosePt + "==2"
if setup.parameters['useTriggers']: MuMuESelection += '&&HLT_2mu1e'
MC_hadronSelection = setup.selection(
'MC',
hadronicSelection=True,
**setup.defaultParameters(
update={
'nJets': self.nJets,
'nBTags': self.nMediumBTags,
'metMin': 0.,
'metSigMin': 0.,
'dPhiJetMet': 0.
}))['cut']
data_hadronSelection = setup.selection(
'Data',
hadronicSelection=True,
**setup.defaultParameters(
update={
'nJets': self.nJets,
'nBTags': self.nMediumBTags,
'metMin': 0.,
'metSigMin': 0.,
'dPhiJetMet': 0.
}))['cut']
#loose bjet selection added here
if self.nLooseBTags[0] >= 0:
MC_hadronSelection += '&&Sum$(Jet_pt>30&&abs(Jet_eta)<2.4&&Jet_id&&Jet_btagCSV>0.605)>=' + str(
self.nLooseBTags[0])
data_hadronSelection += '&&Sum$(Jet_pt>30&&abs(Jet_eta)<2.4&&Jet_id&&Jet_btagCSV>0.605)>=' + str(
self.nLooseBTags[0])
if self.nLooseBTags[1] >= 0:
MC_hadronSelection += '&&Sum$(Jet_pt>30&&abs(Jet_eta)<2.4&&Jet_id&&Jet_btagCSV>0.605)<=' + str(
self.nLooseBTags[1])
data_hadronSelection += '&&Sum$(Jet_pt>30&&abs(Jet_eta)<2.4&&Jet_id&&Jet_btagCSV>0.605)<=' + str(
self.nLooseBTags[1])
MC_MuMuMu = "&&".join([
MC_hadronSelection, MuMuMuSelection, "abs(mlmZ_mass-91.2)<10"
])
MC_EEE = "&&".join(
[MC_hadronSelection, EEESelection, "abs(mlmZ_mass-91.2)<10"])
MC_EEMu = "&&".join(
[MC_hadronSelection, EEMuSelection, "abs(mlmZ_mass-91.2)<10"])
MC_MuMuE = "&&".join(
[MC_hadronSelection, MuMuESelection, "abs(mlmZ_mass-91.2)<10"])
MC_3l = "((" + MC_MuMuMu + ")||(" + MC_EEE + ")||(" + MC_EEMu + ")||(" + MC_MuMuE + "))"
data_MuMuMu = "&&".join([
data_hadronSelection, MuMuMuSelection, "abs(mlmZ_mass-91.2)<10"
])
data_EEE = "&&".join(
[data_hadronSelection, EEESelection, "abs(mlmZ_mass-91.2)<10"])
data_EEMu = "&&".join([
data_hadronSelection, EEMuSelection, "abs(mlmZ_mass-91.2)<10"
])
data_MuMuE = "&&".join([
data_hadronSelection, MuMuESelection, "abs(mlmZ_mass-91.2)<10"
])
######yield_MC_3l computed for ALL channels but lumi changes slightly here depending on channel
yield_MC_3l = setup.lumi[channel] / 1000. * u_float(
getYieldFromChain(setup.sample['TTZ'][channel]['chain'],
cutString=MC_3l,
weight=weight,
returnError=True))
if setup.verbose:
print "yield_MC_looseSelection_3l: %s" % yield_MC_3l
yield_data_MuMuMu = u_float(
getYieldFromChain(setup.sample['Data']['MuMu']['chain'],
cutString=data_MuMuMu,
weight=weight,
returnError=True))
if setup.verbose:
print "yield_data_looseSelection_MuMuMu: %s" % yield_data_MuMuMu
yield_data_EEE = u_float(
getYieldFromChain(setup.sample['Data']['EE']['chain'],
cutString=data_EEE,
weight=weight,
returnError=True))
if setup.verbose:
print "yield_data_looseSelection_EEE: %s" % yield_data_EEE
yield_data_EMu = u_float(
getYieldFromChain(setup.sample['Data']['EMu']['chain'],
cutString="((" + data_MuMuE + ')||(' +
data_EEMu + '))',
weight=weight,
returnError=True))
if setup.verbose:
print "yield_data_looseSelection_EMu: %s" % yield_data_EMu
yield_data_3l = yield_data_MuMuMu + yield_data_EEE + yield_data_EMu
if setup.verbose: print "yield_data_3l: %s" % yield_data_3l
#electroweak subtraction
print "\n Substracting electroweak backgrounds from data: \n"
yield_other = u_float(0., 0.)
for s in ['TTJets', 'DY', 'other']:
yield_other += setup.lumi[channel] / 1000. * u_float(
getYieldFromChain(setup.sample[s][channel]['chain'],
cutString=MC_3l,
weight=weight,
returnError=True))
if setup.verbose:
print "yield_looseSelection_other %s added, now: %s" % (
s, yield_other)
normRegYield = yield_data_3l - yield_other
if normRegYield.val < 0:
print "\n !!!Warning!!! \n Negative normalization region yield data: (%s), MC: (%s) \n" % (
yield_data_3l, yield_other)
print "normRegYield", normRegYield
print "\n Control Region predictys ", normRegYield, " TTZ events in data; ", yield_MC_3l, " TTZ events in MC. Ratio ---> ", (
normRegYield / yield_MC_3l)
print "DD-TTZ ---> ", (normRegYield / yield_MC_3l) * yield_MC_2l
return (normRegYield / yield_MC_3l) * yield_MC_2l
njetscut = [">=4",'2m']
nMedBjetscut = [">=2",'1m']
nLooseBjetscut = [">=2",'2m']
presel_met = 'met_pt>'+str(metcut)
presel_njet = 'nGoodJets'+njetscut[0]
presel_nMedBjet = 'nBTags'+nMedBjetscut[0]
presel_nLooseBjet = 'Sum$(Jet_pt>30&&abs(Jet_eta)<2.4&&Jet_id&&Jet_btagCSV>'+str(LooseBtagcoeff)+')'+nLooseBjetscut[0]
presel_metsig = 'met_pt/sqrt(Sum$(Jet_pt*(Jet_pt>30&&abs(Jet_eta)<2.4&&Jet_id)))>'+str(metsignifcut)
presel_dPhi = 'cos(met_phi-Jet_phi[0])<cos('+str(dphicut)+')&&cos(met_phi-Jet_phi[1])<cos('+str(dphicut)+')'
data = [DoubleMuon_Run2015D,DoubleEG_Run2015D,MuonEG_Run2015D]
presel_flavour_MuMuMu = 'nGoodMuons>=2&&HLT_3mu&&'+looseMuIDString(10)+'==3'
presel_flavour_EEE = 'nGoodElectrons>=2&&HLT_3e&&'+looseEleIDString(10)+'==3'
presel_flavour_MuMuE = '(nGoodElectrons+nGoodMuons)>=2&&HLT_2mu1e&&'+looseEleIDString(10)+'==1&&'+looseMuIDString(10)+'==2'
presel_flavour_EEMu = '(nGoodElectrons+nGoodMuons)>=2&&HLT_2e1mu&&'+looseEleIDString(10)+'==2&&'+looseMuIDString(10)+'==1'
luminosity = data[0]["lumi"]
datacut = "(Flag_HBHENoiseFilter&&Flag_goodVertices&&Flag_CSCTightHaloFilter&&Flag_eeBadScFilter&&weight>0)"
preselection = presel_njet+'&&'+presel_nLooseBjet+'&&'+presel_nMedBjet+'&&'+presel_met+'&&'+presel_metsig+'&&'+presel_dPhi
backgrounds = [DY_HT_LO,TTJets_Lep,TTZ,singleTop, diBoson, triBoson, TTXNoZ, WJetsToLNu_HT]
#backgrounds = [TTZ,TTH,TTW]
#######################################################
# get the TChains for each sample #
#######################################################
)
args = argParser.parse_args()
# Logging
import StopsDilepton.tools.logger as logger
logger = logger.get_logger(args.logLevel, logFile = None )
import RootTools.core.logger as logger_rt
logger_rt = logger_rt.get_logger(args.logLevel, logFile = None )
#make samples
#from StopsDilepton.samples.cmgTuples_Fall15_mAODv2_25ns_2l_postProcessed_dilep import *
#from StopsDilepton.samples.cmgTuples_Data25ns_mAODv2_postProcessed import *
if args.mode=="doubleMu":
leptonSelectionString = "&&".join([looseMuIDString()+"==2", looseEleIDString()+"==0"])
trigger = "HLT_mumuIso"
elif args.mode=="doubleEle":
leptonSelectionString = "&&".join([looseMuIDString()+"==0", looseEleIDString()+"==2"])
trigger = "HLT_ee_DZ"
elif args.mode=="muEle":
leptonSelectionString = "&&".join([looseMuIDString()+"==1", looseEleIDString()+"==1"])
trigger = "HLT_mue"
else:
raise ValueError( "Mode %s not known"%args.mode )
# Extra requirements on data
#dataFilterCut = "(Flag_HBHENoiseIsoFilter&&Flag_HBHENoiseFilter&&Flag_CSCTightHaloFilter&&Flag_goodVertices&&Flag_eeBadScFilter&&Flag_EcalDeadCellTriggerPrimitiveFilter)"
filterCut = "(Flag_HBHENoiseIsoFilter&&Flag_HBHENoiseFilter&&Flag_CSCTightHaloFilter&&Flag_goodVertices&&Flag_eeBadScFilter&&Flag_EcalDeadCellTriggerPrimitiveFilter)"
import StopsDilepton.tools.user as user