def beginFile(self, inputFile, outputFile, inputTree, wrappedOutputTree):
self.out = wrappedOutputTree
self.out.branch("isMC", "I")
self.out.branch("is2016", "I")
self.out.branch("is2017", "I")
self.out.branch("is2018", "I")
self.out.branch("is4m", "I")
self.out.branch("is4e", "I")
self.out.branch("is2e2m", "I")
self.out.branch("isDoubleElectronTrigger", "I")
self.out.branch("isDoubleMuonTrigger", "I")
self.out.branch("isMuonElectronTrigger", "I")
self.out.branch("passedMETFilters", "I")
self.out.branch("nCleanElectron", "I")
self.out.branch("nCleanMuon", "I")
self.out.branch("nCleanTau", "I")
self.out.branch("nCleanPhoton", "I")
self.out.branch("nCleanJet", "I")
self.out.branch("nCleanBTagJet", "I")
self.out.branch("HT30", "F")
self.out.branch("iLepton1", "I")
self.out.branch("iLepton2", "I")
self.out.branch("iLepton3", "I")
self.out.branch("iLepton4", "I")
self.out.branch("Z1_pt", "F")
self.out.branch("Z1_eta", "F")
self.out.branch("Z1_phi", "F")
self.out.branch("Z1_mass", "F")
self.out.branch("Z1_dEta", "F")
self.out.branch("Z1_dPhi", "F")
self.out.branch("Z1_dR", "F")
self.out.branch("Z2_pt", "F")
self.out.branch("Z2_eta", "F")
self.out.branch("Z2_phi", "F")
self.out.branch("Z2_mass", "F")
self.out.branch("Z2_dEta", "F")
self.out.branch("Z2_dPhi", "F")
self.out.branch("Z2_dR", "F")
self.out.branch("H_pt", "F")
self.out.branch("H_eta", "F")
self.out.branch("H_phi", "F")
self.out.branch("H_mass", "F")
self.out.branch("H_dEta", "F")
self.out.branch("H_dPhi", "F")
self.out.branch("H_dR", "F")
self.out.branch("cosThetaStar", "F")
self.out.branch("cosTheta1", "F")
self.out.branch("cosTheta2", "F")
self.out.branch("phi", "F")
self.out.branch("phi1", "F")
#self.out.branch("genCosThetaStar", "F")
#self.out.branch("genCosTheta1", "F")
#self.out.branch("genPhi1", "F")
self.out.branch("lumiWeight", "F")
self.out.branch("lheWeight", "F")
self.out.branch("stitchWeight", "F")
self.out.branch("puWeight", "F")
self.out.branch("topWeight", "F")
self.out.branch("qcdnloWeight", "F")
self.out.branch("qcdnnloWeight", "F")
self.out.branch("ewknloWeight", "F")
self.out.branch("triggerWeight", "F")
self.out.branch("leptonWeight", "F")
self.out.branch("eventWeightLumi", "F")
self.fileName = inputFile.GetName()
self.sampleName = getNameFromFile(self.fileName)
self.isMC = not "Run201" in self.fileName
if self.verbose >= 0: print "+ Opening file", self.fileName
# b-tagging working points for DeepCSV
# https://twiki.cern.ch/twiki/bin/viewauth/CMS/BtagRecommendation2016Legacy
# https://twiki.cern.ch/twiki/bin/viewauth/CMS/BtagRecommendation94X
# https://twiki.cern.ch/twiki/bin/viewauth/CMS/BtagRecommendation102X
if "Run2016" in self.fileName or "Summer16" in self.fileName:
self.year = 2016
self.lumi = 35920.
self.btagLoose = 0.2217 #0.0614
self.btagMedium = 0.6321 #0.3093
self.btagTight = 0.8953 #0.7221
elif "Run2017" in self.fileName or "Fall17" in self.fileName:
self.year = 2017
self.lumi = 41530.
self.btagLoose = 0.1522 #0.0521
self.btagMedium = 0.4941 #0.3033
self.btagTight = 0.8001 #0.7489
elif "Run2018" in self.fileName or "Autumn18" in self.fileName:
self.year = 2018
self.lumi = 59740.
self.btagLoose = 0.1241 #0.0494
self.btagMedium = 0.4184 #0.2770
self.btagTight = 0.7527 #0.7264
else:
if self.verbose >= 0: print "- Unknown year, aborting module"
import sys
sys.exit()
self.xs = XS[self.sampleName] if self.sampleName in XS else 0.
self.nevents = EV[self.sampleName] if self.sampleName in EV else 0.
self.xsWeight = self.xs / self.nevents if self.nevents > 0 else 1.
self.lumiWeight = self.xsWeight * self.lumi if self.isMC else 1.
self.isLO = abs(
self.nevents % 1
) < 1.e-6 # if True, the event count is integer, so the weight should be normalized (+1)
if self.verbose >= 1:
print "+ Module parameters: isMC", self.isMC, ", year", self.year, ", lumi", self.lumi, "pb"
if self.verbose >= 1:
print "+ Sample", self.sampleName, ", XS", self.xs, ", events", self.nevents
if self.verbose >= 1: print "+ Weight", self.lumiWeight
if self.isMC and self.isLO and self.verbose >= 1:
print "+ Sample is LO, gen weight will be set to 1"
# self.puTool = PileupWeightTool(year = year) if self.isMC else None
self.DoubleElectronTriggers = [
"HLT_Ele23_Ele12_CaloIdL_TrackIdL_IsoVL",
"HLT_Ele23_Ele12_CaloIdL_TrackIdL_IsoVL_DZ",
"HLT_DoubleEle33_CaloIdL_MW",
"HLT_Ele16_Ele12_Ele8_CaloIdL_TrackIdL"
]
self.DoubleMuonTriggers = [
"HLT_Mu17_TrkIsoVVL_Mu8_TrkIsoVVL",
"HLT_Mu17_TrkIsoVVL_Mu8_TrkIsoVVL_DZ_Mass3p8",
"HLT_Mu17_TrkIsoVVL_Mu8_TrkIsoVVL_DZ_Mass8",
"HLT_TripleMu_12_10_5", "HLT_TripleMu_10_5_5_DZ"
]
self.MuonElectronTriggers = [
"HLT_Mu8_TrkIsoVVL_Ele23_CaloIdL_TrackIdL_IsoVL",
"HLT_Mu8_TrkIsoVVL_Ele23_CaloIdL_TrackIdL_IsoVL_DZ",
"HLT_Mu23_TrkIsoVVL_Ele12_CaloIdL_TrackIdL_IsoVL",
"HLT_Mu23_TrkIsoVVL_Ele12_CaloIdL_TrackIdL_IsoVL_DZ",
"HLT_Mu12_TrkIsoVVL_Ele23_CaloIdL_TrackIdL_IsoVL_DZ",
"HLT_Mu8_DiEle12_CaloIdL_TrackIdL",
"HLT_Mu8_DiEle12_CaloIdL_TrackIdL_DZ",
"HLT_DiMu9_Ele9_CaloIdL_TrackIdL",
"HLT_DiMu9_Ele9_CaloIdL_TrackIdL_DZ"
]
if self.isMC:
self.muSFs = MuonSFs(year=self.year)
self.elSFs = ElectronSFs(year=self.year)
self.puTool = PileupWeightTool(year=self.year)
class EleTauProducer(Module):
def __init__(self, name, dataType, **kwargs):
year = kwargs.get('year', 2017)
tes = kwargs.get('tes', 1.0)
channel = 'eletau'
self.name = name
self.year = year
self.tes = tes
self.out = TreeProducerEleTau(name)
self.isData = dataType == 'data'
self.doZpt = 'DY' in self.name
setYear(year)
self.vlooseIso = getVLooseTauIso(year)
if year == 2016:
self.trigger = lambda e: e.HLT_Ele25_eta2p1_WPTight_Gsf or e.HLT_Ele45_WPLoose_Gsf_L1JetTauSeeded #or e.HLT_Ele24_eta2p1_WPLoose_Gsf_LooseIsoPFTau20_SingleL1 or e.HLT_Ele24_eta2p1_WPLoose_Gsf_LooseIsoPFTau20 or e.HLT_Ele24_eta2p1_WPLoose_Gsf_LooseIsoPFTau30
self.electronCutPt = 26
else:
# HLT_Ele32_WPTight_Gsf_L1DoubleEG
# HLT_Ele32_WPTight_Gsf
self.trigger = lambda e: e.HLT_Ele35_WPTight_Gsf or e.HLT_Ele32_WPTight_Gsf
self.electronCutPt = 36
self.tauCutPt = 20
if not self.isData:
self.eleSFs = ElectronSFs(year=year)
self.puTool = PileupWeightTool(year=year)
self.ltfSFs = LeptonTauFakeSFs('loose', 'tight', year=year)
self.btagTool = BTagWeightTool('CSVv2',
'medium',
channel=channel,
year=year)
self.btagTool_deep = BTagWeightTool('DeepCSV',
'medium',
channel=channel,
year=year)
if self.doZpt:
self.recoilTool = RecoilCorrectionTool(year=year)
self.csvv2_wp = BTagWPs('CSVv2', year=year)
self.deepcsv_wp = BTagWPs('DeepCSV', year=year)
self.Nocut = 0
self.Trigger = 1
self.GoodElectrons = 2
self.GoodTaus = 3
self.GoodDiLepton = 4
self.TotalWeighted = 15
self.TotalWeighted_no0PU = 16
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.Nocut, "no cut")
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.Trigger, "trigger")
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.GoodElectrons,
"electron object")
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.GoodTaus,
"tau object")
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.GoodDiLepton,
"eletau pair")
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.TotalWeighted,
"no cut, weighted")
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.TotalWeighted_no0PU,
"no cut, weighted, PU>0")
self.out.cutflow.GetXaxis().SetLabelSize(0.041)
def beginJob(self):
pass
def endJob(self):
if not self.isData:
self.btagTool.setDirectory(self.out.outputfile, 'btag')
self.btagTool_deep.setDirectory(self.out.outputfile, 'btag')
self.out.outputfile.Write()
self.out.outputfile.Close()
def beginFile(self, inputFile, outputFile, inputTree, wrappedOutputTree):
pass
def endFile(self, inputFile, outputFile, inputTree, wrappedOutputTree):
pass
def analyze(self, event):
"""process event, return True (go to next module) or False (fail, go to next event)"""
#electrons = Collection(event, "Electron")
#####################################
self.out.cutflow.Fill(self.Nocut)
if self.isData:
self.out.cutflow.Fill(self.TotalWeighted, 1.)
if event.PV_npvs > 0:
self.out.cutflow.Fill(self.TotalWeighted_no0PU, 1.)
else:
return False
else:
self.out.cutflow.Fill(self.TotalWeighted, event.genWeight)
if event.Pileup_nTrueInt > 0:
self.out.cutflow.Fill(self.TotalWeighted_no0PU,
event.genWeight)
else:
return False
#####################################
if not self.trigger(event):
return False
#####################################
self.out.cutflow.Fill(self.Trigger)
#####################################
idx_goodelectrons = []
for ielectron in range(event.nElectron):
if event.Electron_pt[ielectron] < self.electronCutPt: continue
if abs(event.Electron_eta[ielectron]) > 2.1: continue
if abs(event.Electron_dz[ielectron]) > 0.2: continue
if abs(event.Electron_dxy[ielectron]) > 0.045: continue
if event.Electron_convVeto[ielectron] != 1: continue
if ord(event.Electron_lostHits[ielectron]) > 1: continue
if getvar(event, 'Electron_mvaFall17Iso_WP80')[ielectron] < 0.5:
continue
idx_goodelectrons.append(ielectron)
if len(idx_goodelectrons) == 0:
return False
#####################################
self.out.cutflow.Fill(self.GoodElectrons)
#####################################
idx_goodtaus = []
for itau in range(event.nTau):
if event.Tau_pt[itau] < self.tauCutPt: continue
if abs(event.Tau_eta[itau]) > 2.3: continue
if abs(event.Tau_dz[itau]) > 0.2: continue
if event.Tau_decayMode[itau] not in [0, 1, 10]: continue
if abs(event.Tau_charge[itau]) != 1: continue
#if ord(event.Tau_idAntiEle[itau])<1: continue
#if ord(event.Tau_idAntiMu[itau])<1: continue
if not self.vlooseIso(event, itau): continue
idx_goodtaus.append(itau)
if len(idx_goodtaus) == 0:
return False
#####################################
self.out.cutflow.Fill(self.GoodTaus)
#####################################
electrons = Collection(event, 'Electron')
taus = Collection(event, 'Tau')
ltaus = []
for idx1 in idx_goodelectrons:
for idx2 in idx_goodtaus:
dR = taus[idx2].p4().DeltaR(electrons[idx1].p4())
if dR < 0.5: continue
ltau = LeptonTauPair(idx1, event.Electron_pt[idx1],
event.Electron_pfRelIso03_all[idx1], idx2,
event.Tau_pt[idx2],
event.Tau_rawMVAoldDM2017v2[idx2])
ltaus.append(ltau)
if len(ltaus) == 0:
return False
ltau = bestDiLepton(ltaus)
electron = electrons[ltau.id1].p4()
tau = taus[ltau.id2].p4()
#print 'chosen tau1 (idx, pt) = ', ltau.id1, ltau.tau1_pt, 'check', electron.p4().Pt()
#print 'chosen tau2 (idx, pt) = ', ltau.id2, ltau.tau2_pt, 'check', tau.p4().Pt()
#####################################
self.out.cutflow.Fill(self.GoodDiLepton)
#####################################
jetIds = []
bjetIds = []
jets = Collection(event, 'Jet')
nfjets = 0
ncjets = 0
nbtag = 0
for ijet in range(event.nJet):
if event.Jet_pt[ijet] < 30: continue
if abs(event.Jet_eta[ijet]) > 4.7: continue
if electron.DeltaR(jets[ijet].p4()) < 0.5: continue
if tau.DeltaR(jets[ijet].p4()) < 0.5: continue
jetIds.append(ijet)
if abs(event.Jet_eta[ijet]) > 2.4:
nfjets += 1
else:
ncjets += 1
if event.Jet_btagDeepB[ijet] > self.deepcsv_wp.medium:
nbtag += 1
bjetIds.append(ijet)
if not self.isData and self.vlooseIso(
event,
ltau.id2) and event.Electron_pfRelIso03_all[ltau.id1] < 0.50:
self.btagTool.fillEfficiencies(event, jetIds)
self.btagTool_deep.fillEfficiencies(event, jetIds)
#eventSum = ROOT.TLorentzVector()
#
#for lep in electrons :
# eventSum += lep.p4()
#for lep in electrons :
# eventSum += lep.p4()
#for j in filter(self.jetSel,jets):
# eventSum += j.p4()
# ELECTRON
self.out.pt_1[0] = event.Electron_pt[ltau.id1]
self.out.eta_1[0] = event.Electron_eta[ltau.id1]
self.out.phi_1[0] = event.Electron_phi[ltau.id1]
self.out.m_1[0] = event.Electron_mass[ltau.id1]
self.out.dxy_1[0] = event.Electron_dxy[ltau.id1]
self.out.dz_1[0] = event.Electron_dz[ltau.id1]
self.out.q_1[0] = event.Electron_charge[ltau.id1]
self.out.pfRelIso03_all_1[0] = event.Electron_pfRelIso03_all[ltau.id1]
self.out.cutBased_1[0] = event.Electron_cutBased[ltau.id1]
self.out.mvaFall17Iso_1[0] = getvar(event,
'Electron_mvaFall17Iso')[ltau.id1]
self.out.mvaFall17Iso_WP80_1[0] = getvar(
event, 'Electron_mvaFall17Iso_WP80')[ltau.id1]
self.out.mvaFall17Iso_WP90_1[0] = getvar(
event, 'Electron_mvaFall17Iso_WP90')[ltau.id1]
self.out.mvaFall17Iso_WPL_1[0] = getvar(
event, 'Electron_mvaFall17Iso_WPL')[ltau.id1]
# TAU
self.out.pt_2[0] = event.Tau_pt[ltau.id2]
self.out.eta_2[0] = event.Tau_eta[ltau.id2]
self.out.phi_2[0] = event.Tau_phi[ltau.id2]
self.out.m_2[0] = event.Tau_mass[ltau.id2]
self.out.dxy_2[0] = event.Tau_dxy[ltau.id2]
self.out.dz_2[0] = event.Tau_dz[ltau.id2]
self.out.leadTkPtOverTauPt_2[0] = event.Tau_leadTkPtOverTauPt[ltau.id2]
self.out.chargedIso_2[0] = event.Tau_chargedIso[ltau.id2]
self.out.neutralIso_2[0] = event.Tau_neutralIso[ltau.id2]
self.out.photonsOutsideSignalCone_2[
0] = event.Tau_photonsOutsideSignalCone[ltau.id2]
self.out.puCorr_2[0] = event.Tau_puCorr[ltau.id2]
self.out.rawAntiEle_2[0] = event.Tau_rawAntiEle[ltau.id2]
self.out.rawIso_2[0] = event.Tau_rawIso[ltau.id2]
self.out.q_2[0] = event.Tau_charge[ltau.id2]
self.out.decayMode_2[0] = event.Tau_decayMode[ltau.id2]
###self.out.rawAntiEleCat_2[0] = event.Tau_rawAntiEleCat[ltau.id2]
self.out.idAntiEle_2[0] = ord(event.Tau_idAntiEle[ltau.id2])
self.out.idAntiMu_2[0] = ord(event.Tau_idAntiMu[ltau.id2])
self.out.idDecayMode_2[0] = event.Tau_idDecayMode[ltau.id2]
self.out.idDecayModeNewDMs_2[0] = event.Tau_idDecayModeNewDMs[ltau.id2]
self.out.rawMVAoldDM_2[0] = event.Tau_rawMVAoldDM[ltau.id2]
self.out.rawMVAnewDM2017v2_2[0] = event.Tau_rawMVAnewDM2017v2[ltau.id2]
self.out.rawMVAoldDM2017v1_2[0] = event.Tau_rawMVAoldDM2017v1[ltau.id2]
self.out.rawMVAoldDM2017v2_2[0] = event.Tau_rawMVAoldDM2017v2[ltau.id2]
self.out.idMVAoldDM_2[0] = ord(event.Tau_idMVAoldDM[ltau.id2])
self.out.idMVAoldDM2017v1_2[0] = ord(
event.Tau_idMVAoldDM2017v1[ltau.id2])
self.out.idMVAoldDM2017v2_2[0] = ord(
event.Tau_idMVAoldDM2017v2[ltau.id2])
self.out.idMVAnewDM2017v2_2[0] = ord(
event.Tau_idMVAnewDM2017v2[ltau.id2])
# GENERATOR
#print type(event.Tau_genPartFlav[ltau.id2])
if not self.isData:
self.out.genPartFlav_1[0] = ord(
event.Electron_genPartFlav[ltau.id1])
self.out.genPartFlav_2[0] = ord(event.Tau_genPartFlav[ltau.id2])
genvistau = Collection(event, 'GenVisTau')
dRmax = 1000
gendm = -1
genpt = -1
geneta = -1
genphi = -1
for igvt in range(event.nGenVisTau):
dR = genvistau[igvt].p4().DeltaR(tau)
if dR < 0.5 and dR < dRmax:
dRmax = dR
gendm = event.GenVisTau_status[igvt]
genpt = event.GenVisTau_pt[igvt]
geneta = event.GenVisTau_eta[igvt]
genphi = event.GenVisTau_phi[igvt]
self.out.gendecayMode_2[0] = gendm
self.out.genvistaupt_2[0] = genpt
self.out.genvistaueta_2[0] = geneta
self.out.genvistauphi_2[0] = genphi
# EVENT
self.out.isData[0] = self.isData
self.out.run[0] = event.run
self.out.luminosityBlock[0] = event.luminosityBlock
self.out.event[0] = event.event & 0xffffffffffffffff
self.out.met[0] = event.MET_pt
self.out.metphi[0] = event.MET_phi
###self.out.puppimet[0] = event.PuppiMET_pt
###self.out.puppimetphi[0] = event.PuppiMET_phi
###self.out.metsignificance[0] = event.MET_significance
###self.out.metcovXX[0] = event.MET_covXX
###self.out.metcovXY[0] = event.MET_covXY
###self.out.metcovYY[0] = event.MET_covYY
###self.out.fixedGridRhoFastjetAll[0] = event.fixedGridRhoFastjetAll
self.out.npvs[0] = event.PV_npvs
self.out.npvsGood[0] = event.PV_npvsGood
if not self.isData:
self.out.genmet[0] = event.GenMET_pt
self.out.genmetphi[0] = event.GenMET_phi
self.out.nPU[0] = event.Pileup_nPU
self.out.nTrueInt[0] = event.Pileup_nTrueInt
self.out.genWeight[0] = event.genWeight
try:
self.out.LHE_Njets[0] = event.LHE_Njets
except RuntimeError:
self.out.LHE_Njets[0] = -1
# JETS
self.out.njets[0] = len(jetIds)
self.out.njets50[0] = len([j for j in jetIds if event.Jet_pt[j] > 50])
self.out.nfjets[0] = nfjets
self.out.ncjets[0] = ncjets
self.out.nbtag[0] = nbtag
if len(jetIds) > 0:
self.out.jpt_1[0] = event.Jet_pt[jetIds[0]]
self.out.jeta_1[0] = event.Jet_eta[jetIds[0]]
self.out.jphi_1[0] = event.Jet_phi[jetIds[0]]
self.out.jcsvv2_1[0] = event.Jet_btagCSVV2[jetIds[0]]
self.out.jdeepb_1[0] = event.Jet_btagDeepB[jetIds[0]]
else:
self.out.jpt_1[0] = -9.
self.out.jeta_1[0] = -9.
self.out.jphi_1[0] = -9.
self.out.jcsvv2_1[0] = -9.
self.out.jdeepb_1[0] = -9.
if len(jetIds) > 1:
self.out.jpt_2[0] = event.Jet_pt[jetIds[1]]
self.out.jeta_2[0] = event.Jet_eta[jetIds[1]]
self.out.jphi_2[0] = event.Jet_phi[jetIds[1]]
self.out.jcsvv2_2[0] = event.Jet_btagCSVV2[jetIds[1]]
self.out.jdeepb_2[0] = event.Jet_btagDeepB[jetIds[1]]
else:
self.out.jpt_2[0] = -9.
self.out.jeta_2[0] = -9.
self.out.jphi_2[0] = -9.
self.out.jcsvv2_2[0] = -9.
self.out.jdeepb_2[0] = -9.
if len(bjetIds) > 0:
self.out.bpt_1[0] = event.Jet_pt[bjetIds[0]]
self.out.beta_1[0] = event.Jet_eta[bjetIds[0]]
else:
self.out.bpt_1[0] = -9.
self.out.beta_1[0] = -9.
if len(bjetIds) > 1:
self.out.bpt_2[0] = event.Jet_pt[bjetIds[1]]
self.out.beta_2[0] = event.Jet_eta[bjetIds[1]]
else:
self.out.bpt_2[0] = -9.
self.out.beta_2[0] = -9.
self.out.pfmt_1[0] = math.sqrt(
2 * self.out.pt_1[0] * self.out.met[0] *
(1 - math.cos(deltaPhi(self.out.phi_1[0], self.out.metphi[0]))))
self.out.pfmt_2[0] = math.sqrt(
2 * self.out.pt_2[0] * self.out.met[0] *
(1 - math.cos(deltaPhi(self.out.phi_2[0], self.out.metphi[0]))))
self.out.m_vis[0] = (electron + tau).M()
self.out.pt_ll[0] = (electron + tau).Pt()
self.out.dR_ll[0] = electron.DeltaR(tau)
self.out.dphi_ll[0] = deltaPhi(self.out.phi_1[0], self.out.phi_2[0])
# PZETA
leg1 = ROOT.TVector3(electron.Px(), electron.Py(), 0.)
leg2 = ROOT.TVector3(tau.Px(), tau.Py(), 0.)
met_tlv = ROOT.TLorentzVector()
met_tlv = ROOT.TLorentzVector()
met_tlv.SetPxPyPzE(self.out.met[0] * math.cos(self.out.metphi[0]),
self.out.met[0] * math.cos(self.out.metphi[0]), 0,
self.out.met[0])
metleg = met_tlv.Vect()
zetaAxis = ROOT.TVector3(leg1.Unit() + leg2.Unit()).Unit()
pzetaVis = leg1 * zetaAxis + leg2 * zetaAxis
pzetaMET = metleg * zetaAxis
self.out.pzetamiss[0] = pzetaMET
self.out.pzetavis[0] = pzetaVis
self.out.dzeta[0] = pzetaMET - 0.85 * pzetaVis
# VETOS
self.out.extramuon_veto[0], self.out.extraelec_veto[
0], self.out.dilepton_veto[0] = extraLeptonVetos(
event, [-1], [ltau.id1], self.name)
# WEIGHTS
if not self.isData:
if self.doZpt:
zboson = getZPTMass(event)
self.out.m_genboson[0] = zboson.M()
self.out.pt_genboson[0] = zboson.Pt()
self.out.zptweight[0] = self.recoilTool.getZptWeight(
zboson.Pt(), zboson.M())
self.out.genweight[0] = event.genWeight
self.out.puweight[0] = self.puTool.getWeight(event.Pileup_nTrueInt)
self.out.trigweight[0] = self.eleSFs.getTriggerSF(
self.out.pt_1[0], self.out.eta_1[0])
self.out.idisoweight_1[0] = self.eleSFs.getIdIsoSF(
self.out.pt_1[0], self.out.eta_1[0])
self.out.idisoweight_2[0] = self.ltfSFs.getSF(
self.out.genPartFlav_2[0], self.out.eta_2[0])
self.out.btagweight[0] = self.btagTool.getWeight(event, jetIds)
self.out.btagweight_deep[0] = self.btagTool_deep.getWeight(
event, jetIds)
self.out.weight[0] = self.out.genWeight[0] * self.out.puweight[
0] * self.out.trigweight[0] * self.out.idisoweight_1[
0] * self.out.idisoweight_2[0]
self.out.tree.Fill()
return True
def __init__(self, name, dataType, **kwargs):
year = kwargs.get('year', 2017)
tes = kwargs.get('tes', 1.0)
channel = 'eletau'
self.name = name
self.year = year
self.tes = tes
self.out = TreeProducerEleTau(name)
self.isData = dataType == 'data'
self.doZpt = 'DY' in self.name
setYear(year)
self.vlooseIso = getVLooseTauIso(year)
if year == 2016:
self.trigger = lambda e: e.HLT_Ele25_eta2p1_WPTight_Gsf or e.HLT_Ele45_WPLoose_Gsf_L1JetTauSeeded #or e.HLT_Ele24_eta2p1_WPLoose_Gsf_LooseIsoPFTau20_SingleL1 or e.HLT_Ele24_eta2p1_WPLoose_Gsf_LooseIsoPFTau20 or e.HLT_Ele24_eta2p1_WPLoose_Gsf_LooseIsoPFTau30
self.electronCutPt = 26
else:
# HLT_Ele32_WPTight_Gsf_L1DoubleEG
# HLT_Ele32_WPTight_Gsf
self.trigger = lambda e: e.HLT_Ele35_WPTight_Gsf or e.HLT_Ele32_WPTight_Gsf
self.electronCutPt = 36
self.tauCutPt = 20
if not self.isData:
self.eleSFs = ElectronSFs(year=year)
self.puTool = PileupWeightTool(year=year)
self.ltfSFs = LeptonTauFakeSFs('loose', 'tight', year=year)
self.btagTool = BTagWeightTool('CSVv2',
'medium',
channel=channel,
year=year)
self.btagTool_deep = BTagWeightTool('DeepCSV',
'medium',
channel=channel,
year=year)
if self.doZpt:
self.recoilTool = RecoilCorrectionTool(year=year)
self.csvv2_wp = BTagWPs('CSVv2', year=year)
self.deepcsv_wp = BTagWPs('DeepCSV', year=year)
self.Nocut = 0
self.Trigger = 1
self.GoodElectrons = 2
self.GoodTaus = 3
self.GoodDiLepton = 4
self.TotalWeighted = 15
self.TotalWeighted_no0PU = 16
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.Nocut, "no cut")
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.Trigger, "trigger")
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.GoodElectrons,
"electron object")
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.GoodTaus,
"tau object")
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.GoodDiLepton,
"eletau pair")
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.TotalWeighted,
"no cut, weighted")
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.TotalWeighted_no0PU,
"no cut, weighted, PU>0")
self.out.cutflow.GetXaxis().SetLabelSize(0.041)
class EleTauProducer(Module):
def __init__(self, name, dataType, **kwargs):
self.name = name
self.out = TreeProducerEleTau(name)
self.isData = dataType == 'data'
self.year = kwargs.get('year', 2017)
self.tes = kwargs.get('tes', 1.0)
self.ltf = kwargs.get('ltf', 1.0)
self.jtf = kwargs.get('jtf', 1.0)
self.doZpt = kwargs.get('doZpt', 'DY' in name)
self.doRecoil = kwargs.get(
'doRecoil', ('DY' in name or re.search(r"W\d?Jets", name))
and self.year > 2016)
self.doTTpt = kwargs.get('doTTpt', 'TT' in name)
self.doTight = kwargs.get('doTight', self.tes != 1 or self.ltf != 1)
self.channel = 'eletau'
year, channel = self.year, self.channel
self.vlooseIso = getVLooseTauIso(year)
if year == 2016:
self.trigger = lambda e: e.HLT_Ele25_eta2p1_WPTight_Gsf or e.HLT_Ele27_WPTight_Gsf #or e.HLT_Ele45_WPLoose_Gsf_L1JetTauSeeded #or e.HLT_Ele24_eta2p1_WPLoose_Gsf_LooseIsoPFTau20_SingleL1 or e.HLT_Ele24_eta2p1_WPLoose_Gsf_LooseIsoPFTau20 or e.HLT_Ele24_eta2p1_WPLoose_Gsf_LooseIsoPFTau30
self.eleCutPt = 26
elif year == 2017:
self.trigger = lambda e: e.HLT_Ele35_WPTight_Gsf or e.HLT_Ele32_WPTight_Gsf_L1DoubleEG or e.HLT_Ele32_WPTight_Gsf
self.eleCutPt = 36
else:
self.trigger = lambda e: e.HLT_Ele32_WPTight_Gsf or e.HLT_Ele35_WPTight_Gsf
self.eleCutPt = 33
self.tauCutPt = 20
if not self.isData:
self.eleSFs = ElectronSFs(year=year)
self.puTool = PileupWeightTool(year=year)
self.ltfSFs = LeptonTauFakeSFs('loose', 'tight', year=year)
self.btagTool = BTagWeightTool('DeepCSV',
'medium',
channel='mutau',
year=year)
if self.doZpt:
self.zptTool = ZptCorrectionTool(year=year)
if self.doRecoil:
self.recoilTool = RecoilCorrectionTool(year=year)
self.deepcsv_wp = BTagWPs('DeepCSV', year=year)
self.Nocut = 0
self.Trigger = 1
self.GoodElectrons = 2
self.GoodTaus = 3
self.GoodDiLepton = 4
self.TotalWeighted = 15
self.TotalWeighted_no0PU = 16
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.Nocut, "no cut")
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.Trigger, "trigger")
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.GoodElectrons,
"electron object")
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.GoodTaus,
"tau object")
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.GoodDiLepton,
"eletau pair")
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.TotalWeighted,
"no cut, weighted")
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.TotalWeighted_no0PU,
"no cut, weighted, PU>0")
self.out.cutflow.GetXaxis().SetLabelSize(0.041)
def beginJob(self):
pass
def endJob(self):
if not self.isData:
self.btagTool.setDirectory(self.out.outputfile, 'btag')
self.out.endJob()
def beginFile(self, inputFile, outputFile, inputTree, wrappedOutputTree):
sys.stdout.flush()
checkBranches(inputTree)
def endFile(self, inputFile, outputFile, inputTree, wrappedOutputTree):
pass
def analyze(self, event):
"""process event, return True (go to next module) or False (fail, go to next event)"""
sys.stdout.flush()
#####################################
self.out.cutflow.Fill(self.Nocut)
if self.isData:
self.out.cutflow.Fill(self.TotalWeighted, 1.)
if event.PV_npvs > 0:
self.out.cutflow.Fill(self.TotalWeighted_no0PU, 1.)
else:
return False
else:
self.out.cutflow.Fill(self.TotalWeighted, event.genWeight)
if event.Pileup_nTrueInt > 0:
self.out.cutflow.Fill(self.TotalWeighted_no0PU,
event.genWeight)
else:
return False
#####################################
if not self.trigger(event):
return False
#####################################
self.out.cutflow.Fill(self.Trigger)
#####################################
idx_goodelectrons = []
for ielectron in range(event.nElectron):
if event.Electron_pt[ielectron] < self.eleCutPt: continue
if abs(event.Electron_eta[ielectron]) > 2.1: continue
if abs(event.Electron_dz[ielectron]) > 0.2: continue
if abs(event.Electron_dxy[ielectron]) > 0.045: continue
if event.Electron_convVeto[ielectron] != 1: continue
if ord(event.Electron_lostHits[ielectron]) > 1: continue
if event.Electron_mvaFall17V2Iso_WP80[ielectron] < 0.5: continue
idx_goodelectrons.append(ielectron)
if len(idx_goodelectrons) == 0:
return False
#####################################
self.out.cutflow.Fill(self.GoodElectrons)
#####################################
Tau_genmatch = {} # bug in Tau_genPartFlav
idx_goodtaus = []
for itau in range(event.nTau):
if not self.vlooseIso(event, itau): continue
if abs(event.Tau_eta[itau]) > 2.3: continue
if abs(event.Tau_dz[itau]) > 0.2: continue
if event.Tau_decayMode[itau] not in [0, 1, 10]: continue
if abs(event.Tau_charge[itau]) != 1: continue
if not self.isData:
Tau_genmatch[itau] = genmatch(event, itau)
#if self.tes!=1.0 and Tau_genmatch[itau]==5:
# event.Tau_pt[itau] *= self.tes
# event.Tau_mass[itau] *= self.tes
#elif self.ltf!=1.0 and 0<Tau_genmatch[itau]<5:
# event.Tau_pt[itau] *= self.ltf
# event.Tau_mass[itau] *= self.ltf
#elif self.jtf!=1.0 and Tau_genmatch[itau]==0:
# event.Tau_pt[itau] *= self.jtf
# event.Tau_mass[itau] *= self.jtf
if event.Tau_pt[itau] < self.tauCutPt: continue
###if ord(event.Tau_idAntiEle[itau])<1: continue
###if ord(event.Tau_idAntiMu[itau])<1: continue
idx_goodtaus.append(itau)
if len(idx_goodtaus) == 0:
return False
#####################################
self.out.cutflow.Fill(self.GoodTaus)
#####################################
electrons = Collection(event, 'Electron')
taus = Collection(event, 'Tau')
ltaus = []
for idx1 in idx_goodelectrons:
for idx2 in idx_goodtaus:
dR = taus[idx2].p4().DeltaR(electrons[idx1].p4())
if dR < 0.5: continue
ltau = LeptonTauPair(idx1, event.Electron_pt[idx1],
event.Electron_pfRelIso03_all[idx1], idx2,
event.Tau_pt[idx2],
event.Tau_rawMVAoldDM2017v2[idx2])
ltaus.append(ltau)
if len(ltaus) == 0:
return False
ltau = bestDiLepton(ltaus)
electron = electrons[ltau.id1].p4()
tau = taus[ltau.id2].p4()
#print 'chosen tau1 (idx, pt) = ', ltau.id1, ltau.tau1_pt, 'check', electron.p4().Pt()
#print 'chosen tau2 (idx, pt) = ', ltau.id2, ltau.tau2_pt, 'check', tau.p4().Pt()
#####################################
self.out.cutflow.Fill(self.GoodDiLepton)
#####################################
# VETOS
self.out.extramuon_veto[0], self.out.extraelec_veto[
0], self.out.dilepton_veto[0] = extraLeptonVetos(
event, [-1], [ltau.id1], self.channel)
self.out.lepton_vetos[0] = self.out.extramuon_veto[
0] or self.out.extraelec_veto[0] or self.out.dilepton_veto[0]
###if self.doTight and (self.out.lepton_vetos[0] or event.Electron_pfRelIso03_all[ltau.id1]>0.10 or\
### ord(event.Tau_idAntiMu[ltau.id2]<1 or ord(event.Tau_idAntiEle[ltau.id2]<8):
### return False
# JETS
jetIds = []
bjetIds = []
jets = Collection(event, 'Jet')
nfjets = 0
ncjets = 0
nbtag = 0
for ijet in range(event.nJet):
if event.Jet_pt[ijet] < 30: continue
if abs(event.Jet_eta[ijet]) > 4.7: continue
if electron.DeltaR(jets[ijet].p4()) < 0.5: continue
if tau.DeltaR(jets[ijet].p4()) < 0.5: continue
jetIds.append(ijet)
if abs(event.Jet_eta[ijet]) > 2.4:
nfjets += 1
else:
ncjets += 1
if event.Jet_btagDeepB[ijet] > self.deepcsv_wp.medium:
nbtag += 1
bjetIds.append(ijet)
if not self.isData and self.vlooseIso(
event,
ltau.id2) and event.Electron_pfRelIso03_all[ltau.id1] < 0.50:
self.btagTool.fillEfficiencies(event, jetIds)
# EVENT
self.out.isData[0] = self.isData
self.out.run[0] = event.run
self.out.lumi[0] = event.luminosityBlock
self.out.event[0] = event.event & 0xffffffffffffffff
###self.out.puppimet[0] = event.PuppiMET_pt
###self.out.puppimetphi[0] = event.PuppiMET_phi
###self.out.metsignificance[0] = event.MET_significance
###self.out.metcovXX[0] = event.MET_covXX
###self.out.metcovXY[0] = event.MET_covXY
###self.out.metcovYY[0] = event.MET_covYY
self.out.npvs[0] = event.PV_npvs
self.out.npvsGood[0] = event.PV_npvsGood
if not self.isData:
self.out.genmet[0] = event.GenMET_pt
self.out.genmetphi[0] = event.GenMET_phi
self.out.nPU[0] = event.Pileup_nPU
self.out.nTrueInt[0] = event.Pileup_nTrueInt
try:
self.out.LHE_Njets[0] = event.LHE_Njets
except RuntimeError:
self.out.LHE_Njets[0] = -1
# ELECTRON
self.out.pt_1[0] = event.Electron_pt[ltau.id1]
self.out.eta_1[0] = event.Electron_eta[ltau.id1]
self.out.phi_1[0] = event.Electron_phi[ltau.id1]
self.out.m_1[0] = event.Electron_mass[ltau.id1]
self.out.dxy_1[0] = event.Electron_dxy[ltau.id1]
self.out.dz_1[0] = event.Electron_dz[ltau.id1]
self.out.q_1[0] = event.Electron_charge[ltau.id1]
self.out.pfRelIso03_all_1[0] = event.Electron_pfRelIso03_all[ltau.id1]
self.out.cutBased_1[0] = event.Electron_cutBased[ltau.id1]
self.out.mvaFall17Iso_1[0] = event.Electron_mvaFall17V2Iso[ltau.id1]
self.out.mvaFall17Iso_WPL_1[0] = event.Electron_mvaFall17V2Iso_WPL[
ltau.id1]
self.out.mvaFall17Iso_WP80_1[0] = event.Electron_mvaFall17V2Iso_WP80[
ltau.id1]
self.out.mvaFall17Iso_WP90_1[0] = event.Electron_mvaFall17V2Iso_WP90[
ltau.id1]
# TAU
self.out.pt_2[0] = event.Tau_pt[ltau.id2]
self.out.eta_2[0] = event.Tau_eta[ltau.id2]
self.out.phi_2[0] = event.Tau_phi[ltau.id2]
self.out.m_2[0] = event.Tau_mass[ltau.id2]
self.out.dxy_2[0] = event.Tau_dxy[ltau.id2]
self.out.dz_2[0] = event.Tau_dz[ltau.id2]
self.out.leadTkPtOverTauPt_2[0] = event.Tau_leadTkPtOverTauPt[ltau.id2]
self.out.chargedIso_2[0] = event.Tau_chargedIso[ltau.id2]
self.out.neutralIso_2[0] = event.Tau_neutralIso[ltau.id2]
self.out.photonsOutsideSignalCone_2[
0] = event.Tau_photonsOutsideSignalCone[ltau.id2]
self.out.puCorr_2[0] = event.Tau_puCorr[ltau.id2]
self.out.rawAntiEle_2[0] = event.Tau_rawAntiEle[ltau.id2]
self.out.rawIso_2[0] = event.Tau_rawIso[ltau.id2]
self.out.q_2[0] = event.Tau_charge[ltau.id2]
self.out.decayMode_2[0] = event.Tau_decayMode[ltau.id2]
###self.out.rawAntiEleCat_2[0] = event.Tau_rawAntiEleCat[ltau.id2]
self.out.idAntiEle_2[0] = ord(event.Tau_idAntiEle[ltau.id2])
self.out.idAntiMu_2[0] = ord(event.Tau_idAntiMu[ltau.id2])
self.out.idDecayMode_2[0] = event.Tau_idDecayMode[ltau.id2]
self.out.idDecayModeNewDMs_2[0] = event.Tau_idDecayModeNewDMs[ltau.id2]
self.out.rawMVAoldDM_2[0] = event.Tau_rawMVAoldDM[ltau.id2]
self.out.rawMVAnewDM2017v2_2[0] = event.Tau_rawMVAnewDM2017v2[ltau.id2]
self.out.rawMVAoldDM2017v1_2[0] = event.Tau_rawMVAoldDM2017v1[ltau.id2]
self.out.rawMVAoldDM2017v2_2[0] = event.Tau_rawMVAoldDM2017v2[ltau.id2]
self.out.idMVAoldDM_2[0] = ord(event.Tau_idMVAoldDM[ltau.id2])
self.out.idMVAoldDM2017v1_2[0] = ord(
event.Tau_idMVAoldDM2017v1[ltau.id2])
self.out.idMVAoldDM2017v2_2[0] = ord(
event.Tau_idMVAoldDM2017v2[ltau.id2])
self.out.idMVAnewDM2017v2_2[0] = ord(
event.Tau_idMVAnewDM2017v2[ltau.id2])
# GENERATOR
if not self.isData:
self.out.genPartFlav_1[0] = ord(
event.Electron_genPartFlav[ltau.id1])
self.out.genPartFlav_2[0] = Tau_genmatch[
ltau.id2] # ord(event.Tau_genPartFlav[ltau.id2])
genvistau = Collection(event, 'GenVisTau')
dRmax = 1000
gendm = -1
genpt = -1
geneta = -1
genphi = -1
for igvt in range(event.nGenVisTau):
dR = genvistau[igvt].p4().DeltaR(tau)
if dR < 0.5 and dR < dRmax:
dRmax = dR
gendm = event.GenVisTau_status[igvt]
genpt = event.GenVisTau_pt[igvt]
geneta = event.GenVisTau_eta[igvt]
genphi = event.GenVisTau_phi[igvt]
self.out.gendecayMode_2[0] = gendm
self.out.genvistaupt_2[0] = genpt
self.out.genvistaueta_2[0] = geneta
self.out.genvistauphi_2[0] = genphi
# WEIGHTS
met = TLorentzVector()
met.SetPxPyPzE(event.MET_pt * cos(event.MET_phi),
event.MET_pt * sin(event.MET_phi), 0, event.MET_pt)
if not self.isData:
if self.doRecoil:
boson, boson_vis = getBoson(event)
self.recoilTool.CorrectPFMETByMeanResolution(
met, boson, boson_vis, len(jetIds))
event.MET_pt = met.Pt()
event.MET_phi = met.Phi()
self.out.m_genboson[0] = boson.M()
self.out.pt_genboson[0] = boson.Pt()
if self.doZpt:
self.out.zptweight[0] = self.zptTool.getZptWeight(
boson.Pt(), boson.M())
elif self.doZpt:
zboson = getZBoson(event)
self.out.m_genboson[0] = zboson.M()
self.out.pt_genboson[0] = zboson.Pt()
self.out.zptweight[0] = self.zptTool.getZptWeight(
zboson.Pt(), zboson.M())
elif self.doTTpt:
toppt1, toppt2 = getTTPt(event)
self.out.ttptweight[0] = getTTptWeight(toppt1, toppt2)
self.out.genweight[0] = event.genWeight
self.out.puweight[0] = self.puTool.getWeight(event.Pileup_nTrueInt)
self.out.trigweight[0] = self.eleSFs.getTriggerSF(
self.out.pt_1[0], self.out.eta_1[0])
self.out.idisoweight_1[0] = self.eleSFs.getIdIsoSF(
self.out.pt_1[0], self.out.eta_1[0])
self.out.idisoweight_2[0] = self.ltfSFs.getSF(
self.out.genPartFlav_2[0], self.out.eta_2[0])
self.out.btagweight[0] = self.btagTool.getWeight(event, jetIds)
self.out.weight[0] = self.out.genweight[0] * self.out.puweight[
0] * self.out.trigweight[0] * self.out.idisoweight_1[
0] * self.out.idisoweight_2[0]
# JETS
self.out.njets[0] = len(jetIds)
self.out.njets50[0] = len([j for j in jetIds if event.Jet_pt[j] > 50])
self.out.nfjets[0] = nfjets
self.out.ncjets[0] = ncjets
self.out.nbtag[0] = nbtag
if len(jetIds) > 0:
self.out.jpt_1[0] = event.Jet_pt[jetIds[0]]
self.out.jeta_1[0] = event.Jet_eta[jetIds[0]]
self.out.jphi_1[0] = event.Jet_phi[jetIds[0]]
self.out.jdeepb_1[0] = event.Jet_btagDeepB[jetIds[0]]
else:
self.out.jpt_1[0] = -9.
self.out.jeta_1[0] = -9.
self.out.jphi_1[0] = -9.
self.out.jdeepb_1[0] = -9.
if len(jetIds) > 1:
self.out.jpt_2[0] = event.Jet_pt[jetIds[1]]
self.out.jeta_2[0] = event.Jet_eta[jetIds[1]]
self.out.jphi_2[0] = event.Jet_phi[jetIds[1]]
self.out.jdeepb_2[0] = event.Jet_btagDeepB[jetIds[1]]
else:
self.out.jpt_2[0] = -9.
self.out.jeta_2[0] = -9.
self.out.jphi_2[0] = -9.
self.out.jdeepb_2[0] = -9.
if len(bjetIds) > 0:
self.out.bpt_1[0] = event.Jet_pt[bjetIds[0]]
self.out.beta_1[0] = event.Jet_eta[bjetIds[0]]
else:
self.out.bpt_1[0] = -9.
self.out.beta_1[0] = -9.
if len(bjetIds) > 1:
self.out.bpt_2[0] = event.Jet_pt[bjetIds[1]]
self.out.beta_2[0] = event.Jet_eta[bjetIds[1]]
else:
self.out.bpt_2[0] = -9.
self.out.beta_2[0] = -9.
self.out.met[0] = event.MET_pt
self.out.metphi[0] = event.MET_phi
self.out.pfmt_1[0] = sqrt(
2 * self.out.pt_1[0] * self.out.met[0] *
(1 - cos(deltaPhi(self.out.phi_1[0], self.out.metphi[0]))))
self.out.pfmt_2[0] = sqrt(
2 * self.out.pt_2[0] * self.out.met[0] *
(1 - cos(deltaPhi(self.out.phi_2[0], self.out.metphi[0]))))
self.out.m_vis[0] = (electron + tau).M()
self.out.pt_ll[0] = (electron + tau).Pt()
self.out.dR_ll[0] = electron.DeltaR(tau)
self.out.dphi_ll[0] = deltaPhi(self.out.phi_1[0], self.out.phi_2[0])
self.out.deta_ll[0] = abs(self.out.eta_1[0] - self.out.eta_2[0])
# PZETA
leg1 = TVector3(electron.Px(), electron.Py(), 0.)
leg2 = TVector3(tau.Px(), tau.Py(), 0.)
zetaAxis = TVector3(leg1.Unit() + leg2.Unit()).Unit()
pzeta_vis = leg1 * zetaAxis + leg2 * zetaAxis
pzeta_miss = met.Vect() * zetaAxis
self.out.pzetamiss[0] = pzeta_miss
self.out.pzetavis[0] = pzeta_vis
self.out.dzeta[0] = pzeta_miss - 0.85 * pzeta_vis
self.out.tree.Fill()
return True
def __init__(self, name, dataType, **kwargs):
self.name = name
self.out = TreeProducerEleTau(name)
self.isData = dataType == 'data'
self.year = kwargs.get('year', 2017)
self.tes = kwargs.get('tes', 1.0)
self.ltf = kwargs.get('ltf', 1.0)
self.jtf = kwargs.get('jtf', 1.0)
self.doZpt = kwargs.get('doZpt', 'DY' in name)
self.doRecoil = kwargs.get(
'doRecoil', ('DY' in name or re.search(r"W\d?Jets", name))
and self.year > 2016)
self.doTTpt = kwargs.get('doTTpt', 'TT' in name)
self.doTight = kwargs.get('doTight', self.tes != 1 or self.ltf != 1)
self.channel = 'eletau'
year, channel = self.year, self.channel
self.vlooseIso = getVLooseTauIso(year)
if year == 2016:
self.trigger = lambda e: e.HLT_Ele25_eta2p1_WPTight_Gsf or e.HLT_Ele27_WPTight_Gsf #or e.HLT_Ele45_WPLoose_Gsf_L1JetTauSeeded #or e.HLT_Ele24_eta2p1_WPLoose_Gsf_LooseIsoPFTau20_SingleL1 or e.HLT_Ele24_eta2p1_WPLoose_Gsf_LooseIsoPFTau20 or e.HLT_Ele24_eta2p1_WPLoose_Gsf_LooseIsoPFTau30
self.eleCutPt = 26
elif year == 2017:
self.trigger = lambda e: e.HLT_Ele35_WPTight_Gsf or e.HLT_Ele32_WPTight_Gsf_L1DoubleEG or e.HLT_Ele32_WPTight_Gsf
self.eleCutPt = 36
else:
self.trigger = lambda e: e.HLT_Ele32_WPTight_Gsf or e.HLT_Ele35_WPTight_Gsf
self.eleCutPt = 33
self.tauCutPt = 20
if not self.isData:
self.eleSFs = ElectronSFs(year=year)
self.puTool = PileupWeightTool(year=year)
self.ltfSFs = LeptonTauFakeSFs('loose', 'tight', year=year)
self.btagTool = BTagWeightTool('DeepCSV',
'medium',
channel='mutau',
year=year)
if self.doZpt:
self.zptTool = ZptCorrectionTool(year=year)
if self.doRecoil:
self.recoilTool = RecoilCorrectionTool(year=year)
self.deepcsv_wp = BTagWPs('DeepCSV', year=year)
self.Nocut = 0
self.Trigger = 1
self.GoodElectrons = 2
self.GoodTaus = 3
self.GoodDiLepton = 4
self.TotalWeighted = 15
self.TotalWeighted_no0PU = 16
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.Nocut, "no cut")
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.Trigger, "trigger")
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.GoodElectrons,
"electron object")
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.GoodTaus,
"tau object")
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.GoodDiLepton,
"eletau pair")
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.TotalWeighted,
"no cut, weighted")
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.TotalWeighted_no0PU,
"no cut, weighted, PU>0")
self.out.cutflow.GetXaxis().SetLabelSize(0.041)
def beginFile(self, inputFile, outputFile, inputTree, wrappedOutputTree):
self.out = wrappedOutputTree
self.out.branch("iSkim", "I")
self.out.branch("isMC", "I")
self.out.branch("isSingleMuIsoTrigger", "I")
self.out.branch("isSingleMuTrigger", "I")
self.out.branch("isDoubleMuonTrigger", "I")
self.out.branch("isSingleEleIsoTrigger", "I")
self.out.branch("isSingleEleTrigger", "I")
self.out.branch("passedMETFilters", "I")
self.out.branch("nCleanElectron", "I")
self.out.branch("nCleanMuon", "I")
self.out.branch("nCleanJet", "I")
self.out.branch("Z_mass", "F")
self.out.branch("Z_pt", "F")
self.out.branch("W_mass", "F")
self.out.branch("W_tmass", "F")
self.out.branch("W_pt", "F")
self.out.branch("ll_dEta", "F")
self.out.branch("ll_dPhi", "F")
self.out.branch("ll_dR", "F")
self.out.branch("Wqq_mass", "F")
self.out.branch("Tlvb1_mass", "F")
self.out.branch("Tlvb2_mass", "F")
self.out.branch("Tqqb1_mass", "F")
self.out.branch("Tqqb2_mass", "F")
self.out.branch("jj_mass", "F")
self.out.branch("jj_pt", "F")
self.out.branch("jj_dEta", "F")
self.out.branch("jj_dPhi", "F")
self.out.branch("jj_dR", "F")
self.out.branch("nj_mass", "F")
self.out.branch("vis_mass", "F")
self.out.branch("lljj_mass", "F")
self.out.branch("minMuonIso", "F")
self.out.branch("maxMuonIso", "F")
self.out.branch("minMuonMetDPhi", "F")
self.out.branch("maxMuonMetDPhi", "F")
self.out.branch("minMuonJetDR", "F")
self.out.branch("HT30", "F")
self.out.branch("nj20", "I")
self.out.branch("nj30", "I")
self.out.branch("nj40", "I")
self.out.branch("nBJet", "I")
self.out.branch("CSV1", "F")
self.out.branch("CSV2", "F")
self.out.branch("CSV3", "F")
self.out.branch("CSV4", "F")
self.out.branch("iCSV1", "I")
self.out.branch("iCSV2", "I")
self.out.branch("iCSV3", "I")
self.out.branch("iCSV4", "I")
self.out.branch("lumiWeight", "F")
self.out.branch("lheWeight", "F")
self.out.branch("stitchWeight", "F")
self.out.branch("puWeight", "F")
self.out.branch("topWeight", "F")
self.out.branch("qcdnloWeight", "F")
self.out.branch("qcdnnloWeight", "F")
self.out.branch("ewknloWeight", "F")
self.out.branch("triggerWeight", "F")
self.out.branch("leptonWeight", "F")
self.out.branch("eventWeightLumi", "F")
self.fileName = inputFile.GetName()
self.sampleName = getNameFromFile(self.fileName)
self.isMC = not "Run201" in self.fileName
print "+ Opening file", self.fileName
# b-tagging working points for DeepCSV
# https://twiki.cern.ch/twiki/bin/viewauth/CMS/BtagRecommendation2016Legacy
# https://twiki.cern.ch/twiki/bin/viewauth/CMS/BtagRecommendation94X
# https://twiki.cern.ch/twiki/bin/viewauth/CMS/BtagRecommendation102X
if "Run2016" in self.fileName or "Summer16" in self.fileName:
self.year = 2016
self.lumi = 35920.
self.btagLoose = 0.2217 #0.0614
self.btagMedium = 0.6321 #0.3093
self.btagTight = 0.8953 #0.7221
elif "Run2017" in self.fileName or "Fall17" in self.fileName:
self.year = 2017
self.lumi = 41530.
self.btagLoose = 0.1522 #0.0521
self.btagMedium = 0.4941 #0.3033
self.btagTight = 0.8001 #0.7489
elif "Run2018" in self.fileName or "Autumn18" in self.fileName:
self.year = 2018
self.lumi = 59740.
self.btagLoose = 0.1241 #0.0494
self.btagMedium = 0.4184 #0.2770
self.btagTight = 0.7527 #0.7264
else:
print "- Unknown year, aborting module"
import sys
sys.exit()
self.xs = XS[self.sampleName] if self.sampleName in XS else 0.
self.nevents = EV[self.sampleName] if self.sampleName in EV else 0.
self.xsWeight = self.xs / self.nevents if self.nevents > 0 else 1.
self.lumiWeight = self.xsWeight * self.lumi if self.isMC else 1.
self.isLO = abs(
self.nevents % 1
) < 1.e-6 # if True, the event count is integer, so the weight should be normalized (+1)
print "+ Module parameters: isMC", self.isMC, ", year", self.year, ", lumi", self.lumi, "pb"
print "+ Sample", self.sampleName, (
"is LO" if self.isLO else
"is not LO"), ", XS", self.xs, ", events", self.nevents
print "+ Weight", self.lumiWeight
if self.isMC and self.isLO:
print "+ Sample is LO, gen weight will be set to 1"
# self.puTool = PileupWeightTool(year = year) if self.isMC else None
self.SingleMuIsoTriggers = [
"HLT_IsoMu24", "HLT_IsoTkMu24", "HLT_IsoMu27", "HLT_IsoTkMu27"
]
self.SingleMuTriggers = [
"HLT_Mu50", "HLT_TkMu50", "HLT_Mu100", "HLT_TkMu100"
]
self.DoubleMuonTriggers = [
"HLT_Mu17_Mu8", "HLT_Mu17_Mu8_DZ", "HLT_Mu17_TkMu8_DZ"
]
self.SingleEleIsoTriggers = [
"HLT_Ele27_WPTight_Gsf", "HLT_Ele27_WPLoose_Gsf",
"HLT_Ele32_WPTight_Gsf", "HLT_Ele35_WPTight_Gsf"
]
self.SingleEleTriggers = [
"HLT_Ele105_CaloIdVT_GsfTrkIdT", "HLT_Ele115_CaloIdVT_GsfTrkIdT",
"HLT_Photon175", "HLT_Photon200"
]
if self.isMC:
self.muSFs = MuonSFs(year=self.year)
self.elSFs = ElectronSFs(year=self.year)
self.puTool = PileupWeightTool(year=self.year)
# self.btagToolAK8 = BTagWeightTool('CSVv2','AK8','loose',sigma='central',channel='ll',year=year)
# self.btagToolAK4 = BTagWeightTool('CSVv2','AK4','loose',sigma='central',channel='ll',year=year)
# self.btagToolAK8_deep = BTagWeightTool('DeepCSV','AK8','loose',sigma='central',channel='ll',year=year)
# self.btagToolAK8_deep_up = BTagWeightTool('DeepCSV','AK8','loose',sigma='up',channel='ll',year=year)
# self.btagToolAK8_deep_down = BTagWeightTool('DeepCSV','AK8','loose',sigma='down',channel='ll',year=year)
# self.btagToolAK4_deep = BTagWeightTool('DeepCSV','AK4','loose',sigma='central',channel='ll',year=year)
# self.btagToolAK4_deep_up = BTagWeightTool('DeepCSV','AK4','loose',sigma='up',channel='ll',year=year)
# self.btagToolAK4_deep_down = BTagWeightTool('DeepCSV','AK4','loose',sigma='down',channel='ll',year=year)
if 'DYJetsToLL_M-50' in self.fileName:
self.VptCorr = DYCorrection('DYJetsToLL')
elif 'ZJetsToNuNu' in self.fileName:
self.VptCorr = DYCorrection('ZJetsToNuNu')
elif 'WJetsToLNu' in self.fileName:
self.VptCorr = DYCorrection('WJetsToLNu')
else:
self.VptCorr = None
class SameSign(Module):
def __init__(self):
self.writeHistFile = True
def beginJob(self, histFile=None, histDirName=None):
Module.beginJob(self, histFile, histDirName)
self.event = 0
self.hists = {}
self.hists["Events"] = ROOT.TH1F("Events", "Events", 1, 0, 1)
# self.addObject(self.h_events)
# self.h_events.SetDirectory
def endJob(self):
Module.endJob(self)
print "+ Module ended successfully," #, self.h_events.GetEntries(), "events analyzed"
pass
def beginFile(self, inputFile, outputFile, inputTree, wrappedOutputTree):
self.out = wrappedOutputTree
self.out.branch("iSkim", "I")
self.out.branch("isMC", "I")
self.out.branch("isSingleMuIsoTrigger", "I")
self.out.branch("isSingleMuTrigger", "I")
self.out.branch("isDoubleMuonTrigger", "I")
self.out.branch("isSingleEleIsoTrigger", "I")
self.out.branch("isSingleEleTrigger", "I")
self.out.branch("passedMETFilters", "I")
self.out.branch("nCleanElectron", "I")
self.out.branch("nCleanMuon", "I")
self.out.branch("nCleanJet", "I")
self.out.branch("Z_mass", "F")
self.out.branch("Z_pt", "F")
self.out.branch("W_mass", "F")
self.out.branch("W_tmass", "F")
self.out.branch("W_pt", "F")
self.out.branch("ll_dEta", "F")
self.out.branch("ll_dPhi", "F")
self.out.branch("ll_dR", "F")
self.out.branch("Wqq_mass", "F")
self.out.branch("Tlvb1_mass", "F")
self.out.branch("Tlvb2_mass", "F")
self.out.branch("Tqqb1_mass", "F")
self.out.branch("Tqqb2_mass", "F")
self.out.branch("jj_mass", "F")
self.out.branch("jj_pt", "F")
self.out.branch("jj_dEta", "F")
self.out.branch("jj_dPhi", "F")
self.out.branch("jj_dR", "F")
self.out.branch("nj_mass", "F")
self.out.branch("vis_mass", "F")
self.out.branch("lljj_mass", "F")
self.out.branch("minMuonIso", "F")
self.out.branch("maxMuonIso", "F")
self.out.branch("minMuonMetDPhi", "F")
self.out.branch("maxMuonMetDPhi", "F")
self.out.branch("minMuonJetDR", "F")
self.out.branch("HT30", "F")
self.out.branch("nj20", "I")
self.out.branch("nj30", "I")
self.out.branch("nj40", "I")
self.out.branch("nBJet", "I")
self.out.branch("CSV1", "F")
self.out.branch("CSV2", "F")
self.out.branch("CSV3", "F")
self.out.branch("CSV4", "F")
self.out.branch("iCSV1", "I")
self.out.branch("iCSV2", "I")
self.out.branch("iCSV3", "I")
self.out.branch("iCSV4", "I")
self.out.branch("lumiWeight", "F")
self.out.branch("lheWeight", "F")
self.out.branch("stitchWeight", "F")
self.out.branch("puWeight", "F")
self.out.branch("topWeight", "F")
self.out.branch("qcdnloWeight", "F")
self.out.branch("qcdnnloWeight", "F")
self.out.branch("ewknloWeight", "F")
self.out.branch("triggerWeight", "F")
self.out.branch("leptonWeight", "F")
self.out.branch("eventWeightLumi", "F")
self.fileName = inputFile.GetName()
self.sampleName = getNameFromFile(self.fileName)
self.isMC = not "Run201" in self.fileName
print "+ Opening file", self.fileName
# b-tagging working points for DeepCSV
# https://twiki.cern.ch/twiki/bin/viewauth/CMS/BtagRecommendation2016Legacy
# https://twiki.cern.ch/twiki/bin/viewauth/CMS/BtagRecommendation94X
# https://twiki.cern.ch/twiki/bin/viewauth/CMS/BtagRecommendation102X
if "Run2016" in self.fileName or "Summer16" in self.fileName:
self.year = 2016
self.lumi = 35920.
self.btagLoose = 0.2217 #0.0614
self.btagMedium = 0.6321 #0.3093
self.btagTight = 0.8953 #0.7221
elif "Run2017" in self.fileName or "Fall17" in self.fileName:
self.year = 2017
self.lumi = 41530.
self.btagLoose = 0.1522 #0.0521
self.btagMedium = 0.4941 #0.3033
self.btagTight = 0.8001 #0.7489
elif "Run2018" in self.fileName or "Autumn18" in self.fileName:
self.year = 2018
self.lumi = 59740.
self.btagLoose = 0.1241 #0.0494
self.btagMedium = 0.4184 #0.2770
self.btagTight = 0.7527 #0.7264
else:
print "- Unknown year, aborting module"
import sys
sys.exit()
self.xs = XS[self.sampleName] if self.sampleName in XS else 0.
self.nevents = EV[self.sampleName] if self.sampleName in EV else 0.
self.xsWeight = self.xs / self.nevents if self.nevents > 0 else 1.
self.lumiWeight = self.xsWeight * self.lumi if self.isMC else 1.
self.isLO = abs(
self.nevents % 1
) < 1.e-6 # if True, the event count is integer, so the weight should be normalized (+1)
print "+ Module parameters: isMC", self.isMC, ", year", self.year, ", lumi", self.lumi, "pb"
print "+ Sample", self.sampleName, (
"is LO" if self.isLO else
"is not LO"), ", XS", self.xs, ", events", self.nevents
print "+ Weight", self.lumiWeight
if self.isMC and self.isLO:
print "+ Sample is LO, gen weight will be set to 1"
# self.puTool = PileupWeightTool(year = year) if self.isMC else None
self.SingleMuIsoTriggers = [
"HLT_IsoMu24", "HLT_IsoTkMu24", "HLT_IsoMu27", "HLT_IsoTkMu27"
]
self.SingleMuTriggers = [
"HLT_Mu50", "HLT_TkMu50", "HLT_Mu100", "HLT_TkMu100"
]
self.DoubleMuonTriggers = [
"HLT_Mu17_Mu8", "HLT_Mu17_Mu8_DZ", "HLT_Mu17_TkMu8_DZ"
]
self.SingleEleIsoTriggers = [
"HLT_Ele27_WPTight_Gsf", "HLT_Ele27_WPLoose_Gsf",
"HLT_Ele32_WPTight_Gsf", "HLT_Ele35_WPTight_Gsf"
]
self.SingleEleTriggers = [
"HLT_Ele105_CaloIdVT_GsfTrkIdT", "HLT_Ele115_CaloIdVT_GsfTrkIdT",
"HLT_Photon175", "HLT_Photon200"
]
if self.isMC:
self.muSFs = MuonSFs(year=self.year)
self.elSFs = ElectronSFs(year=self.year)
self.puTool = PileupWeightTool(year=self.year)
# self.btagToolAK8 = BTagWeightTool('CSVv2','AK8','loose',sigma='central',channel='ll',year=year)
# self.btagToolAK4 = BTagWeightTool('CSVv2','AK4','loose',sigma='central',channel='ll',year=year)
# self.btagToolAK8_deep = BTagWeightTool('DeepCSV','AK8','loose',sigma='central',channel='ll',year=year)
# self.btagToolAK8_deep_up = BTagWeightTool('DeepCSV','AK8','loose',sigma='up',channel='ll',year=year)
# self.btagToolAK8_deep_down = BTagWeightTool('DeepCSV','AK8','loose',sigma='down',channel='ll',year=year)
# self.btagToolAK4_deep = BTagWeightTool('DeepCSV','AK4','loose',sigma='central',channel='ll',year=year)
# self.btagToolAK4_deep_up = BTagWeightTool('DeepCSV','AK4','loose',sigma='up',channel='ll',year=year)
# self.btagToolAK4_deep_down = BTagWeightTool('DeepCSV','AK4','loose',sigma='down',channel='ll',year=year)
if 'DYJetsToLL_M-50' in self.fileName:
self.VptCorr = DYCorrection('DYJetsToLL')
elif 'ZJetsToNuNu' in self.fileName:
self.VptCorr = DYCorrection('ZJetsToNuNu')
elif 'WJetsToLNu' in self.fileName:
self.VptCorr = DYCorrection('WJetsToLNu')
else:
self.VptCorr = None
def endFile(self, inputFile, outputFile, inputTree, wrappedOutputTree):
outputFile.mkdir("Hists")
outputFile.cd("Hists")
for histname, hist in self.hists.iteritems():
hist.Write()
outputFile.cd("..")
print "+ File closed successfully"
pass
def analyze(self, event):
iSkim = 0
eventWeightLumi, lheWeight, stitchWeight, puWeight, qcdnloWeight, qcdnnloWeight, ewknloWeight, topWeight = 1., 1., 1., 1., 1., 1., 1., 1.
triggerWeight, leptonWeight = 1., 1.
MinMuonIso, MaxMuonIso, MinMuonMetDPhi, MaxMuonMetDPhi, MinMuonJetDR = -1., -1., -1., -1., -1.
mZ, mW, mT, ptZ, ptW = -1., -1., -1., -1., -1.
dEtaLL, dPhiLL, dRLL = -1., -1., -1.
mJJ, ptJJ, dEtaJJ, dPhiJJ, dRJJ = -1., -1., -1., -1., -1.
mLLJJ, mNJ = -1., -1.
mHadW, mLepTop1, mLepTop2, mHadTop1, mHadTop2 = -1., -1., -1., -1., -1.
isSingleMuIsoTrigger, isSingleMuTrigger, isDoubleMuonTrigger, isSingleEleIsoTrigger, isSingleEleTrigger = False, False, False, False, False
for t in self.SingleMuIsoTriggers:
if hasattr(event, t) and getattr(event, t):
isSingleMuIsoTrigger = True
for t in self.SingleMuTriggers:
if hasattr(event, t) and getattr(event, t):
isSingleMuTrigger = True
for t in self.DoubleMuonTriggers:
if hasattr(event, t) and getattr(event, t):
isDoubleMuonTrigger = True
for t in self.SingleEleIsoTriggers:
if hasattr(event, t) and getattr(event, t):
isSingleEleIsoTrigger = True
for t in self.SingleEleTriggers:
if hasattr(event, t) and getattr(event, t):
isSingleEleTrigger = True
lheWeight = 1.
if self.isMC:
# Event weight
if not self.isLO and hasattr(event, "LHEWeight_originalXWGTUP"):
lheWeight = event.LHEWeight_originalXWGTUP
GenVpt = self.VptCorr.getGenVpt(
event) if not self.VptCorr is None else 0.
# MC stitching weight
# W+jets inclusive and exclusive
if 'WJetsToLNu_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8' in self.fileName and 'Summer16' in self.fileName:
if event.LHE_Vpt > 100.: stitchWeight = 0.
# Z+jets and Z+gamma
if self.fileName.startswith(
'ZGTo2LG_TuneCUETP8M1') or self.fileName.startswith(
'ZGToLLG_01J_5f_TuneCP5') or self.fileName.startswith(
'DYJetsToLL_M-50_Tune'):
nGenPhotons = 0
photonPtTh = 15. if self.fileName.startswith(
'ZGTo2LG_TuneCUETP8M1') or self.fileName.startswith(
'DYJetsToLL_M-50_TuneCUETP8M1') else 20.
for i in range(event.nGenPart):
if GenPart_pdgId[i] == 22 and TMath.Odd(
GenPart_statusFlags[i]
) and GenPart_pt > photonPtTh:
nGenPhotons += 1
if self.fileName.startswith('ZG') and nGenPhotons <= 0:
stitchWeight = 0.
if self.fileName.startswith(
'DYJetsToLL_M-50') and nGenPhotons >= 1:
stitchWeight = 0.
# PU weight
puWeight = self.puTool.getWeight(event.Pileup_nTrueInt)
# Higher order correction weights
if not self.VptCorr is None:
if not 'amcatnlo' in self.fileName:
qcdnloWeight = self.VptCorr.getWeightQCDNLO(GenVpt)
qcdnnloWeight = self.VptCorr.getWeightQCDNNLO(GenVpt)
ewknloWeight = self.VptCorr.getWeightEWKNLO(GenVpt)
if 'TTTo' in self.fileName:
Top1_pt, Top2_pt = getTTPt(event)
topWeight = getTTptWeight(Top1_pt, Top2_pt)
self.hists["Events"].Fill(0, lheWeight)
# electrons = Collection(event, "Electron")
# muons = Collection(event, "Muon")
# jets = Collection(event, "Jet")
# eventSum = ROOT.TLorentzVector()
# #select events with at least 2 muons
# if len(muons) >=2 :
# for lep in muons : #loop on muons
# eventSum += lep.p4()
# for lep in electrons : #loop on electrons
# eventSum += lep.p4()
# for j in jets : #loop on jets
# eventSum += j.p4()
# Clean collections
# Electrons
cleanElectron = []
for i in range(event.nElectron):
if event.Electron_pt[i] > 10. and event.Electron_cutBased[i] >= 2:
p4 = TLorentzVector()
p4.SetPtEtaPhiM(event.Electron_pt[i], event.Electron_eta[i],
event.Electron_phi[i], event.Electron_mass[i])
cleanElectron.append(p4)
# Muons
cleanMuon = []
for i in range(event.nMuon):
if event.Muon_pt[i] > 10. and event.Muon_mediumId[
i] and event.Muon_pfRelIso03_all[i] < 0.15:
p4 = TLorentzVector()
p4.SetPtEtaPhiM(event.Muon_pt[i], event.Muon_eta[i],
event.Muon_phi[i], event.Muon_mass[i])
cleanMuon.append(p4)
# Jets and Event variables
cleanJet, cleanBJet, cleanNonBJet = [], [], []
HT30, Nj20, Nj30, Nj40, nBJet = 0., 0, 0, 0, 0
CSVs = []
twoJets, allJets = TLorentzVector(), TLorentzVector()
for i in range(event.nJet):
if event.Jet_jetId[i] >= 6 and abs(event.Jet_eta[i]) < 2.5:
p4 = TLorentzVector()
p4.SetPtEtaPhiM(event.Jet_pt[i], event.Jet_eta[i],
event.Jet_phi[i], event.Jet_mass[i])
# remove overlaps with electrons and muons
cleanFromLeptons = True
for e in range(len(cleanElectron)):
if cleanElectron[e].DeltaR(p4) < 0.4:
cleanFromLeptons = False
for m in range(len(cleanMuon)):
if cleanMuon[m].DeltaR(p4) < 0.4: cleanFromLeptons = False
# fill variables
if cleanFromLeptons:
if event.Jet_pt[i] > 20.:
Nj20 += 1
if event.Jet_pt[i] > 30.:
HT30 += event.Jet_pt[i]
Nj30 += 1
cleanJet.append(p4)
if event.Jet_btagDeepB[i] >= self.btagMedium:
cleanBJet.append(p4)
else:
cleanNonBJet.append(p4)
CSVs.append([i, event.Jet_btagDeepB[i]])
if len(cleanJet) < 2: twoJets += p4
allJets += p4
if event.Jet_pt[i] > 40.:
Nj40 += 1
# b-tag ranking
nBJet = len(cleanBJet)
CSV1, CSV2, CSV3, CSV4, iCSV1, iCSV2, iCSV3, iCSV4 = 0., 0., 0., 0., -1, -1, -1, -1
CSVs.sort(key=lambda x: x[1], reverse=True)
if len(CSVs) > 0: iCSV1, CSV1 = CSVs[0][0], CSVs[0][1]
if len(CSVs) > 1: iCSV2, CSV2 = CSVs[1][0], CSVs[1][1]
if len(CSVs) > 2: iCSV3, CSV3 = CSVs[2][0], CSVs[2][1]
if len(CSVs) > 3: iCSV4, CSV4 = CSVs[3][0], CSVs[3][1]
if len(cleanJet) >= 2:
mJJ = (cleanJet[0] + cleanJet[1]).M()
ptJJ = (cleanJet[0] + cleanJet[1]).Pt()
dEtaJJ = abs(cleanJet[0].Eta() - cleanJet[1].Eta())
dPhiJJ = abs(cleanJet[0].DeltaPhi(cleanJet[1]))
dRJJ = cleanJet[0].DeltaR(cleanJet[1])
mNJ = allJets.M()
Lepton1, Lepton2, Lepton3, Neutrino, MET, LLJJ, Vis = TLorentzVector(
), TLorentzVector(), TLorentzVector(), TLorentzVector(
), TLorentzVector(), TLorentzVector(), TLorentzVector()
MET.SetPtEtaPhiM(event.MET_pt, 0., event.MET_phi, 0.)
# Categorization:
# iSkim = 1: 2 muons (OS or SS)
# iSkim = 2: 1 muon and 1 electron (no OS requirement)
# iSkim = 3: 3 muons (assumed that they are not 3 same-sign)
# iSkim = 4: 2 muons (OS) and 1 electron
# iSkim = 5: 1 muons, 3 jets, >= 1 btag
# iSkim = 6: 2 electrons
# case 3a: 3 lepton CR (3mu)
if iSkim == 0 and event.nMuon >= 3:
if (isSingleMuIsoTrigger or isSingleMuTrigger) and event.Muon_pt[
0] > 27. and event.Muon_pt[1] > 15. and event.Muon_pt[
2] > 15. and event.Muon_mediumId[
0] and event.Muon_mediumId[
1] and event.Muon_mediumId[2]:
Lepton1.SetPtEtaPhiM(event.Muon_pt[0], event.Muon_eta[0],
event.Muon_phi[0], event.Muon_mass[0])
Lepton2.SetPtEtaPhiM(event.Muon_pt[1], event.Muon_eta[1],
event.Muon_phi[1], event.Muon_mass[1])
Lepton3.SetPtEtaPhiM(event.Muon_pt[2], event.Muon_eta[2],
event.Muon_phi[2], event.Muon_mass[2])
if event.Muon_charge[0] * event.Muon_charge[1] < 0:
mll = (Lepton1 + Lepton2).M()
ptll = (Lepton1 + Lepton2).Pt()
detall = abs(Lepton1.Eta() - Lepton2.Eta())
dphill = abs(Lepton1.DeltaPhi(Lepton2))
drll = Lepton1.DeltaR(Lepton2)
else:
mll = (Lepton1 + Lepton3).M()
ptll = (Lepton1 + Lepton3).Pt()
detall = abs(Lepton1.Eta() - Lepton3.Eta())
dphill = abs(Lepton1.DeltaPhi(Lepton3))
drll = Lepton1.DeltaR(Lepton3)
if mll > 15.:
iSkim = 3
# Variables
mZ, ptZ = mll, ptll
dEtaLL, dPhiLL, dRLL = detall, dphill, drll
MinMuonIso = min(
min(event.Muon_pfRelIso03_all[0],
event.Muon_pfRelIso03_all[1]),
event.Muon_pfRelIso03_all[2])
MaxMuonIso = max(
max(event.Muon_pfRelIso03_all[0],
event.Muon_pfRelIso03_all[1]),
event.Muon_pfRelIso03_all[2])
MinMuonMetDPhi = min(
min(abs(Lepton1.DeltaPhi(MET)),
abs(Lepton2.DeltaPhi(MET))),
abs(Lepton3.DeltaPhi(MET)))
MaxMuonMetDPhi = max(
max(abs(Lepton1.DeltaPhi(MET)),
abs(Lepton2.DeltaPhi(MET))),
abs(Lepton3.DeltaPhi(MET)))
MinMuonJetDR = min(
min(getMinMuonJetDR(cleanJet, Lepton1),
getMinMuonJetDR(cleanJet, Lepton2)),
getMinMuonJetDR(cleanJet, Lepton3))
LLJJ = twoJets + Lepton1 + Lepton2
Vis = allJets + Lepton1 + Lepton2 + Lepton3
# Weights
if self.isMC:
triggerWeight = self.muSFs.getTriggerSF(
event.Muon_pt[0], event.Muon_eta[0])
IdSF1 = self.muSFs.getIdSF(event.Muon_pt[0],
event.Muon_eta[0], 2)
IsoSF1 = self.muSFs.getIsoSF(event.Muon_pt[0],
event.Muon_eta[0])
IdSF2 = self.muSFs.getIdSF(event.Muon_pt[1],
event.Muon_eta[1], 2)
IsoSF2 = self.muSFs.getIsoSF(event.Muon_pt[1],
event.Muon_eta[1])
IdSF3 = self.muSFs.getIdSF(event.Muon_pt[2],
event.Muon_eta[2], 2)
IsoSF3 = self.muSFs.getIsoSF(event.Muon_pt[2],
event.Muon_eta[2])
leptonWeight = IdSF1 * IsoSF1 * IdSF2 * IsoSF2 * IdSF3 * IsoSF3
# case 3b: 3 lepton CR (2mu 1e)
if iSkim == 0 and event.nMuon >= 2 and event.nElectron >= 1:
if (isSingleMuIsoTrigger or isSingleMuTrigger) and event.Muon_pt[
0] > 27. and event.Muon_pt[1] > 15. and event.Electron_pt[
0] > 15. and event.Muon_mediumId[
0] and event.Muon_mediumId[
1] and event.Electron_cutBased[
0] >= 2 and event.Muon_charge[
0] * event.Muon_charge[1] < 0:
Lepton1.SetPtEtaPhiM(event.Muon_pt[0], event.Muon_eta[0],
event.Muon_phi[0], event.Muon_mass[0])
Lepton2.SetPtEtaPhiM(event.Muon_pt[1], event.Muon_eta[1],
event.Muon_phi[1], event.Muon_mass[1])
Lepton3.SetPtEtaPhiM(event.Electron_pt[0],
event.Electron_eta[0],
event.Electron_phi[0],
event.Electron_mass[0])
mll = (Lepton1 + Lepton2).M()
ptll = (Lepton1 + Lepton2).Pt()
if mll > 15.:
iSkim = 4
# Variables
mZ, ptZ = mll, ptll
dEtaLL = abs(Lepton1.Eta() - Lepton2.Eta())
dPhiLL = abs(Lepton1.DeltaPhi(Lepton2))
dRLL = Lepton1.DeltaR(Lepton2)
MinMuonIso = min(event.Muon_pfRelIso03_all[0],
event.Muon_pfRelIso03_all[1])
MaxMuonIso = max(event.Muon_pfRelIso03_all[0],
event.Muon_pfRelIso03_all[1])
MinMuonMetDPhi = min(
min(abs(Lepton1.DeltaPhi(MET)),
abs(Lepton2.DeltaPhi(MET))),
abs(Lepton3.DeltaPhi(MET)))
MaxMuonMetDPhi = max(
max(abs(Lepton1.DeltaPhi(MET)),
abs(Lepton2.DeltaPhi(MET))),
abs(Lepton3.DeltaPhi(MET)))
MinMuonJetDR = min(getMinMuonJetDR(cleanJet, Lepton1),
getMinMuonJetDR(cleanJet, Lepton2))
LLJJ = twoJets + Lepton1 + Lepton2
Vis = allJets + Lepton1 + Lepton2 + Lepton3
# Weights
if self.isMC:
triggerWeight = self.muSFs.getTriggerSF(
event.Muon_pt[0], event.Muon_eta[0])
IdSF1 = self.muSFs.getIdSF(event.Muon_pt[0],
event.Muon_eta[0], 2)
IsoSF1 = self.muSFs.getIsoSF(event.Muon_pt[0],
event.Muon_eta[0])
IdSF2 = self.muSFs.getIdSF(event.Muon_pt[1],
event.Muon_eta[1], 2)
IsoSF2 = self.muSFs.getIsoSF(event.Muon_pt[1],
event.Muon_eta[1])
IdIsoSF3 = self.elSFs.getIdIsoSF(
event.Electron_pt[0], event.Electron_eta[0])
leptonWeight = IdSF1 * IsoSF1 * IdSF2 * IsoSF2 * IdIsoSF3
# case 1: Z->mumu CR (2 mu)
if iSkim == 0 and event.nMuon >= 2:
if (isSingleMuIsoTrigger or isSingleMuTrigger) and event.Muon_pt[
0] > 27. and event.Muon_pt[1] > 7. and event.Muon_mediumId[
0] and event.Muon_mediumId[1]:
Lepton1.SetPtEtaPhiM(event.Muon_pt[0], event.Muon_eta[0],
event.Muon_phi[0], event.Muon_mass[0])
Lepton2.SetPtEtaPhiM(event.Muon_pt[1], event.Muon_eta[1],
event.Muon_phi[1], event.Muon_mass[1])
mll = (Lepton1 + Lepton2).M()
ptll = (Lepton1 + Lepton2).Pt()
if mll > 15.:
iSkim = 1
# Variables
mZ, ptZ = mll, ptll
dEtaLL = abs(Lepton1.Eta() - Lepton2.Eta())
dPhiLL = abs(Lepton1.DeltaPhi(Lepton2))
dRLL = Lepton1.DeltaR(Lepton2)
MinMuonIso = min(event.Muon_pfRelIso03_all[0],
event.Muon_pfRelIso03_all[1])
MaxMuonIso = max(event.Muon_pfRelIso03_all[0],
event.Muon_pfRelIso03_all[1])
MinMuonMetDPhi = min(abs(Lepton1.DeltaPhi(MET)),
abs(Lepton2.DeltaPhi(MET)))
MaxMuonMetDPhi = max(abs(Lepton1.DeltaPhi(MET)),
abs(Lepton2.DeltaPhi(MET)))
MinMuonJetDR = min(getMinMuonJetDR(cleanJet, Lepton1),
getMinMuonJetDR(cleanJet, Lepton2))
LLJJ = twoJets + Lepton1 + Lepton2
Vis = allJets + Lepton1 + Lepton2
# Weights
if self.isMC:
triggerWeight = self.muSFs.getTriggerSF(
event.Muon_pt[0], event.Muon_eta[0])
IdSF1 = self.muSFs.getIdSF(event.Muon_pt[0],
event.Muon_eta[0], 2)
IdSF2 = self.muSFs.getIdSF(event.Muon_pt[1],
event.Muon_eta[1], 2)
IsoSF1 = self.muSFs.getIsoSF(event.Muon_pt[0],
event.Muon_eta[0])
IsoSF2 = self.muSFs.getIsoSF(event.Muon_pt[1],
event.Muon_eta[1])
leptonWeight = IdSF1 * IdSF2 * IsoSF1 * IsoSF2
# case 4: Z->ee CR (2 electrons)
if iSkim == 0 and event.nElectron >= 2:
if isSingleEleIsoTrigger and event.Electron_pt[
0] > 35. and event.Electron_pt[
1] > 10. and event.Electron_cutBased[
0] > 0 and event.Electron_cutBased[1] > 0:
Lepton1.SetPtEtaPhiM(event.Electron_pt[0],
event.Electron_eta[0],
event.Electron_phi[0],
event.Electron_mass[0])
Lepton2.SetPtEtaPhiM(event.Electron_pt[1],
event.Electron_eta[1],
event.Electron_phi[1],
event.Electron_mass[1])
mll = (Lepton1 + Lepton2).M()
ptll = (Lepton1 + Lepton2).Pt()
if mll > 15.:
iSkim = 6
# Variables
mZ, ptZ = mll, ptll
dEtaLL = abs(Lepton1.Eta() - Lepton2.Eta())
dPhiLL = abs(Lepton1.DeltaPhi(Lepton2))
dRLL = Lepton1.DeltaR(Lepton2)
MinMuonIso = event.Electron_pfRelIso03_all[0]
MaxMuonIso = event.Electron_pfRelIso03_all[0]
MinMuonMetDPhi = min(abs(Lepton1.DeltaPhi(MET)),
abs(Lepton2.DeltaPhi(MET)))
MaxMuonMetDPhi = max(abs(Lepton1.DeltaPhi(MET)),
abs(Lepton2.DeltaPhi(MET)))
MinMuonJetDR = min(getMinMuonJetDR(cleanJet, Lepton1),
getMinMuonJetDR(cleanJet, Lepton2))
LLJJ = twoJets + Lepton1 + Lepton2
Vis = allJets + Lepton1 + Lepton2
# Weights
if self.isMC:
triggerWeight = self.elSFs.getTriggerSF(
event.Electron_pt[0], event.Electron_eta[0])
leptonWeight = self.elSFs.getIdIsoSF(
event.Electron_pt[0],
event.Electron_eta[0]) * self.elSFs.getIdIsoSF(
event.Electron_pt[1], event.Electron_eta[1])
# case 2: ttbar and Z OF CR (1 muon and 1 electron)
if iSkim == 0 and event.nMuon >= 1 and event.nElectron >= 1:
if (isSingleMuIsoTrigger or isSingleMuTrigger
) and event.Muon_pt[0] > 27. and event.Electron_pt[
0] > 20. and event.Muon_mediumId[
0] and event.Electron_cutBased[0] >= 2:
Lepton1.SetPtEtaPhiM(event.Muon_pt[0], event.Muon_eta[0],
event.Muon_phi[0], event.Muon_mass[0])
Lepton2.SetPtEtaPhiM(event.Electron_pt[0],
event.Electron_eta[0],
event.Electron_phi[0],
event.Electron_mass[0])
mll = (Lepton1 + Lepton2).M()
ptll = (Lepton1 + Lepton2).Pt()
if mll > 15.:
iSkim = 2
# Variables
mZ, ptZ = mll, ptll
dEtaLL = abs(Lepton1.Eta() - Lepton2.Eta())
dPhiLL = abs(Lepton1.DeltaPhi(Lepton2))
dRLL = Lepton1.DeltaR(Lepton2)
MinMuonIso = event.Muon_pfRelIso03_all[0]
MaxMuonIso = event.Muon_pfRelIso03_all[0]
MinMuonMetDPhi = abs(Lepton1.DeltaPhi(MET))
MaxMuonMetDPhi = abs(Lepton1.DeltaPhi(MET))
MinMuonJetDR = min(getMinMuonJetDR(cleanJet, Lepton1),
getMinMuonJetDR(cleanJet, Lepton2))
LLJJ = twoJets + Lepton1 + Lepton2
Vis = allJets + Lepton1 + Lepton2
# Weights
if self.isMC:
triggerWeight = self.muSFs.getTriggerSF(
Lepton1.Pt(), Lepton1.Eta())
IdSF1 = self.muSFs.getIdSF(event.Muon_pt[0],
event.Muon_eta[0], 2)
IsoSF1 = self.muSFs.getIsoSF(event.Muon_pt[0],
event.Muon_eta[0])
IdIsoSF2 = self.elSFs.getIdIsoSF(
event.Electron_pt[0], event.Electron_eta[0])
leptonWeight = IdSF1 * IsoSF1 * IdIsoSF2
# case 4: ttbar CR (1 muon and >= 3 jets)
if iSkim == 0 and event.nMuon >= 1:
if (isSingleMuIsoTrigger or isSingleMuTrigger
) and event.Muon_pt[0] > 27. and event.Muon_mediumId[
0] and event.Muon_pfRelIso03_all[0] < 0.15:
Lepton1.SetPtEtaPhiM(event.Muon_pt[0], event.Muon_eta[0],
event.Muon_phi[0], event.Muon_mass[0])
pzN = recoverNeutrinoPz(Lepton1, MET)
Neutrino.SetPxPyPzE(MET.Px(), MET.Py(), pzN, MET.Pt())
mW = (Lepton1 + Neutrino).M()
ptW = (Lepton1 + Neutrino).Pt()
mT = math.sqrt(2. * Lepton1.Pt() * MET.Pt() *
(1. - math.cos(Lepton1.DeltaPhi(MET))))
if Nj30 >= 3: # and nBJet >= 1:
iSkim = 5
# Variables
MinMuonIso = event.Muon_pfRelIso03_all[0]
MaxMuonIso = event.Muon_pfRelIso03_all[0]
MinMuonMetDPhi = abs(Lepton1.DeltaPhi(MET))
MaxMuonMetDPhi = abs(Lepton1.DeltaPhi(MET))
MinMuonJetDR = getMinMuonJetDR(cleanJet, Lepton1)
LLJJ = twoJets + Lepton1
Vis = allJets + Lepton1
# W and Top reconstruction
if len(cleanBJet) >= 2:
mLepTop1 = (Lepton1 + Neutrino + cleanBJet[0]).M()
mLepTop2 = (Lepton1 + Neutrino + cleanBJet[1]).M()
elif len(cleanBJet) >= 1:
mLepTop1 = (Lepton1 + Neutrino + cleanBJet[0]).M()
mHadTop2 = mHadTop1
if len(cleanNonBJet) >= 2:
mHadW = (cleanNonBJet[0] + cleanNonBJet[1]).M()
if len(cleanBJet) >= 2:
mHadTop1 = (cleanNonBJet[0] + cleanNonBJet[1] +
cleanBJet[0]).M()
mHadTop2 = (cleanNonBJet[0] + cleanNonBJet[1] +
cleanBJet[1]).M()
elif len(cleanBJet) >= 1:
mHadTop1 = (cleanNonBJet[0] + cleanNonBJet[1] +
cleanBJet[0]).M()
mHadTop2 = mHadTop1
# Weights
if self.isMC:
triggerWeight = self.muSFs.getTriggerSF(
Lepton1.Pt(), Lepton1.Eta())
IdSF1 = self.muSFs.getIdSF(event.Muon_pt[0],
event.Muon_eta[0], 2)
IsoSF1 = self.muSFs.getIsoSF(event.Muon_pt[0],
event.Muon_eta[0])
leptonWeight = IdSF1 * IsoSF1
passedMETFilters = True
filters = [
"Flag_goodVertices", "Flag_globalSuperTightHalo2016Filter",
"Flag_BadPFMuonFilter", "Flag_EcalDeadCellTriggerPrimitiveFilter",
"Flag_HBHENoiseFilter", "Flag_HBHENoiseIsoFilter",
"Flag_ecalBadCalibFilter", "Flag_ecalBadCalibFilterV2"
]
if not self.isMC: filters += ["Flag_eeBadScFilter"]
for f in filters:
if hasattr(event, f) and getattr(event, f) == False:
passedMETFilters = False
# try:
## if event.Flag_goodVertices: print "Flag_goodVertices"
## if event.Flag_globalSuperTightHalo2016Filter: print "Flag_globalSuperTightHalo2016Filter"
## if event.Flag_BadPFMuonFilter: print "Flag_BadPFMuonFilter"
## if event.Flag_EcalDeadCellTriggerPrimitiveFilter: print "Flag_EcalDeadCellTriggerPrimitiveFilter"
## if event.Flag_HBHENoiseFilter: print "Flag_HBHENoiseFilter"
## if event.Flag_HBHENoiseIsoFilter: print "Flag_HBHENoiseIsoFilter"
### if (self.isMC or event.Flag_eeBadScFilter): print "Flag_eeBadScFilter"
## if event.Flag_ecalBadCalibFilter: print "Flag_ecalBadCalibFilterV2"
# if event.Flag_goodVertices and event.Flag_globalSuperTightHalo2016Filter and event.Flag_BadPFMuonFilter and event.Flag_EcalDeadCellTriggerPrimitiveFilter and event.Flag_HBHENoiseFilter and event.Flag_HBHENoiseIsoFilter: # and event.Flag_ecalBadCalibFilter: #and (self.isMC or event.Flag_eeBadScFilter): FIXME
# passedMETFilters = True
## if not self.isMC:
## if not event.Flag_eeBadScFilter:
## passedMETFilters = False
# except:
# passedMETFilters = False
if iSkim == 0: return False
if self.isMC:
eventWeightLumi = self.lumiWeight * lheWeight * puWeight * topWeight * qcdnloWeight * qcdnnloWeight * ewknloWeight * triggerWeight * leptonWeight
self.out.fillBranch("iSkim", iSkim)
self.out.fillBranch("isMC", self.isMC)
self.out.fillBranch("isSingleMuIsoTrigger", isSingleMuIsoTrigger)
self.out.fillBranch("isSingleMuTrigger", isSingleMuTrigger)
self.out.fillBranch("isDoubleMuonTrigger", isDoubleMuonTrigger)
self.out.fillBranch("isSingleEleIsoTrigger", isSingleEleIsoTrigger)
self.out.fillBranch("isSingleEleTrigger", isSingleEleTrigger)
self.out.fillBranch("passedMETFilters", passedMETFilters)
self.out.fillBranch("nCleanElectron", len(cleanElectron))
self.out.fillBranch("nCleanMuon", len(cleanMuon))
self.out.fillBranch("nCleanJet", len(cleanJet))
self.out.fillBranch("Z_mass", mZ)
self.out.fillBranch("Z_pt", ptZ)
self.out.fillBranch("W_mass", mW)
self.out.fillBranch("W_tmass", mT)
self.out.fillBranch("W_pt", ptW)
self.out.fillBranch("ll_dEta", dEtaLL)
self.out.fillBranch("ll_dPhi", dPhiLL)
self.out.fillBranch("ll_dR", dRLL)
self.out.fillBranch("Wqq_mass", mHadW)
self.out.fillBranch("Tlvb1_mass", mLepTop1)
self.out.fillBranch("Tlvb2_mass", mLepTop2)
self.out.fillBranch("Tqqb1_mass", mHadTop1)
self.out.fillBranch("Tqqb2_mass", mHadTop2)
self.out.fillBranch("jj_mass", mJJ)
self.out.fillBranch("jj_pt", ptJJ)
self.out.fillBranch("jj_dEta", dEtaJJ)
self.out.fillBranch("jj_dPhi", dPhiJJ)
self.out.fillBranch("jj_dR", dRJJ)
self.out.fillBranch("nj_mass", mNJ)
self.out.fillBranch("vis_mass", Vis.M())
self.out.fillBranch("lljj_mass", LLJJ.M())
self.out.fillBranch("minMuonIso", MinMuonIso)
self.out.fillBranch("maxMuonIso", MaxMuonIso)
self.out.fillBranch("minMuonMetDPhi", MinMuonMetDPhi)
self.out.fillBranch("maxMuonMetDPhi", MaxMuonMetDPhi)
self.out.fillBranch("minMuonJetDR", MinMuonJetDR)
self.out.fillBranch("HT30", HT30)
self.out.fillBranch("nj20", Nj20)
self.out.fillBranch("nj30", Nj30)
self.out.fillBranch("nj40", Nj40)
self.out.fillBranch("nBJet", nBJet)
self.out.fillBranch("CSV1", CSV1)
self.out.fillBranch("CSV2", CSV2)
self.out.fillBranch("CSV3", CSV3)
self.out.fillBranch("CSV4", CSV4)
self.out.fillBranch("iCSV1", iCSV1)
self.out.fillBranch("iCSV2", iCSV2)
self.out.fillBranch("iCSV3", iCSV3)
self.out.fillBranch("iCSV4", iCSV4)
self.out.fillBranch("lumiWeight", self.lumiWeight)
self.out.fillBranch("lheWeight", lheWeight)
self.out.fillBranch("stitchWeight", stitchWeight)
self.out.fillBranch("puWeight", puWeight)
self.out.fillBranch("topWeight", topWeight)
self.out.fillBranch("qcdnloWeight", qcdnloWeight)
self.out.fillBranch("qcdnnloWeight", qcdnnloWeight)
self.out.fillBranch("ewknloWeight", ewknloWeight)
self.out.fillBranch("triggerWeight", triggerWeight)
self.out.fillBranch("leptonWeight", leptonWeight)
self.out.fillBranch("eventWeightLumi", eventWeightLumi)
return True
def __init__(self, name, dataType, **kwargs):
self.name = name
self.out = TreeProducerEleMu(name)
self.isData = dataType == 'data'
self.year = kwargs.get('year', 2017)
self.tes = kwargs.get('tes', 1.0)
self.ltf = kwargs.get('ltf', 1.0)
self.doZpt = kwargs.get('doZpt', 'DY' in name)
self.doRecoil = kwargs.get(
'doRecoil', ('DY' in name or re.search(r"W\d?Jets", name))
and self.year > 2016)
self.doTTpt = kwargs.get('doTTpt', 'TT' in name)
self.doTight = kwargs.get('doTight', self.tes != 1 or self.ltf != 1)
self.channel = 'elemu'
year, channel = self.year, self.channel
if year == 2016:
self.trigger = lambda e: e.HLT_IsoMu22 or e.HLT_IsoMu22_eta2p1 or e.HLT_IsoTkMu22 or e.HLT_IsoTkMu22_eta2p1 #or e.HLT_IsoMu19_eta2p1_LooseIsoPFTau20_SingleL1
self.muonCutPt = lambda e: 23
elif year == 2017:
self.trigger = lambda e: e.HLT_IsoMu24 or e.HLT_IsoMu27 #or e.HLT_IsoMu20_eta2p1_LooseChargedIsoPFTau27_eta2p1_CrossL1
self.muonCutPt = lambda e: 25 if e.HLT_IsoMu24 else 28
else:
self.trigger = lambda e: e.HLT_IsoMu24 or e.HLT_IsoMu27 #or e.HLT_IsoMu20_eta2p1_LooseChargedIsoPFTau27_eta2p1_CrossL1
self.muonCutPt = lambda e: 25
self.eleCutPt = 15
if not self.isData:
self.eleSFs = ElectronSFs(year=year)
self.muonSFs = MuonSFs(year=year)
self.puTool = PileupWeightTool(year=year)
self.btagTool = BTagWeightTool('DeepCSV',
'medium',
channel='mutau',
year=year)
if self.doZpt:
self.zptTool = ZptCorrectionTool(year=year)
if self.doRecoil:
self.recoilTool = RecoilCorrectionTool(year=year)
self.deepcsv_wp = BTagWPs('DeepCSV', year=year)
self.Nocut = 0
self.Trigger = 1
self.GoodMuons = 2
self.GoodElectrons = 3
self.GoodDiLepton = 4
self.TotalWeighted = 15
self.TotalWeighted_no0PU = 16
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.Nocut, "no cut")
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.Trigger, "trigger")
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.GoodMuons,
"muon object")
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.GoodElectrons,
"electron object")
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.GoodDiLepton,
"mumu pair")
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.TotalWeighted,
"no cut, weighted")
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.TotalWeighted_no0PU,
"no cut, weighted, PU>0")
self.out.cutflow.GetXaxis().SetLabelSize(0.041)
class EleMuProducer(Module):
def __init__(self, name, dataType, **kwargs):
self.name = name
self.out = TreeProducerEleMu(name)
self.isData = dataType == 'data'
self.year = kwargs.get('year', 2017)
self.tes = kwargs.get('tes', 1.0)
self.ltf = kwargs.get('ltf', 1.0)
self.doZpt = kwargs.get('doZpt', 'DY' in name)
self.doRecoil = kwargs.get(
'doRecoil', ('DY' in name or re.search(r"W\d?Jets", name))
and self.year > 2016)
self.doTTpt = kwargs.get('doTTpt', 'TT' in name)
self.doTight = kwargs.get('doTight', self.tes != 1 or self.ltf != 1)
self.channel = 'elemu'
year, channel = self.year, self.channel
if year == 2016:
self.trigger = lambda e: e.HLT_IsoMu22 or e.HLT_IsoMu22_eta2p1 or e.HLT_IsoTkMu22 or e.HLT_IsoTkMu22_eta2p1 #or e.HLT_IsoMu19_eta2p1_LooseIsoPFTau20_SingleL1
self.muonCutPt = lambda e: 23
elif year == 2017:
self.trigger = lambda e: e.HLT_IsoMu24 or e.HLT_IsoMu27 #or e.HLT_IsoMu20_eta2p1_LooseChargedIsoPFTau27_eta2p1_CrossL1
self.muonCutPt = lambda e: 25 if e.HLT_IsoMu24 else 28
else:
self.trigger = lambda e: e.HLT_IsoMu24 or e.HLT_IsoMu27 #or e.HLT_IsoMu20_eta2p1_LooseChargedIsoPFTau27_eta2p1_CrossL1
self.muonCutPt = lambda e: 25
self.eleCutPt = 15
if not self.isData:
self.eleSFs = ElectronSFs(year=year)
self.muonSFs = MuonSFs(year=year)
self.puTool = PileupWeightTool(year=year)
self.btagTool = BTagWeightTool('DeepCSV',
'medium',
channel='mutau',
year=year)
if self.doZpt:
self.zptTool = ZptCorrectionTool(year=year)
if self.doRecoil:
self.recoilTool = RecoilCorrectionTool(year=year)
self.deepcsv_wp = BTagWPs('DeepCSV', year=year)
self.Nocut = 0
self.Trigger = 1
self.GoodMuons = 2
self.GoodElectrons = 3
self.GoodDiLepton = 4
self.TotalWeighted = 15
self.TotalWeighted_no0PU = 16
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.Nocut, "no cut")
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.Trigger, "trigger")
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.GoodMuons,
"muon object")
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.GoodElectrons,
"electron object")
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.GoodDiLepton,
"mumu pair")
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.TotalWeighted,
"no cut, weighted")
self.out.cutflow.GetXaxis().SetBinLabel(1 + self.TotalWeighted_no0PU,
"no cut, weighted, PU>0")
self.out.cutflow.GetXaxis().SetLabelSize(0.041)
def beginJob(self):
pass
def endJob(self):
if not self.isData:
self.btagTool.setDirectory(self.out.outputfile, 'btag')
self.out.endJob()
def beginFile(self, inputFile, outputFile, inputTree, wrappedOutputTree):
sys.stdout.flush()
checkBranches(inputTree)
def endFile(self, inputFile, outputFile, inputTree, wrappedOutputTree):
pass
def analyze(self, event):
"""process event, return True (go to next module) or False (fail, go to next event)"""
sys.stdout.flush()
#####################################
self.out.cutflow.Fill(self.Nocut)
if self.isData:
self.out.cutflow.Fill(self.TotalWeighted, 1.)
if event.PV_npvs > 0:
self.out.cutflow.Fill(self.TotalWeighted_no0PU, 1.)
else:
return False
else:
self.out.cutflow.Fill(self.TotalWeighted, event.genWeight)
self.out.pileup.Fill(event.Pileup_nTrueInt)
if event.Pileup_nTrueInt > 0:
self.out.cutflow.Fill(self.TotalWeighted_no0PU,
event.genWeight)
else:
return False
#####################################
if not self.trigger(event):
return False
#####################################
self.out.cutflow.Fill(self.Trigger)
#####################################
idx_goodmuons = []
for imuon in range(event.nMuon):
if event.Muon_pt[imuon] < self.muonCutPt(event): continue
if abs(event.Muon_eta[imuon]) > 2.4: continue
if abs(event.Muon_dz[imuon]) > 0.2: continue
if abs(event.Muon_dxy[imuon]) > 0.045: continue
if not event.Muon_mediumId[imuon]: continue
idx_goodmuons.append(imuon)
if len(idx_goodmuons) == 0:
return False
#####################################
self.out.cutflow.Fill(self.GoodMuons)
#####################################
idx_goodelectrons = []
for ielectron in range(event.nElectron):
if event.Electron_pt[ielectron] < self.eleCutPt: continue
if abs(event.Electron_eta[ielectron]) > 2.1: continue
if abs(event.Electron_dz[ielectron]) > 0.2: continue
if abs(event.Electron_dxy[ielectron]) > 0.045: continue
if event.Electron_convVeto[ielectron] != 1: continue
if ord(event.Electron_lostHits[ielectron]) > 1: continue
#if event.Electron_mvaFall17V2Iso_WP80[ielectron] < 0.5: continue
idx_goodelectrons.append(ielectron)
if len(idx_goodelectrons) == 0:
return False
#####################################
self.out.cutflow.Fill(self.GoodElectrons)
#####################################
muons = Collection(event, 'Muon')
electrons = Collection(event, 'Electron')
dileptons = []
for idx1 in idx_goodelectrons:
for idx2 in idx_goodmuons:
if electrons[idx1].p4().DeltaR(muons[idx2].p4()) < 0.5:
continue
dilepton = DiLeptonBasicClass(
idx1, event.Electron_pt[idx1],
event.Electron_pfRelIso03_all[idx1], idx2,
event.Muon_pt[idx2], event.Muon_pfRelIso04_all[idx2])
dileptons.append(dilepton)
if len(dileptons) == 0:
return False
dilepton = bestDiLepton(dileptons)
electron = electrons[dilepton.id1].p4()
muon = muons[dilepton.id2].p4()
#####################################
self.out.cutflow.Fill(self.GoodDiLepton)
#####################################
# VETOS
self.out.extramuon_veto[0], self.out.extraelec_veto[
0], self.out.dilepton_veto[0] = extraLeptonVetos(
event, [dilepton.id2], [dilepton.id1], self.channel)
self.out.lepton_vetos[0] = self.out.extramuon_veto[
0] or self.out.extraelec_veto[0] or self.out.dilepton_veto[0]
###if self.doTight and (self.out.lepton_vetos[0] or event.Electron_pfRelIso03_all[dilepton.id1]>0.10 or\
### ord(event.Tau_idAntiMu[dilepton.id2]<1 or ord(event.Tau_idAntiEle[dilepton.id2]<8):
### return False
# JETS
jetIds = []
bjetIds = []
jets = Collection(event, 'Jet')
nfjets = 0
ncjets = 0
nbtag = 0
for ijet in range(event.nJet):
if event.Jet_pt[ijet] < 20:
continue # 20 for tau -> j fake measurement
if abs(event.Jet_eta[ijet]) > 4.7: continue
if muon.DeltaR(jets[ijet].p4()) < 0.5: continue
if electron.DeltaR(jets[ijet].p4()) < 0.5: continue
jetIds.append(ijet)
if abs(event.Jet_eta[ijet]) > 2.4:
nfjets += 1
else:
ncjets += 1
if event.Jet_btagDeepB[ijet] > self.deepcsv_wp.medium:
nbtag += 1
bjetIds.append(ijet)
if not self.isData and event.Electron_pfRelIso03_all[
dilepton.id1] < 0.50 and event.Muon_pfRelIso04_all[
dilepton.id2] < 0.50:
self.btagTool.fillEfficiencies(event, jetIds)
# EVENT
self.out.isData[0] = self.isData
self.out.run[0] = event.run
self.out.lumi[0] = event.luminosityBlock
self.out.event[0] = event.event & 0xffffffffffffffff
###self.out.puppimet[0] = event.PuppiMET_pt
###self.out.puppimetphi[0] = event.PuppiMET_phi
###self.out.metsignificance[0] = event.MET_significance
###self.out.metcovXX[0] = event.MET_covXX
###self.out.metcovXY[0] = event.MET_covXY
###self.out.metcovYY[0] = event.MET_covYY
self.out.npvs[0] = event.PV_npvs
self.out.npvsGood[0] = event.PV_npvsGood
if not self.isData:
self.out.genPartFlav_1[0] = ord(
event.Electron_genPartFlav[dilepton.id1])
self.out.genPartFlav_2[0] = ord(
event.Muon_genPartFlav[dilepton.id2])
self.out.genmet[0] = event.GenMET_pt
self.out.genmetphi[0] = event.GenMET_phi
self.out.nPU[0] = event.Pileup_nPU
self.out.nTrueInt[0] = event.Pileup_nTrueInt
try:
self.out.LHE_Njets[0] = event.LHE_Njets
except RuntimeError:
self.out.LHE_Njets[0] = -1
# ELECTRON
self.out.pt_1[0] = event.Electron_pt[dilepton.id1]
self.out.eta_1[0] = event.Electron_eta[dilepton.id1]
self.out.phi_1[0] = event.Electron_phi[dilepton.id1]
self.out.m_1[0] = event.Electron_mass[dilepton.id1]
self.out.dxy_1[0] = event.Electron_dxy[dilepton.id1]
self.out.dz_1[0] = event.Electron_dz[dilepton.id1]
self.out.q_1[0] = event.Electron_charge[dilepton.id1]
self.out.pfRelIso03_all_1[0] = event.Electron_pfRelIso03_all[
dilepton.id1]
self.out.cutBased_1[0] = event.Electron_cutBased[dilepton.id1]
self.out.mvaFall17Iso_1[0] = event.Electron_mvaFall17V2Iso[
dilepton.id1]
self.out.mvaFall17Iso_WPL_1[0] = event.Electron_mvaFall17V2Iso_WPL[
dilepton.id1]
self.out.mvaFall17Iso_WP80_1[0] = event.Electron_mvaFall17V2Iso_WP80[
dilepton.id1]
self.out.mvaFall17Iso_WP90_1[0] = event.Electron_mvaFall17V2Iso_WP90[
dilepton.id1]
# MUON
self.out.pt_2[0] = event.Muon_pt[dilepton.id2]
self.out.eta_2[0] = event.Muon_eta[dilepton.id2]
self.out.phi_2[0] = event.Muon_phi[dilepton.id2]
self.out.m_2[0] = event.Muon_mass[dilepton.id2]
self.out.dxy_2[0] = event.Muon_dxy[dilepton.id2]
self.out.dz_2[0] = event.Muon_dz[dilepton.id2]
self.out.q_2[0] = event.Muon_charge[dilepton.id2]
self.out.pfRelIso04_all_2[0] = event.Muon_pfRelIso04_all[dilepton.id2]
# TAU for jet -> tau fake control region
maxId = -1
maxPt = 20
taus = Collection(event, 'Tau')
for itau in range(event.nTau):
if event.Tau_pt[itau] < maxPt: continue
if electron.DeltaR(taus[itau].p4()) < 0.5: continue
if muon.DeltaR(taus[itau].p4()) < 0.5: continue
if abs(event.Tau_eta[itau]) > 2.3: continue
if abs(event.Tau_dz[itau]) > 0.2: continue
if event.Tau_decayMode[itau] not in [0, 1, 10, 11]: continue
if abs(event.Tau_charge[itau]) != 1: continue
if ord(event.Tau_idAntiEle[itau]) < 1: continue # VLoose
if ord(event.Tau_idAntiMu[itau]) < 1: continue # Loose
#if not self.vlooseIso(event,itau): continue
maxId = itau
maxPt = event.Tau_pt[itau]
if maxId > -1:
self.out.pt_3[0] = event.Tau_pt[maxId]
self.out.eta_3[0] = event.Tau_eta[maxId]
self.out.m_3[0] = event.Tau_mass[maxId]
self.out.decayMode_3[0] = event.Tau_decayMode[maxId]
self.out.idAntiEle_3[0] = ord(event.Tau_idAntiEle[maxId])
self.out.idAntiMu_3[0] = ord(event.Tau_idAntiMu[maxId])
self.out.idMVAoldDM_3[0] = ord(event.Tau_idMVAoldDM[maxId])
self.out.idMVAoldDM2017v1_3[0] = ord(
event.Tau_idMVAoldDM2017v1[maxId])
self.out.idMVAoldDM2017v2_3[0] = ord(
event.Tau_idMVAoldDM2017v2[maxId])
self.out.idMVAnewDM2017v2_3[0] = ord(
event.Tau_idMVAnewDM2017v2[maxId])
self.out.idIso_3[0] = Tau_idIso(event, maxId)
if not self.isData:
self.out.genPartFlav_3[0] = ord(event.Tau_genPartFlav[maxId])
else:
self.out.pt_3[0] = -1
self.out.eta_3[0] = -9
self.out.m_3[0] = -1
self.out.decayMode_3[0] = -1
self.out.idAntiEle_3[0] = -1
self.out.idAntiMu_3[0] = -1
self.out.idMVAoldDM_3[0] = -1
self.out.idMVAoldDM2017v1_3[0] = -1
self.out.idMVAoldDM2017v2_3[0] = -1
self.out.idMVAnewDM2017v2_3[0] = -1
self.out.idIso_3[0] = -1
self.out.genPartFlav_3[0] = -1
# WEIGHTS
met = TLorentzVector()
met.SetPxPyPzE(event.MET_pt * cos(event.MET_phi),
event.MET_pt * sin(event.MET_phi), 0, event.MET_pt)
if not self.isData:
if self.doRecoil:
boson, boson_vis = getBoson(event)
self.recoilTool.CorrectPByMeanResolution(
met, boson, boson_vis, len(jetIds))
event.MET_pt = met.Pt()
event.MET_phi = met.Phi()
self.out.m_genboson[0] = boson.M()
self.out.pt_genboson[0] = boson.Pt()
if self.doZpt:
self.out.zptweight[0] = self.zptTool.getZptWeight(
boson.Pt(), boson.M())
elif self.doZpt:
zboson = getZBoson(event)
self.out.m_genboson[0] = zboson.M()
self.out.pt_genboson[0] = zboson.Pt()
self.out.zptweight[0] = self.zptTool.getZptWeight(
zboson.Pt(), zboson.M())
elif self.doTTpt:
toppt1, toppt2 = getTTPt(event)
self.out.ttptweight[0] = getTTptWeight(toppt1, toppt2)
self.out.genweight[0] = event.genWeight
self.out.puweight[0] = self.puTool.getWeight(event.Pileup_nTrueInt)
self.out.trigweight[0] = 1.
self.out.idisoweight_1[0] = self.eleSFs.getIdIsoSF(
self.out.pt_1[0], self.out.eta_1[0])
self.out.idisoweight_2[0] = self.muonSFs.getIdIsoSF(
self.out.pt_2[0], self.out.eta_2[0])
self.out.btagweight[0] = self.btagTool.getWeight(event, jetIds)
self.out.weight[0] = self.out.genweight[0] * self.out.puweight[
0] * self.out.trigweight[0] * self.out.idisoweight_1[
0] * self.out.idisoweight_2[0]
# JETS
self.out.njets[0] = len(jetIds)
self.out.njets50[0] = len([j for j in jetIds if event.Jet_pt[j] > 50])
self.out.nfjets[0] = nfjets
self.out.ncjets[0] = ncjets
self.out.nbtag[0] = nbtag
if len(jetIds) > 0:
self.out.jpt_1[0] = event.Jet_pt[jetIds[0]]
self.out.jeta_1[0] = event.Jet_eta[jetIds[0]]
self.out.jphi_1[0] = event.Jet_phi[jetIds[0]]
self.out.jdeepb_1[0] = event.Jet_btagDeepB[jetIds[0]]
else:
self.out.jpt_1[0] = -9.
self.out.jeta_1[0] = -9.
self.out.jphi_1[0] = -9.
self.out.jdeepb_1[0] = -9.
if len(jetIds) > 1:
self.out.jpt_2[0] = event.Jet_pt[jetIds[1]]
self.out.jeta_2[0] = event.Jet_eta[jetIds[1]]
self.out.jphi_2[0] = event.Jet_phi[jetIds[1]]
self.out.jdeepb_2[0] = event.Jet_btagDeepB[jetIds[1]]
else:
self.out.jpt_2[0] = -9.
self.out.jeta_2[0] = -9.
self.out.jphi_2[0] = -9.
self.out.jdeepb_2[0] = -9.
if len(bjetIds) > 0:
self.out.bpt_1[0] = event.Jet_pt[bjetIds[0]]
self.out.beta_1[0] = event.Jet_eta[bjetIds[0]]
else:
self.out.bpt_1[0] = -9.
self.out.beta_1[0] = -9.
if len(bjetIds) > 1:
self.out.bpt_2[0] = event.Jet_pt[bjetIds[1]]
self.out.beta_2[0] = event.Jet_eta[bjetIds[1]]
else:
self.out.bpt_2[0] = -9.
self.out.beta_2[0] = -9.
self.out.njets[0] = len(jetIds)
self.out.nfjets[0] = nfjets
self.out.ncjets[0] = ncjets
self.out.nbtag[0] = nbtag
self.out.met[0] = event.MET_pt
self.out.metphi[0] = event.MET_phi
self.out.pfmt_1[0] = sqrt(
2 * self.out.pt_1[0] * self.out.met[0] *
(1 - cos(deltaPhi(self.out.phi_1[0], self.out.metphi[0]))))
self.out.pfmt_2[0] = sqrt(
2 * self.out.pt_2[0] * self.out.met[0] *
(1 - cos(deltaPhi(self.out.phi_2[0], self.out.metphi[0]))))
self.out.m_vis[0] = (electron + muon).M()
self.out.pt_ll[0] = (electron + muon).Pt()
self.out.dR_ll[0] = electron.DeltaR(muon)
self.out.dphi_ll[0] = deltaPhi(self.out.phi_1[0], self.out.phi_2[0])
self.out.deta_ll[0] = abs(self.out.eta_1[0] - self.out.eta_2[0])
# PZETA
leg1 = TVector3(electron.Px(), electron.Py(), 0.)
leg2 = TVector3(muon.Px(), muon.Py(), 0.)
zetaAxis = TVector3(leg1.Unit() + leg2.Unit()).Unit()
pzeta_vis = leg1 * zetaAxis + leg2 * zetaAxis
pzeta_miss = met.Vect() * zetaAxis
self.out.pzetamiss[0] = pzeta_miss
self.out.pzetavis[0] = pzeta_vis
self.out.dzeta[0] = pzeta_miss - 0.85 * pzeta_vis
self.out.tree.Fill()
return True
def beginFile(self, inputFile, outputFile, inputTree, wrappedOutputTree):
self.out = wrappedOutputTree
self.out.branch("isMC", "I")
self.out.branch("is2016", "I")
self.out.branch("is2017", "I")
self.out.branch("is2018", "I")
self.out.branch("isSingleMuonTrigger", "I")
self.out.branch("isSingleMuonPhotonTrigger", "I")
self.out.branch("isSingleMuonNoFiltersPhotonTrigger", "I")
self.out.branch("isDoubleMuonTrigger", "I")
self.out.branch("isDoubleMuonPhotonTrigger", "I")
self.out.branch("isJPsiTrigger", "I")
self.out.branch("isDisplacedTrigger", "I")
self.out.branch("passedMETFilters", "I")
self.out.branch("nCleanElectron", "I")
self.out.branch("nCleanMuon", "I")
self.out.branch("nCleanTau", "I")
self.out.branch("nCleanPhoton", "I")
self.out.branch("nCleanJet", "I")
self.out.branch("nCleanBTagJet", "I")
self.out.branch("HT30", "F")
self.out.branch("iPhoton", "I")
self.out.branch("iMuon1", "I")
self.out.branch("iMuon2", "I")
self.out.branch("Muon1_pt", "F")
self.out.branch("Muon1_eta", "F")
self.out.branch("Muon2_pt", "F")
self.out.branch("Muon2_eta", "F")
self.out.branch("Muon1_pfRelIso03_all", "F")
self.out.branch("Muon2_pfRelIso03_all", "F")
self.out.branch("Muon1_mediumId", "I")
self.out.branch("Muon2_mediumId", "I")
self.out.branch("Muon1_ip3d", "F")
self.out.branch("Muon2_ip3d", "F")
self.out.branch("minMuonIso", "F")
self.out.branch("maxMuonIso", "F")
self.out.branch("minMuonTrkIso", "F")
self.out.branch("maxMuonTrkIso", "F")
self.out.branch("Muon12_diffdxy", "F")
self.out.branch("Muon12_diffdz", "F")
self.out.branch("Muon12_signdxy", "F")
self.out.branch("Muon12_signdz", "F")
self.out.branch("Photon1_pt", "F")
self.out.branch("Photon1_eta", "F")
self.out.branch("Photon1_mvaID_WP80", "I")
self.out.branch("Photon1_pfRelIso03_all", "F")
self.out.branch("JPsi_pt", "F")
self.out.branch("JPsi_eta", "F")
self.out.branch("JPsi_phi", "F")
self.out.branch("JPsi_mass", "F")
self.out.branch("JPsi_dEta", "F")
self.out.branch("JPsi_dPhi", "F")
self.out.branch("JPsi_dR", "F")
self.out.branch("H_pt", "F")
self.out.branch("H_eta", "F")
self.out.branch("H_phi", "F")
self.out.branch("H_mass", "F")
self.out.branch("H_dEta", "F")
self.out.branch("H_dPhi", "F")
self.out.branch("H_dR", "F")
self.out.branch("minMuonMetDPhi", "F")
self.out.branch("maxMuonMetDPhi", "F")
self.out.branch("photonMetDPhi", "F")
self.out.branch("metPlusPhotonDPhi", "F")
self.out.branch("cosThetaStar", "F")
self.out.branch("cosTheta1", "F")
self.out.branch("phi1", "F")
self.out.branch("genCosThetaStar", "F")
self.out.branch("genCosTheta1", "F")
self.out.branch("genPhi1", "F")
self.out.branch("lumiWeight", "F")
self.out.branch("lheWeight", "F")
self.out.branch("stitchWeight", "F")
self.out.branch("puWeight", "F")
self.out.branch("topWeight", "F")
self.out.branch("qcdnloWeight", "F")
self.out.branch("qcdnnloWeight", "F")
self.out.branch("ewknloWeight", "F")
self.out.branch("triggerWeight", "F")
self.out.branch("leptonWeight", "F")
self.out.branch("eventWeightLumi", "F")
self.fileName = inputFile.GetName()
self.sampleName = getNameFromFile(self.fileName)
self.isMC = not "Run201" in self.fileName
if self.verbose >= 0: print "+ Opening file", self.fileName
self.thMuons = [10., 3.]
self.thPhoton = [10.]
# b-tagging working points for DeepCSV
# https://twiki.cern.ch/twiki/bin/viewauth/CMS/BtagRecommendation2016Legacy
# https://twiki.cern.ch/twiki/bin/viewauth/CMS/BtagRecommendation94X
# https://twiki.cern.ch/twiki/bin/viewauth/CMS/BtagRecommendation102X
if "Run2016" in self.fileName or "UL16" in self.fileName or "Summer16" in self.fileName:
self.year = 2016
self.lumi = 35920.
self.btagLoose = 0.2217 #0.0614
self.btagMedium = 0.6321 #0.3093
self.btagTight = 0.8953 #0.7221
elif "Run2017" in self.fileName or "UL17" in self.fileName or "Fall17" in self.fileName:
self.year = 2017
self.lumi = 41530.
self.btagLoose = 0.1522 #0.0521
self.btagMedium = 0.4941 #0.3033
self.btagTight = 0.8001 #0.7489
elif "Run2018" in self.fileName or "UL18" in self.fileName or "Autumn18" in self.fileName:
self.year = 2018
self.lumi = 59740.
self.btagLoose = 0.1241 #0.0494
self.btagMedium = 0.4184 #0.2770
self.btagTight = 0.7527 #0.7264
else:
if self.verbose >= 0: print "- Unknown year, aborting module"
import sys
sys.exit()
self.xs = XS[self.sampleName] if self.sampleName in XS else 0.
self.nevents = EV[self.sampleName] if self.sampleName in EV else 0.
self.xsWeight = self.xs / self.nevents if self.nevents > 0 else 1.
self.lumiWeight = self.xsWeight * self.lumi if self.isMC else 1.
self.isLO = abs(self.nevents % 1) < 1.e-6 # if True, the event count is integer, so the weight should be normalized (+1)
self.isSignal = ('JPsiG' in self.sampleName)
if self.verbose >= 1: print "+ Module parameters: isMC", self.isMC, ", year", self.year, ", lumi", self.lumi, "pb"
if self.verbose >= 1: print "+ Sample", self.sampleName, ", XS", self.xs, ", events", self.nevents
if self.verbose >= 1: print "+ Weight", self.lumiWeight
if self.isMC and self.isLO and self.verbose >= 1: print "+ Sample is LO, gen weight will be set to 1"
# self.puTool = PileupWeightTool(year = year) if self.isMC else None
self.SingleMuonTriggers = ["HLT_IsoMu27"]
self.SingleMuonPhotonTriggers = ["HLT_Mu17_Photon30_CaloIdL_L1ISO", "HLT_Mu17_Photon30_IsoCaloId", "HLT_Mu17_Photon30_CaloIdL"] # 27.13 in 2017
self.SingleMuonNoFiltersPhotonTriggers = ["HLT_Mu38NoFiltersNoVtxDisplaced_Photon38_CaloIdL", "HLT_Mu38NoFiltersNoVtx_Photon38_CaloIdL", "HLT_Mu43NoFiltersNoVtx_Photon43_CaloIdL"]
self.DoubleMuonTriggers = ["HLT_Mu17_Mu8", "HLT_Mu17_Mu8_DZ", "HLT_Mu17_TkMu8_DZ", "HLT_Mu17_TrkIsoVVL_Mu8_TrkIsoVVL_DZ_Mass", "HLT_Mu37_TkMu27", ] #, "HLT_DoubleMu33NoFiltersNoVtxDisplaced"]
self.DoubleMuonPhotonTriggers = ["HLT_DoubleMu20_7_Mass0to30_Photon23"]
self.JPsiTriggers = ["HLT_Dimuon16_Jpsi", "HLT_Dimuon18_PsiPrime", "HLT_Dimuon18_PsiPrime_noCorrL1", "HLT_Dimuon25_Jpsi", "HLT_Dimuon25_Jpsi_noCorrL1", "HLT_Dimuon20_Jpsi", ]
self.DisplacedTriggers = ["HLT_DoubleMu4_JpsiTrk_Displaced", "HLT_DoubleMu4_PsiPrimeTrk_Displaced"]
if self.isMC:
self.muSFs = MuonSFs(year = self.year)
self.elSFs = ElectronSFs(year = self.year)
self.puTool = PileupWeightTool(year = self.year)