ConfigurableTrimmedVertexFinder Class Reference

#include <ConfigurableTrimmedVertexFinder.h>

Inheritance diagram for ConfigurableTrimmedVertexFinder:

Public Member Functions
virtual ConfigurableTrimmedVertexFinder *	clone () const
	ConfigurableTrimmedVertexFinder (const VertexFitter< 5 > *vf, const VertexUpdator< 5 > *vu, const VertexTrackCompatibilityEstimator< 5 > *ve)
int	maxNbOfVertices () const
float	ptCut () const
void	setMaxNbOfVertices (int max)
void	setParameters (const edm::ParameterSet &)
void	setPtCut (float cut)
void	setTrackCompatibilityCut (float cut)
void	setTrackCompatibilityToSV (float cut)
void	setVertexFitProbabilityCut (float cut)
float	trackCompatibilityCut () const
float	trackCompatibilityToSV () const
const TrimmedTrackFilter &	trackFilter () const
float	vertexFitProbabilityCut () const
virtual std::vector < TransientVertex >	vertices (const std::vector< reco::TransientTrack > &tracks) const
virtual std::vector < TransientVertex >	vertices (const std::vector< reco::TransientTrack > &tracks, const reco::BeamSpot &spot) const
std::vector< TransientVertex >	vertices (const std::vector< reco::TransientTrack > &tracks, std::vector< reco::TransientTrack > &unused, const reco::BeamSpot &spot, bool use_spot) const
virtual	~ConfigurableTrimmedVertexFinder ()
Public Member Functions inherited from VertexReconstructor
	VertexReconstructor ()
virtual std::vector < TransientVertex >	vertices (const std::vector< reco::TransientTrack > &primaries, const std::vector< reco::TransientTrack > &tracks, const reco::BeamSpot &spot) const
virtual	~VertexReconstructor ()

Protected Member Functions
virtual void	analyseClusterFinder (const std::vector< TransientVertex > &vts, const std::vector< reco::TransientTrack > &remain) const
virtual void	analyseFoundVertices (const std::vector< TransientVertex > &vts) const
virtual void	analyseInputTracks (const std::vector< reco::TransientTrack > &tracks) const
virtual void	analyseVertexCandidates (const std::vector< TransientVertex > &vts) const
virtual void	resetEvent (const std::vector< reco::TransientTrack > &tracks) const

Private Member Functions
std::vector< TransientVertex >	clean (const std::vector< TransientVertex > &candidates) const
std::vector< TransientVertex >	vertexCandidates (const std::vector< reco::TransientTrack > &tracks, std::vector< reco::TransientTrack > &unused, const reco::BeamSpot &spot, bool use_spot) const

Private Attributes
TrimmedVertexFinder	theClusterFinder
TrimmedTrackFilter	theFilter
int	theMaxNbOfVertices
float	theTrackCompatibilityToPV
float	theTrackCompatibilityToSV
float	theVtxFitProbCut

Detailed Description

Algorithm to find a series of distinct vertices among the given set of tracks. The technique is:
1) use TrimmedTrackFilter to select tracks above a certain pT;
2) use TrimmedVertexFinder to split the set of tracks into a cluster of compatible tracks and a set of remaining tracks;
3) repeat 2) with the remaining set, and repeat as long as (a) a cluster of compatible tracks can be found or (b) the maximum number of clusters asked for is reached;
4) reject vertices with a low fit probability.

This algorithm has 5 parameters that can be set at runtime via the corresponding set() methods, or a ParameterSet:

• "ptCut" (default: 1.5 GeV/c) which defines the minimum pT of the tracks used to make vertices. This value overrides the corresponding configurable parameter of the TrimmedTrackFilter used internally.
  
• "trackCompatibilityCut" (default: 0.05) which defines the probability below which a track is considered incompatible with the 1st vertex candidate formed. This value overrides the corresponding configurable parameter of the TrimmedVertexFinder used internally.
  
• "trackCompatibilityToSV" (default: 0.01) which defines the probability below which a track is considered incompatible with the next vertex candidates formed. This value overrides the corresponding configurable parameter of the TrimmedVertexFinder used internally.
  
• "vertexFitProbabilityCut" (default: 0.01) which defines the probability below which a vertex is rejected.
  
• "maxNbOfVertices" (default: 0) which defines the maximum number of vertices searched for. 0 means all vertices which can be found.

Definition at line 44 of file ConfigurableTrimmedVertexFinder.h.

Constructor & Destructor Documentation

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

Definition at line 6 of file ConfigurableTrimmedVertexFinder.cc.

References TrimmedTrackFilter::setPtCut(), and theFilter.

