---

EcalTBWeightUncalibRecHitProducer.cc

Go to the documentation of this file.

#include "RecoTBCalo/EcalTBRecProducers/interface/EcalTBWeightUncalibRecHitProducer.h"
#include "DataFormats/EcalDigi/interface/EcalDigiCollections.h"
#include "DataFormats/EcalDigi/interface/EcalMGPASample.h"
#include "DataFormats/Common/interface/Handle.h"

#include <iostream>
#include <iomanip>
#include <cmath>

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

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

#include "CondFormats/EcalObjects/interface/EcalPedestals.h"
#include "CondFormats/DataRecord/interface/EcalPedestalsRcd.h"

#include "DataFormats/EcalRecHit/interface/EcalUncalibratedRecHit.h"
#include "DataFormats/EcalRecHit/interface/EcalRecHitCollections.h"

#include "CondFormats/EcalObjects/interface/EcalXtalGroupId.h"
#include "CondFormats/EcalObjects/interface/EcalWeightXtalGroups.h"
#include "CondFormats/DataRecord/interface/EcalWeightXtalGroupsRcd.h"

#include "CondFormats/EcalObjects/interface/EcalWeight.h"
#include "CondFormats/EcalObjects/interface/EcalWeightSet.h"
#include "CondFormats/EcalObjects/interface/EcalTBWeights.h"
#include "CondFormats/DataRecord/interface/EcalTBWeightsRcd.h"

#include "CondFormats/EcalObjects/interface/EcalMGPAGainRatio.h"
#include "CondFormats/EcalObjects/interface/EcalGainRatios.h"
#include "CondFormats/DataRecord/interface/EcalGainRatiosRcd.h"

#include "CLHEP/Matrix/Matrix.h"
#include "CLHEP/Matrix/SymMatrix.h"
#include <vector>

EcalTBWeightUncalibRecHitProducer::EcalTBWeightUncalibRecHitProducer(const edm::ParameterSet& ps) {

   EBdigiCollection_ = ps.getParameter<std::string>("EBdigiCollection");
   digiProducer_   = ps.getParameter<std::string>("digiProducer");
   tdcRecInfoCollection_ = ps.getParameter<std::string>("tdcRecInfoCollection");
   tdcRecInfoProducer_   = ps.getParameter<std::string>("tdcRecInfoProducer");
   EBhitCollection_  = ps.getParameter<std::string>("EBhitCollection");
   nbTimeBin_  = ps.getParameter<int>("nbTimeBin");
   use2004OffsetConvention_ = ps.getUntrackedParameter< bool >("use2004OffsetConvention",false);
   produces< EBUncalibratedRecHitCollection >(EBhitCollection_);
}

EcalTBWeightUncalibRecHitProducer::~EcalTBWeightUncalibRecHitProducer() {
}

