DivisiveVertexFinder.h

Go to the documentation of this file.

#ifndef RecoVertex_PixelVertexFinding_DivisiveVertexFinder_h
#define RecoVertex_PixelVertexFinding_DivisiveVertexFinder_h

#include <vector>
#include "DataFormats/VertexReco/interface/Vertex.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
//#include "CommonTools/Clustering1D/interface/DivisiveClusterizer1D.h"
#include "RecoVertex/PixelVertexFinding/interface/DivisiveClusterizer1D.h"

#include "RecoVertex/PixelVertexFinding/interface/PVClusterComparer.h"

class DivisiveVertexFinder {
public:
  DivisiveVertexFinder(double track_pt_min,
                       double track_pt_max,
                       double track_chi2_max,
                       double track_prob_min,
                       double zOffset = 5.0,
                       int ntrkMin = 5,
                       bool useError = true,
                       double zSeparation = 0.05,
                       bool wtAverage = true,
                       int verbosity = 0);
  ~DivisiveVertexFinder();

  bool findVertexes(const reco::TrackRefVector &trks,     // input
                    reco::VertexCollection &vertexes);    // output
  bool findVertexesAlt(const reco::TrackRefVector &trks,  // input
                       reco::VertexCollection &vertexes,
                       const math::XYZPoint &bs);  // output
private:
  double zOffset_, zSeparation_;
  int ntrkMin_;
  bool useError_, wtAverage_;

  pixeltemp::DivisiveClusterizer1D<reco::Track> divmeth_;
  //  DivisiveClusterizer1D< reco::Track > divmeth_;

  // How loud should I be?
  int verbose_;

  PVClusterComparer *pvComparer_;
};
#endif