Referenced by clone().

  : theClusterFinder(vf, vu, ve), theVtxFitProbCut(0.01), 
    theTrackCompatibilityToPV(0.05), theTrackCompatibilityToSV(0.01), 
    theMaxNbOfVertices(0)
{
  // default pt cut is 1.5 GeV
  theFilter.setPtCut(1.5);
}

virtual ConfigurableTrimmedVertexFinder::~ConfigurableTrimmedVertexFinder ( )

inlinevirtual

Definition at line 52 of file ConfigurableTrimmedVertexFinder.h.

{}

Member Function Documentation

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

inlineprotectedvirtual

Definition at line 105 of file ConfigurableTrimmedVertexFinder.h.

Referenced by vertexCandidates().

          {}

virtual void ConfigurableTrimmedVertexFinder::analyseFoundVertices ( const std::vector< TransientVertex > &  vts ) const

inlineprotectedvirtual

Definition at line 112 of file ConfigurableTrimmedVertexFinder.h.

Referenced by vertices().

          {}

virtual void ConfigurableTrimmedVertexFinder::analyseInputTracks ( const std::vector< reco::TransientTrack > &  tracks ) const

inlineprotectedvirtual

Definition at line 101 of file ConfigurableTrimmedVertexFinder.h.

Referenced by vertices().

          {}

virtual void ConfigurableTrimmedVertexFinder::analyseVertexCandidates ( const std::vector< TransientVertex > &  vts ) const

inlineprotectedvirtual

Definition at line 109 of file ConfigurableTrimmedVertexFinder.h.

Referenced by vertices().

          {}

std::vector< TransientVertex > ConfigurableTrimmedVertexFinder::clean ( const std::vector< TransientVertex > &  candidates ) const

private

Definition at line 138 of file ConfigurableTrimmedVertexFinder.cc.

References ChiSquaredProbability(), i, EgammaValidation_Wenu_cff::sel, and theVtxFitProbCut.

Referenced by vertices().

{
  std::vector<TransientVertex> sel;
  for (std::vector<TransientVertex>::const_iterator i = candidates.begin(); 
       i != candidates.end(); i++) {

    if (ChiSquaredProbability((*i).totalChiSquared(), (*i).degreesOfFreedom())
    > theVtxFitProbCut) { sel.push_back(*i); }
  }

  return sel;
}

virtual ConfigurableTrimmedVertexFinder* ConfigurableTrimmedVertexFinder::clone ( void  ) const

inlinevirtual

Clone method

Implements VertexReconstructor.

Definition at line 93 of file ConfigurableTrimmedVertexFinder.h.

References ConfigurableTrimmedVertexFinder().

Referenced by KalmanTrimmedVertexFinder::KalmanTrimmedVertexFinder().

                                                          {
    return new ConfigurableTrimmedVertexFinder(*this);
  }

int ConfigurableTrimmedVertexFinder::maxNbOfVertices ( ) const

inline

Definition at line 75 of file ConfigurableTrimmedVertexFinder.h.

References theMaxNbOfVertices.

Referenced by KalmanTrimmedVertexFinder::maxNbOfVertices().

{ return theMaxNbOfVertices; }

float ConfigurableTrimmedVertexFinder::ptCut ( ) const

inline

Access to parameters

Definition at line 68 of file ConfigurableTrimmedVertexFinder.h.

References TrimmedTrackFilter::ptCut(), and theFilter.

Referenced by KalmanTrimmedVertexFinder::ptCut().

{ return theFilter.ptCut(); }

virtual void ConfigurableTrimmedVertexFinder::resetEvent ( const std::vector< reco::TransientTrack > &  tracks ) const

inlineprotectedvirtual

Definition at line 99 of file ConfigurableTrimmedVertexFinder.h.

Referenced by vertices().

{}

void ConfigurableTrimmedVertexFinder::setMaxNbOfVertices ( int  max )

inline

Definition at line 89 of file ConfigurableTrimmedVertexFinder.h.

References max(), and theMaxNbOfVertices.

Referenced by KalmanTrimmedVertexFinder::setMaxNbOfVertices().

{ theMaxNbOfVertices = max; }

void ConfigurableTrimmedVertexFinder::setParameters

(

const edm::ParameterSet &

s

)

Set parameters

Definition at line 18 of file ConfigurableTrimmedVertexFinder.cc.

References edm::ParameterSet::getParameter(), TrimmedTrackFilter::setPtCut(), theFilter, theMaxNbOfVertices, theTrackCompatibilityToPV, theTrackCompatibilityToSV, and theVtxFitProbCut.