void
EcalTBWeightUncalibRecHitProducer::produce(edm::Event& evt, const edm::EventSetup& es) {

   using namespace edm;
   
   Handle< EBDigiCollection > pEBDigis;
   const EBDigiCollection* EBdigis =0;
   
   //     evt.getByLabel( digiProducer_, EBdigiCollection_, pEBDigis);
   evt.getByLabel( digiProducer_, pEBDigis);
   if (!pEBDigis.isValid()) {
     edm::LogError("EcalUncalibRecHitError") << "Error! can't get the product " << EBdigiCollection_.c_str() ;
   } else {
     EBdigis = pEBDigis.product(); // get a ptr to the produc
#ifdef DEBUG
     LogDebug("EcalUncalibRecHitInfo") << "total # EBdigis: " << EBdigis->size() ;
#endif
   }

   if (!EBdigis)
     return;

   Handle< EcalTBTDCRecInfo > pRecTDC;
   const EcalTBTDCRecInfo* recTDC =0;

   //     evt.getByLabel( digiProducer_, EBdigiCollection_, pEBDigis);
   evt.getByLabel( tdcRecInfoProducer_, tdcRecInfoCollection_, pRecTDC);
   if (pRecTDC.isValid()) {
     recTDC = pRecTDC.product(); // get a ptr to the product
   } 

   // fetch map of groups of xtals
   edm::ESHandle<EcalWeightXtalGroups> pGrp;
   es.get<EcalWeightXtalGroupsRcd>().get(pGrp);
   const EcalWeightXtalGroups* grp = pGrp.product();
   if (!grp)
     return;

   const EcalXtalGroupsMap& grpMap = grp->getMap();
   
   
   // Gain Ratios
   edm::ESHandle<EcalGainRatios> pRatio;
   es.get<EcalGainRatiosRcd>().get(pRatio);
   const EcalGainRatioMap& gainMap = pRatio.product()->getMap(); // map of gain ratios


   // fetch TB weights
#ifdef DEBUG
   LogDebug("EcalUncalibRecHitDebug") <<"Fetching EcalTBWeights from DB " ;
#endif
   edm::ESHandle<EcalTBWeights> pWgts;
   es.get<EcalTBWeightsRcd>().get(pWgts);
   const EcalTBWeights* wgts = pWgts.product();

   if (!wgts)
     return;

#ifdef DEBUG
   LogDebug("EcalUncalibRecHitDebug") << "EcalTBWeightMap.size(): " << std::setprecision(3) << wgts->getMap().size() ;
#endif   

   // fetch the pedestals from the cond DB via EventSetup
#ifdef DEBUG
   LogDebug("EcalUncalibRecHitDebug") << "fetching pedestals....";
#endif
   edm::ESHandle<EcalPedestals> pedHandle;
   es.get<EcalPedestalsRcd>().get( pedHandle );
   const EcalPedestalsMap& pedMap = pedHandle.product()->getMap(); // map of pedestals
#ifdef DEBUG
   LogDebug("EcalUncalibRecHitDebug") << "done." ;
#endif
   // collection of reco'ed ampltudes to put in the event

   std::auto_ptr< EBUncalibratedRecHitCollection > EBuncalibRechits( new EBUncalibratedRecHitCollection );

   EcalPedestalsMapIterator pedIter; // pedestal iterator

   EcalGainRatioMap::const_iterator gainIter; // gain iterator

   EcalXtalGroupsMap::const_iterator git; // group iterator

   EcalTBWeights::EcalTBWeightMap::const_iterator wit; //weights iterator 
   // loop over EB digis
     //Getting the TDC bin
   EcalTBWeights::EcalTDCId tdcid(int(nbTimeBin_/2)+1);
   
     if (recTDC)
       if (recTDC->offset() == -999.)
         {
           edm::LogError("EcalUncalibRecHitError") << "TDC bin completely out of range. Returning" ;
           return;
         }
   
   
   for(unsigned int idig = 0; idig < EBdigis->size(); ++idig) {

     EBDataFrame itdg = (*EBdigis)[idig];

     
     //     counter_++; // verbosity counter
     
     // find pedestals for this channel
#ifdef DEBUG
         LogDebug("EcalUncalibRecHitDebug") << "looking up pedestal for crystal: " << EBDetId(itdg.id()) ;
#endif
         pedIter = pedMap.find(itdg.id().rawId());
         if( pedIter == pedMap.end() ) {
           edm::LogError("EcalUncalibRecHitError") << "error!! could not find pedestals for channel: " << EBDetId(itdg.id()) 
                                                   << "\n  no uncalib rechit will be made for this digi!"
             ;
           continue;
         }
         const EcalPedestals::Item& aped = (*pedIter);
         double pedVec[3];
         pedVec[0]=aped.mean_x12;pedVec[1]=aped.mean_x6;pedVec[2]=aped.mean_x1;

         // find gain ratios
#ifdef DEBUG
         LogDebug("EcalUncalibRecHitDebug") << "looking up gainRatios for crystal: " << EBDetId(itdg.id()) ;
#endif
         gainIter = gainMap.find(itdg.id().rawId());
         if( gainIter == gainMap.end() ) {
           edm::LogError("EcalUncalibRecHitError") << "error!! could not find gain ratios for channel: " << EBDetId(itdg.id()) 
                                                   << "\n  no uncalib rechit will be made for this digi!"
             ;
           continue;
         }
         const EcalMGPAGainRatio& aGain = (*gainIter);
         double gainRatios[3];
         gainRatios[0]=1.;gainRatios[1]=aGain.gain12Over6();gainRatios[2]=aGain.gain6Over1()*aGain.gain12Over6();
         
         
         
         // lookup group ID for this channel
         git = grpMap.find( itdg.id().rawId() );
         if( git == grpMap.end() ) {
           edm::LogError("EcalUncalibRecHitError") << "No group id found for this crystal. something wrong with EcalWeightXtalGroups in your DB?"
                                                   << "\n  no uncalib rechit will be made for digi with id: " << EBDetId(itdg.id())
             ;
           continue;
         }
         const EcalXtalGroupId& gid = (*git);


     //GAIN SWITCHING DETECTION ///////////////////////////////////////////////////////////////////////////////////////////////////     
     double sampleGainRef = 1;
     int    sampleSwitch  = 999;
     for (int sample = 0; sample < itdg.size(); ++sample)
       {
         double gainSample = itdg.sample(sample).gainId();
         if(gainSample != sampleGainRef) {sampleGainRef = gainSample; sampleSwitch = sample;}
       }//loop sample
     
     if (recTDC)
     {
       int tdcBin=0;
       if (recTDC->offset() <= 0.)
         tdcBin = 1;
       if (recTDC->offset() >= 1.)
         tdcBin = nbTimeBin_;
       else
         tdcBin = int(recTDC->offset()*float(nbTimeBin_))+1;
       
       if (tdcBin < 1 || tdcBin > nbTimeBin_ )
         {
           edm::LogError("EcalUncalibRecHitError") << "TDC bin out of range " << tdcBin << " offset " << recTDC->offset();
           continue;
         }

       // In case gain switching happens at the sample 4 (5th sample) 
       // (sample 5 (6th sample) in 2004 TDC convention) an extra
       // set of weights has to be used. This set of weights is assigned to 
       // TDC values going from 25 and up.
       if (use2004OffsetConvention_ && sampleSwitch == 5)
         tdcid=EcalTBWeights::EcalTDCId(tdcBin+25);
       else if (!use2004OffsetConvention_ && sampleSwitch == 4)
         tdcid=EcalTBWeights::EcalTDCId(tdcBin+25);
       else 
         tdcid=EcalTBWeights::EcalTDCId(tdcBin);
     }//check TDC     
     
     // now lookup the correct weights in the map
     wit = wgts->getMap().find( std::make_pair(gid,tdcid) );
     if( wit == wgts->getMap().end() ) {  // no weights found for this group ID
       edm::LogError("EcalUncalibRecHitError") << "No weights found for EcalGroupId: " << gid.id() << " and  EcalTDCId: " << tdcid
                                               << "\n  skipping digi with id: " << EBDetId(itdg.id())
         ;
       continue;
     }
     const EcalWeightSet& wset = wit->second; // this is the EcalWeightSet


     // EcalWeightMatrix is vec<vec:double>>

#ifdef DEBUG
     LogDebug("EcalUncalibRecHitDebug") << "accessing matrices of weights...";
#endif
     const EcalWeightSet::EcalWeightMatrix& mat1 = wset.getWeightsBeforeGainSwitch();
     const EcalWeightSet::EcalWeightMatrix& mat2 = wset.getWeightsAfterGainSwitch();
     const EcalWeightSet::EcalChi2WeightMatrix& mat3 = wset.getChi2WeightsBeforeGainSwitch();
     const EcalWeightSet::EcalChi2WeightMatrix& mat4 = wset.getChi2WeightsAfterGainSwitch();
     const EcalWeightSet::EcalWeightMatrix* weights[2];
     weights[0]=&mat1;
     weights[1]=&mat2;
     //          weights.push_back(clmat1);
     //          weights.push_back(clmat2);
     //          LogDebug("EcalUncalibRecHitDebug") << "weights before switch:\n" << clmat1 ;
     //          LogDebug("EcalUncalibRecHitDebug") << "weights after switch:\n" << clmat2 ;
     
     
     // build CLHEP chi2  matrices
     const EcalWeightSet::EcalChi2WeightMatrix* chi2mat[2];
     chi2mat[0]=&mat3;
     chi2mat[1]=&mat4;

     EcalUncalibratedRecHit aHit =
       EBalgo_.makeRecHit(itdg, pedVec, gainRatios, weights, chi2mat);
     EBuncalibRechits->push_back( aHit );
#ifdef DEBUG
      if(aHit.amplitude()>0.) {
       LogDebug("EcalUncalibRecHitDebug") << "processed EBDataFrame with id: "
                                          << EBDetId(itdg.id()) << "\n"
                                          << "uncalib rechit amplitude: " << aHit.amplitude()
         ;
      }
#endif
   }
   // put the collection of recunstructed hits in the event
   evt.put( EBuncalibRechits, EBhitCollection_ );
}

// HepMatrix
// EcalTBWeightUncalibRecHitProducer::makeMatrixFromVectors(const std::vector< std::vector<EcalWeight> >& vecvec) {
//   int nrow = vecvec.size();
//   int ncol = (vecvec[0]).size();
//   HepMatrix clmat(nrow,ncol);
//   //LogDebug("EcalUncalibRecHitDebug") << "created HepMatrix(" << nrow << "," << ncol << ")" ;
//   for(int irow=0;irow<nrow;++irow) {
//     for(int icol=0;icol<ncol;++icol) {
//         clmat[irow][icol] = ((vecvec[irow])[icol]).value();
//     }
//   }
//   return clmat;
// }

// HepMatrix 
// EcalTBWeightUncalibRecHitProducer::makeDummySymMatrix(int size)
// {
//   HepMatrix clmat(10,10);
//   //LogDebug("EcalUncalibRecHitDebug") << "created HepMatrix(" << nrow << "," << ncol << ")" ;
//   for(int irow=0; irow<size; ++irow) {
//     for(int icol=0 ; icol<size; ++icol) {
//       clmat[irow][icol] = irow+icol;
//     }
//   }
//   return clmat;
// }

---