KinematicParticleVertexFitter.h

Go to the documentation of this file.

#ifndef KinematicParticleVertexFitter_H
#define KinematicParticleVertexFitter_H

#include "RecoVertex/KinematicFitPrimitives/interface/RefCountedKinematicTree.h"
#include "RecoVertex/VertexTools/interface/LinearizationPointFinder.h"
#include "RecoVertex/VertexTools/interface/VertexTrackFactory.h"
#include "RecoVertex/KinematicFit/interface/InputSort.h"
#include "RecoVertex/VertexPrimitives/interface/VertexFitter.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"

class KinematicParticleVertexFitter {
public:
  KinematicParticleVertexFitter();

  KinematicParticleVertexFitter(const edm::ParameterSet &pSet);

  /*
 * Constructor with the LinearizationPointFinder
 * Linearization point finder should have an 
 * ability to find point out of set of FreeTrajectoryStates
 * LMSLinearizationPointFinder is used as default.
 */
  //  KinematicParticleVertexFitter(const LinearizationPointFinder& finder);

  ~KinematicParticleVertexFitter();

  RefCountedKinematicTree fit(const std::vector<RefCountedKinematicParticle> &particles) const;

private:
  edm::ParameterSet defaultParameters() const;
  void setup(const edm::ParameterSet &pSet);

  VertexFitter<6> *fitter;
  LinearizationPointFinder *pointFinder;
  VertexTrackFactory<6> *vFactory;
};
#endif