ElectronIDExternalProducer.h

Go to the documentation of this file.

#ifndef RecoEgamma_ElectronIdentification_ElectronIDExternalProducer_h
#define RecoEgamma_ElectronIdentification_ElectronIDExternalProducer_h
 
#include <memory>
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/InputTag.h"
 
#include "DataFormats/EgammaCandidates/interface/GsfElectron.h"
#include "DataFormats/EgammaCandidates/interface/GsfElectronFwd.h"
 
#include "DataFormats/Common/interface/ValueMap.h"
 
template <class algo>
class ElectronIDExternalProducer : public edm::stream::EDProducer<> {
public:
  explicit ElectronIDExternalProducer(const edm::ParameterSet& iConfig)
      : srcToken_(consumes<reco::GsfElectronCollection>(iConfig.getParameter<edm::InputTag>("src"))),
        select_(iConfig, consumesCollector()) {
    produces<edm::ValueMap<float>>();
  }
 
  ~ElectronIDExternalProducer() override {}
 
  void produce(edm::Event& iEvent, const edm::EventSetup& iSetup) override;
 
private:
  edm::EDGetTokenT<reco::GsfElectronCollection> srcToken_;
  algo select_;
};
 
template <typename algo>
void ElectronIDExternalProducer<algo>::produce(edm::Event& iEvent, const edm::EventSetup& iSetup) {
  // read input collection
  edm::Handle<reco::GsfElectronCollection> electrons;
  iEvent.getByToken(srcToken_, electrons);
 
  // initialize common selector
  select_.newEvent(iEvent, iSetup);
 
  // prepare room for output
  std::vector<float> values;
  values.reserve(electrons->size());
  for (reco::GsfElectronCollection::const_iterator eleIt = electrons->begin(); eleIt != electrons->end(); ++eleIt) {
    values.push_back(float(select_((*eleIt), iEvent, iSetup)));
  }
 
  // fill in the ValueMap
  auto out = std::make_unique<edm::ValueMap<float>>();
  edm::ValueMap<float>::Filler filler(*out);
  filler.insert(electrons, values.begin(), values.end());
  filler.fill();
  // and put it into the event
  iEvent.put(std::move(out));
}
#endif