def reapplyPUJetID(process, srcJets=cms.InputTag("slimmedJets")):
from RecoJets.JetProducers.PileupJetID_cfi import pileupJetId
process.pileupJetIdUpdated = pileupJetId.clone(
jets=srcJets,
inputIsCorrected=True,
applyJec=True,
vertexes=cms.InputTag("offlineSlimmedPrimaryVertices"))
process.analysisSequence *= process.pileupJetIdUpdated
def ReclusterAK4CHSJets(proc, recoJA, runOnMC):
"""
Recluster AK4 CHS jets and replace slimmedJets
that is used as default to save AK4 CHS jets
in NanoAODs.
"""
print("custom_jme_cff::ReclusterAK4CHSJets: Recluster AK4 PF CHS jets")
#
# Recluster AK4 CHS jets
#
cfg = {
"jet": "ak4pfchs",
"inputCollection": "",
"genJetsCollection": "AK4GenJetsNoNu",
"bTagDiscriminators": bTagDiscriminatorsForAK4
}
recoJetInfo = recoJA.addRecoJetCollection(proc, **cfg)
jetName = recoJetInfo.jetUpper
patJetFinalColl = "selectedUpdatedPatJets{}Final".format(jetName)
#
# Change the input jet source for jetCorrFactorsNano
# and updatedJets
#
proc.jetCorrFactorsNano.src = patJetFinalColl
proc.updatedJets.jetSource = patJetFinalColl
#
# Change pt cut
#
proc.finalJets.cut = "pt > 2"
proc.simpleCleanerTable.jetSel = "pt > 10" # Change this from 15 -> 10
#
# Add variables
#
proc.jetTable.variables.hfHEF = PFJETVARS.hfHEF
proc.jetTable.variables.hfEmEF = PFJETVARS.hfEmEF
proc.jetTable.variables.nConstChHads = PFJETVARS.nConstChHads
proc.jetTable.variables.nConstNeuHads = PFJETVARS.nConstNeuHads
proc.jetTable.variables.nConstHFHads = PFJETVARS.nConstHFHads
proc.jetTable.variables.nConstHFEMs = PFJETVARS.nConstHFEMs
proc.jetTable.variables.nConstMuons = PFJETVARS.nConstMuons
proc.jetTable.variables.nConstElecs = PFJETVARS.nConstElecs
proc.jetTable.variables.nConstPhotons = PFJETVARS.nConstPhotons
proc.jetTable.doc = cms.string(
"AK4 PF CHS Jets with JECs applied, after basic selection (pt > 2)")
#
# Setup pileup jet ID with 80X training.
#
pileupJetId80X = "pileupJetId80X"
setattr(
proc, pileupJetId80X,
pileupJetId.clone(jets="updatedJets",
algos=cms.VPSet(_chsalgos_81x),
inputIsCorrected=True,
applyJec=False,
vertexes="offlineSlimmedPrimaryVertices"))
proc.jetSequence.insert(proc.jetSequence.index(proc.pileupJetId94X),
getattr(proc, pileupJetId80X))
proc.updatedJetsWithUserData.userInts.puId80XfullId = cms.InputTag(
'pileupJetId80X:fullId')
run2_jme_2016.toModify(
proc.updatedJetsWithUserData.userFloats,
puId80XDisc=cms.InputTag("pileupJetId80X:fullDiscriminant"))
proc.jetTable.variables.puId = Var(
"userInt('puId80XfullId')",
int,
doc="Pilup ID flags with 80X (2016) training")
run2_jme_2016.toModify(
proc.jetTable.variables,
puIdDisc=Var("userFloat('puId80XDisc')",
float,
doc="Pilup ID discriminant with 80X (2016) training",
precision=10))
#
# Add variables for pileup jet ID studies.
#
proc = AddPileUpJetIDVars(proc,
jetName="",
jetSrc="updatedJets",
jetTableName="jetTable",
jetSequenceName="jetSequence")
#
# Add variables for quark guon likelihood tagger studies.
# Save variables as userFloats and userInts in each jet
#
proc.updatedJetsWithUserData.userFloats.qgl_axis2 = cms.InputTag(
"qgtagger:axis2")
proc.updatedJetsWithUserData.userFloats.qgl_ptD = cms.InputTag(
"qgtagger:ptD")
proc.updatedJetsWithUserData.userInts.qgl_mult = cms.InputTag(
"qgtagger:mult")
#
# Specfiy variables in the jetTable to save in NanoAOD
#
proc.jetTable.variables.qgl_axis2 = QGLVARS.qgl_axis2
proc.jetTable.variables.qgl_ptD = QGLVARS.qgl_ptD
proc.jetTable.variables.qgl_mult = QGLVARS.qgl_mult
#
# Save DeepJet raw score for gluon and light quarks
#
proc.jetTable.variables.btagDeepFlavG = DEEPJETVARS.btagDeepFlavG
proc.jetTable.variables.btagDeepFlavUDS = DEEPJETVARS.btagDeepFlavUDS
#Adding hf shower shape producer to the jet sequence. By default this producer is not automatically rerun at the NANOAOD step
#The following lines make sure it is.
hfJetShowerShapeforCustomNanoAOD = "hfJetShowerShapeforCustomNanoAOD"
setattr(
proc, hfJetShowerShapeforCustomNanoAOD,
hfJetShowerShapeforNanoAOD.clone(
jets="updatedJets", vertices="offlineSlimmedPrimaryVertices"))
proc.jetSequence.insert(
proc.jetSequence.index(proc.updatedJetsWithUserData),
getattr(proc, hfJetShowerShapeforCustomNanoAOD))
proc.updatedJetsWithUserData.userFloats.hfsigmaEtaEta = cms.InputTag(
'hfJetShowerShapeforCustomNanoAOD:sigmaEtaEta')
proc.updatedJetsWithUserData.userFloats.hfsigmaPhiPhi = cms.InputTag(
'hfJetShowerShapeforCustomNanoAOD:sigmaPhiPhi')
proc.updatedJetsWithUserData.userInts.hfcentralEtaStripSize = cms.InputTag(
'hfJetShowerShapeforCustomNanoAOD:centralEtaStripSize')
proc.updatedJetsWithUserData.userInts.hfadjacentEtaStripsSize = cms.InputTag(
'hfJetShowerShapeforCustomNanoAOD:adjacentEtaStripsSize')
proc.jetTable.variables.hfsigmaEtaEta = Var(
"userFloat('hfsigmaEtaEta')",
float,
doc="sigmaEtaEta for HF jets (noise discriminating variable)",
precision=10)
proc.jetTable.variables.hfsigmaPhiPhi = Var(
"userFloat('hfsigmaPhiPhi')",
float,
doc="sigmaPhiPhi for HF jets (noise discriminating variable)",
precision=10)
proc.jetTable.variables.hfcentralEtaStripSize = Var(
"userInt('hfcentralEtaStripSize')",
int,
doc=
"eta size of the central tower strip in HF (noise discriminating variable) "
)
proc.jetTable.variables.hfadjacentEtaStripsSize = Var(
"userInt('hfadjacentEtaStripsSize')",
int,
doc=
"eta size of the strips next to the central tower strip in HF (noise discriminating variable) "
)
return proc
process.updatedPatJetsUndoneJEC.userData.userFloats.src = []
## An example where the jet corrections are reapplied
updateJetCollection(
process,
labelName = 'ReappliedJEC',
jetSource = cms.InputTag('selectedUpdatedPatJetsUndoneJEC'),
jetCorrections = ('AK4PFchs', cms.vstring(['L1FastJet', 'L2Relative', 'L3Absolute']), 'None')
)
process.updatedPatJetsReappliedJEC.userData.userFloats.src = []
## An example where the pileup jet id is recomputed
from RecoJets.JetProducers.PileupJetID_cfi import pileupJetId
process.pileupJetIdUpdated = pileupJetId.clone(
jets=cms.InputTag("slimmedJets"),
inputIsCorrected=True,
applyJec=True,
vertexes=cms.InputTag("offlineSlimmedPrimaryVertices")
)
patAlgosToolsTask.add(process.pileupJetIdUpdated)
updateJetCollection(
process,
labelName = 'PileupJetID',
jetSource = cms.InputTag('slimmedJets'),
)
process.updatedPatJetsPileupJetID.userData.userInts.src = ['pileupJetIdUpdated:fullId']
process.updatedPatJetsPileupJetID.userData.userFloats.src = ['pileupJetIdUpdated:fullDiscriminant']
## An example where the jet energy corrections are updated to the current GlobalTag
## and specified b-tag discriminators are rerun and added to SoftDrop subjets
updateJetCollection(
## An example where the jet corrections are reapplied
updateJetCollection(
process,
labelName='ReappliedJEC',
jetSource=cms.InputTag('selectedUpdatedPatJetsUndoneJEC'),
jetCorrections=('AK4PFchs',
cms.vstring(['L1FastJet', 'L2Relative',
'L3Absolute']), 'None'))
process.updatedPatJetsReappliedJEC.userData.userFloats.src = []
## An example where the pileup jet id is recomputed
from RecoJets.JetProducers.PileupJetID_cfi import pileupJetId
process.pileupJetIdUpdated = pileupJetId.clone(
jets=cms.InputTag("slimmedJets"),
inputIsCorrected=True,
applyJec=True,
vertexes=cms.InputTag("offlineSlimmedPrimaryVertices"))
patAlgosToolsTask.add(process.pileupJetIdUpdated)
updateJetCollection(
process,
labelName='PileupJetID',
jetSource=cms.InputTag('slimmedJets'),
)
process.updatedPatJetsPileupJetID.userData.userInts.src = [
'pileupJetIdUpdated:fullId'
]
process.updatedPatJetsPileupJetID.userData.userFloats.src = [
'pileupJetIdUpdated:fullDiscriminant'
]
run2_jme_2017.toModify( cjetNN, weightFile = cms.FileInPath("PhysicsTools/NanoAOD/data/creg_training_2017.onnx") )
run2_jme_2017.toModify( cjetNN,outputFormulas = cms.vstring(["at(0)*0.24718524515628815+0.9927206635475159","0.5*(at(2)-at(1))*0.24718524515628815"]))
#
# Quark-Gluon Likelihood (QGL)
#
from RecoJets.JetProducers.QGTagger_cfi import QGTagger
qgtagger=QGTagger.clone(srcJets="updatedJets",srcVertexCollection="offlineSlimmedPrimaryVertices")
#
# PileUp ID
#
from RecoJets.JetProducers.PileupJetID_cfi import pileupJetId, _chsalgos_94x, _chsalgos_102x, _chsalgos_106X_UL16, _chsalgos_106X_UL16APV, _chsalgos_106X_UL17, _chsalgos_106X_UL18
pileupJetIdNano=pileupJetId.clone(jets="updatedJets",algos = cms.VPSet(_chsalgos_106X_UL18),inputIsCorrected=True,applyJec=False,vertexes="offlineSlimmedPrimaryVertices")
run2_jme_2017.toModify(pileupJetIdNano, algos = _chsalgos_106X_UL17)
(run2_jme_2016 & ~tracker_apv_vfp30_2016 ).toModify(pileupJetIdNano, algos = _chsalgos_106X_UL16)
(run2_jme_2016 & tracker_apv_vfp30_2016 ).toModify(pileupJetIdNano, algos = _chsalgos_106X_UL16APV)
run2_nanoAOD_102Xv1.toModify(pileupJetIdNano, algos = _chsalgos_102x)
for modifier in run2_nanoAOD_94XMiniAODv1, run2_nanoAOD_94XMiniAODv2:
modifier.toModify(pileupJetIdNano, algos = _chsalgos_94x)
##############################################################
## DeepInfoAK4CHS:Start
## - To be used in nanoAOD_customizeCommon() in nano_cff.py
###############################################################
from PhysicsTools.PatAlgos.tools.jetTools import updateJetCollection
def nanoAOD_addDeepInfoAK4CHS(process,addDeepBTag,addDeepFlavour):
_btagDiscriminators=[]
if addDeepBTag:
))
### Era dependent customization
run2_miniAOD_80XLegacy.toModify(
genJetFlavourTable,
jetFlavourInfos=cms.InputTag("genJetFlavourAssociation"),
)
from RecoJets.JetProducers.QGTagger_cfi import QGTagger
qgtagger = QGTagger.clone(srcJets="updatedJets",
srcVertexCollection="offlineSlimmedPrimaryVertices")
from RecoJets.JetProducers.PileupJetID_cfi import pileupJetId, _chsalgos_94x, _chsalgos_102x
pileupJetId94X = pileupJetId.clone(jets="updatedJets",
algos=cms.VPSet(_chsalgos_94x),
inputIsCorrected=True,
applyJec=False,
vertexes="offlineSlimmedPrimaryVertices")
pileupJetId102X = pileupJetId.clone(jets="updatedJets",
algos=cms.VPSet(_chsalgos_102x),
inputIsCorrected=True,
applyJec=False,
vertexes="offlineSlimmedPrimaryVertices")
#before cross linking
jetSequence = cms.Sequence(jetCorrFactorsNano + updatedJets + tightJetId +
tightJetIdLepVeto + bJetVars + qgtagger + jercVars +
pileupJetId94X + pileupJetId102X +
updatedJetsWithUserData + jetCorrFactorsAK8 +
updatedJetsAK8 + tightJetIdAK8 +
tightJetIdLepVetoAK8 + updatedJetsAK8WithUserData +
def addLeptonSubtractedAK4Jets(process, runOnMC, era, useFakeable):
assert (era in ["2016", "2017", "2018"])
suffix = "Fakeable" if useFakeable else "Loose"
#----------------------------------------------------------------------------
# produce collection of packedPFCandidates not associated to loose or fakeable electrons or muons
(leptonSubtractedPFCandsSequence,
leptonLessPU_str) = addLeptonSubtractedPFCands(process, era, useFakeable,
'chs', runOnMC)
#----------------------------------------------------------------------------
#----------------------------------------------------------------------------
# reconstruct lepton-subtracted AK4 jets
bTagDiscriminators = [
'pfCombinedInclusiveSecondaryVertexV2BJetTags',
'pfCombinedMVAV2BJetTags',
]
JETCorrLevels = ['L1FastJet', 'L2Relative', 'L3Absolute']
if not runOnMC:
JETCorrLevels.append('L2L3Residual')
jetSequenceAK4LS_str = 'jetSequenceAK4LS%s' % suffix
NoLep_str = 'NoLep%s' % suffix
jetToolbox(
proc=process,
jetType='ak4',
jetSequence=jetSequenceAK4LS_str,
outputFile='out',
PUMethod='CHS',
JETCorrPayload='AK4PFchs',
postFix=NoLep_str,
JETCorrLevels=JETCorrLevels,
miniAOD=True,
runOnMC=runOnMC,
newPFCollection=True,
nameNewPFCollection=leptonLessPU_str,
bTagDiscriminators=bTagDiscriminators,
)
slimmedJetCollectionAK4LS_str = 'selectedPatJetsAK4PFCHS%s' % NoLep_str
bTagDiscriminators_ = [
'pfDeepCSVJetTags:probb',
'pfDeepCSVJetTags:probbb',
'pfDeepCSVJetTags:probc',
'pfDeepFlavourJetTags:probb',
'pfDeepFlavourJetTags:probbb',
'pfDeepFlavourJetTags:problepb',
'pfDeepFlavourJetTags:probc',
]
deepInfoSuffix = 'PlusDeepInfo%s' % NoLep_str
updateJetCollection(
process,
jetSource=cms.InputTag(slimmedJetCollectionAK4LS_str),
jetCorrections=('AK4PFchs', cms.vstring(JETCorrLevels), 'None'),
btagDiscriminators=bTagDiscriminators_,
postfix=deepInfoSuffix,
)
jetCollectionAK4LS_str = "selectedUpdatedPatJets%s" % deepInfoSuffix
getattr(process, jetCollectionAK4LS_str).cut = cms.string('pt > 15')
#----------------------------------------------------------------------------
#----------------------------------------------------------------------------
# CV: add 'patJetPartons' module to 'genParticleSequence' (which runs at beginning of event processing),
# to avoid run-time exception of type:
#
# ----- Begin Fatal Exception 22-Feb-2018 10:16:02 EET-----------------------
# An exception of category 'ScheduleExecutionFailure' occurred while
# [0] Calling beginJob
# Exception Message:
# Unrunnable schedule
# Module run order problem found:
# ...
# Running in the threaded framework would lead to indeterminate results.
# Please change order of modules in mentioned Path(s) to avoid inconsistent module ordering.
# ----- End Fatal Exception -------------------------------------------------
if hasattr(process, "patJetPartons") and hasattr(
process, "genParticleSequence") and runOnMC:
process.genParticleSequence += process.patJetPartons
#----------------------------------------------------------------------------
#----------------------------------------------------------------------------
# add PF jet ID flags and jet energy corrections for AK4 pat::Jet collection,
# following what is done for non-lepton-subtracted AK4 pat::Jets in https://github.com/cms-sw/cmssw/blob/master/PhysicsTools/NanoAOD/python/jets_cff.py
looseJetIdAK4LS_str = 'looseJetIdAK4LS%s' % suffix
setattr(process, looseJetIdAK4LS_str,
process.looseJetId.clone(src=cms.InputTag(jetCollectionAK4LS_str)))
tightJetIdAK4LS_str = 'tightJetIdAK4LS%s' % suffix
setattr(process, tightJetIdAK4LS_str,
process.tightJetId.clone(src=cms.InputTag(jetCollectionAK4LS_str)))
for modifier in run2_miniAOD_80XLegacy, run2_nanoAOD_94X2016:
tightJetIdAK4LS = getattr(process, tightJetIdAK4LS_str)
modifier.toModify(tightJetIdAK4LS.filterParams, version="WINTER16")
tightJetIdLepVetoAK4LS_str = 'tightJetIdLepVetoAK4LS%s' % suffix
setattr(
process, tightJetIdLepVetoAK4LS_str,
process.tightJetIdLepVeto.clone(
src=cms.InputTag(jetCollectionAK4LS_str)))
#----------------------------------------------------------------------------
bJetVars_str = 'bJetVarsAK4LS%s' % suffix
setattr(process, bJetVars_str,
process.bJetVars.clone(src=cms.InputTag(jetCollectionAK4LS_str)))
qgtagger_str = 'qgtaggerAK4LS%s' % suffix
setattr(
process, qgtagger_str,
process.qgtagger.clone(srcJets=cms.InputTag(jetCollectionAK4LS_str)))
# ----------------------------------------------------------------------------
jetSubStructureVars_str = 'jetSubStructureVars%s' % NoLep_str
setattr(
process, jetSubStructureVars_str,
cms.EDProducer(
"JetSubstructureObservableProducer",
src=cms.InputTag(jetCollectionAK4LS_str),
kappa=cms.double(1.),
))
# ----------------------------------------------------------------------------
pileupJetId_str = 'pileupJetId%s' % NoLep_str
setattr(
process, pileupJetId_str,
pileupJetId.clone(
jets=cms.InputTag(jetCollectionAK4LS_str),
inputIsCorrected=True,
applyJec=False,
vertexes="offlineSlimmedPrimaryVertices",
))
# ----------------------------------------------------------------------------
jetsAK4LSWithUserData_str = 'jetsAK4LSWithUserData%s' % suffix
setattr(
process, jetsAK4LSWithUserData_str,
process.updatedJetsWithUserData.clone(
src=cms.InputTag(jetCollectionAK4LS_str),
userFloats=cms.PSet(
leadTrackPt=cms.InputTag("%s:leadTrackPt" % bJetVars_str),
leptonPtRel=cms.InputTag("%s:leptonPtRel" % bJetVars_str),
leptonPtRatio=cms.InputTag("%s:leptonPtRatio" % bJetVars_str),
leptonPtRelInv=cms.InputTag("%s:leptonPtRelInv" %
bJetVars_str),
leptonPtRelv0=cms.InputTag("%s:leptonPtRelv0" % bJetVars_str),
leptonPtRatiov0=cms.InputTag("%s:leptonPtRatiov0" %
bJetVars_str),
leptonPtRelInvv0=cms.InputTag("%s:leptonPtRelInvv0" %
bJetVars_str),
leptonDeltaR=cms.InputTag("%s:leptonDeltaR" % bJetVars_str),
leptonPt=cms.InputTag("%s:leptonPt" % bJetVars_str),
vtxPt=cms.InputTag("%s:vtxPt" % bJetVars_str),
vtxMass=cms.InputTag("%s:vtxMass" % bJetVars_str),
vtx3dL=cms.InputTag("%s:vtx3dL" % bJetVars_str),
vtx3deL=cms.InputTag("%s:vtx3deL" % bJetVars_str),
ptD=cms.InputTag("%s:ptD" % bJetVars_str),
genPtwNu=cms.InputTag("%s:genPtwNu" % bJetVars_str),
qgl=cms.InputTag("%s:qgLikelihood" % qgtagger_str),
jetCharge=cms.InputTag("%s:jetCharge" %
jetSubStructureVars_str),
pull_dEta=cms.InputTag("%s:pullDEta" %
jetSubStructureVars_str),
pull_dPhi=cms.InputTag("%s:pullDPhi" %
jetSubStructureVars_str),
pull_dR=cms.InputTag("%s:pullDR" % jetSubStructureVars_str),
puIdDisc=cms.InputTag("%s:fullDiscriminant" % pileupJetId_str),
),
userInts=cms.PSet(
tightId=cms.InputTag(tightJetIdAK4LS_str),
tightIdLepVeto=cms.InputTag(tightJetIdLepVetoAK4LS_str),
vtxNtrk=cms.InputTag("%s:vtxNtrk" % bJetVars_str),
leptonPdgId=cms.InputTag("%s:leptonPdgId" % bJetVars_str),
puId=cms.InputTag("%s:fullId" % pileupJetId_str),
)))
for modifier in run2_miniAOD_80XLegacy, run2_nanoAOD_94X2016:
modifier.toModify(
getattr(process, jetsAK4LSWithUserData_str).userInts,
looseId=cms.InputTag(looseJetIdAK4LS_str),
)
#----------------------------------------------------------------------------
bjetNN_str = 'bjetNN%s' % NoLep_str
setattr(
process, bjetNN_str,
process.bjetNN.clone(src=cms.InputTag(jetsAK4LSWithUserData_str), ))
#----------------------------------------------------------------------------
# add lepton-subtracted AK4 jets to nanoAOD Ntuple
jetAK4LSTable_str = 'jetAK4LS%sTable' % suffix # NB! must end with 'Table'
setattr(
process, jetAK4LSTable_str,
process.jetTable.clone(
src=cms.InputTag(jetsAK4LSWithUserData_str),
name=cms.string("JetAK4LS%s" % suffix),
doc=cms.string("lepton-subtracted ak4 jets"),
externalVariables=cms.PSet(
bRegCorr=ExtVar(
cms.InputTag("%s:corr" % bjetNN_str),
float,
doc="pt correction for b-jet energy regression",
precision=12),
bRegRes=ExtVar(cms.InputTag("%s:res" % bjetNN_str),
float,
doc="res on pt corrected with b-jet regression",
precision=8),
)))
getattr(process,
jetAK4LSTable_str).variables.puId = Var("userInt('puId')",
int,
doc="Pilup ID flags")
getattr(process, jetAK4LSTable_str).variables.puIdDisc = Var(
"userFloat('puIdDisc')", float, doc="Pilup ID discriminant")
if runOnMC:
getattr(process, jetAK4LSTable_str).variables.partonFlavour = Var(
"partonFlavour()", int, doc="flavour from parton matching")
getattr(process, jetAK4LSTable_str).variables.hadronFlavour = Var(
"hadronFlavour()", int, doc="flavour from hadron ghost clustering")
getattr(process, jetAK4LSTable_str).variables.genJetIdx = Var(
"?genJetFwdRef().backRef().isNonnull()?genJetFwdRef().backRef().key():-1",
int,
doc="index of matched gen jet")
#----------------------------------------------------------------------------
# produce lepton-subtracted generator-level jets
genjetAK4LS_str = 'genJetAK4LS'
if not hasattr(process, genjetAK4LS_str):
setattr(
process, genjetAK4LS_str,
ak4GenJets.clone(
src=cms.InputTag(LEPTONLESSGENPARTICLEPRODUCER_STR)))
# add lepton-subtracted generator-level jets to nanoAOD Ntuple
genjetAK4LSTable_str = 'genJetAK4LSTable'
if not hasattr(process, genjetAK4LSTable_str):
setattr(
process, genjetAK4LSTable_str,
process.genJetTable.clone(
src=cms.InputTag(genjetAK4LS_str),
name=cms.string("GenJetAK4LS"),
doc=cms.string(
"genJetsAK4LS, i.e. ak4 Jets made with visible genparticles excluding prompt leptons and leptons from tau decays"
),
))
# add information on generator-level parton flavor to reconstructed jets
genJetFlavourAssociationAK4LS_str = 'genJetFlavourAssociationAK4LS%s' % suffix
setattr(
process, genJetFlavourAssociationAK4LS_str,
process.genJetFlavourAssociation.clone(
jets=cms.InputTag(genjetAK4LS_str)))
genJetFlavourAK4LSTable = 'genJetFlavourAK4LS%sTable' % suffix # NB! must end with 'Table'
setattr(
process, genJetFlavourAK4LSTable,
process.genJetFlavourTable.clone(
src=cms.InputTag(genjetAK4LS_str),
name=cms.string("GenJetAK4LS"),
jetFlavourInfos=cms.InputTag(
genJetFlavourAssociationAK4LS_str)))
#----------------------------------------------------------------------------
leptonSubtractedJetSequence = cms.Sequence(
leptonSubtractedPFCandsSequence +
getattr(process, jetSequenceAK4LS_str) +
getattr(process, slimmedJetCollectionAK4LS_str) +
getattr(process, tightJetIdAK4LS_str) +
getattr(process, tightJetIdLepVetoAK4LS_str) +
getattr(process, bJetVars_str) + getattr(process, qgtagger_str) +
getattr(process, jetSubStructureVars_str) +
getattr(process, pileupJetId_str) +
getattr(process, jetsAK4LSWithUserData_str) +
getattr(process, bjetNN_str) + getattr(process, jetAK4LSTable_str))
if runOnMC:
leptonSubtractedJetSequence += getattr(
process, genjetAK4LS_str) + getattr(process, genjetAK4LSTable_str)
leptonSubtractedJetSequence += getattr(
process, genJetFlavourAssociationAK4LS_str) + getattr(
process, genJetFlavourAK4LSTable)
#----------------------------------------------------------------------------
_leptonSubtractedJetSequence_80X = leptonSubtractedJetSequence.copy()
_leptonSubtractedJetSequence_80X.insert(
_leptonSubtractedJetSequence_80X.index(
getattr(process, tightJetIdAK4LS_str)),
getattr(process, looseJetIdAK4LS_str))
run2_miniAOD_80XLegacy.toReplaceWith(leptonSubtractedJetSequence,
_leptonSubtractedJetSequence_80X)
_leptonSubtractedJetSequence_94X2016 = leptonSubtractedJetSequence.copy()
_leptonSubtractedJetSequence_94X2016.insert(
_leptonSubtractedJetSequence_94X2016.index(
getattr(process, tightJetIdAK4LS_str)),
getattr(process, looseJetIdAK4LS_str))
run2_nanoAOD_94X2016.toReplaceWith(leptonSubtractedJetSequence,
_leptonSubtractedJetSequence_94X2016)
leptonSubtractedJetSequence_str = 'leptonSubtractedJetSequenceAK4LS%s' % suffix
setattr(process, leptonSubtractedJetSequence_str,
leptonSubtractedJetSequence)
process.nanoSequence += getattr(process, leptonSubtractedJetSequence_str)
process.nanoSequenceMC += getattr(process, leptonSubtractedJetSequence_str)
process.nanoSequenceFS += getattr(process, leptonSubtractedJetSequence_str)
tauSrc_InputTag = cms.InputTag('slimmedTausNewID')
jetSrc_InputTag = cms.InputTag('selectedUpdatedPatJetsNewDFTraining')
objects_InputTag = cms.InputTag('slimmedPatTrigger')
# Update pileup jet ID following recommendations from https://twiki.cern.ch/twiki/bin/viewauth/CMS/PileupJetID
from RecoJets.JetProducers.PileupJetID_cfi import pileupJetId, _chsalgos_81x, _chsalgos_94x, _chsalgos_102x
pu_jet_id_wp = {
'Run2016': _chsalgos_81x,
'Run2017': _chsalgos_94x,
'Run2018': _chsalgos_102x
}
process.updatedPileupJetId = pileupJetId.clone(
jets=jetSrc_InputTag,
inputIsCorrected=True,
applyJec=False,
vertexes=cms.InputTag("offlineSlimmedPrimaryVertices"),
algos=cms.VPSet(pu_jet_id_wp[period]),
)
### Top gen level info
process.topGenSequence = cms.Sequence()
if options.saveGenTopInfo:
process.load("TopQuarkAnalysis.TopEventProducers.sequences.ttGenEvent_cff")
process.decaySubset.fillMode = cms.string("kME")
process.initSubset.src = cms.InputTag('prunedGenParticles')
process.decaySubset.src = cms.InputTag('prunedGenParticles')
process.decaySubset.runMode = cms.string("Run2")
process.topGenSequence += process.makeGenEvt
if options.anaChannels == 'all':