{
  theFilter.setPtCut(s.getParameter<double>("ptCut"));
  theTrackCompatibilityToPV = s.getParameter<double>("trackCompatibilityToPVcut");
  theTrackCompatibilityToSV = s.getParameter<double>("trackCompatibilityToSVcut");
  theVtxFitProbCut = s.getParameter<double>("vtxFitProbCut"); 
  theMaxNbOfVertices =  s.getParameter<int>("maxNbOfVertices");
}

void ConfigurableTrimmedVertexFinder::setPtCut ( float  cut )

inline

Definition at line 81 of file ConfigurableTrimmedVertexFinder.h.

References TrimmedTrackFilter::setPtCut(), and theFilter.

Referenced by KalmanTrimmedVertexFinder::setPtCut().

{ theFilter.setPtCut(cut); }

void ConfigurableTrimmedVertexFinder::setTrackCompatibilityCut ( float  cut )

inline

Definition at line 82 of file ConfigurableTrimmedVertexFinder.h.

References GOODCOLL_filter_cfg::cut, and theTrackCompatibilityToPV.

Referenced by KalmanTrimmedVertexFinder::setTrackCompatibilityCut().

                                           {
    theTrackCompatibilityToPV = cut;
  }

void ConfigurableTrimmedVertexFinder::setTrackCompatibilityToSV ( float  cut )

inline

Definition at line 85 of file ConfigurableTrimmedVertexFinder.h.

References GOODCOLL_filter_cfg::cut, and theTrackCompatibilityToSV.

Referenced by KalmanTrimmedVertexFinder::setTrackCompatibilityToSV().

                                            {
    theTrackCompatibilityToSV = cut;
  }

void ConfigurableTrimmedVertexFinder::setVertexFitProbabilityCut ( float  cut )

inline

Definition at line 88 of file ConfigurableTrimmedVertexFinder.h.

References GOODCOLL_filter_cfg::cut, and theVtxFitProbCut.

Referenced by KalmanTrimmedVertexFinder::setVertexFitProbabilityCut().

{ theVtxFitProbCut = cut; }

float ConfigurableTrimmedVertexFinder::trackCompatibilityCut ( ) const

inline

Definition at line 72 of file ConfigurableTrimmedVertexFinder.h.

References theTrackCompatibilityToPV.

Referenced by KalmanTrimmedVertexFinder::trackCompatibilityCut().

{ return theTrackCompatibilityToPV; }

float ConfigurableTrimmedVertexFinder::trackCompatibilityToSV ( ) const

inline

Definition at line 73 of file ConfigurableTrimmedVertexFinder.h.

References theTrackCompatibilityToSV.

Referenced by KalmanTrimmedVertexFinder::trackCompatibilityToSV().

{ return theTrackCompatibilityToSV; }

const TrimmedTrackFilter& ConfigurableTrimmedVertexFinder::trackFilter ( ) const

inline

Definition at line 69 of file ConfigurableTrimmedVertexFinder.h.

References theFilter.

                                                 { 
    return theFilter; 
  }

std::vector< TransientVertex > ConfigurableTrimmedVertexFinder::vertexCandidates ( const std::vector< reco::TransientTrack > &  tracks, std::vector< reco::TransientTrack > &  unused, const reco::BeamSpot &  spot, bool  use_spot  ) const

private

Definition at line 76 of file ConfigurableTrimmedVertexFinder.cc.

References analyseClusterFinder(), TrimmedVertexFinder::setTrackCompatibilityCut(), theClusterFinder, theMaxNbOfVertices, theTrackCompatibilityToPV, theTrackCompatibilityToSV, testEve_cfg::tracks, and TrimmedVertexFinder::vertices().

Referenced by vertices().

{

  std::vector<TransientVertex> cand;

  std::vector<TransientTrack> remain = tracks;

  while (true) {

    float tkCompCut = (cand.size() == 0 ? 
               theTrackCompatibilityToPV 
               : theTrackCompatibilityToSV);

    //    std::cout << "PVR:compat cut " << tkCompCut << std::endl;
    theClusterFinder.setTrackCompatibilityCut(tkCompCut);
    //    std::cout << "PVCF:compat cut after setting " 
    //   << theClusterFinder.trackCompatibilityCut() << std::endl;

    std::vector<TransientVertex> newVertices;
    if ( cand.size() == 0 && use_spot )
    {
      newVertices = theClusterFinder.vertices(remain, spot );
    } else {
      newVertices = theClusterFinder.vertices(remain);
    }
    if (newVertices.empty()) break;

    analyseClusterFinder(newVertices, remain);
    
    for (std::vector<TransientVertex>::const_iterator iv = newVertices.begin();
         iv != newVertices.end(); iv++) {
      if ( iv->originalTracks().size() > 1 ) {
        cand.push_back(*iv);
      } 
      else {
        // candidate has too few tracks - get them back into the vector
        for ( std::vector< TransientTrack >::const_iterator trk
        = iv->originalTracks().begin();
              trk != iv->originalTracks().end(); ++trk ) {
          unused.push_back ( *trk );
        }
      }
    }

    // when max number of vertices reached, stop
    if (theMaxNbOfVertices != 0) {
      if (cand.size() >= (unsigned int) theMaxNbOfVertices) break;
    }
  }

  for (std::vector<TransientTrack>::const_iterator it = remain.begin();
       it != remain.end(); it++) {
    unused.push_back(*it);
  }

  return cand;
}

