File indexing completed on 2025-02-05 23:51:40
0001 #ifndef PatAlgos_PATPrimaryVertexSelector_H_
0002 #define PatAlgos_PATPrimaryVertexSelector_H_
0003
0004
0005
0006
0007
0008
0009
0010
0011
0012
0013
0014
0015
0016 #include "FWCore/Framework/interface/Event.h"
0017 #include "FWCore/Framework/interface/EventSetup.h"
0018 #include "FWCore/Framework/interface/ConsumesCollector.h"
0019 #include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
0020 #include "DataFormats/Common/interface/Handle.h"
0021 #include "DataFormats/VertexReco/interface/Vertex.h"
0022 #include "DataFormats/VertexReco/interface/VertexFwd.h"
0023
0024 class PATPrimaryVertexSelector {
0025 public:
0026 typedef reco::VertexCollection collection;
0027 typedef std::vector<const reco::Vertex*> container;
0028 typedef container::const_iterator const_iterator;
0029 PATPrimaryVertexSelector(const edm::ParameterSet& cfg, edm::ConsumesCollector&& iC);
0030
0031 const_iterator begin() const { return selected_.begin(); }
0032
0033 const_iterator end() const { return selected_.end(); }
0034
0035 void select(const edm::Handle<collection>&, const edm::Event&, const edm::EventSetup&);
0036
0037 size_t size() const { return selected_.size(); }
0038
0039 bool operator()(const reco::Vertex*, const reco::Vertex*) const;
0040
0041 static void fillPSetDescription(edm::ParameterSetDescription& desc);
0042
0043 private:
0044
0045 void getVertexVariables(const reco::Vertex&, unsigned int&, double&) const;
0046
0047 bool acceptTrack(const reco::Track&) const;
0048
0049 private:
0050 container selected_;
0051 unsigned int multiplicityCut_;
0052 float ptSumCut_;
0053 float trackEtaCut_;
0054 float chi2Cut_;
0055 float dr2Cut_;
0056 float dzCut_;
0057 };
0058
0059 #endif