KalmanVertexFitter.cc

Go to the documentation of this file.

#include "RecoVertex/KalmanVertexFit/interface/KalmanVertexFitter.h"
#include "RecoVertex/KalmanVertexFit/interface/KalmanVertexUpdator.h"
#include "RecoVertex/VertexTools/interface/SequentialVertexSmoother.h"
#include "RecoVertex/VertexTools/interface/DummyVertexSmoother.h"
#include "RecoVertex/KalmanVertexFit/interface/KalmanVertexTrackUpdator.h"
#include "RecoVertex/KalmanVertexFit/interface/KalmanSmoothedVertexChi2Estimator.h"
#include "RecoVertex/KalmanVertexFit/interface/KalmanTrackToTrackCovCalculator.h"
#include "RecoVertex/LinearizationPointFinders/interface/FsmwLinearizationPointFinder.h"


KalmanVertexFitter::KalmanVertexFitter(bool useSmoothing )
{
  edm::ParameterSet pSet = defaultParameters();
  setup(pSet, useSmoothing);
}

KalmanVertexFitter::KalmanVertexFitter(const edm::ParameterSet& pSet,  bool useSmoothing )
{
  setup(pSet, useSmoothing);
}

void KalmanVertexFitter::setup(const edm::ParameterSet& pSet,  bool useSmoothing )
{
  if (useSmoothing) {
    KalmanVertexTrackUpdator<5> vtu;
    KalmanSmoothedVertexChi2Estimator<5> vse;
    KalmanTrackToTrackCovCalculator<5> covCalc;
    SequentialVertexSmoother<5> smoother(vtu, vse, covCalc);
    theSequentialFitter 
      = new SequentialVertexFitter<5>(pSet, FsmwLinearizationPointFinder(20, -2., 0.4, 10.), 
                   KalmanVertexUpdator<5>(), 
                   smoother, LinearizedTrackStateFactory());
  }
  else {
    DummyVertexSmoother<5> smoother;
    theSequentialFitter 
      = new SequentialVertexFitter<5>(pSet, FsmwLinearizationPointFinder(20, -2., 0.4, 10.), 
                   KalmanVertexUpdator<5>(), 
                   smoother, LinearizedTrackStateFactory());
  }
}


edm::ParameterSet KalmanVertexFitter::defaultParameters() const 
{
  edm::ParameterSet pSet;
  pSet.addParameter<double>("maxDistance", 0.01);
  pSet.addParameter<int>("maxNbrOfIterations", 10); //10
  return pSet;
}