float ConfigurableTrimmedVertexFinder::vertexFitProbabilityCut ( ) const

inline

Definition at line 74 of file ConfigurableTrimmedVertexFinder.h.

References theVtxFitProbCut.

Referenced by KalmanTrimmedVertexFinder::vertexFitProbabilityCut().

{ return theVtxFitProbCut; }

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

virtual

Reconstruct vertices

Implements VertexReconstructor.

Definition at line 28 of file ConfigurableTrimmedVertexFinder.cc.

Referenced by KalmanTrimmedVertexFinder::vertices(), and vertices().

{
  std::vector<TransientTrack> remaining;

  return vertices(tracks, remaining, reco::BeamSpot(), false );

}

std::vector< TransientVertex > ConfigurableTrimmedVertexFinder::vertices ( const std::vector< reco::TransientTrack > &  t, const reco::BeamSpot &    ) const

virtual

Reconstruct vertices, exploiting the beamspot constraint for the primary vertex

Reimplemented from VertexReconstructor.

Definition at line 37 of file ConfigurableTrimmedVertexFinder.cc.

References vertices().

{
  std::vector<TransientTrack> remaining;
  return vertices ( tracks, remaining, spot, true );
}

std::vector< TransientVertex > ConfigurableTrimmedVertexFinder::vertices	(	const std::vector< reco::TransientTrack > &	tracks,
		std::vector< reco::TransientTrack > &	unused,
		const reco::BeamSpot &	spot,
		bool	use_spot
	)		const

Definition at line 44 of file ConfigurableTrimmedVertexFinder.cc.

References cond::ecalcond::all, analyseFoundVertices(), analyseInputTracks(), analyseVertexCandidates(), clean(), filtered(), resetEvent(), EgammaValidation_Wenu_cff::sel, theFilter, and vertexCandidates().

{
  resetEvent(tracks);
  analyseInputTracks(tracks);

  std::vector<TransientTrack> filtered;
  for (std::vector<TransientTrack>::const_iterator it = tracks.begin();
       it != tracks.end(); it++) {
    if (theFilter(*it)) { 
      filtered.push_back(*it);
    }
    else {
      unused.push_back(*it);
    }
  }

  std::vector<TransientVertex> all = vertexCandidates(filtered, unused,
      spot, use_spot );

  analyseVertexCandidates(all);

  std::vector<TransientVertex> sel = clean(all);

  analyseFoundVertices(sel);

  return sel;

}

Member Data Documentation

TrimmedVertexFinder ConfigurableTrimmedVertexFinder::theClusterFinder

mutableprivate

Definition at line 133 of file ConfigurableTrimmedVertexFinder.h.

Referenced by vertexCandidates().

TrimmedTrackFilter ConfigurableTrimmedVertexFinder::theFilter

private

Definition at line 138 of file ConfigurableTrimmedVertexFinder.h.

Referenced by ConfigurableTrimmedVertexFinder(), ptCut(), setParameters(), setPtCut(), trackFilter(), and vertices().

int ConfigurableTrimmedVertexFinder::theMaxNbOfVertices

private

Definition at line 137 of file ConfigurableTrimmedVertexFinder.h.

Referenced by maxNbOfVertices(), setMaxNbOfVertices(), setParameters(), and vertexCandidates().

float ConfigurableTrimmedVertexFinder::theTrackCompatibilityToPV

private

Definition at line 135 of file ConfigurableTrimmedVertexFinder.h.

Referenced by setParameters(), setTrackCompatibilityCut(), trackCompatibilityCut(), and vertexCandidates().

float ConfigurableTrimmedVertexFinder::theTrackCompatibilityToSV

private

Definition at line 136 of file ConfigurableTrimmedVertexFinder.h.

Referenced by setParameters(), setTrackCompatibilityToSV(), trackCompatibilityToSV(), and vertexCandidates().

float ConfigurableTrimmedVertexFinder::theVtxFitProbCut

private

Definition at line 134 of file ConfigurableTrimmedVertexFinder.h.

Referenced by clean(), setParameters(), setVertexFitProbabilityCut(), and vertexFitProbabilityCut().