def __init__(self, cfg_ana, cfg_comp, looperName):
super(JetAnalyzer, self).__init__(cfg_ana, cfg_comp, looperName)
self.btagSF = BTagSF(0, wp='medium')
self.recalibrateJets = getattr(cfg_ana, 'recalibrateJets', False)
mcGT = getattr(cfg_ana, 'mcGT', 'Spring16_25nsV6_MC')
dataGT = getattr(cfg_ana, 'dataGT', 'Spring16_25nsV6_DATA')
if self.recalibrateJets:
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)}"
GT = getattr(cfg_comp, 'jecGT',
mcGT if self.cfg_comp.isMC else dataGT)
# instantiate the jet re-calibrator
self.jetReCalibrator = JetReCalibrator(
GT,
'AK4PFchs',
doResidual,
jecPath="%s/src/CMGTools/RootTools/data/jec" %
os.environ['CMSSW_BASE'])
def __init__(self, cfg_ana, cfg_comp, looperName):
super(JetAnalyzer, self).__init__(cfg_ana, cfg_comp, looperName)
self.btagSF = BTagSF(0, wp='medium')
self.recalibrateJets = self.cfg_ana.recalibrateJets if hasattr(
cfg_ana, 'recalibrateJets') else False
mcGT = cfg_ana.mcGT if hasattr(
cfg_ana, 'mcGT') else "76X_mcRun2_asymptotic_RunIIFall15DR76_v1"
dataGT = cfg_ana.dataGT if hasattr(
cfg_ana, 'dataGT') else "76X_dataRun2_16Dec2015_v0"
if self.recalibrateJets:
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)}"
GT = getattr(cfg_comp, 'jecGT',
mcGT if self.cfg_comp.isMC else dataGT)
# instantiate the jet re-calibrator
self.jetReCalibrator = JetReCalibrator(
GT,
'AK4PFchs',
doResidual,
jecPath="%s/src/CMGTools/RootTools/data/jec" %
os.environ['CMSSW_BASE'])
class BJetAnalyzer(Analyzer):
def __init__(self, cfg_ana, cfg_comp, looperName):
super(BJetAnalyzer, self).__init__(cfg_ana, cfg_comp, looperName)
self.btagSF = BTagSF(0, wp='medium')
def process(self, event):
'''Adds the is_btagged attribute to the jets of the
given jets collection.
'''
jets = getattr(event, self.cfg_ana.jets)
for jet in jets:
jet.is_btagged = self.btagSF.isBTagged(pt=jet.pt(),
eta=jet.eta(),
csv=jet.btag(
"pfDeepCSVDiscriminatorsJetTags:BvsAll"),
jetflavor=abs(jet.hadronFlavour()),
is_data=not self.cfg_comp.isMC,
csv_cut=0.4941)
class JetAnalyzer(Analyzer):
"""Analyze jets.
Copied from heppy examples and edit to not rely on heppy examples.
This analyzer filters the jets that do not correspond to the leptons
stored in event.selectedLeptons, and puts in the event:
- jets: all jets passing the pt and eta cuts
- cleanJets: the collection of jets away from the leptons
- cleanBJets: the jets passing testBJet, and away from the leptons
Example configuration:
jetAna = cfg.Analyzer(
'JetAnalyzer',
jetCol = 'slimmedJets'
# cmg jet input collection
# pt threshold
jetPt = 30,
# eta range definition
jetEta = 5.0,
# seed for the btag scale factor
btagSFseed = 0xdeadbeef,
# if True, the PF and PU jet ID are not applied, and the jets get flagged
relaxJetId = False,
relaxPuJetId = False,
)
"""
def __init__(self, cfg_ana, cfg_comp, looperName):
super(JetAnalyzer, self).__init__(cfg_ana, cfg_comp, looperName)
self.btagSF = BTagSF(0, wp='medium')
self.recalibrateJets = getattr(cfg_ana, 'recalibrateJets', False)
mcGT = getattr(cfg_ana, 'mcGT', 'Spring16_25nsV6_MC')
dataGT = getattr(cfg_ana, 'dataGT', 'Spring16_25nsV6_DATA')
if self.recalibrateJets:
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)}"
GT = getattr(cfg_comp, 'jecGT',
mcGT if self.cfg_comp.isMC else dataGT)
# instantiate the jet re-calibrator
self.jetReCalibrator = JetReCalibrator(
GT,
'AK4PFchs',
doResidual,
jecPath="%s/src/CMGTools/RootTools/data/jec" %
os.environ['CMSSW_BASE'])
def declareHandles(self):
super(JetAnalyzer, self).declareHandles()
self.handles['jets'] = AutoHandle(self.cfg_ana.jetCol,
'std::vector<pat::Jet>')
if hasattr(self.cfg_ana, 'leptonCollections'):
for coll, cms_type in self.cfg_ana.leptonCollections.items():
self.handles[coll] = AutoHandle(coll, cms_type)
if self.cfg_comp.isMC:
self.mchandles['genParticles'] = AutoHandle(
'packedGenParticles', 'std::vector<pat::PackedGenParticle>')
self.mchandles['genJets'] = AutoHandle(
'slimmedGenJets', 'std::vector<reco::GenJet>')
def beginLoop(self, setup):
super(JetAnalyzer, self).beginLoop(setup)
self.counters.addCounter('jets')
count = self.counters.counter('jets')
count.register('all events')
count.register('at least 2 good jets')
count.register('at least 2 clean jets')
count.register('at least 1 b jet')
count.register('at least 2 b jets')
def process(self, event):
self.readCollections(event.input)
miniaodjets = self.handles['jets'].product()
allJets = []
event.jets = []
event.bJets = []
event.cleanJets = []
event.cleanBJets = []
leptons = []
if hasattr(event, 'selectedLeptons'):
leptons = event.selectedLeptons
if hasattr(self.cfg_ana, 'toClean'):
leptons = getattr(event, self.cfg_ana.toClean)
if hasattr(self.cfg_ana, 'leptonCollections'):
for coll in self.cfg_ana.leptonCollections:
leptons += self.handles[coll].product()
genJets = None
if self.cfg_comp.isMC:
genJets = map(GenJet, self.mchandles['genJets'].product())
allJets = [Jet(jet) for jet in miniaodjets]
if self.recalibrateJets:
self.jetReCalibrator.correctAll(allJets,
event.rho,
delta=0.,
addCorr=True,
addShifts=True)
for jet in allJets:
if genJets:
# Use DeltaR = 0.25 matching like JetMET
pairs = matchObjectCollection([jet], genJets, 0.25 * 0.25)
if pairs[jet] is None:
pass
else:
jet.matchedGenJet = pairs[jet]
# Add JER correction for MC jets. Requires gen-jet matching.
if self.cfg_comp.isMC and hasattr(
self.cfg_ana, 'jerCorr') and self.cfg_ana.jerCorr:
self.jerCorrection(jet)
# Add JES correction for MC jets.
if self.cfg_comp.isMC and hasattr(self.cfg_ana, 'jesCorr'):
self.jesCorrection(jet, self.cfg_ana.jesCorr)
if self.testJet(jet):
event.jets.append(jet)
if self.testBJet(jet):
event.bJets.append(jet)
self.counters.counter('jets').inc('all events')
event.cleanJets, dummy = cleanObjectCollection(event.jets,
masks=leptons,
deltaRMin=0.5)
event.cleanBJets, dummy = cleanObjectCollection(event.bJets,
masks=leptons,
deltaRMin=0.5)
# Attach matched jets to selected + other leptons
if hasattr(event, 'otherLeptons'):
leptons += event.otherLeptons
pairs = matchObjectCollection(leptons, allJets, 0.5 * 0.5)
# associating a jet to each lepton
for lepton in leptons:
jet = pairs[lepton]
if jet is None:
lepton.jet = lepton
else:
lepton.jet = jet
# associating a leg to each clean jet
invpairs = matchObjectCollection(event.cleanJets, leptons, 99999.)
for jet in event.cleanJets:
leg = invpairs[jet]
jet.leg = leg
for jet in event.cleanJets:
jet.matchGenParton = 999.0
event.jets30 = [jet for jet in event.jets if jet.pt() > 30]
event.cleanJets30 = [jet for jet in event.cleanJets if jet.pt() > 30]
if len(event.jets30) >= 2:
self.counters.counter('jets').inc('at least 2 good jets')
if len(event.cleanJets30) >= 2:
self.counters.counter('jets').inc('at least 2 clean jets')
if len(event.cleanBJets) > 0:
self.counters.counter('jets').inc('at least 1 b jet')
if len(event.cleanBJets) > 1:
self.counters.counter('jets').inc('at least 2 b jets')
# save HTs
event.HT_allJets = sum([jet.pt() for jet in allJets])
event.HT_jets = sum([jet.pt() for jet in event.jets])
event.HT_bJets = sum([jet.pt() for jet in event.bJets])
event.HT_cleanJets = sum([jet.pt() for jet in event.cleanJets])
event.HT_jets30 = sum([jet.pt() for jet in event.jets30])
event.HT_cleanJets30 = sum([jet.pt() for jet in event.cleanJets30])
return True
def jerCorrection(self, jet):
''' Adds JER correction according to first method at
https://twiki.cern.ch/twiki/bin/view/CMS/JetResolution
Requires some attention when genJet matching fails.
'''
if not hasattr(jet, 'matchedGenJet'):
return
#import pdb; pdb.set_trace()
corrections = [0.052, 0.057, 0.096, 0.134, 0.288]
maxEtas = [0.5, 1.1, 1.7, 2.3, 5.0]
eta = abs(jet.eta())
for i, maxEta in enumerate(maxEtas):
if eta < maxEta:
pt = jet.pt()
deltaPt = (pt - jet.matchedGenJet.pt()) * corrections[i]
totalScale = (pt + deltaPt) / pt
if totalScale < 0.:
totalScale = 0.
jet.scaleEnergy(totalScale)
break
def jesCorrection(self, jet, scale=0.):
''' Adds JES correction in number of sigmas (scale)
'''
# Do nothing if nothing to change
if scale == 0.:
return
unc = jet.uncOnFourVectorScale()
totalScale = 1. + scale * unc
if totalScale < 0.:
totalScale = 0.
jet.scaleEnergy(totalScale)
def testJetID(self, jet):
jet.puJetIdPassed = jet.puJetId()
jet.pfJetIdPassed = jet.jetID("POG_PFID_Loose")
puJetId = self.cfg_ana.relaxPuJetId or jet.puJetIdPassed
pfJetId = self.cfg_ana.relaxJetId or jet.pfJetIdPassed
return puJetId and pfJetId
def testJet(self, jet):
pt = jet.pt()
if hasattr(self.cfg_ana,
'ptUncTolerance') and self.cfg_ana.ptUncTolerance:
pt = max(pt, pt * jet.corrJECUp / jet.corr,
pt * jet.corrJECDown / jet.corr)
return pt > self.cfg_ana.jetPt and \
abs( jet.eta() ) < self.cfg_ana.jetEta and \
self.testJetID(jet)
def testBJet(self, jet, csv_cut=0.8):
# medium csv working point
# https://twiki.cern.ch/twiki/bin/viewauth/CMS/BtagRecommendation74X
jet.btagMVA = jet.btag('pfCombinedInclusiveSecondaryVertexV2BJetTags')
# jet.btagFlag = jet.btagMVA > csv_cut
# Use the following once we start applying data-MC scale factors:
jet.btagFlag = self.btagSF.isBTagged(
pt=jet.pt(),
eta=jet.eta(),
csv=jet.btag("pfCombinedInclusiveSecondaryVertexV2BJetTags"),
jetflavor=abs(jet.partonFlavour()),
is_data=not self.cfg_comp.isMC,
csv_cut=csv_cut)
return self.testJet(jet) and \
abs(jet.eta()) < 2.4 and \
jet.btagFlag and \
self.testJetID(jet)
def __init__(self, cfg_ana, cfg_comp, looperName):
super(BJetAnalyzer, self).__init__(cfg_ana, cfg_comp, looperName)
self.btagSF = BTagSF(0, wp='medium')
class JetSelector(Analyzer):
"""Analyze jets.
Copied from heppy examples and edit to not rely on heppy examples.
This analyzer filters the jets that do not correspond to the leptons
stored in event.selectedLeptons, and puts in the event:
- jets: all jets passing the pt and eta cuts
- cleanJets: the collection of jets away from the leptons
- cleanBJets: the jets passing testBJet, and away from the leptons
Example configuration:
jetAna = cfg.Analyzer(
'JetSelector',
jetCol = 'slimmedJets'
# cmg jet input collection
# pt threshold
jetPt = 30,
# eta range definition
jetEta = 5.0,
# seed for the btag scale factor
btagSFseed = 0xdeadbeef,
# if True, the PF and PU jet ID are not applied, and the jets get flagged
relaxJetId = False,
relaxPuJetId = False,
)
"""
def __init__(self, cfg_ana, cfg_comp, looperName):
super(JetSelector, self).__init__(cfg_ana, cfg_comp, looperName)
self.btagSF = BTagSF(0, wp='medium')
self.recalibrateJets = getattr(cfg_ana, 'recalibrateJets', False)
mcGT = getattr(cfg_ana, 'mcGT', 'Spring16_25nsV6_MC')
dataGT = getattr(cfg_ana, 'dataGT', 'Spring16_25nsV6_DATA')
if self.recalibrateJets:
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)}"
GT = getattr(cfg_comp, 'jecGT',
mcGT if self.cfg_comp.isMC else dataGT)
# instantiate the jet re-calibrator
self.jetReCalibrator = JetReCalibrator(
GT,
'AK4PFchs',
doResidual,
jecPath="%s/src/CMGTools/RootTools/data/jec" %
os.environ['CMSSW_BASE'])
def declareHandles(self):
super(JetSelector, self).declareHandles()
self.handles['jets'] = AutoHandle(self.cfg_ana.jetCol,
'std::vector<pat::Jet>')
if self.cfg_comp.isMC:
self.mchandles['genParticles'] = AutoHandle(
'packedGenParticles', 'std::vector<pat::PackedGenParticle>')
self.mchandles['genJets'] = AutoHandle(
'slimmedGenJets', 'std::vector<reco::GenJet>')
def beginLoop(self, setup):
super(JetSelector, self).beginLoop(setup)
self.counters.addCounter('jets')
count = self.counters.counter('jets')
count.register('all events')
count.register('at least 2 good jets')
count.register('at least 2 clean jets')
count.register('exactly 2 b jets')
def process(self, event):
self.counters.counter('jets').inc('all events')
self.readCollections(event.input)
miniaodjets = self.handles['jets'].product()
allJets = []
to_clean_against = []
if hasattr(self.cfg_ana, 'toClean'):
for collection in self.cfg_ana.toClean:
to_clean_against += getattr(event, collection)
genJets = None
if self.cfg_comp.isMC:
genJets = map(GenJet, self.mchandles['genJets'].product())
allJets = [Jet(jet) for jet in miniaodjets]
if self.recalibrateJets:
self.jetReCalibrator.correctAll(allJets,
event.rho,
delta=0.,
addCorr=True,
addShifts=True)
for jet in allJets:
if genJets:
# Use DeltaR = 0.25 matching like JetMET
pairs = matchObjectCollection([jet], genJets, 0.25 * 0.25)
if pairs[jet] is None:
pass
else:
jet.matchedGenJet = pairs[jet]
# Add JER correction for MC jets. Requires gen-jet matching.
if self.cfg_comp.isMC and hasattr(
self.cfg_ana, 'jerCorr') and self.cfg_ana.jerCorr:
self.jerCorrection(jet)
# Add JES correction for MC jets.
if self.cfg_comp.isMC and hasattr(self.cfg_ana, 'jesCorr'):
self.jesCorrection(jet, self.cfg_ana.jesCorr)
jet.btagMVA = jet.btag(
'pfCombinedInclusiveSecondaryVertexV2BJetTags'
) #TODO switch to deepCSV
jet.btagged = self.btagSF.isBTagged(
pt=jet.pt(),
eta=jet.eta(),
csv=jet.btag("pfCombinedInclusiveSecondaryVertexV2BJetTags"),
jetflavor=abs(jet.hadronFlavour()),
is_data=not self.cfg_comp.isMC,
csv_cut=
0.800 #CSVM from https://twiki.cern.ch/twiki/bin/viewauth/CMS/BtagRecommendation80X
)
## if self.testJet(jet):
## event.jets.append(jet)
## if self.testBJet(jet):
## event.bJets.append(jet)
allJets = [
jet for jet in allJets if bool(jet.jetID("POG_PFID_Loose"))
if jet.pt() > 20. if abs(jet.eta()) < 2.4
]
if len(allJets) < 2: return False
self.counters.counter('jets').inc('at least 2 good jets')
event.jets, dummy = cleanObjectCollection(allJets,
masks=to_clean_against,
deltaRMin=0.5)
if len(event.jets) < 2: return False
self.counters.counter('jets').inc('at least 2 clean jets')
event.bJets = [jet for jet in event.jets if jet.btagged]
if len(event.bJets) != 2: return False
self.counters.counter('jets').inc('exactly 2 b jets')
# save HTs
event.HT_jets = sum([jet.pt() for jet in event.jets])
event.HT_bJets = sum([jet.pt() for jet in event.bJets])
return True
def jerCorrection(self, jet):
''' Adds JER correction according to first method at
https://twiki.cern.ch/twiki/bin/view/CMS/JetResolution
Requires some attention when genJet matching fails.
'''
if not hasattr(jet, 'matchedGenJet'):
return
#import pdb; pdb.set_trace()
corrections = [0.052, 0.057, 0.096, 0.134, 0.288]
maxEtas = [0.5, 1.1, 1.7, 2.3, 5.0]
eta = abs(jet.eta())
for i, maxEta in enumerate(maxEtas):
if eta < maxEta:
pt = jet.pt()
deltaPt = (pt - jet.matchedGenJet.pt()) * corrections[i]
totalScale = (pt + deltaPt) / pt
if totalScale < 0.:
totalScale = 0.
jet.scaleEnergy(totalScale)
break
def jesCorrection(self, jet, scale=0.):
''' Adds JES correction in number of sigmas (scale)
'''
# Do nothing if nothing to change
if scale == 0.:
return
unc = jet.uncOnFourVectorScale()
totalScale = 1. + scale * unc
if totalScale < 0.:
totalScale = 0.
jet.scaleEnergy(totalScale)