Merge pull request #46752 from martinamalberti/mm_fixHitPositionTrack…

…Extender MTD reconstruction: save correct outermost hit position also for tracks w/o last hit in mtd
cms-sw · Dec 2, 2024 · 81e5ef6 · 81e5ef6
2 parents 3be60c7 + 550e9be
commit 81e5ef6
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/RecoMTD/TrackExtender/plugins/TrackExtenderWithMTD.cc b/RecoMTD/TrackExtender/plugins/TrackExtenderWithMTD.cc
@@ -620,6 +620,8 @@ class TrackExtenderWithMTDT : public edm::stream::EDProducer<> {
   const bool useSimVertex_;
   const float dzCut_;
   const float bsTimeSpread_;
+
+  static constexpr float trackMaxBtlEta_ = 1.5;
 };
 
 template <class TrackCollection>
@@ -1004,9 +1006,15 @@ void TrackExtenderWithMTDT<TrackCollection>::produce(edm::Event& ev, const edm::
 #endif
         npixBarrel.push_back(backtrack.hitPattern().numberOfValidPixelBarrelHits());
         npixEndcap.push_back(backtrack.hitPattern().numberOfValidPixelEndcapHits());
-        outermostHitPosition.push_back(
-            mBTL.hit ? (float)(*track).outerRadius()
-                     : (float)(*track).outerZ());  // save R of the outermost hit for BTL, z for ETL.
+
+        if (mBTL.hit || mETL.hit) {
+          outermostHitPosition.push_back(
+              mBTL.hit ? (float)(*track).outerRadius()
+                       : (float)(*track).outerZ());  // save R of the outermost hit for BTL, z for ETL.
+        } else {
+          outermostHitPosition.push_back(std::abs(track->eta()) < trackMaxBtlEta_ ? (float)(*track).outerRadius()
+                                                                                  : (float)(*track).outerZ());
+        }
 
         LogTrace("TrackExtenderWithMTD") << "TrackExtenderWithMTD: tmtd " << tmtdMap << " +/- " << sigmatmtdMap
                                          << " t0 " << t0Map << " +/- " << sigmat0Map << " tof pi/K/p " << tofpiMap