PrimaryVertexProducer/interface/VertexHigherPtSquared.h

0001 #ifndef VertexHigherPtSquared_H
0002 #define VertexHigherPtSquared_H
0003
0004 #include "RecoVertex/VertexPrimitives/interface/TransientVertex.h"
0005 #include "TrackingTools/TransientTrack/interface/TransientTrack.h"
0006 #include "DataFormats/VertexReco/interface/Vertex.h"
0007 #include <vector>
0008
0009 /** \class VertexHigherPtSquared
0010  * operator for sorting TransientVertex objects
0011  * in decreasing order of the sum of the squared track pT's
0012  */
0013 struct VertexHigherPtSquared {
0014   bool operator()(const TransientVertex& v1, const TransientVertex& v2) const;
0015
0016   bool operator()(const reco::Vertex& v1, const reco::Vertex& v2) const;
0017
0018 public:
0019   double sumPtSquared(const reco::Vertex& v) const;
0020   double sumPtSquared(const std::vector<reco::TransientTrack>& tks) const;
0021 };
0022
0023 #endif