VertexAssociationProducer.cc

Go to the documentation of this file.

 
#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/InputTag.h"
 
#include "DataFormats/Common/interface/ValueMap.h"
#include "DataFormats/Common/interface/View.h"
#include "DataFormats/PatCandidates/interface/Vertexing.h"
#include "PhysicsTools/PatAlgos/interface/VertexingHelper.h"
 
namespace pat {
 
  class PATVertexAssociationProducer : public edm::stream::EDProducer<> {
    typedef edm::ValueMap<pat::VertexAssociation> VertexAssociationMap;
 
  public:
    explicit PATVertexAssociationProducer(const edm::ParameterSet& iConfig);
    ~PATVertexAssociationProducer() override;
 
    void produce(edm::Event& iEvent, const edm::EventSetup& iSetup) override;
 
  private:
    typedef std::vector<edm::InputTag> VInputTag;
    // configurables
    std::vector<edm::InputTag> particles_;
    std::vector<edm::EDGetTokenT<edm::View<reco::Candidate> > > particlesTokens_;
    pat::helper::VertexingHelper vertexing_;
  };
 
}  // namespace pat
 
using pat::PATVertexAssociationProducer;
 
PATVertexAssociationProducer::PATVertexAssociationProducer(const edm::ParameterSet& iConfig)
    : particles_(iConfig.existsAs<VInputTag>("candidates")
                     ?  // if it's a VInputTag
                     iConfig.getParameter<VInputTag>("candidates")
                     : VInputTag(1, iConfig.getParameter<edm::InputTag>("candidates"))),
      vertexing_(iConfig, consumesCollector()) {
  for (VInputTag::const_iterator it = particles_.begin(), end = particles_.end(); it != end; ++it) {
    particlesTokens_.push_back(consumes<edm::View<reco::Candidate> >(*it));
  }
  produces<VertexAssociationMap>();
}
 
PATVertexAssociationProducer::~PATVertexAssociationProducer() {}
 
void PATVertexAssociationProducer::produce(edm::Event& iEvent, const edm::EventSetup& iSetup) {
  using namespace edm;
  using namespace std;
  // read in vertices and EventSetup
  vertexing_.newEvent(iEvent, iSetup);
 
  // prepare room and tools for output
  auto result = std::make_unique<VertexAssociationMap>();
  VertexAssociationMap::Filler filler(*result);
  vector<pat::VertexAssociation> assos;
 
  // loop on input tags
  for (std::vector<edm::EDGetTokenT<edm::View<reco::Candidate> > >::const_iterator it = particlesTokens_.begin(),
                                                                                   end = particlesTokens_.end();
       it != end;
       ++it) {
    // read candidates
    Handle<View<reco::Candidate> > cands;
    iEvent.getByToken(*it, cands);
    assos.clear();
    assos.reserve(cands->size());
    // loop on candidates
    for (size_t i = 0, n = cands->size(); i < n; ++i) {
      assos.push_back(vertexing_(cands->refAt(i)));
    }
    // insert into ValueMap
    filler.insert(cands, assos.begin(), assos.end());
  }
 
  // do the real filling
  filler.fill();
 
  // put our produced stuff in the event
  iEvent.put(std::move(result));
}
 
#include "FWCore/Framework/interface/MakerMacros.h"
 
DEFINE_FWK_MODULE(PATVertexAssociationProducer);