Fixed bug with threshold and only compute cxPtR2 once.

rtschmitz · Apr 27, 2023 · 7b74c03 · 7b74c03
1 parent 98a11b8
commit 7b74c03
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/RecoVertex/V0Producer/src/V0Fitter.cc b/RecoVertex/V0Producer/src/V0Fitter.cc
@@ -186,10 +186,12 @@ void V0Fitter::fitAll(const edm::Event& iEvent,
 
       // the POCA should at least be in the sensitive volume
       GlobalPoint cxPt = cApp.crossingPoint();
-      if ((cxPt.x() * cxPt.x() + cxPt.y() * cxPt.y()) > 120. * 120. || std::abs(cxPt.z()) > 300.)
+      const double cxPtR2 = cxPt.x() * cxPt.x() + cxPt.y() * cxPt.y();
+      if (cxPtR2 > 120. * 120. || std::abs(cxPt.z()) > 300.)
         continue;
 
-      if (cxPt.x() * cxPt.x() + cxPt.y() * cxPt.y() < innerOuterTkDCAThreshold_) {
+      // allow for different DCA cuts depending on position of POCA
+      if (cxPtR2 < innerOuterTkDCAThreshold_ * innerOuterTkDCAThreshold_) {
         if (dca > innerTkDCACut_)
           continue;
       } else {