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