PVClusterComparer Class Reference

#include <PVClusterComparer.h>

Public Member Functions
bool	operator() (const PVCluster &v1, const PVCluster &v2)
	Use this operator in a std::sort to sort them in decreasing sumPt. More...
bool	operator() (const reco::Vertex &v1, const reco::Vertex &v2)
double	pTSquaredSum (const PVCluster &v)
	Calculate sum of square of the pT's of the tracks in the vertex. More...
double	pTSquaredSum (const reco::Vertex &v)
	PVClusterComparer ()
	Constructor does nothing, no data members. More...
	PVClusterComparer (double track_pt_min, double track_pt_max, double track_chi2_max, double track_prob_min)
void	setChisquareQuantile ()
void	updateChisquareQuantile (size_t ndof)

Public Attributes
std::vector< double >	maxChi2_
const double	track_chi2_max_
const double	track_prob_min_
const double	track_pT_max_
const double	track_pT_min_

Detailed Description

RecoPixelVertexing/PixelVertexFinding/PVClusterComparer.h This helper class is used to sort the collection of vertexes by sumPt. It is used in DivisiveVertexFinder. The sum of the squares of the pT is only done for tracks with pT>2.5 GeV. If the pT>10 GeV, then the max value of 10 is used. (The pT of pixel tracks is not very precise.)

Author

Aaron Dominguez (UNL)

Definition at line 16 of file PVClusterComparer.h.

Constructor & Destructor Documentation

PVClusterComparer::PVClusterComparer

(

)

Constructor does nothing, no data members.

Definition at line 7 of file PVClusterComparer.cc.

References setChisquareQuantile().

  : track_pT_min_  ( 2.5)
  , track_pT_max_  (10.)
  , track_chi2_max_(9999999.)
  , track_prob_min_(-1.)
{
  setChisquareQuantile();
}

PVClusterComparer::PVClusterComparer	(	double	track_pt_min,
		double	track_pt_max,
		double	track_chi2_max,
		double	track_prob_min
	)

Definition at line 16 of file PVClusterComparer.cc.

References setChisquareQuantile().

  : track_pT_min_  (track_pt_min)
  , track_pT_max_  (track_pt_max)
  , track_chi2_max_(track_chi2_max)
  , track_prob_min_(track_prob_min)
{
  setChisquareQuantile();
}

Member Function Documentation

bool PVClusterComparer::operator()	(	const PVCluster &	v1,
		const PVCluster &	v2
	)

Use this operator in a std::sort to sort them in decreasing sumPt.

Definition at line 97 of file PVClusterComparer.cc.

References pTSquaredSum().

                                                                            {
  return ( pTSquaredSum(v1) > pTSquaredSum(v2) );
}

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

Definition at line 100 of file PVClusterComparer.cc.

References pTSquaredSum().

                                                                              {
  return ( pTSquaredSum(v1) > pTSquaredSum(v2) );
}

double PVClusterComparer::pTSquaredSum

(

const PVCluster &

v

)

Calculate sum of square of the pT's of the tracks in the vertex.

Definition at line 25 of file PVClusterComparer.cc.

References mps_fire::i, maxChi2_, ndof, EnergyCorrector::pt, track_chi2_max_, track_prob_min_, track_pT_max_, track_pT_min_, and updateChisquareQuantile().

Referenced by operator()(), and PixelVertexCollectionTrimmer::produce().

                                                         {
  double sum=0;
  for (size_t i=0; i<v.tracks().size(); ++i) {

    double pt = v.tracks()[i]->pt();
    if (pt < track_pT_min_) continue; // Don't count tracks below track_pT_min_ (2.5 GeV)

    // RM : exclude badly reconstructed tracks from the sum
    //    if (track_prob_min_ >= 0. && track_prob_min_ <= 1.) 
    //      if (TMath::Prob(v.tracks()[i]->chi2(),v.tracks()[i]->ndof()) < track_prob_min_) continue ; 
    if (track_prob_min_ >= 0. && track_prob_min_ <= 1.) {      
      size_t ndof = v.tracks()[i]->ndof();
      if (ndof >= maxChi2_.size()) 
    updateChisquareQuantile(ndof);
      // cut on chi2 which corresponds to the configured probability
      if (v.tracks()[i]->chi2() > maxChi2_[ndof] ) continue ;   
    }
    if (v.tracks()[i]->normalizedChi2() > track_chi2_max_) continue;
    if (pt > track_pT_max_) pt = track_pT_max_;
    sum += pt*pt;
  }
  return sum;
}

