HGCalUncalibRecHitProducer.cc

Go to the documentation of this file.

#include "RecoLocalCalo/HGCalRecProducers/plugins/HGCalUncalibRecHitProducer.h"

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

#include "FWCore/MessageLogger/interface/MessageLogger.h"

#include "RecoLocalCalo/HGCalRecProducers/interface/HGCalUncalibRecHitWorkerFactory.h"

#include "DataFormats/HGCRecHit/interface/HGCRecHitCollections.h"

HGCalUncalibRecHitProducer::HGCalUncalibRecHitProducer(const edm::ParameterSet& ps) :
  eeDigiCollection_( consumes<HGCEEDigiCollection>( ps.getParameter<edm::InputTag>("HGCEEdigiCollection") ) ),
  hefDigiCollection_( consumes<HGCHEDigiCollection>( ps.getParameter<edm::InputTag>("HGCHEFdigiCollection") ) ),
  hebDigiCollection_( consumes<HGCHEDigiCollection>( ps.getParameter<edm::InputTag>("HGCHEBdigiCollection") ) ),
  eeHitCollection_( ps.getParameter<std::string>("HGCEEhitCollection") ),
  hefHitCollection_( ps.getParameter<std::string>("HGCHEFhitCollection") ),
  hebHitCollection_( ps.getParameter<std::string>("HGCHEBhitCollection") ) {
  
  produces< HGCeeUncalibratedRecHitCollection >(eeHitCollection_);
  produces< HGChefUncalibratedRecHitCollection >(hefHitCollection_);
  produces< HGChebUncalibratedRecHitCollection >(hebHitCollection_);
  
  const std::string& componentType = ps.getParameter<std::string>("algo");
  worker_.reset( HGCalUncalibRecHitWorkerFactory::get()->create(componentType, ps) );
}

HGCalUncalibRecHitProducer::~HGCalUncalibRecHitProducer() {
}

void
HGCalUncalibRecHitProducer::produce(edm::Event& evt, const edm::EventSetup& es) {
  using namespace edm;
  
  Handle< HGCEEDigiCollection > pHGCEEDigis;
  Handle< HGCHEDigiCollection > pHGCHEFDigis;
  Handle< HGCHEDigiCollection > pHGCHEBDigis;
    
  evt.getByToken( eeDigiCollection_, pHGCEEDigis);
  const HGCEEDigiCollection* eeDigis = 
    pHGCEEDigis.product(); // get a ptr to the product
  
  evt.getByToken( hefDigiCollection_, pHGCHEFDigis);
  const HGCHEDigiCollection* hefDigis = 
    pHGCHEFDigis.product(); // get a ptr to the product
  
  evt.getByToken( hebDigiCollection_, pHGCHEBDigis);
  const HGCHEDigiCollection* hebDigis = 
    pHGCHEBDigis.product(); // get a ptr to the product
  
  // tranparently get things from event setup
  worker_->set(es);
  
  // prepare output
  std::auto_ptr< HGCeeUncalibratedRecHitCollection > eeUncalibRechits( new HGCeeUncalibratedRecHitCollection );
  std::auto_ptr< HGChefUncalibratedRecHitCollection > hefUncalibRechits( new HGChefUncalibratedRecHitCollection );
  std::auto_ptr< HGChefUncalibratedRecHitCollection > hebUncalibRechits( new HGChebUncalibratedRecHitCollection );
  
  // loop over HGCEE digis
  eeUncalibRechits->reserve(eeDigis->size());
  for(auto itdg = eeDigis->begin(); itdg != eeDigis->end(); ++itdg) {
    worker_->run1(evt, itdg, *eeUncalibRechits);
  }

  hefUncalibRechits->reserve(hefDigis->size());
  for(auto itdg = hefDigis->begin(); itdg != hefDigis->end(); ++itdg) {
    worker_->run2(evt, itdg, *hefUncalibRechits);
  }
  
  hebUncalibRechits->reserve(hebDigis->size());
  for(auto itdg = hebDigis->begin(); itdg != hebDigis->end(); ++itdg) {
    worker_->run3(evt, itdg, *hebUncalibRechits);
  }
  
  // put the collection of recunstructed hits in the event
  evt.put( eeUncalibRechits, eeHitCollection_ );
  evt.put( hefUncalibRechits, hefHitCollection_ );
  evt.put( hebUncalibRechits, hebHitCollection_ );
}

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