PATPrimaryVertexSelector.h

Go to the documentation of this file.

#ifndef PatAlgos_PATPrimaryVertexSelector_H_
#define PatAlgos_PATPrimaryVertexSelector_H_

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "DataFormats/Common/interface/Handle.h"
#include "DataFormats/VertexReco/interface/Vertex.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"

class PATPrimaryVertexSelector {
public:
  typedef reco::VertexCollection collection;
  typedef std::vector<const reco::Vertex*> container;
  typedef container::const_iterator const_iterator;
  PATPrimaryVertexSelector (const edm::ParameterSet& cfg);
  const_iterator begin() const { return selected_.begin(); }
  const_iterator end() const { return selected_.end(); }
  void select(const edm::Handle<collection>&, const edm::Event&, const edm::EventSetup&);
  size_t size() const { return selected_.size(); }
  bool operator() (const reco::Vertex*, const reco::Vertex*) const;
private:
  void getVertexVariables (const reco::Vertex&, unsigned int&, double&) const;
  bool acceptTrack (const reco::Track&) const;

private:
  container selected_;               
  unsigned int multiplicityCut_;     
  float ptSumCut_;                   
  float trackEtaCut_;                
  float chi2Cut_;                    
  float dr2Cut_;                     
  float dzCut_;                      
};

#endif