---

ConfigurableTrimmedVertexFinder.h

Go to the documentation of this file.

#ifndef _ConfigurableTrimmedVertexFinder_H_
#define _ConfigurableTrimmedVertexFinder_H_

#include "RecoVertex/VertexPrimitives/interface/VertexReconstructor.h"
#include "RecoVertex/TrimmedKalmanVertexFinder/interface/TrimmedVertexFinder.h"
#include "RecoVertex/TrimmedKalmanVertexFinder/interface/TrimmedTrackFilter.h"
#include <vector>

class ConfigurableTrimmedVertexFinder : public VertexReconstructor {

public:

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

  virtual ~ConfigurableTrimmedVertexFinder() {}

  virtual vector<TransientVertex> 
    vertices(const vector<reco::TransientTrack> & tracks) const;

  vector<TransientVertex> 
    vertices( const vector<reco::TransientTrack> & tracks,
              vector<reco::TransientTrack>& unused) const;

  float ptCut() const { return theFilter.ptCut(); }
  const TrimmedTrackFilter & trackFilter() const { 
    return theFilter; 
  }
  float trackCompatibilityCut() const { return theTrackCompatibilityToPV; }
  float trackCompatibilityToSV() const { return theTrackCompatibilityToSV; }
  float vertexFitProbabilityCut() const { return theVtxFitProbCut; }
  int maxNbOfVertices() const { return theMaxNbOfVertices; }

  void setPtCut(float cut) { theFilter.setPtCut(cut); }
  void setTrackCompatibilityCut(float cut) {
    theTrackCompatibilityToPV = cut;
  }
  void setTrackCompatibilityToSV(float cut) {
    theTrackCompatibilityToSV = cut;
  }
  void setVertexFitProbabilityCut(float cut) { theVtxFitProbCut = cut; }
  void setMaxNbOfVertices(int max) { theMaxNbOfVertices = max; }

  virtual ConfigurableTrimmedVertexFinder * clone() const {
    return new ConfigurableTrimmedVertexFinder(*this);
  }

protected:

  virtual void resetEvent(const vector<reco::TransientTrack> & tracks) const {}

  virtual void analyseInputTracks(const vector<reco::TransientTrack> & tracks) 
    const {}

  virtual void 
    analyseClusterFinder(const vector<TransientVertex> & vts, 
                         const vector<reco::TransientTrack> & remain) 
    const {}

  virtual void analyseVertexCandidates(const vector<TransientVertex> & vts) 
    const {}

  virtual void analyseFoundVertices(const vector<TransientVertex> & vts) 
    const {}


private:

  // separate tracks passing the filter
  //  void separateTracks(vector<TransientTrack>& filtered, 
  //                  vector<TransientTrack>& unused) const;

  // find vertex candidates
  vector<TransientVertex> 
    vertexCandidates(const vector<reco::TransientTrack> & tracks, 
                     vector<reco::TransientTrack>& unused) const;

  // remove bad candidates
  vector<TransientVertex> 
    clean(const vector<TransientVertex> & candidates) const;
  
  // 
  mutable TrimmedVertexFinder theClusterFinder;
  float theVtxFitProbCut;
  float theTrackCompatibilityToPV;
  float theTrackCompatibilityToSV;
  int theMaxNbOfVertices;
  TrimmedTrackFilter theFilter;

};

#endif

---