Test option for the KalmanVertexFitter with useMuonSystemBounds

Author: Lovisa

RecoVertex/KalmanVertexFit/interface/KalmanVertexFitter.h

@@ -27,24 +27,20 @@ class KalmanVertexFitter : public VertexFitter<5> {
    * \param useSmoothing Specifies whether the tracks should be refit or not.
    */
 
-  KalmanVertexFitter(bool useSmoothing = false);
+  KalmanVertexFitter(bool useSmoothing = false, bool useMuonSystemBounds = false);
 
   /**
    * Same as above, using a ParameterSet to set the convergence criteria
    */
 
-  KalmanVertexFitter(const edm::ParameterSet& pSet, bool useSmoothing = false);
+  KalmanVertexFitter(const edm::ParameterSet& pSet, bool useSmoothing = false, bool useMuonSystemBounds = false);
 
   KalmanVertexFitter(const KalmanVertexFitter& other) : theSequentialFitter(other.theSequentialFitter->clone()) {}
 
   ~KalmanVertexFitter() override { delete theSequentialFitter; }
 
   KalmanVertexFitter* clone() const override { return new KalmanVertexFitter(*this); }
 
-  void setTrackerBounds(float trackerBoundsRadius, float trackerBoundsHalfLength) {
-    theSequentialFitter->setTrackerBounds(trackerBoundsRadius, trackerBoundsHalfLength);
-  }
-
 public:
   typedef CachingVertex<5>::RefCountedVertexTrack RefCountedVertexTrack;
 
@@ -108,11 +104,14 @@ class KalmanVertexFitter : public VertexFitter<5> {
   //  edm::ParameterSet defaultParameters() const;
 
 private:
-  void setup(const edm::ParameterSet& pSet, bool useSmoothing);
+  void setup(const edm::ParameterSet& pSet, bool useSmoothing, bool useMuonSystemBounds);
 
   edm::ParameterSet defaultParameters() const;
 
   SequentialVertexFitter<5>* theSequentialFitter;
+
+  float muonSystemBoundsRadius{740.};
+  float muonSystemBoundsHalfLength{960.};
 };
 
 #endif

RecoVertex/KalmanVertexFit/src/KalmanVertexFitter.cc

@@ -7,14 +7,16 @@
 #include "RecoVertex/KalmanVertexFit/interface/KalmanTrackToTrackCovCalculator.h"
 #include "RecoVertex/LinearizationPointFinders/interface/FsmwLinearizationPointFinder.h"
 
-KalmanVertexFitter::KalmanVertexFitter(bool useSmoothing) {
+KalmanVertexFitter::KalmanVertexFitter(bool useSmoothing, bool useMuonSystemBounds) {
   edm::ParameterSet pSet = defaultParameters();
-  setup(pSet, useSmoothing);
+  setup(pSet, useSmoothing, useMuonSystemBounds);
 }
 
-KalmanVertexFitter::KalmanVertexFitter(const edm::ParameterSet& pSet, bool useSmoothing) { setup(pSet, useSmoothing); }
+KalmanVertexFitter::KalmanVertexFitter(const edm::ParameterSet& pSet, bool useSmoothing, bool useMuonSystemBounds) {
+  setup(pSet, useSmoothing, useMuonSystemBounds);
+}
 
-void KalmanVertexFitter::setup(const edm::ParameterSet& pSet, bool useSmoothing) {
+void KalmanVertexFitter::setup(const edm::ParameterSet& pSet, bool useSmoothing, bool useMuonSystemBounds) {
   if (useSmoothing) {
     KalmanVertexTrackUpdator<5> vtu;
     KalmanSmoothedVertexChi2Estimator<5> vse;
@@ -33,6 +35,9 @@ void KalmanVertexFitter::setup(const edm::ParameterSet& pSet, bool useSmoothing)
                                                         smoother,
                                                         LinearizedTrackStateFactory());
   }
+  if (useMuonSystemBounds) {
+    theSequentialFitter->setTrackerBounds(muonSystemBoundsRadius, muonSystemBoundsHalfLength);
+  }
 }
 
 edm::ParameterSet KalmanVertexFitter::defaultParameters() const {