HitReCalibrator.cc

Go to the documentation of this file.

#include "Calibration/HcalCalibAlgos/plugins/HitReCalibrator.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "DataFormats/GeometryVector/interface/GlobalPoint.h"
#include "DataFormats/CaloTowers/interface/CaloTowerDetId.h"
#include "DataFormats/EcalRecHit/interface/EcalRecHitCollections.h"
#include "RecoTracker/TrackProducer/interface/TrackProducerBase.h"
#include "TrackingTools/TransientTrack/interface/TransientTrack.h"
#include "DataFormats/JetReco/interface/CaloJetCollection.h"
#include "CondFormats/HcalObjects/interface/HcalRespCorrs.h"
#include "CondFormats/DataRecord/interface/HcalRespCorrsRcd.h"
#include "FWCore/Utilities/interface/Exception.h"


using namespace edm;
using namespace std;
using namespace reco;

namespace cms
{

HitReCalibrator::HitReCalibrator(const edm::ParameterSet& iConfig)
{
   tok_hbhe_ = consumes<HBHERecHitCollection>(iConfig.getParameter<edm::InputTag>("hbheInput"));
   tok_ho_  = consumes<HORecHitCollection>(iConfig.getParameter<edm::InputTag>("hoInput"));
   tok_hf_ = consumes<HFRecHitCollection>(iConfig.getParameter<edm::InputTag>("hfInput")); 
   allowMissingInputs_ = true;
//register your products

   produces<HBHERecHitCollection>("DiJetsHBHEReRecHitCollection");
   produces<HORecHitCollection>("DiJetsHOReRecHitCollection");
   produces<HFRecHitCollection>("DiJetsHFReRecHitCollection");

}
void HitReCalibrator::beginJob()
{
}

HitReCalibrator::~HitReCalibrator()
{

}


// ------------ method called to produce the data  ------------
void
HitReCalibrator::produce(edm::Event& iEvent, const edm::EventSetup& iSetup)
{

  auto miniDiJetsHBHERecHitCollection = std::make_unique<HBHERecHitCollection>();
  auto miniDiJetsHORecHitCollection = std::make_unique<HORecHitCollection>();
  auto miniDiJetsHFRecHitCollection = std::make_unique<HFRecHitCollection>();

 
  edm::ESHandle <HcalRespCorrs> recalibCorrs;
  iSetup.get<HcalRespCorrsRcd>().get("recalibrate",recalibCorrs);
  const HcalRespCorrs* jetRecalib = recalibCorrs.product();
 
   
  try {
  edm::Handle<HBHERecHitCollection> hbhe;
  iEvent.getByToken(tok_hbhe_,hbhe);
  const HBHERecHitCollection Hithbhe = *(hbhe.product());
  for(HBHERecHitCollection::const_iterator hbheItr=Hithbhe.begin(); hbheItr!=Hithbhe.end(); hbheItr++)
        {
           DetId id = hbheItr->detid();
           float recal; 
           if (jetRecalib->exists(id))
             recal = jetRecalib->getValues(id)->getValue();
       else recal = 1.; 
           float energy = hbheItr->energy(); 
           float time = hbheItr->time();
           HBHERecHit* hbhehit = new HBHERecHit(id,recal*energy,time);  
           miniDiJetsHBHERecHitCollection->push_back(*hbhehit);
        }
  } catch (cms::Exception& e) { // can't find it!
  if (!allowMissingInputs_) {cout<<"No HBHE collection "<<endl; throw e;}
  }

  try{  
  edm::Handle<HORecHitCollection> ho;
  iEvent.getByToken(tok_ho_,ho);
  const HORecHitCollection Hitho = *(ho.product());
  for(HORecHitCollection::const_iterator hoItr=Hitho.begin(); hoItr!=Hitho.end(); hoItr++)
        {
           DetId id = hoItr->detid();
           float recal; 
           if (jetRecalib->exists(id))
             recal = jetRecalib->getValues(id)->getValue();
       else recal = 1.; 
           float energy = hoItr->energy(); 
           float time = hoItr->time();
           HORecHit* hohit = new HORecHit(id,recal*energy,time); 
           miniDiJetsHORecHitCollection->push_back(*hohit);
        }
  } catch (cms::Exception& e) { // can't find it!
  if (!allowMissingInputs_) {cout<<" No HO collection "<<endl; throw e;}
  }

  try {
  edm::Handle<HFRecHitCollection> hf;
  iEvent.getByToken(tok_hf_,hf);
  const HFRecHitCollection Hithf = *(hf.product());
  for(HFRecHitCollection::const_iterator hfItr=Hithf.begin(); hfItr!=Hithf.end(); hfItr++)
      {
           DetId id = hfItr->detid();
           float recal; 
           if (jetRecalib->exists(id))
             recal = jetRecalib->getValues(id)->getValue();
       else recal = 1.; 
           float energy = hfItr->energy(); 
           float time = hfItr->time();
           HFRecHit* hfhit = new HFRecHit(id,recal*energy,time); 
           miniDiJetsHFRecHitCollection->push_back(*hfhit);
      }
  } catch (cms::Exception& e) { // can't find it!
  if (!allowMissingInputs_) throw e;
  }

  //Put selected information in the event

  iEvent.put(std::move(miniDiJetsHBHERecHitCollection), "DiJetsHBHEReRecHitCollection");

  iEvent.put(std::move(miniDiJetsHORecHitCollection), "DiJetsHOReRecHitCollection");

  iEvent.put(std::move(miniDiJetsHFRecHitCollection), "DiJetsHFReRecHitCollection");

}
}

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

using cms::HitReCalibrator;
DEFINE_FWK_MODULE(HitReCalibrator);