Back to home page

Project CMSSW displayed by LXR

 
 

    


File indexing completed on 2022-08-30 22:39:12

0001 #include "ESRecHitWorker.h"
0002 
0003 #include "DataFormats/Common/interface/Handle.h"
0004 #include "DataFormats/EcalDigi/interface/EcalDigiCollections.h"
0005 #include "DataFormats/EcalRecHit/interface/EcalUncalibratedRecHit.h"
0006 #include "DataFormats/EcalRecHit/interface/EcalRecHitCollections.h"
0007 #include "FWCore/MessageLogger/interface/MessageLogger.h"
0008 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0009 
0010 #include <cmath>
0011 #include <iomanip>
0012 #include <iostream>
0013 
0014 ESRecHitWorker::ESRecHitWorker(const edm::ParameterSet &ps, edm::ConsumesCollector cc) : ESRecHitWorkerBaseClass(ps) {
0015   recoAlgo_ = ps.getParameter<int>("ESRecoAlgo");
0016   esgainToken_ = cc.esConsumes<ESGain, ESGainRcd>();
0017   esMIPToGeVToken_ = cc.esConsumes<ESMIPToGeVConstant, ESMIPToGeVConstantRcd>();
0018   esWeightsToken_ = cc.esConsumes<ESTimeSampleWeights, ESTimeSampleWeightsRcd>();
0019   esPedestalsToken_ = cc.esConsumes<ESPedestals, ESPedestalsRcd>();
0020   esMIPsToken_ = cc.esConsumes<ESIntercalibConstants, ESIntercalibConstantsRcd>();
0021   esChannelStatusToken_ = cc.esConsumes<ESChannelStatus, ESChannelStatusRcd>();
0022   esRatioCutsToken_ = cc.esConsumes<ESRecHitRatioCuts, ESRecHitRatioCutsRcd>();
0023   esAngleCorrFactorsToken_ = cc.esConsumes<ESAngleCorrectionFactors, ESAngleCorrectionFactorsRcd>();
0024 
0025   if (recoAlgo_ == 0)
0026     algoW_ = new ESRecHitSimAlgo();
0027   else if (recoAlgo_ == 1)
0028     algoF_ = new ESRecHitFitAlgo();
0029   else
0030     algoA_ = new ESRecHitAnalyticAlgo();
0031 }
0032 
0033 ESRecHitWorker::~ESRecHitWorker() {
0034   if (recoAlgo_ == 0)
0035     delete algoW_;
0036   else if (recoAlgo_ == 1)
0037     delete algoF_;
0038   else
0039     delete algoA_;
0040 }
0041 
0042 void ESRecHitWorker::set(const edm::EventSetup &es) {
0043   esgain_ = es.getHandle(esgainToken_);
0044   const ESGain *gain = esgain_.product();
0045 
0046   esMIPToGeV_ = es.getHandle(esMIPToGeVToken_);
0047   const ESMIPToGeVConstant *mipToGeV = esMIPToGeV_.product();
0048 
0049   double ESGain = gain->getESGain();
0050   double ESMIPToGeV = (ESGain == 1) ? mipToGeV->getESValueLow() : mipToGeV->getESValueHigh();
0051 
0052   esWeights_ = es.getHandle(esWeightsToken_);
0053   const ESTimeSampleWeights *wgts = esWeights_.product();
0054 
0055   float w0 = wgts->getWeightForTS0();
0056   float w1 = wgts->getWeightForTS1();
0057   float w2 = wgts->getWeightForTS2();
0058 
0059   esPedestals_ = es.getHandle(esPedestalsToken_);
0060   const ESPedestals *peds = esPedestals_.product();
0061 
0062   esMIPs_ = es.getHandle(esMIPsToken_);
0063   const ESIntercalibConstants *mips = esMIPs_.product();
0064 
0065   esAngleCorrFactors_ = es.getHandle(esAngleCorrFactorsToken_);
0066   const ESAngleCorrectionFactors *ang = esAngleCorrFactors_.product();
0067 
0068   esChannelStatus_ = es.getHandle(esChannelStatusToken_);
0069   const ESChannelStatus *channelStatus = esChannelStatus_.product();
0070 
0071   esRatioCuts_ = es.getHandle(esRatioCutsToken_);
0072   const ESRecHitRatioCuts *ratioCuts = esRatioCuts_.product();
0073 
0074   if (recoAlgo_ == 0) {
0075     algoW_->setESGain(ESGain);
0076     algoW_->setMIPGeV(ESMIPToGeV);
0077     algoW_->setW0(w0);
0078     algoW_->setW1(w1);
0079     algoW_->setW2(w2);
0080     algoW_->setPedestals(peds);
0081     algoW_->setIntercalibConstants(mips);
0082     algoW_->setChannelStatus(channelStatus);
0083     algoW_->setRatioCuts(ratioCuts);
0084     algoW_->setAngleCorrectionFactors(ang);
0085   } else if (recoAlgo_ == 1) {
0086     algoF_->setESGain(ESGain);
0087     algoF_->setMIPGeV(ESMIPToGeV);
0088     algoF_->setPedestals(peds);
0089     algoF_->setIntercalibConstants(mips);
0090     algoF_->setChannelStatus(channelStatus);
0091     algoF_->setRatioCuts(ratioCuts);
0092     algoF_->setAngleCorrectionFactors(ang);
0093   } else {
0094     algoA_->setESGain(ESGain);
0095     algoA_->setMIPGeV(ESMIPToGeV);
0096     algoA_->setPedestals(peds);
0097     algoA_->setIntercalibConstants(mips);
0098     algoA_->setChannelStatus(channelStatus);
0099     algoA_->setRatioCuts(ratioCuts);
0100     algoA_->setAngleCorrectionFactors(ang);
0101   }
0102 }
0103 
0104 bool ESRecHitWorker::run(const ESDigiCollection::const_iterator &itdg, ESRecHitCollection &result) {
0105   if (recoAlgo_ == 0)
0106     result.push_back(algoW_->reconstruct(*itdg));
0107   else if (recoAlgo_ == 1)
0108     result.push_back(algoF_->reconstruct(*itdg));
0109   else
0110     result.push_back(algoA_->reconstruct(*itdg));
0111   return true;
0112 }
0113 
0114 #include "FWCore/Framework/interface/MakerMacros.h"
0115 #include "RecoLocalCalo/EcalRecProducers/interface/ESRecHitWorkerFactory.h"
0116 DEFINE_EDM_PLUGIN(ESRecHitWorkerFactory, ESRecHitWorker, "ESRecHitWorker");