TrimmedVertexFinder.h

Go to the documentation of this file.

#ifndef _RecoVertex_TrimmedVertexFinder_H_
#define _RecoVertex_TrimmedVertexFinder_H_

#include "RecoVertex/VertexPrimitives/interface/TransientVertex.h"
#include "TrackingTools/TransientTrack/interface/TransientTrack.h"
#include "RecoVertex/VertexPrimitives/interface/CachingVertex.h"
#include "RecoVertex/VertexPrimitives/interface/VertexFitter.h"
#include "RecoVertex/VertexPrimitives/interface/VertexUpdator.h"
#include "RecoVertex/VertexPrimitives/interface/VertexTrackCompatibilityEstimator.h"
#include <vector>

class TrimmedVertexFinder {
public:
  typedef ReferenceCountingPointer<VertexTrack<5> > RefCountedVertexTrack;
  typedef ReferenceCountingPointer<LinearizedTrackState<5> > RefCountedLinearizedTrackState;

  TrimmedVertexFinder(const VertexFitter<5>* vf,
                      const VertexUpdator<5>* vu,
                      const VertexTrackCompatibilityEstimator<5>* ve);

  TrimmedVertexFinder(const TrimmedVertexFinder& other);
  ~TrimmedVertexFinder();

  std::vector<TransientVertex> vertices(std::vector<reco::TransientTrack>& remain) const;

  std::vector<TransientVertex> vertices(std::vector<reco::TransientTrack>& remain,
                                        const reco::BeamSpot& s,
                                        bool use_beamspot = true) const;

  float trackCompatibilityCut() const { return theMinProb; }

  void setTrackCompatibilityCut(float cut) { theMinProb = cut; }

  TrimmedVertexFinder* clone() const { return new TrimmedVertexFinder(*this); }

private:
  // finds least compatible track
  // returns vtxTracks.end() if all tracks are compatible
  //
  std::vector<RefCountedVertexTrack>::iterator theWorst(const CachingVertex<5>& vtx,
                                                        std::vector<RefCountedVertexTrack>& vtxTracks,
                                                        float cut) const;

  VertexFitter<5>* theFitter;
  VertexUpdator<5>* theUpdator;
  VertexTrackCompatibilityEstimator<5>* theEstimator;
  float theMinProb;
};

#endif