ElectronIDExternalProducerRemapper.cc

Go to the documentation of this file.

#include "RecoEgamma/ElectronIdentification/plugins/ElectronIDExternalProducerRemapper.h"

#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "DataFormats/Common/interface/Handle.h"

ElectronIDExternalProducerRemapper::ElectronIDExternalProducerRemapper(const edm::ParameterSet& config) : 
  emObjectProducer_       (consumes<reco::GsfElectronCollection >(config.getParameter<edm::InputTag>("candidateProducer"))),
  newToOldObjectMap_      (consumes<CRefMap>(config.getParameter<edm::InputTag>("newToOldObjectMap"))),
  idMap_           (consumes<edm::ValueMap<float> >(config.getParameter<edm::InputTag>("idMap")))
{
  produces<edm::ValueMap<float> >();
}

ElectronIDExternalProducerRemapper::~ElectronIDExternalProducerRemapper()
{
}

void ElectronIDExternalProducerRemapper::produce(edm::Event& iEvent, const edm::EventSetup&)
{
  edm::Handle<reco::GsfElectronCollection> emObjectHandle;
  iEvent.getByToken(emObjectProducer_, emObjectHandle);

  edm::Handle<CRefMap> objectMapHandle;
  iEvent.getByToken(newToOldObjectMap_, objectMapHandle);
  auto& objectMap(*objectMapHandle);

  edm::Handle<edm::ValueMap<float> > idMapHandle;
  iEvent.getByToken(idMap_, idMapHandle);
  auto& idMap(*idMapHandle);

  std::vector<float> retV(emObjectHandle->size(), 0.);

  for (unsigned iO(0); iO != emObjectHandle->size(); ++iO) {
    CRef ref(emObjectHandle, iO);
    auto oldRef(objectMap[ref]);
    retV[iO] = idMap[oldRef];
  }

  std::auto_ptr<edm::ValueMap<float> > isoMap(new edm::ValueMap<float>);
  edm::ValueMap<float>::Filler filler(*isoMap);
  
  filler.insert(emObjectHandle, retV.begin(), retV.end());
  filler.fill();

  iEvent.put(isoMap);
}


DEFINE_FWK_MODULE(ElectronIDExternalProducerRemapper);