TrimmedVertexFitter.h

Go to the documentation of this file.

#ifndef _TrimmedVertexFitter_H_
#define _TrimmedVertexFitter_H_

#include "RecoVertex/VertexPrimitives/interface/VertexFitter.h"
#include "RecoVertex/TrimmedKalmanVertexFinder/interface/KalmanTrimmedVertexFinder.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "DataFormats/BeamSpot/interface/BeamSpot.h"
/*
 *  Turn the TrimmedVertexFinder into a VertexFitter.
 */

class TrimmedVertexFitter : public VertexFitter<5> {

public:

  typedef CachingVertex<5>::RefCountedVertexTrack RefCountedVertexTrack;
  typedef ReferenceCountingPointer<LinearizedTrackState<5> > RefCountedLinearizedTrackState;

  TrimmedVertexFitter();
  TrimmedVertexFitter(const edm::ParameterSet & pSet);

  virtual ~TrimmedVertexFitter(){}

  virtual CachingVertex<5> vertex(const std::vector<reco::TransientTrack> & tracks) const;

  virtual CachingVertex<5> vertex(const std::vector<RefCountedVertexTrack> & tracks) const;
  
  virtual CachingVertex<5> vertex(const std::vector<RefCountedVertexTrack> & tracks,
      const reco::BeamSpot & spot ) const;

  virtual CachingVertex<5> vertex(const std::vector<reco::TransientTrack> & tracks,
            const GlobalPoint& linPoint) const;

  virtual CachingVertex<5> vertex(const std::vector<reco::TransientTrack> & tracks,
            const GlobalPoint& priorPos,
            const GlobalError& priorError) const;

  virtual CachingVertex<5> vertex(const std::vector<RefCountedVertexTrack> & tracks,
            const GlobalPoint& priorPos,
            const GlobalError& priorError) const;

  virtual CachingVertex<5> vertex(const std::vector<reco::TransientTrack> & tracks,
        const reco::BeamSpot& beamSpot) const;


   // Clone method
  TrimmedVertexFitter * clone() const;


  void setPtCut ( float cut );
  void setTrackCompatibilityCut ( float cut );
  void setVertexFitProbabilityCut ( float cut );

private:
  KalmanTrimmedVertexFinder theRector;
  double ptcut;
};

#endif