double PVClusterComparer::pTSquaredSum

(

const reco::Vertex &

v

)

Definition at line 49 of file PVClusterComparer.cc.

References mps_fire::i, maxChi2_, ndof, EnergyCorrector::pt, track_chi2_max_, track_prob_min_, track_pT_max_, track_pT_min_, reco::Vertex::tracks_begin(), reco::Vertex::tracks_end(), and updateChisquareQuantile().

                                                          {
  double sum=0;
  for (reco::Vertex::trackRef_iterator i=v.tracks_begin(), ie =v.tracks_end();
       i!=ie; ++i) {
    
    double pt = (*i)->pt();
    if (pt < track_pT_min_) continue; // Don't count tracks below track_pT_min_ (2.5 GeV)

    // RM : exclude badly reconstructed tracks from the sum
    //    if (track_prob_min_ >= 0. && track_prob_min_ <= 1.)
    //      if (TMath::Prob((*i)->chi2(),(*i)->ndof()) < track_prob_min_) continue ; 
    if (track_prob_min_ >= 0. && track_prob_min_ <= 1.) {      
      unsigned int ndof = (*i)->ndof();
      if (ndof >= maxChi2_.size()) 
    updateChisquareQuantile(ndof);
      // cut on chi2 which corresponds to the configured probability
      if ((*i)->chi2() > maxChi2_[ndof] ) continue ;
    }
    if ((*i)->normalizedChi2() > track_chi2_max_) continue;
    
    if (pt > track_pT_max_) pt = track_pT_max_;
    sum += pt*pt;
  }
  return sum;
}

void PVClusterComparer::setChisquareQuantile

(

)

Definition at line 75 of file PVClusterComparer.cc.

References btvTracks_cfi::maxChi2, maxChi2_, ndof, and track_prob_min_.

Referenced by PVClusterComparer().

                                             {
  std::vector<double> maxChi2(20, 0.);
  if (track_prob_min_ >= 0. && track_prob_min_ <= 1.)
    for (size_t ndof = 0; ndof < maxChi2_.size(); ++ndof)
      // http://root.cern.ch/root/html/TMath.html#TMath:ChisquareQuantile  
      maxChi2[ndof] = TMath::ChisquareQuantile(1- track_prob_min_, ndof);
  
  maxChi2_ = maxChi2;

}

void PVClusterComparer::updateChisquareQuantile

(

size_t

ndof

)

Definition at line 86 of file PVClusterComparer.cc.

References HiEvtPlane_cfi::chi2, mps_fire::i, maxChi2_, ndof, and track_prob_min_.

Referenced by pTSquaredSum().

                                                           {
  
  size_t oldsize = maxChi2_.size();
  //    maxChi2_.resize(ndof+1);
  for (size_t i = oldsize; i <= ndof; ++i) {
    double chi2 = TMath::ChisquareQuantile(1- track_prob_min_, i);
    maxChi2_.push_back(chi2);
  }
}

Member Data Documentation

std::vector<double> PVClusterComparer::maxChi2_

Definition at line 32 of file PVClusterComparer.h.

Referenced by pTSquaredSum(), setChisquareQuantile(), and updateChisquareQuantile().

const double PVClusterComparer::track_chi2_max_

Definition at line 36 of file PVClusterComparer.h.

Referenced by pTSquaredSum().

const double PVClusterComparer::track_prob_min_

Definition at line 37 of file PVClusterComparer.h.

Referenced by pTSquaredSum(), setChisquareQuantile(), and updateChisquareQuantile().

const double PVClusterComparer::track_pT_max_

Definition at line 35 of file PVClusterComparer.h.

Referenced by pTSquaredSum().

const double PVClusterComparer::track_pT_min_

Definition at line 34 of file PVClusterComparer.h.

Referenced by pTSquaredSum().