class JetAnalyzer( Analyzer ):
"""Taken from RootTools.JetAnalyzer, simplified, modified, added corrections """
def __init__(self, cfg_ana, cfg_comp, looperName):
super(JetAnalyzer,self).__init__(cfg_ana, cfg_comp, looperName)
mcGT = cfg_ana.mcGT if hasattr(cfg_ana,'mcGT') else "PHYS14_25_V2"
dataGT = cfg_ana.dataGT if hasattr(cfg_ana,'dataGT') else "GR_70_V2_AN1"
self.shiftJEC = self.cfg_ana.shiftJEC if hasattr(self.cfg_ana, 'shiftJEC') else 0
self.recalibrateJets = self.cfg_ana.recalibrateJets
self.addJECShifts = self.cfg_ana.addJECShifts
if self.recalibrateJets == "MC" : self.recalibrateJets = self.cfg_comp.isMC
elif self.recalibrateJets == "Data": self.recalibrateJets = not self.cfg_comp.isMC
elif self.recalibrateJets not in [True,False]: raise RuntimeError, "recalibrateJets must be any of { True, False, 'MC', 'Data' }, while it is %r " % self.recalibrateJets
self.doJEC = self.recalibrateJets or (self.shiftJEC != 0) or self.addJECShifts
if self.doJEC:
doResidual = getattr(cfg_ana, 'applyL2L3Residual', 'Data')
if doResidual == "MC": doResidual = self.cfg_comp.isMC
elif doResidual == "Data": doResidual = not self.cfg_comp.isMC
elif doResidual not in [True,False]: raise RuntimeError, "If specified, applyL2L3Residual must be any of { True, False, 'MC', 'Data'(default)}"
if self.cfg_comp.isMC:
self.jetReCalibrator = JetReCalibrator(mcGT,self.cfg_ana.recalibrationType, doResidual, cfg_ana.jecPath, calculateSeparateCorrections=getattr(cfg_ana,"calculateSeparateCorrections",False))
else:
self.jetReCalibrator = JetReCalibrator(dataGT,self.cfg_ana.recalibrationType, doResidual, cfg_ana.jecPath, calculateSeparateCorrections=getattr(cfg_ana,"calculateSeparateCorrections",False))
self.doPuId = getattr(self.cfg_ana, 'doPuId', True)
self.jetLepDR = getattr(self.cfg_ana, 'jetLepDR', 0.4)
self.jetLepArbitration = getattr(self.cfg_ana, 'jetLepArbitration', lambda jet,lepton: lepton)
self.lepPtMin = getattr(self.cfg_ana, 'minLepPt', -1)
self.lepSelCut = getattr(self.cfg_ana, 'lepSelCut', lambda lep : True)
self.jetGammaDR = getattr(self.cfg_ana, 'jetGammaDR', 0.4)
if(self.cfg_ana.doQG):
qgdefname="{CMSSW_BASE}/src/PhysicsTools/Heppy/data/pdfQG_AK4chs_antib_13TeV_v1.root"
self.qglcalc = QGLikelihoodCalculator(getattr(self.cfg_ana,"QGpath",qgdefname).format(CMSSW_BASE= os.environ['CMSSW_BASE']))
if not hasattr(self.cfg_ana ,"collectionPostFix"):self.cfg_ana.collectionPostFix=""
def declareHandles(self):
super(JetAnalyzer, self).declareHandles()
self.handles['jets'] = AutoHandle( self.cfg_ana.jetCol, 'std::vector<pat::Jet>' )
self.handles['genJet'] = AutoHandle( self.cfg_ana.genJetCol, 'vector<reco::GenJet>' )
self.shiftJER = self.cfg_ana.shiftJER if hasattr(self.cfg_ana, 'shiftJER') else 0
self.handles['rho'] = AutoHandle( self.cfg_ana.rho, 'double' )
def beginLoop(self, setup):
super(JetAnalyzer,self).beginLoop(setup)
def process(self, event):
self.readCollections( event.input )
rho = float(self.handles['rho'].product()[0])
self.rho = rho
## Read jets, if necessary recalibrate and shift MET
if self.cfg_ana.copyJetsByValue:
import ROOT
allJets = map(lambda j:Jet(ROOT.pat.Jet(ROOT.edm.Ptr(ROOT.pat.Jet)(ROOT.edm.ProductID(),j,0))), self.handles['jets'].product()) #copy-by-value is safe if JetAnalyzer is ran more than once
else:
allJets = map(Jet, self.handles['jets'].product())
self.deltaMetFromJEC = [0.,0.]
# print "before. rho",self.rho,self.cfg_ana.collectionPostFix,'allJets len ',len(allJets),'pt', [j.pt() for j in allJets]
if self.doJEC:
# print "\nCalibrating jets %s for lumi %d, event %d" % (self.cfg_ana.jetCol, event.lumi, event.eventId)
self.jetReCalibrator.correctAll(allJets, rho, delta=self.shiftJEC, metShift=self.deltaMetFromJEC)
if self.addJECShifts:
for delta, shift in [(1.0, "JECUp"), (0.0, ""), (-1.0, "JECDown")]:
for j1 in allJets:
corr = self.jetReCalibrator.getCorrection(j1, rho, delta, self.deltaMetFromJEC)
setattr(j1, "corr"+shift, corr)
self.allJetsUsedForMET = allJets
# print "after. rho",self.rho,self.cfg_ana.collectionPostFix,'allJets len ',len(allJets),'pt', [j.pt() for j in allJets]
if self.cfg_comp.isMC:
self.genJets = [ x for x in self.handles['genJet'].product() ]
if self.cfg_ana.do_mc_match:
self.matchJets(event, allJets)
if getattr(self.cfg_ana, 'smearJets', False):
self.smearJets(event, allJets)
##Sort Jets by pT
allJets.sort(key = lambda j : j.pt(), reverse = True)
## Apply jet selection
self.jets = []
self.jetsFailId = []
self.jetsAllNoID = []
self.jetsIdOnly = []
for jet in allJets:
if self.testJetNoID( jet ):
self.jetsAllNoID.append(jet)
# temporary fix since the jetID it's not good for eta>3
if abs(jet.eta()) <3:
if self.testJetID (jet ):
if(self.cfg_ana.doQG):
jet.qgl_calc = self.qglcalc.computeQGLikelihood
jet.qgl_rho = rho
self.jets.append(jet)
self.jetsIdOnly.append(jet)
else:
self.jetsFailId.append(jet)
else:
self.jets.append(jet)
elif self.testJetID (jet ):
self.jetsIdOnly.append(jet)
## Clean Jets from leptons
leptons = []
if hasattr(event, 'selectedLeptons'):
leptons = [ l for l in event.selectedLeptons if l.pt() > self.lepPtMin and self.lepSelCut(l) ]
if self.cfg_ana.cleanJetsFromTaus and hasattr(event, 'selectedTaus'):
leptons = leptons[:] + event.selectedTaus
if self.cfg_ana.cleanJetsFromIsoTracks and hasattr(event, 'selectedIsoCleanTrack'):
leptons = leptons[:] + event.selectedIsoCleanTrack
jetsEtaCut = [j for j in self.jets if abs(j.eta()) < self.cfg_ana.jetEta ]
self.cleanJetsAll, cleanLeptons = cleanJetsAndLeptons(jetsEtaCut, leptons, self.jetLepDR, self.jetLepArbitration)
#self.cleanJetsAll, cleanLeptons = cleanJetsAndLeptons(self.jets, leptons, self.jetLepDR, self.jetLepArbitration) ##from central
self.cleanJets = [j for j in self.cleanJetsAll if abs(j.eta()) < self.cfg_ana.jetEtaCentral ]
self.cleanJetsFwd = [j for j in self.cleanJetsAll if abs(j.eta()) >= self.cfg_ana.jetEtaCentral ]
self.discardedJets = [j for j in self.jets if j not in self.cleanJetsAll]
if hasattr(event, 'selectedLeptons') and self.cfg_ana.cleanSelectedLeptons:
event.discardedLeptons = [ l for l in leptons if l not in cleanLeptons ]
event.selectedLeptons = [ l for l in event.selectedLeptons if l not in event.discardedLeptons ]
## Clean Jets from photons
photons = []
if hasattr(event, 'selectedPhotons'):
if self.cfg_ana.cleanJetsFromFirstPhoton:
photons = event.selectedPhotons[:1]
else:
photons = [ g for g in event.selectedPhotons ]
self.gamma_cleanJetsAll = cleanNearestJetOnly(self.cleanJetsAll, photons, self.jetGammaDR)
self.gamma_cleanJets = [j for j in self.gamma_cleanJetsAll if abs(j.eta()) < self.cfg_ana.jetEtaCentral ]
self.gamma_cleanJetsFwd = [j for j in self.gamma_cleanJetsAll if abs(j.eta()) >= self.cfg_ana.jetEtaCentral ]
###
if self.cfg_ana.alwaysCleanPhotons:
self.cleanJets = self.gamma_cleanJets
self.cleanJetsAll = self.gamma_cleanJetsAll
self.cleanJetsFwd = self.gamma_cleanJetsFwd
## Associate jets to leptons
leptons = event.inclusiveLeptons if hasattr(event, 'inclusiveLeptons') else event.selectedLeptons
jlpairs = matchObjectCollection( leptons, allJets, self.jetLepDR**2)
for jet in allJets:
jet.leptons = [l for l in jlpairs if jlpairs[l] == jet ]
for lep in leptons:
jet = jlpairs[lep]
if jet is None:
setattr(lep,"jet"+self.cfg_ana.collectionPostFix,lep)
else:
setattr(lep,"jet"+self.cfg_ana.collectionPostFix,jet)
## Associate jets to taus
taus = getattr(event,'selectedTaus',[])
jtaupairs = matchObjectCollection( taus, allJets, self.jetLepDR**2)
for jet in allJets:
jet.taus = [l for l in jtaupairs if jtaupairs[l] == jet ]
for tau in taus:
setattr(tau,"jet"+self.cfg_ana.collectionPostFix,jtaupairs[tau])
#MC stuff
if self.cfg_comp.isMC:
self.deltaMetFromJetSmearing = [0, 0]
for j in self.cleanJetsAll:
if hasattr(j, 'deltaMetFromJetSmearing'):
self.deltaMetFromJetSmearing[0] += j.deltaMetFromJetSmearing[0]
self.deltaMetFromJetSmearing[1] += j.deltaMetFromJetSmearing[1]
self.cleanGenJets = cleanNearestJetOnly(self.genJets, leptons, self.jetLepDR)
if self.cfg_ana.cleanGenJetsFromPhoton:
self.cleanGenJets = cleanNearestJetOnly(self.cleanGenJets, photons, self.jetLepDR)
#event.nGenJets25 = 0
#event.nGenJets25Cen = 0
#event.nGenJets25Fwd = 0
#for j in event.cleanGenJets:
# event.nGenJets25 += 1
# if abs(j.eta()) <= 2.4: event.nGenJets25Cen += 1
# else: event.nGenJets25Fwd += 1
if self.cfg_ana.do_mc_match:
self.jetFlavour(event)
if hasattr(event,"jets"+self.cfg_ana.collectionPostFix): raise RuntimeError, "Event already contains a jet collection with the following postfix: "+self.cfg_ana.collectionPostFix
setattr(event,"rho" +self.cfg_ana.collectionPostFix, self.rho )
setattr(event,"deltaMetFromJEC" +self.cfg_ana.collectionPostFix, self.deltaMetFromJEC )
setattr(event,"allJetsUsedForMET" +self.cfg_ana.collectionPostFix, self.allJetsUsedForMET )
setattr(event,"jets" +self.cfg_ana.collectionPostFix, self.jets )
setattr(event,"jetsFailId" +self.cfg_ana.collectionPostFix, self.jetsFailId )
setattr(event,"jetsAllNoID" +self.cfg_ana.collectionPostFix, self.jetsAllNoID )
setattr(event,"jetsIdOnly" +self.cfg_ana.collectionPostFix, self.jetsIdOnly )
setattr(event,"cleanJetsAll" +self.cfg_ana.collectionPostFix, self.cleanJetsAll )
setattr(event,"cleanJets" +self.cfg_ana.collectionPostFix, self.cleanJets )
setattr(event,"cleanJetsFwd" +self.cfg_ana.collectionPostFix, self.cleanJetsFwd )
setattr(event,"discardedJets" +self.cfg_ana.collectionPostFix, self.discardedJets )
setattr(event,"gamma_cleanJetsAll" +self.cfg_ana.collectionPostFix, self.gamma_cleanJetsAll )
setattr(event,"gamma_cleanJets" +self.cfg_ana.collectionPostFix, self.gamma_cleanJets )
setattr(event,"gamma_cleanJetsFwd" +self.cfg_ana.collectionPostFix, self.gamma_cleanJetsFwd )
if self.cfg_comp.isMC:
setattr(event,"deltaMetFromJetSmearing"+self.cfg_ana.collectionPostFix, self.deltaMetFromJetSmearing)
setattr(event,"cleanGenJets" +self.cfg_ana.collectionPostFix, self.cleanGenJets )
setattr(event,"genJets" +self.cfg_ana.collectionPostFix, self.genJets )
if self.cfg_ana.do_mc_match:
setattr(event,"bqObjects" +self.cfg_ana.collectionPostFix, self.bqObjects )
setattr(event,"cqObjects" +self.cfg_ana.collectionPostFix, self.cqObjects )
setattr(event,"partons" +self.cfg_ana.collectionPostFix, self.partons )
setattr(event,"heaviestQCDFlavour" +self.cfg_ana.collectionPostFix, self.heaviestQCDFlavour )
return True
def testJetID(self, jet):
jet.puJetIdPassed = jet.puJetId()
jet.pfJetIdPassed = jet.jetID('POG_PFID_Loose')
if self.cfg_ana.relaxJetId:
return True
else:
return jet.pfJetIdPassed and (jet.puJetIdPassed or not(self.doPuId))
def testJetNoID( self, jet ):
# 2 is loose pile-up jet id
return jet.pt() > self.cfg_ana.jetPt and \
abs( jet.eta() ) < self.cfg_ana.jetEta;
def jetFlavour(self,event):
def isFlavour(x,f):
id = abs(x.pdgId())
if id > 999: return (id/1000)%10 == f
if id > 99: return (id/100)%10 == f
return id % 100 == f
self.bqObjects = [ p for p in event.genParticles if (p.status() == 2 and isFlavour(p,5)) ]
self.cqObjects = [ p for p in event.genParticles if (p.status() == 2 and isFlavour(p,4)) ]
self.partons = [ p for p in event.genParticles if ((p.status() == 23 or p.status() == 3) and abs(p.pdgId())>0 and (abs(p.pdgId()) in [1,2,3,4,5,21]) ) ]
match = matchObjectCollection2(self.cleanJetsAll,
self.partons,
deltaRMax = 0.3)
for jet in self.cleanJetsAll:
parton = match[jet]
jet.partonId = (parton.pdgId() if parton != None else 0)
jet.partonMotherId = (parton.mother(0).pdgId() if parton != None and parton.numberOfMothers()>0 else 0)
for jet in self.jets:
(bmatch, dr) = bestMatch(jet, self.bqObjects)
if dr < 0.4:
jet.mcFlavour = 5
else:
(cmatch, dr) = bestMatch(jet, self.cqObjects)
if dr < 0.4:
jet.mcFlavour = 4
else:
jet.mcFlavour = 0
self.heaviestQCDFlavour = 5 if len(self.bqObjects) else (4 if len(self.cqObjects) else 1);
def matchJets(self, event, jets):
match = matchObjectCollection2(jets,
event.genbquarks + event.genwzquarks,
deltaRMax = 0.3)
for jet in jets:
gen = match[jet]
jet.mcParton = gen
jet.mcMatchId = (gen.sourceId if gen != None else 0)
jet.mcMatchFlav = (abs(gen.pdgId()) if gen != None else 0)
match = matchObjectCollection2(jets,
self.genJets,
deltaRMax = 0.3)
for jet in jets:
jet.mcJet = match[jet]
def smearJets(self, event, jets):
# https://twiki.cern.ch/twiki/bin/viewauth/CMS/TWikiTopRefSyst#Jet_energy_resolution
for jet in jets:
gen = jet.mcJet
if gen != None:
genpt, jetpt, aeta = gen.pt(), jet.pt(), abs(jet.eta())
# from https://twiki.cern.ch/twiki/bin/view/CMS/JetResolution
factor = 1.052 + self.shiftJER*math.hypot(0.012,0.062);
if aeta > 2.3: factor = 1.288 + self.shiftJER*math.hypot(0.127,0.154)
elif aeta > 1.7: factor = 1.134 + self.shiftJER*math.hypot(0.035,0.066)
elif aeta > 1.1: factor = 1.096 + self.shiftJER*math.hypot(0.017,0.063)
elif aeta > 0.5: factor = 1.057 + self.shiftJER*math.hypot(0.012,0.056)
ptscale = max(0.0, (jetpt + (factor-1)*(jetpt-genpt))/jetpt)
#print "get with pt %.1f (gen pt %.1f, ptscale = %.3f)" % (jetpt,genpt,ptscale)
jet.deltaMetFromJetSmearing = [ -(ptscale-1)*jet.rawFactor()*jet.px(), -(ptscale-1)*jet.rawFactor()*jet.py() ]
if ptscale != 0:
jet.setP4(jet.p4()*ptscale)
# leave the uncorrected unchanged for sync
jet._rawFactorMultiplier *= (1.0/ptscale) if ptscale != 0 else 1
class AdditionalBoost( Analyzer ):
def __init__(self, cfg_ana, cfg_comp, looperName):
super(AdditionalBoost,self).__init__(cfg_ana, cfg_comp, looperName)
# Get the config parameters
skip_ca15 = cfg_ana.skip_ca15 if hasattr(cfg_ana,'skip_ca15') else False
GT = cfg_ana.GT if hasattr(cfg_ana,'GT') else "Summer15_25nsV6_DATA"
jecPath = cfg_ana.jecPath if hasattr(cfg_ana,'jecPath') else "."
isMC = cfg_ana.isMC if hasattr(cfg_ana,'isMC') else False
self.skip_ca15 = skip_ca15
# Prepare re-calibrator
recalibrationType = "AK8PFchs"
# Following instructions from:
# https://twiki.cern.ch/twiki/bin/viewauth/CMS/JetWtagging
# L2L3
if isMC:
doResidual = False
# L2L3 Residual
else:
doResidual = True
self.jetReCalibrator = JetReCalibrator(GT,
recalibrationType,
doResidual,
jecPath,
skipLevel1=True)
def declareHandles(self):
super(AdditionalBoost, self).declareHandles()
self.handles['rho'] = AutoHandle( ('fixedGridRhoFastjetAll',""), 'double' )
self.handles['ak08'] = AutoHandle( ("slimmedJetsAK8",""), "std::vector<pat::Jet>")
self.handles['ak08pruned'] = AutoHandle( ("ak08PFPrunedJetsCHS","","EX"), "std::vector<reco::BasicJet>")
self.handles['ak08prunedsubjets'] = AutoHandle( ("ak08PFPrunedJetsCHS","SubJets","EX"), "std::vector<reco::PFJet>")
self.handles['ak08softdropsubjets'] = AutoHandle( ("slimmedJetsAK8PFCHSSoftDropPacked","SubJets"), "std::vector<pat::Jet>")
self.handles['ak08bbtag'] = AutoHandle( ("slimmedJetsAK8pfBoostedDoubleSecondaryVertexBJetTags","","EX"),
"edm::AssociationVector<edm::RefToBaseProd<reco::Jet>,vector<float>,edm::RefToBase<reco::Jet>,unsigned int,edm::helper::AssociationIdenticalKeyReference>")
self.handles['ak08prunedsubjetbtag'] = AutoHandle( ("ak08PFPrunedJetsCHSpfCombinedInclusiveSecondaryVertexV2BJetTags","","EX"),
"edm::AssociationVector<edm::RefToBaseProd<reco::Jet>,vector<float>,edm::RefToBase<reco::Jet>,unsigned int,edm::helper::AssociationIdenticalKeyReference>")
self.handles['ak08ipTagInfos'] = AutoHandle( ("slimmedJetsAK8ImpactParameterTagInfos","","EX"), "vector<reco::IPTagInfo<vector<edm::Ptr<reco::Candidate> >,reco::JetTagInfo> >")
self.handles['ak08svTagInfos'] = AutoHandle( ("slimmedJetsAK8pfInclusiveSecondaryVertexFinderTagInfos", "","EX"), "vector<reco::TemplatedSecondaryVertexTagInfo<reco::IPTagInfo<vector<edm::Ptr<reco::Candidate> >,reco::JetTagInfo>,reco::VertexCompositePtrCandidate> >")
self.handles['ak08muonTagInfos'] = AutoHandle( ("slimmedJetsAK8softPFMuonsTagInfos", "","EX"), "vector<reco::TemplatedSoftLeptonTagInfo<edm::Ptr<reco::Candidate> > >")
self.handles['ak08elecTagInfos'] = AutoHandle( ("slimmedJetsAK8softPFElectronsTagInfos", "","EX"), "vector<reco::TemplatedSoftLeptonTagInfo<edm::Ptr<reco::Candidate> > >")
if not self.skip_ca15:
self.handles['ca15ipTagInfos'] = AutoHandle( ("ca15PFJetsCHSImpactParameterTagInfos","","EX"), "vector<reco::IPTagInfo<vector<edm::Ptr<reco::Candidate> >,reco::JetTagInfo> >")
self.handles['ca15svTagInfos'] = AutoHandle( ("ca15PFJetsCHSpfInclusiveSecondaryVertexFinderTagInfos", "","EX"), "vector<reco::TemplatedSecondaryVertexTagInfo<reco::IPTagInfo<vector<edm::Ptr<reco::Candidate> >,reco::JetTagInfo>,reco::VertexCompositePtrCandidate> >")
self.handles['ca15muonTagInfos'] = AutoHandle( ("ca15PFJetsCHSsoftPFMuonsTagInfos", "","EX"), "vector<reco::TemplatedSoftLeptonTagInfo<edm::Ptr<reco::Candidate> > >")
self.handles['ca15elecTagInfos'] = AutoHandle( ("ca15PFJetsCHSsoftPFElectronsTagInfos", "","EX"), "vector<reco::TemplatedSoftLeptonTagInfo<edm::Ptr<reco::Candidate> > >")
self.handles['ca15ungroomed'] = AutoHandle( ("ca15PFJetsCHS","","EX"), "std::vector<reco::PFJet>")
self.handles['ca15trimmed'] = AutoHandle( ("ca15PFTrimmedJetsCHS","","EX"), "std::vector<reco::PFJet>")
self.handles['ca15softdrop'] = AutoHandle( ("ca15PFSoftdropJetsCHS","","EX"), "std::vector<reco::PFJet>")
self.handles['ca15softdropz2b1'] = AutoHandle( ("ca15PFSoftdropZ2B1JetsCHS","","EX"), "std::vector<reco::PFJet>")
self.handles['ca15pruned'] = AutoHandle( ("ca15PFPrunedJetsCHS","","EX"), "std::vector<reco::BasicJet>")
self.handles['ca15prunedsubjets'] = AutoHandle( ("ca15PFPrunedJetsCHS","SubJets","EX"), "std::vector<reco::PFJet>")
self.handles['ca15tau1'] = AutoHandle( ("ca15PFJetsCHSNSubjettiness","tau1","EX"), "edm::ValueMap<float>")
self.handles['ca15tau2'] = AutoHandle( ("ca15PFJetsCHSNSubjettiness","tau2","EX"), "edm::ValueMap<float>")
self.handles['ca15tau3'] = AutoHandle( ("ca15PFJetsCHSNSubjettiness","tau3","EX"), "edm::ValueMap<float>")
self.handles['ca15softdropz2b1tau1'] = AutoHandle( ("ca15PFSoftdropZ2B1JetsCHSNSubjettiness","tau1","EX"), "edm::ValueMap<float>")
self.handles['ca15softdropz2b1tau2'] = AutoHandle( ("ca15PFSoftdropZ2B1JetsCHSNSubjettiness","tau2","EX"), "edm::ValueMap<float>")
self.handles['ca15softdropz2b1tau3'] = AutoHandle( ("ca15PFSoftdropZ2B1JetsCHSNSubjettiness","tau3","EX"), "edm::ValueMap<float>")
self.handles['httCandJets'] = AutoHandle( ("looseOptRHTT","","EX"), "std::vector<reco::BasicJet>")
self.handles['httCandInfos'] = AutoHandle( ("looseOptRHTT","","EX"), "vector<reco::HTTTopJetTagInfo>")
self.handles['httSubjetBtags'] = AutoHandle( ("looseOptRHTTpfCombinedInclusiveSecondaryVertexV2BJetTags","","EX"),
"edm::AssociationVector<edm::RefToBaseProd<reco::Jet>,vector<float>,edm::RefToBase<reco::Jet>,unsigned int,edm::helper::AssociationIdenticalKeyReference>")
self.handles['ca15bbtag'] = AutoHandle( ("ca15PFJetsCHSpfBoostedDoubleSecondaryVertexBJetTags","","EX"),
"edm::AssociationVector<edm::RefToBaseProd<reco::Jet>,vector<float>,edm::RefToBase<reco::Jet>,unsigned int,edm::helper::AssociationIdenticalKeyReference>")
self.handles['ca15prunedsubjetbtag'] = AutoHandle( ("ca15PFPrunedJetsCHSpfCombinedInclusiveSecondaryVertexV2BJetTags","","EX"),
"edm::AssociationVector<edm::RefToBaseProd<reco::Jet>,vector<float>,edm::RefToBase<reco::Jet>,unsigned int,edm::helper::AssociationIdenticalKeyReference>")
self.regressionsAK08={}
print 'Initialize regression AK08'
regressionAK08 = JetRegressionAK08('regAK08.weights.xml', 'Jet0RegressionAK08')
#/afs/cern.ch/user/d/degrutto/scratch3/VHbbRun2/CMSSW_7_4_15/src/run2_boostedZH_161115/weights/TMVARegression_BDTG.weights.xml', 'Jet0RegressionAK08')
self.regressionsAK08[0] = regressionAK08
def process(self, event):
run = event.input.eventAuxiliary().id().run()
lumi = event.input.eventAuxiliary().id().luminosityBlock()
eventId = event.input.eventAuxiliary().id().event()
self.readCollections( event.input )
# Will need who for jet calibration later
rho = self.handles["rho"].product()[0]
########
# AK8 Jets from MiniAOD + Subjet btags
########
setattr(event, "ak08", map(PhysicsObject, self.handles["ak08"].product()))
setattr(event, "ak0softdropsubjets", map(PhysicsObject, self.handles["ak08softdropsubjets"].product()))
# bb-tag Output
newtags = self.handles['ak08bbtag'].product()
# Loop over jets
for ij, jet in enumerate(getattr(event, "ak08")):
# Fill bb-tag
for i in xrange(len(newtags)) :
if jet.physObj == newtags.key(i).get():
jet.bbtag = newtags.value(i)
# bb-tag Inputs
muonTagInfos = self.handles['ak08muonTagInfos'].product()[ij]
elecTagInfos = self.handles['ak08elecTagInfos'].product()[ij]
ipTagInfo = self.handles['ak08ipTagInfos'].product()[ij]
svTagInfo = self.handles['ak08svTagInfos'].product()[ij]
calcBBTagVariables(jet,
muonTagInfos,
elecTagInfos,
ipTagInfo,
svTagInfo,
njettiness_08,
maxSVDeltaRToJet = 0.7,)
# end of loop over jets
########
# Ungroomed Fatjets + NSubjettiness + Hbb Tagging
########
for prefix in ["ca15"]:
if self.skip_ca15 and ("ca15" in prefix):
continue
# N-Subjettiness
tau1 = self.handles[prefix+'tau1'].product()
tau2 = self.handles[prefix+'tau2'].product()
tau3 = self.handles[prefix+'tau3'].product()
# bb-tag Output
newtags = self.handles[prefix+'bbtag'].product()
# Four Vector
setattr(event, prefix+"ungroomed", map(PhysicsObject, self.handles[prefix+'ungroomed'].product()))
# Loop over jets
for ij, jet in enumerate(getattr(event, prefix+"ungroomed")):
# Fill N-Subjettiness
jet.tau1 = tau1.get(ij)
jet.tau2 = tau2.get(ij)
jet.tau3 = tau3.get(ij)
# Fill bb-tag
for i in xrange(len(newtags)) :
if jet.physObj == newtags.key(i).get():
jet.bbtag = newtags.value(i)
# bb-tag Inputs
muonTagInfos = self.handles['ca15muonTagInfos'].product()[ij]
elecTagInfos = self.handles['ca15elecTagInfos'].product()[ij]
ipTagInfo = self.handles['ca15ipTagInfos'].product()[ij]
svTagInfo = self.handles['ca15svTagInfos'].product()[ij]
calcBBTagVariables(jet,
muonTagInfos,
elecTagInfos,
ipTagInfo,
svTagInfo,
njettiness_15,
maxSVDeltaRToJet = 1.3)
# end of loop over jets
########
# Softdrop Fatjets + NSubjettiness
########
for fj_name in ["ca15softdropz2b1"]:
if self.skip_ca15 and ("ca15" in fj_name):
continue
# Set the four-vector
setattr(event, fj_name, map(PhysicsObject, self.handles[fj_name].product()))
# N-Subjettiness
tau1 = self.handles[fj_name+'tau1'].product()
tau2 = self.handles[fj_name+'tau2'].product()
tau3 = self.handles[fj_name+'tau3'].product()
# Loop over jets
for ij, jet in enumerate(getattr(event, fj_name)):
# Fill N-Subjettiness
jet.tau1 = tau1.get(ij)
jet.tau2 = tau2.get(ij)
jet.tau3 = tau3.get(ij)
# end of loop over jets
########
# Groomed Uncalibrated Fatjets
########
for fj_name in ['ak08pruned', 'ca15trimmed', 'ca15softdrop', 'ca15pruned']:
if self.skip_ca15 and ("ca15" in fj_name):
continue
setattr(event, fj_name, map(PhysicsObject, self.handles[fj_name].product()))
########
# Groomed Fatjets to calibrate
########
pruned_cal_jets = []
for groomed_fj in self.handles['ak08pruned'].product():
# We need the closest ungroomed fatjet to get the JEC:
# - Make a list of pairs: deltaR(ungroomed fj, groomed fj) for all ungroomed fatjets
# - Sort by deltaR
# - And take the minimum
if len(getattr(event, "ak08")):
closest_ung_fj_and_dr = sorted(
[(ung_fj, deltaR2(ung_fj, groomed_fj)) for ung_fj in getattr(event, "ak08")],
key=lambda x:x[1])[0]
else:
print "WARNING: No ungroomed fatjets found in event with groomed fatjet. Skipping"
continue
# Use the jet cone size for matching
minimal_dr_groomed_ungroomed = 0.8
if closest_ung_fj_and_dr[1] > minimal_dr_groomed_ungroomed:
print "WARNING: No ungroomed fatjet found close to groomed fatjet. Skipping"
continue
ungroomed_jet = Jet(closest_ung_fj_and_dr[0])
c = self.jetReCalibrator.getCorrection(ungroomed_jet, rho)
# Need to do a deep-copy. Otherwise the original jet will be modified
cal_groomed_fj = PhysicsObject(groomed_fj).__copy__()
cal_groomed_fj.scaleEnergy(c)
pruned_cal_jets.append(cal_groomed_fj)
setattr(event, 'ak08prunedcal', pruned_cal_jets)
########
# AK08 Regression
#######
self.regressionsAK08[0].evaluateRegressionAK08(event)
########
# Subjets
########
for fj_name in ['ak08pruned','ca15pruned']:
if self.skip_ca15 and ("ca15" in fj_name):
continue
setattr(event, fj_name + "subjets", map(PhysicsObject, self.handles[fj_name+"subjets"].product()))
newtags = self.handles[fj_name+'subjetbtag'].product()
for i in xrange(0,len(newtags)) :
for j in getattr(event, fj_name+"subjets"):
if j.physObj == newtags.key(i).get():
j.btag = newtags.value(i)
########
# HEPTopTagger
########
if not self.skip_ca15:
candJets = self.handles['httCandJets'].product()
candInfos = self.handles['httCandInfos'].product()
event.httCandidates = map(PhysicsObject, candJets)
sjbtags = self.handles['httSubjetBtags'].product()
for i in xrange(0, len(candJets)):
event.httCandidates[i].fRec = candInfos[i].properties().fRec
event.httCandidates[i].Ropt = candInfos[i].properties().Ropt
event.httCandidates[i].RoptCalc = candInfos[i].properties().RoptCalc
event.httCandidates[i].ptForRoptCalc = candInfos[i].properties().ptForRoptCalc
# HTT return the subjet-pair closest to the W-mass as W-subjets
# Could be improved by b-tagging if we run into a problem
[sj_w1, sj_w2, sj_nonw] = [con.__deref__() for con in candJets[i].getJetConstituents() if not con.isNull()]
event.httCandidates[i].sjW1pt = sj_w1.pt()
event.httCandidates[i].sjW1eta = sj_w1.eta()
event.httCandidates[i].sjW1phi = sj_w1.phi()
event.httCandidates[i].sjW1mass = sj_w1.mass()
# Get the correct b-tag
for ib in xrange(0, len(sjbtags)) :
if sj_w1 == sjbtags.key(ib).get():
event.httCandidates[i].sjW1btag = sjbtags.value(ib)
event.httCandidates[i].sjW2pt = sj_w2.pt()
event.httCandidates[i].sjW2eta = sj_w2.eta()
event.httCandidates[i].sjW2phi = sj_w2.phi()
event.httCandidates[i].sjW2mass = sj_w2.mass()
# Get the correct b-tag
for ib in xrange(0, len(sjbtags)) :
if sj_w2 == sjbtags.key(ib).get():
event.httCandidates[i].sjW2btag = sjbtags.value(ib)
event.httCandidates[i].sjNonWpt = sj_nonw.pt()
event.httCandidates[i].sjNonWeta = sj_nonw.eta()
event.httCandidates[i].sjNonWphi = sj_nonw.phi()
event.httCandidates[i].sjNonWmass = sj_nonw.mass()
# Get the correct b-tag
for ib in xrange(0, len(sjbtags)) :
if sj_nonw == sjbtags.key(ib).get():
event.httCandidates[i].sjNonWbtag = sjbtags.value(ib)
return True
class AdditionalBoost( Analyzer ):
def __init__(self, cfg_ana, cfg_comp, looperName):
super(AdditionalBoost,self).__init__(cfg_ana, cfg_comp, looperName)
# Get the config parameters
skip_ca15 = cfg_ana.skip_ca15 if hasattr(cfg_ana,'skip_ca15') else False
GT = cfg_ana.GT if hasattr(cfg_ana,'GT') else "Summer15_25nsV6_DATA"
jecPath = cfg_ana.jecPath if hasattr(cfg_ana,'jecPath') else "."
isMC = cfg_ana.isMC if hasattr(cfg_ana,'isMC') else False
self.skip_ca15 = skip_ca15
# Prepare re-calibrator
recalibrationTypeAK8 = "AK8PFchs"
recalibrationTypeAK4 = "AK4PFchs"
# Following instructions from:
# https://twiki.cern.ch/twiki/bin/viewauth/CMS/JetWtagging
# L2L3
if isMC:
doResidual = False
# L2L3 Residual
else:
doResidual = True
self.jetReCalibratorAK8L2L3 = JetReCalibrator(GT,
recalibrationTypeAK8,
doResidual,
jecPath,
skipLevel1=True)
self.jetReCalibratorAK8L1L2L3 = JetReCalibrator(GT,
recalibrationTypeAK8,
doResidual,
jecPath,
skipLevel1=False)
self.jetReCalibratorAK4 = JetReCalibrator(GT,
recalibrationTypeAK4,
doResidual,
jecPath,
skipLevel1=False)
def declareHandles(self):
super(AdditionalBoost, self).declareHandles()
self.handles['rho'] = AutoHandle( ('fixedGridRhoFastjetAll',""), 'double' )
self.handles['ak08'] = AutoHandle( ("slimmedJetsAK8",""), "std::vector<pat::Jet>")
self.handles['ak08softdropsubjets'] = AutoHandle( ("slimmedJetsAK8PFCHSSoftDropPacked","SubJets"), "std::vector<pat::Jet>")
self.handles['ak08bbtag'] = AutoHandle( ("slimmedJetsAK8pfBoostedDoubleSecondaryVertexBJetTags","","EX"),
"edm::AssociationVector<edm::RefToBaseProd<reco::Jet>,vector<float>,edm::RefToBase<reco::Jet>,unsigned int,edm::helper::AssociationIdenticalKeyReference>")
self.handles['ak08ipTagInfos'] = AutoHandle( ("slimmedJetsAK8ImpactParameterTagInfos","","EX"), "vector<reco::IPTagInfo<vector<edm::Ptr<reco::Candidate> >,reco::JetTagInfo> >")
self.handles['ak08svTagInfos'] = AutoHandle( ("slimmedJetsAK8pfInclusiveSecondaryVertexFinderTagInfos", "","EX"), "vector<reco::TemplatedSecondaryVertexTagInfo<reco::IPTagInfo<vector<edm::Ptr<reco::Candidate> >,reco::JetTagInfo>,reco::VertexCompositePtrCandidate> >")
self.handles['ak08muonTagInfos'] = AutoHandle( ("slimmedJetsAK8softPFMuonsTagInfos", "","EX"), "vector<reco::TemplatedSoftLeptonTagInfo<edm::Ptr<reco::Candidate> > >")
self.handles['ak08elecTagInfos'] = AutoHandle( ("slimmedJetsAK8softPFElectronsTagInfos", "","EX"), "vector<reco::TemplatedSoftLeptonTagInfo<edm::Ptr<reco::Candidate> > >")
if not self.skip_ca15:
self.handles['ca15ipTagInfos'] = AutoHandle( ("ca15PFJetsCHSImpactParameterTagInfos","","EX"), "vector<reco::IPTagInfo<vector<edm::Ptr<reco::Candidate> >,reco::JetTagInfo> >")
self.handles['ca15svTagInfos'] = AutoHandle( ("ca15PFJetsCHSpfInclusiveSecondaryVertexFinderTagInfos", "","EX"), "vector<reco::TemplatedSecondaryVertexTagInfo<reco::IPTagInfo<vector<edm::Ptr<reco::Candidate> >,reco::JetTagInfo>,reco::VertexCompositePtrCandidate> >")
self.handles['ca15muonTagInfos'] = AutoHandle( ("ca15PFJetsCHSsoftPFMuonsTagInfos", "","EX"), "vector<reco::TemplatedSoftLeptonTagInfo<edm::Ptr<reco::Candidate> > >")
self.handles['ca15elecTagInfos'] = AutoHandle( ("ca15PFJetsCHSsoftPFElectronsTagInfos", "","EX"), "vector<reco::TemplatedSoftLeptonTagInfo<edm::Ptr<reco::Candidate> > >")
self.handles['ca15ungroomed'] = AutoHandle( ("ca15PFJetsCHS","","EX"), "std::vector<reco::PFJet>")
self.handles['ca15trimmed'] = AutoHandle( ("ca15PFTrimmedJetsCHS","","EX"), "std::vector<reco::PFJet>")
self.handles['ca15softdrop'] = AutoHandle( ("ca15PFSoftdropJetsCHS","","EX"), "std::vector<reco::PFJet>")
self.handles['ca15softdropz2b1'] = AutoHandle( ("ca15PFSoftdropZ2B1JetsCHS","","EX"), "std::vector<reco::PFJet>")
self.handles['ca15pruned'] = AutoHandle( ("ca15PFPrunedJetsCHS","","EX"), "std::vector<reco::BasicJet>")
self.handles['ca15prunedsubjets'] = AutoHandle( ("ca15PFPrunedJetsCHS","SubJets","EX"), "std::vector<reco::PFJet>")
self.handles['ca15tau1'] = AutoHandle( ("ca15PFJetsCHSNSubjettiness","tau1","EX"), "edm::ValueMap<float>")
self.handles['ca15tau2'] = AutoHandle( ("ca15PFJetsCHSNSubjettiness","tau2","EX"), "edm::ValueMap<float>")
self.handles['ca15tau3'] = AutoHandle( ("ca15PFJetsCHSNSubjettiness","tau3","EX"), "edm::ValueMap<float>")
self.handles['ca15softdropz2b1tau1'] = AutoHandle( ("ca15PFSoftdropZ2B1JetsCHSNSubjettiness","tau1","EX"), "edm::ValueMap<float>")
self.handles['ca15softdropz2b1tau2'] = AutoHandle( ("ca15PFSoftdropZ2B1JetsCHSNSubjettiness","tau2","EX"), "edm::ValueMap<float>")
self.handles['ca15softdropz2b1tau3'] = AutoHandle( ("ca15PFSoftdropZ2B1JetsCHSNSubjettiness","tau3","EX"), "edm::ValueMap<float>")
self.handles['httCandJets'] = AutoHandle( ("looseOptRHTT","","EX"), "std::vector<reco::BasicJet>")
self.handles['httCandInfos'] = AutoHandle( ("looseOptRHTT","","EX"), "vector<reco::HTTTopJetTagInfo>")
self.handles['httSubjetBtags'] = AutoHandle( ("looseOptRHTTpfCombinedInclusiveSecondaryVertexV2BJetTags","","EX"),
"edm::AssociationVector<edm::RefToBaseProd<reco::Jet>,vector<float>,edm::RefToBase<reco::Jet>,unsigned int,edm::helper::AssociationIdenticalKeyReference>")
self.handles['ca15bbtag'] = AutoHandle( ("ca15PFJetsCHSpfBoostedDoubleSecondaryVertexBJetTags","","EX"),
"edm::AssociationVector<edm::RefToBaseProd<reco::Jet>,vector<float>,edm::RefToBase<reco::Jet>,unsigned int,edm::helper::AssociationIdenticalKeyReference>")
self.handles['ca15prunedsubjetbtag'] = AutoHandle( ("ca15PFPrunedJetsCHSpfCombinedInclusiveSecondaryVertexV2BJetTags","","EX"),
"edm::AssociationVector<edm::RefToBaseProd<reco::Jet>,vector<float>,edm::RefToBase<reco::Jet>,unsigned int,edm::helper::AssociationIdenticalKeyReference>")
def process(self, event):
run = event.input.eventAuxiliary().id().run()
lumi = event.input.eventAuxiliary().id().luminosityBlock()
eventId = event.input.eventAuxiliary().id().event()
self.readCollections( event.input )
# Will need who for jet calibration later
rho = self.handles["rho"].product()[0]
########
# Softdrop Fatjets + NSubjettiness
########
for fj_name in ["ca15softdropz2b1"]:
if self.skip_ca15 and ("ca15" in fj_name):
continue
# Set the four-vector
setattr(event, fj_name, map(PhysicsObject, self.handles[fj_name].product()))
# N-Subjettiness
tau1 = self.handles[fj_name+'tau1'].product()
tau2 = self.handles[fj_name+'tau2'].product()
tau3 = self.handles[fj_name+'tau3'].product()
# Loop over jets
for ij, jet in enumerate(getattr(event, fj_name)):
# Fill N-Subjettiness
jet.tau1 = tau1.get(ij)
jet.tau2 = tau2.get(ij)
jet.tau3 = tau3.get(ij)
# end of loop over jets
########
# Groomed Uncalibrated Fatjets
########
for fj_name in ['ca15trimmed', 'ca15softdrop', 'ca15pruned']:
setattr(event, fj_name, map(PhysicsObject, self.handles[fj_name].product()))
#
# ########
# # Groomed Fatjets to calibrate
# ########
#
# pruned_cal_jets = []
#
# for groomed_fj in self.handles['ak08pruned'].product():
#
# # We need the closest ungroomed fatjet to get the JEC:
# # - Make a list of pairs: deltaR(ungroomed fj, groomed fj) for all ungroomed fatjets
# # - Sort by deltaR
# # - And take the minimum
#
# if len(getattr(event, "ak08")):
# closest_ung_fj_and_dr = sorted(
# [(ung_fj, deltaR2(ung_fj, groomed_fj)) for ung_fj in getattr(event, "ak08")],
# key=lambda x:x[1])[0]
# else:
# print "WARNING: No ungroomed fatjets found in event with groomed fatjet. Skipping"
# continue
#
# # Use the jet cone size for matching
# minimal_dr_groomed_ungroomed = 0.8
# if closest_ung_fj_and_dr[1] > minimal_dr_groomed_ungroomed:
# print "WARNING: No ungroomed fatjet found close to groomed fatjet. Skipping"
# continue
#
# ungroomed_jet = Jet(closest_ung_fj_and_dr[0])
#
# c = self.jetReCalibratorAK8L2L3.getCorrection(ungroomed_jet, rho)
#
#
# # Need to do a deep-copy. Otherwise the original jet will be modified
# cal_groomed_fj = PhysicsObject(groomed_fj).__copy__()
# cal_groomed_fj.scaleEnergy(c)
#
# pruned_cal_jets.append(cal_groomed_fj)
#
# setattr(event, 'ak08prunedcal', pruned_cal_jets)
#
########
# Subjets
########
for fj_name in ['ca15pruned']:
if self.skip_ca15 and ("ca15" in fj_name):
continue
setattr(event, fj_name + "subjets", map(PhysicsObject, self.handles[fj_name+"subjets"].product()))
newtags = self.handles[fj_name+'subjetbtag'].product()
for i in xrange(0,len(newtags)) :
for j in getattr(event, fj_name+"subjets"):
if j.physObj == newtags.key(i).get():
j.btag = newtags.value(i)
########
# HEPTopTagger
########
if not self.skip_ca15:
candJets = self.handles['httCandJets'].product()
candInfos = self.handles['httCandInfos'].product()
event.httCandidates = map(PhysicsObject, candJets)
sjbtags = self.handles['httSubjetBtags'].product()
for i in xrange(0, len(candJets)):
event.httCandidates[i].fRec = candInfos[i].properties().fRec
event.httCandidates[i].Ropt = candInfos[i].properties().Ropt
event.httCandidates[i].RoptCalc = candInfos[i].properties().RoptCalc
event.httCandidates[i].ptForRoptCalc = candInfos[i].properties().ptForRoptCalc
# HTT return the subjet-pair closest to the W-mass as W-subjets
# Could be improved by b-tagging if we run into a problem
[sj_w1, sj_w2, sj_nonw] = [con.__deref__() for con in candJets[i].getJetConstituents() if not con.isNull()]
# Calibrate the subjets: W1
sj_w1_uncal = Jet(sj_w1)
c = self.jetReCalibratorAK4.getCorrection(sj_w1_uncal, rho, isHttSubjet=True)
sj_w1_cal = PhysicsObject(sj_w1).__copy__()
sj_w1_cal.scaleEnergy(c)
# Calibrate the subjets: W2
sj_w2_uncal = Jet(sj_w2)
c = self.jetReCalibratorAK4.getCorrection(sj_w2_uncal, rho, isHttSubjet=True)
sj_w2_cal = PhysicsObject(sj_w2).__copy__()
sj_w2_cal.scaleEnergy(c)
# Calibrate the subjets: NonW
sj_nonw_uncal = Jet(sj_nonw)
c = self.jetReCalibratorAK4.getCorrection(sj_nonw_uncal, rho, isHttSubjet=True)
sj_nonw_cal = PhysicsObject(sj_nonw).__copy__()
sj_nonw_cal.scaleEnergy(c)
# Make TLVs so we can add them and get the top quark
# candidate
tlv_sj_w1_cal = ROOT.TLorentzVector()
tlv_sj_w2_cal = ROOT.TLorentzVector()
tlv_sj_nonw_cal = ROOT.TLorentzVector()
tlv_sj_w1_cal.SetPtEtaPhiM(sj_w1_cal.pt(),
sj_w1_cal.eta(),
sj_w1_cal.phi(),
sj_w1_cal.mass())
tlv_sj_w2_cal.SetPtEtaPhiM(sj_w2_cal.pt(),
sj_w2_cal.eta(),
sj_w2_cal.phi(),
sj_w2_cal.mass())
tlv_sj_nonw_cal.SetPtEtaPhiM(sj_nonw_cal.pt(),
sj_nonw_cal.eta(),
sj_nonw_cal.phi(),
sj_nonw_cal.mass())
tlv_top_cal = tlv_sj_w1_cal + tlv_sj_w2_cal + tlv_sj_nonw_cal
# Store calibrated top candidate variables
event.httCandidates[i].ptcal = tlv_top_cal.Pt()
event.httCandidates[i].etacal = tlv_top_cal.Eta()
event.httCandidates[i].phical = tlv_top_cal.Phi()
event.httCandidates[i].masscal = tlv_top_cal.M()
# Store SJ W1 Variables
event.httCandidates[i].sjW1ptcal = sj_w1_cal.pt()
event.httCandidates[i].sjW1masscal = sj_w1_cal.mass()
event.httCandidates[i].sjW1pt = sj_w1.pt()
event.httCandidates[i].sjW1eta = sj_w1.eta()
event.httCandidates[i].sjW1phi = sj_w1.phi()
event.httCandidates[i].sjW1mass = sj_w1.mass()
# Get the correct b-tag
for ib in xrange(0, len(sjbtags)) :
if sj_w1 == sjbtags.key(ib).get():
event.httCandidates[i].sjW1btag = sjbtags.value(ib)
# Store SJ W2 Variables
event.httCandidates[i].sjW2ptcal = sj_w2_cal.pt()
event.httCandidates[i].sjW2masscal = sj_w2_cal.mass()
event.httCandidates[i].sjW2pt = sj_w2.pt()
event.httCandidates[i].sjW2eta = sj_w2.eta()
event.httCandidates[i].sjW2phi = sj_w2.phi()
event.httCandidates[i].sjW2mass = sj_w2.mass()
# Get the correct b-tag
for ib in xrange(0, len(sjbtags)) :
if sj_w2 == sjbtags.key(ib).get():
event.httCandidates[i].sjW2btag = sjbtags.value(ib)
# Store SJ Non W Variables
event.httCandidates[i].sjNonWptcal = sj_nonw_cal.pt()
event.httCandidates[i].sjNonWmasscal = sj_nonw_cal.mass()
event.httCandidates[i].sjNonWpt = sj_nonw.pt()
event.httCandidates[i].sjNonWeta = sj_nonw.eta()
event.httCandidates[i].sjNonWphi = sj_nonw.phi()
event.httCandidates[i].sjNonWmass = sj_nonw.mass()
# Get the correct b-tag
for ib in xrange(0, len(sjbtags)) :
if sj_nonw == sjbtags.key(ib).get():
event.httCandidates[i].sjNonWbtag = sjbtags.value(ib)
########
# AK8 Jets from MiniAOD + Subjet btags
########
setattr(event, "ak08", map(PhysicsObject, self.handles["ak08"].product()))
setattr(event, "ak08softdropsubjets", map(PhysicsObject, self.handles["ak08softdropsubjets"].product()))
do_calc_bb = False
# Calc BB tag
# -- if there is a HTT candidate and at least one lepton
if len(event.selectedLeptons):
for cand in event.httCandidates:
if cand.fRec < 0.2 and cand.mass() > 100 and cand.mass() < 200:
do_calc_bb = True
# -- or if there are at least two fatjets with pT > 250 GeV and the scalar sum of the jet pTs is > 600
fj_pt250 = [j for j in event.ak08 if j.pt() > 250]
if len(fj_pt250) >= 2 and sum([j.pt() for j in fj_pt250]) > 600:
do_calc_bb = True
# bb-tag Output
newtags = self.handles['ak08bbtag'].product()
# Loop over jets
for ij, jet in enumerate(getattr(event, "ak08")):
# Fill bb-tag
for i in xrange(len(newtags)) :
if jet.physObj == newtags.key(i).get():
jet.bbtag = newtags.value(i)
corr = self.jetReCalibratorAK8L2L3.getCorrection(Jet(jet),rho)
jet.mprunedcorr= jet.userFloat("ak8PFJetsCHSPrunedMass")*corr
jet.JEC_L2L3 = corr
jet.JEC_L1L2L3 = self.jetReCalibratorAK8L1L2L3.getCorrection(Jet(jet),rho)
# bb-tag Inputs
muonTagInfos = self.handles['ak08muonTagInfos'].product()[ij]
elecTagInfos = self.handles['ak08elecTagInfos'].product()[ij]
ipTagInfo = self.handles['ak08ipTagInfos'].product()[ij]
svTagInfo = self.handles['ak08svTagInfos'].product()[ij]
orig_jet = self.handles["ak08"].product()[ij]
if do_calc_bb:
calcBBTagVariables(jet,
orig_jet,
muonTagInfos,
elecTagInfos,
ipTagInfo,
svTagInfo,
njettiness_08,
maxSVDeltaRToJet = 0.7,)
else:
calcBBTagVariables_dummy(jet)
# end of loop over jets
########
# Ungroomed Fatjets + NSubjettiness + Hbb Tagging
########
for prefix in ["ca15"]:
if self.skip_ca15 and ("ca15" in prefix):
continue
# N-Subjettiness
tau1 = self.handles[prefix+'tau1'].product()
tau2 = self.handles[prefix+'tau2'].product()
tau3 = self.handles[prefix+'tau3'].product()
# bb-tag Output
newtags = self.handles[prefix+'bbtag'].product()
# Four Vector
setattr(event, prefix+"ungroomed", map(PhysicsObject, self.handles[prefix+'ungroomed'].product()))
# Loop over jets
for ij, jet in enumerate(getattr(event, prefix+"ungroomed")):
# Fill N-Subjettiness
jet.tau1 = tau1.get(ij)
jet.tau2 = tau2.get(ij)
jet.tau3 = tau3.get(ij)
# Fill bb-tag
for i in xrange(len(newtags)) :
if jet.physObj == newtags.key(i).get():
jet.bbtag = newtags.value(i)
# bb-tag Inputs
muonTagInfos = self.handles['ca15muonTagInfos'].product()[ij]
elecTagInfos = self.handles['ca15elecTagInfos'].product()[ij]
ipTagInfo = self.handles['ca15ipTagInfos'].product()[ij]
svTagInfo = self.handles['ca15svTagInfos'].product()[ij]
orig_jet = self.handles[prefix+'ungroomed'].product()[ij]
if do_calc_bb:
calcBBTagVariables(jet,
orig_jet,
muonTagInfos,
elecTagInfos,
ipTagInfo,
svTagInfo,
njettiness_15,
maxSVDeltaRToJet = 1.3)
else:
calcBBTagVariables_dummy(jet)
# end of loop over jets
return True
