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> >();
   }

   virtual ~ElectronIDExternalProducer() {}

   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