/data/refman/pasoursint/CMSSW_4_4_5_patch3/src/RecoLocalCalo/HcalRecProducers/src/HcalSimpleReconstructor.cc

Go to the documentation of this file.

#include "HcalSimpleReconstructor.h"
#include "DataFormats/HcalDigi/interface/HcalDigiCollections.h"
#include "DataFormats/HcalRecHit/interface/HcalRecHitCollections.h"
#include "DataFormats/Common/interface/EDCollection.h"
#include "DataFormats/Common/interface/Handle.h"
#include "FWCore/Framework/interface/Selector.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "CalibFormats/HcalObjects/interface/HcalCoderDb.h"
#include "CalibFormats/HcalObjects/interface/HcalCalibrations.h"
#include "CalibFormats/HcalObjects/interface/HcalDbService.h"
#include "CalibFormats/HcalObjects/interface/HcalDbRecord.h"

#include <iostream>
    
HcalSimpleReconstructor::HcalSimpleReconstructor(edm::ParameterSet const& conf):
  reco_(conf.getParameter<bool>("correctForTimeslew"),
        conf.getParameter<bool>("correctForPhaseContainment"),conf.getParameter<double>("correctionPhaseNS")),
  det_(DetId::Hcal),
  inputLabel_(conf.getParameter<edm::InputTag>("digiLabel")),
  dropZSmarkedPassed_(conf.getParameter<bool>("dropZSmarkedPassed")),
  firstSample_(conf.getParameter<int>("firstSample")),
  samplesToAdd_(conf.getParameter<int>("samplesToAdd")),
  tsFromDB_(conf.getParameter<bool>("tsFromDB"))
{

  std::string subd=conf.getParameter<std::string>("Subdetector");
  if (!strcasecmp(subd.c_str(),"HBHE")) {
    subdet_=HcalBarrel;
    produces<HBHERecHitCollection>();
  } else if (!strcasecmp(subd.c_str(),"HO")) {
    subdet_=HcalOuter;
    produces<HORecHitCollection>();
  } else if (!strcasecmp(subd.c_str(),"HF")) {
    subdet_=HcalForward;
    produces<HFRecHitCollection>();
  } else {
    std::cout << "HcalSimpleReconstructor is not associated with a specific subdetector!" << std::endl;
  }       
  
}

HcalSimpleReconstructor::~HcalSimpleReconstructor() { }

void HcalSimpleReconstructor::beginRun(edm::Run&r, edm::EventSetup const & es){

  if (tsFromDB_==true)
    {
      edm::ESHandle<HcalRecoParams> p;
      es.get<HcalRecoParamsRcd>().get(p);
      paramTS = new HcalRecoParams(*p.product());
    }
}

void HcalSimpleReconstructor::endRun(edm::Run&r, edm::EventSetup const & es){
  if (tsFromDB_==true)
    {
      if (paramTS) delete paramTS;
    }
}




