backport of 42646

cms-sw · Sep 12, 2023 · 7135a5c · 7135a5c
1 parent 382249d
commit 7135a5c
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/PhysicsTools/NanoAOD/python/muons_cff.py b/PhysicsTools/NanoAOD/python/muons_cff.py
@@ -119,6 +119,11 @@
     weightFile = "PhysicsTools/NanoAOD/data/mu_BDTG_2016.weights.xml",
 )
 
+from TrackingTools.TransientTrack.TransientTrackBuilder_cfi import *
+muonBSConstrain = cms.EDProducer("MuonBeamspotConstraintValueMapProducer",
+    src = cms.InputTag("linkedObjects","muons"),
+)
+
 muonTable = simpleCandidateFlatTableProducer.clone(
     src = cms.InputTag("linkedObjects","muons"),
     name = cms.string("Muon"),

diff --git a/PhysicsTools/NanoAOD/python/nanoDQM_cfi.py b/PhysicsTools/NanoAOD/python/nanoDQM_cfi.py
@@ -504,6 +504,9 @@
             ),
             plots = cms.VPSet(
                 Count1D('_size', 5, -0.5, 4.5, 'slimmedMuons after basic selection (pt > 3 && track.isNonnull && isLooseMuon)'),
+                Plot1D('bsConstrainedPt', '', 20, 0., 200., 'pt with beamspot constraint'),
+                Plot1D('bsConstrainedPtErr', '', 20, 0., 20., 'pt error w. beamspot constraint'),
+                Plot1D('bsConstrainedChi2', '', 20, 0., 40., 'chi2 of beamspot constraint'),
                 Plot1D('charge', 'charge', 3, -1.5, 1.5, 'electric charge'),
                 Plot1D('dxy', 'dxy', 20, -0.1, 0.1, 'dxy (with sign) wrt first PV, in cm'),
                 Plot1D('dxyErr', 'dxyErr', 20, 0, 0.1, 'dxy uncertainty, in cm'),

diff --git a/RecoMuon/GlobalTrackingTools/plugins/BuildFile.xml b/RecoMuon/GlobalTrackingTools/plugins/BuildFile.xml
@@ -8,6 +8,7 @@
   <use name="Geometry/Records"/>
   <use name="RecoMuon/TrackingTools"/>
   <use name="RecoMuon/GlobalTrackingTools"/>
+  <use name="RecoVertex/KalmanVertexFit"/>
   <use name="boost_regex"/>
   <flags EDM_PLUGIN="1"/>
 </library>