Merge pull request #34860 from lathomas/TkMet_NanoFix

Bug fix for track MET in MET correction tools
cms-sw · Aug 27, 2021 · 1cbe017 · 1cbe017
2 parents a75f99d + 0e55061
commit 1cbe017
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 6 deletions.
diff --git a/PhysicsTools/PatAlgos/python/slimming/miniAOD_tools.py b/PhysicsTools/PatAlgos/python/slimming/miniAOD_tools.py
@@ -6,6 +6,8 @@
 
 from Configuration.ProcessModifiers.pp_on_AA_cff import pp_on_AA
 
+from PhysicsTools.PatUtils.tools.pfforTrkMET_cff import *
+
 def miniAOD_customizeCommon(process):
     process.patMuons.isoDeposits = cms.PSet()
     process.patElectrons.isoDeposits = cms.PSet()
@@ -216,10 +218,7 @@ def _add_slimmedMETsNoHF(process):
     #  ==================  CHSMET 
 
     #  ==================  TrkMET 
-    process.TrkCands = cms.EDFilter("CandPtrSelector",
-                                    src=cms.InputTag("packedPFCandidates"),
-                                    cut=cms.string("charge()!=0 && pvAssociationQuality()>=4 && vertexRef().key()==0")
-                                    )
+    process.TrkCands = chargedPackedCandsForTkMet.clone()
     task.add(process.TrkCands)
 
     process.pfMetTrk = pfMet.clone(src = 'TrkCands')

diff --git a/PhysicsTools/PatUtils/python/tools/pfforTrkMET_cff.py b/PhysicsTools/PatUtils/python/tools/pfforTrkMET_cff.py
@@ -0,0 +1,6 @@
+import FWCore.ParameterSet.Config as cms
+
+chargedPackedCandsForTkMet = cms.EDFilter("CandPtrSelector",
+                                          src=cms.InputTag("packedPFCandidates"),
+                                          cut=cms.string("charge()!=0 && pvAssociationQuality()>=4 && vertexRef().key()==0")
+                                      )
diff --git a/PhysicsTools/PatUtils/python/tools/runMETCorrectionsAndUncertainties.py b/PhysicsTools/PatUtils/python/tools/runMETCorrectionsAndUncertainties.py
@@ -6,6 +6,7 @@
 from PhysicsTools.PatAlgos.tools.helpers import getPatAlgosToolsTask, addToProcessAndTask
 from PhysicsTools.PatAlgos.tools.jetTools import switchJetCollection
 import CommonTools.CandAlgos.candPtrProjector_cfi as _mod
+from PhysicsTools.PatUtils.tools.pfforTrkMET_cff import *
 
 
 def isValidInputTag(input):
@@ -1727,8 +1728,8 @@ def miniAODConfigurationPre(self, process, patMetModuleSequence, pfCandCollectio
         patMetModuleSequence += getattr(process, "pfCHS")
         patMetModuleSequence += getattr(process, "pfMetCHS")
         patMetModuleSequence += getattr(process, "patCHSMet")
-
-        pfTrk = cms.EDFilter("CandPtrSelector", src = cms.InputTag("packedPFCandidates"), cut = cms.string("fromPV(0) > 0 && charge()!=0"))
+        
+        pfTrk = chargedPackedCandsForTkMet.clone()
         addToProcessAndTask("pfTrk", pfTrk, process, task)
         pfMetTrk = pfMet.clone(src = 'pfTrk')
         addToProcessAndTask("pfMetTrk", pfMetTrk, process, task)