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 "FWCore/Framework/interface/ConsumesCollector.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, edm::ConsumesCollector&& iC);
  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