Add Pileup Jet ID for AK4 Puppi jets

cms-sw · Sep 26, 2024 · 298c232 · 298c232
1 parent fbed391
commit 298c232
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/PhysicsTools/PatAlgos/python/slimming/miniAOD_tools.py b/PhysicsTools/PatAlgos/python/slimming/miniAOD_tools.py
@@ -361,7 +361,7 @@ def _add_deepFlavour(process):
     (pp_on_AA).toReplaceWith(
         process.makePatTausTask, _makePatTausTaskWithTauReReco
         )
-    
+
     # Adding puppi jets
     process.load('CommonTools.PileupAlgos.Puppi_cff')
     process.load('RecoJets.JetProducers.ak4PFJets_cfi')
@@ -383,6 +383,10 @@ def _add_deepFlavour(process):
     )
     task.add(process.patJetPuppiCharge)
 
+    ## PUJetID for ak4PFJetsPuppi
+    process.load("RecoJets.JetProducers.PileupJetID_cfi")
+    task.add(process.pileUpJetIDPuppiTask)
+
     def _add_jetsPuppi(process):
         from PhysicsTools.PatAlgos.tools.jetTools import addJetCollection
         noDeepFlavourDiscriminators = [x.value() if isinstance(x, cms.InputTag) else x for x in process.patJets.discriminatorSources 
@@ -394,9 +398,13 @@ def _add_jetsPuppi(process):
                      )
 
         process.patJetGenJetMatchPuppi.matched = 'slimmedGenJets'
-    
+
         process.patJetsPuppi.jetChargeSource = cms.InputTag("patJetPuppiCharge")
-
+
+        ## Store PUJetID variables in patJetsPuppi
+        process.patJetsPuppi.userData.userFloats.src += [cms.InputTag("pileupJetIdPuppi:fullDiscriminant")]
+        process.patJetsPuppi.userData.userInts.src += [cms.InputTag("pileupJetIdPuppi:fullId")]
+
         process.selectedPatJetsPuppi.cut = cms.string("pt > 10")
 
         from PhysicsTools.PatAlgos.slimming.applyDeepBtagging_cff import applyDeepBtagging