ESRecHitProducer.cc

Go to the documentation of this file.

#include "RecoLocalCalo/EcalRecProducers/plugins/ESRecHitProducer.h"

#include "RecoLocalCalo/EcalRecProducers/interface/ESRecHitWorkerFactory.h"

#include "DataFormats/EcalDigi/interface/EcalDigiCollections.h"
#include "DataFormats/EcalRecHit/interface/EcalRecHit.h"
#include "DataFormats/EcalRecHit/interface/EcalRecHitCollections.h"
#include "DataFormats/Common/interface/EDCollection.h"
#include "DataFormats/Common/interface/Handle.h"

#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"

ESRecHitProducer::ESRecHitProducer(edm::ParameterSet const& ps) :
  digiToken_( consumes<ESDigiCollection>(ps.getParameter<edm::InputTag>("ESdigiCollection")) ),
  rechitCollection_( ps.getParameter<std::string>("ESrechitCollection") ),
  worker_{ESRecHitWorkerFactory::get()->create(ps.getParameter<std::string>("algo"), ps )}
{
  produces<ESRecHitCollection>(rechitCollection_);
}

ESRecHitProducer::~ESRecHitProducer()  = default;

void ESRecHitProducer::produce(edm::Event& e, const edm::EventSetup& es) {

  edm::Handle<ESDigiCollection> digiHandle;  
  const ESDigiCollection* digi=nullptr;
  e.getByToken( digiToken_, digiHandle);
 
  digi = digiHandle.product();
  LogDebug("ESRecHitInfo") << "total # ESdigis: " << digi->size();
  
  // Create empty output
  auto rec = std::make_unique<ESRecHitCollection>();
  
  if ( digi ) {
    rec->reserve(digi->size()); 
    
    worker_->set( es );
    
    // run the algorithm
    for (ESDigiCollection::const_iterator i (digi->begin());
     i!=digi->end(); i++) {    
      worker_->run( i, *rec );
    }
  }
  
  e.put(std::move(rec),rechitCollection_);
}

#include "FWCore/Framework/interface/MakerMacros.h"  
DEFINE_FWK_MODULE( ESRecHitProducer );