PVClusterComparer.h

Go to the documentation of this file.

#ifndef RecoPixelVertexing_PVClusterComparer_h
#define RecoPixelVertexing_PVClusterComparer_h

#include "RecoPixelVertexing/PixelVertexFinding/interface/PVCluster.h"
#include "DataFormats/VertexReco/interface/Vertex.h"

class PVClusterComparer {
public:
  PVClusterComparer();
  PVClusterComparer(double track_pt_min, double track_pt_max, double track_chi2_max, double track_prob_min);

  double pTSquaredSum(const PVCluster &v);
  double pTSquaredSum(const reco::Vertex &v);
  void setChisquareQuantile();
  void updateChisquareQuantile(size_t ndof);

  bool operator()(const PVCluster &v1, const PVCluster &v2);
  bool operator()(const reco::Vertex &v1, const reco::Vertex &v2);

  std::vector<double> maxChi2_;

  const double track_pT_min_;
  const double track_pT_max_;
  const double track_chi2_max_;
  const double track_prob_min_;
};
#endif