void HcalSimpleReconstructor::produce(edm::Event& e, const edm::EventSetup& eventSetup)
{
  // get conditions
  edm::ESHandle<HcalDbService> conditions;
  eventSetup.get<HcalDbRecord>().get(conditions);
  const HcalQIEShape* shape = conditions->getHcalShape (); // this one is generic

  // HACK related to HB- corrections
  if(e.isRealData()) reco_.setForData();
 
  
  if (det_==DetId::Hcal) {
    if (subdet_==HcalBarrel || subdet_==HcalEndcap) {
      edm::Handle<HBHEDigiCollection> digi;
      
      e.getByLabel(inputLabel_,digi);
      
      // create empty output
      std::auto_ptr<HBHERecHitCollection> rec(new HBHERecHitCollection);
      rec->reserve(digi->size());
      // run the algorithm
      int toaddMem = 0;
      int first = firstSample_;
      int toadd = samplesToAdd_;
      HBHEDigiCollection::const_iterator i;
      for (i=digi->begin(); i!=digi->end(); i++) {
        HcalDetId cell = i->id();
        DetId detcell=(DetId)cell;
        // rof 27.03.09: drop ZS marked and passed digis:
        if (dropZSmarkedPassed_)
          if (i->zsMarkAndPass()) continue;

        const HcalCalibrations& calibrations=conditions->getHcalCalibrations(cell);
        const HcalQIECoder* channelCoder = conditions->getHcalCoder (cell);
        HcalCoderDb coder (*channelCoder, *shape);

        //>>> firstSample & samplesToAdd
        if(tsFromDB_) {
          const HcalRecoParam* param_ts = paramTS->getValues(detcell.rawId());
          first = param_ts->firstSample();    
          toadd = param_ts->samplesToAdd();
        }    
        if(toaddMem != toadd) {
          reco_.initPulseCorr(toadd);
          toaddMem = toadd;
        }

        rec->push_back(reco_.reconstruct(*i,first,toadd,coder,calibrations));

      }
      // return result
      e.put(rec);
    } else if (subdet_==HcalOuter) {
      edm::Handle<HODigiCollection> digi;
      e.getByLabel(inputLabel_,digi);
      
      // create empty output
      std::auto_ptr<HORecHitCollection> rec(new HORecHitCollection);
      rec->reserve(digi->size());
      // run the algorithm
      int toaddMem = 0;
      int first = firstSample_;
      int toadd = samplesToAdd_;
      HODigiCollection::const_iterator i;
      for (i=digi->begin(); i!=digi->end(); i++) {
        HcalDetId cell = i->id();
        DetId detcell=(DetId)cell;
        // rof 27.03.09: drop ZS marked and passed digis:
        if (dropZSmarkedPassed_)
          if (i->zsMarkAndPass()) continue;

        const HcalCalibrations& calibrations=conditions->getHcalCalibrations(cell);
        const HcalQIECoder* channelCoder = conditions->getHcalCoder (cell);
        HcalCoderDb coder (*channelCoder, *shape);
       

        //>>> firstSample & samplesToAdd
        if(tsFromDB_) {
          const HcalRecoParam* param_ts = paramTS->getValues(detcell.rawId());
          first = param_ts->firstSample();    
          toadd = param_ts->samplesToAdd();    
        }
        if(toaddMem != toadd) {
          reco_.initPulseCorr(toadd);
          toaddMem = toadd;
        }

        rec->push_back(reco_.reconstruct(*i,first,toadd,coder,calibrations));
      }
      // return result
      e.put(rec);    
    } else if (subdet_==HcalForward) {
      edm::Handle<HFDigiCollection> digi;
      e.getByLabel(inputLabel_,digi);
      
      // create empty output
      std::auto_ptr<HFRecHitCollection> rec(new HFRecHitCollection);
      rec->reserve(digi->size());
      // run the algorithm
      int toaddMem = 0;
      int first = firstSample_;
      int toadd = samplesToAdd_;
      HFDigiCollection::const_iterator i;
      for (i=digi->begin(); i!=digi->end(); i++) {
        HcalDetId cell = i->id();        
        DetId detcell=(DetId)cell;
        // rof 27.03.09: drop ZS marked and passed digis:
        if (dropZSmarkedPassed_)
          if (i->zsMarkAndPass()) continue;
 
        const HcalCalibrations& calibrations=conditions->getHcalCalibrations(cell);
        const HcalQIECoder* channelCoder = conditions->getHcalCoder (cell);
        HcalCoderDb coder (*channelCoder, *shape);

        //>>> firstSample & samplesToAdd
        if(tsFromDB_) {
          const HcalRecoParam* param_ts = paramTS->getValues(detcell.rawId());
          first = param_ts->firstSample();    
          toadd = param_ts->samplesToAdd();    
        }
        if(toaddMem != toadd) {
          reco_.initPulseCorr(toadd);
          toaddMem = toadd;
        }

        rec->push_back(reco_.reconstruct(*i,first,toadd,coder,calibrations));
      }
      // return result
      e.put(rec);     
    } else if (subdet_==HcalOther && subdetOther_==HcalCalibration) {
      edm::Handle<HcalCalibDigiCollection> digi;
      e.getByLabel(inputLabel_,digi);
      
      // create empty output
      std::auto_ptr<HcalCalibRecHitCollection> rec(new HcalCalibRecHitCollection);
      rec->reserve(digi->size());
      // run the algorithm
      int toaddMem = 0;
      int first = firstSample_;
      int toadd = samplesToAdd_;
      HcalCalibDigiCollection::const_iterator i;
      for (i=digi->begin(); i!=digi->end(); i++) {
        HcalCalibDetId cell = i->id();    
        DetId detcell=(DetId)cell;
        // rof 27.03.09: drop ZS marked and passed digis:
        if (dropZSmarkedPassed_)
          if (i->zsMarkAndPass()) continue;

        const HcalCalibrations& calibrations=conditions->getHcalCalibrations(cell);
        const HcalQIECoder* channelCoder = conditions->getHcalCoder (cell);
        HcalCoderDb coder (*channelCoder, *shape);

        //>>> firstSample & samplesToAdd
        if(tsFromDB_) {
          const HcalRecoParam* param_ts = paramTS->getValues(detcell.rawId());
          first = param_ts->firstSample();    
          toadd = param_ts->samplesToAdd();    
        }
        if(toaddMem != toadd) {
          reco_.initPulseCorr(toadd);
          toaddMem = toadd;
        }

        rec->push_back(reco_.reconstruct(*i,first,toadd,coder,calibrations));
      }
      // return result
      e.put(rec);     
    }
  } 
}