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