FixLArElecSCCalib.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#include "FixLArElecSCCalib.h"
#include "Gaudi/Property.h"
#include "GaudiKernel/IToolSvc.h"
 
#include "CaloIdentifier/CaloCell_Base_ID.h"
#include "CaloIdentifier/CaloCell_SuperCell_ID.h"
#include "CaloIdentifier/CaloGain.h"
#include "CaloIdentifier/LArEM_ID.h"
#include "LArIdentifier/LArOnline_SuperCellID.h"
#include "StoreGate/StoreGateSvc.h"
 
#include "LArCOOLConditions/LArPedestalSC.h"
#include "LArCOOLConditions/LArRampSC.h"
 
#include "AthenaPoolUtilities/CondAttrListCollection.h"
#include "AthenaPoolUtilities/AthenaAttributeList.h"
#include "CoralBase/Blob.h"
 
#include "TTree.h"
#include "TFile.h"
 
#include <fstream>
 
FixLArElecSCCalib::FixLArElecSCCalib(const std::string& name, ISvcLocator* pSvcLocator) : 
  AthAlgorithm(name,pSvcLocator){ 
 
}
 
FixLArElecSCCalib::~FixLArElecSCCalib()
{ }
 
StatusCode FixLArElecSCCalib::initialize() {
  ATH_MSG_INFO ( " in initialize " );
  
  ATH_CHECK( detStore()->retrieve(m_sem_idhelper) );
  ATH_CHECK( detStore()->retrieve(m_shec_idhelper) );
  ATH_CHECK( detStore()->retrieve(m_sfcal_idhelper) );
  ATH_CHECK( detStore()->retrieve(m_sonline_idhelper) );
  ATH_CHECK( detStore()->retrieve(m_scell_idhelper) );
 
  ATH_CHECK( m_cablingKeySC.initialize() );
  ATH_CHECK( m_CLKeySC.initialize() );
  ATH_CHECK( m_mcSymKey.initialize() );
 
  return StatusCode::SUCCESS;
}
 
StatusCode FixLArElecSCCalib::execute() {
 
  ATH_MSG_INFO ( " in execute  " );
  return StatusCode::SUCCESS;
}
 
StatusCode FixLArElecSCCalib::stop() {
 
   ATH_MSG_INFO ( " in stop " );
 
    switch(m_fixFlag) {
       case 1: return fix1(); 
       case 2: {
                  SG::ReadCondHandle<LArOnOffIdMapping> cablingHdl{m_cablingKeySC};
                  const LArOnOffIdMapping* cabling{*cablingHdl};
                  if(!cabling) {
                      ATH_MSG_ERROR( "Do not have cabling mapping from key " << m_cablingKeySC.key() );
                      return StatusCode::FAILURE;
                  }
                  SG::ReadCondHandle<LArCalibLineMapping> clHdl{m_CLKeySC};
                  const LArCalibLineMapping *clCont {*clHdl};
                  if(!clCont) {
                     ATH_MSG_ERROR( "Do not have calib line mapping !!!" );
                     return StatusCode::FAILURE;
                  }
                  return fix2(cabling, clCont);
               }
       case 3: {
                  SG::ReadCondHandle<LArOnOffIdMapping> cablingHdl{m_cablingKeySC};
                  const LArOnOffIdMapping* cabling{*cablingHdl};
                  if(!cabling) {
                      ATH_MSG_ERROR( "Do not have cabling mapping from key " << m_cablingKeySC.key() );
                      return StatusCode::FAILURE;
                  }
                  SG::ReadCondHandle<LArMCSym> symHdl{m_mcSymKey};
                  const LArMCSym* sym{*symHdl};
                  if(!sym) {
                      ATH_MSG_ERROR( "Do not have MCSym from key " << m_mcSymKey.key() );
                      return StatusCode::FAILURE;
                  }
                  return fix3(cabling,sym);
               }
       default: return StatusCode::SUCCESS;
    }        
}
 
StatusCode FixLArElecSCCalib::fix1() {
 
   ATH_MSG_INFO ( " in fix1() " );
 
   // Fix1 is for scaling the Pedestals and Rams 
   int n=0;
 
   const LArPedestalSC *ped = nullptr;
   CHECK(detStore()->retrieve(ped, "Pedestal_orig"));
 
  coral::AttributeListSpecification* spec = new coral::AttributeListSpecification();
  spec->extend("Pedestal", "blob");
  spec->extend("PedestalRMS", "blob");
  spec->extend<unsigned>("version");
 
  CondAttrListCollection* collPed=new CondAttrListCollection(true);
 
  unsigned gain=0;
  unsigned hashMax=m_sonline_idhelper->channelHashMax();
    coral::AttributeList* attrList = new coral::AttributeList(*spec);               
    (*attrList)["version"].setValue(0U);                               
    coral::Blob& blobPed=(*attrList)["Pedestal"].data<coral::Blob>();
    coral::Blob& blobRMS=(*attrList)["PedestalRMS"].data<coral::Blob>();
    blobPed.resize(hashMax*sizeof(float));
    blobRMS.resize(hashMax*sizeof(float));
    float* pblobPed=static_cast<float*>(blobPed.startingAddress());
    float* pblobRMS=static_cast<float*>(blobRMS.startingAddress());
 
    for (unsigned hs=0;hs<hashMax;++hs) {
      const HWIdentifier chid=m_sonline_idhelper->channel_Id(hs);
      float p=ped->pedestal(chid,gain);
      float rms=ped->pedestalRMS(chid,gain);
 
      pblobPed[hs]=p*m_fixFactor;
      pblobRMS[hs]=rms;
      ++n; 
    }//end loop over hash ids
    collPed->add(gain,*attrList);
   
  StatusCode sc=detStore()->record(collPed,"/LAR/ElecCalibMCSC/Pedestal");
  if (sc.isFailure()) {
    ATH_MSG_ERROR( "Failed to record CondAttrListCollection for pedestal "  );
    return sc;
  }
 
   ATH_MSG_DEBUG("  Number of entries changes in pedestals =  " <<n );
 
   // Now ramp
   n=0;
   const LArRampSC *ramp = nullptr;
   CHECK(detStore()->retrieve(ramp, "Ramp_orig"));
 
  spec = new coral::AttributeListSpecification();
  spec->extend("RampVec", "blob");
  spec->extend<unsigned>("nPoints");
  spec->extend<unsigned>("version");
  CondAttrListCollection* coll=new CondAttrListCollection(true);
 
 
  std::vector<float> defaultRamp={0.0,1.0};
 
  gain=0;
 
    unsigned nPoints=0;
    unsigned nDefault=0;
    for (unsigned hs=0;hs<hashMax && nPoints==0;++hs) {
      const HWIdentifier chid=m_sonline_idhelper->channel_Id(hs);
      const ILArRamp::RampRef_t rampref= ramp->ADC2DAC(chid,gain);
      nPoints=rampref.size();
    }
    if (nPoints==0) {
      ATH_MSG_ERROR( "All input Ramps for gain " << gain << " have 0 points!" );
      return StatusCode::FAILURE;
    }
 
    defaultRamp.resize(nPoints,0.0); //fill remaining points if needed
    ATH_MSG_DEBUG( "Gain " << gain << ": Found a ramp polynom of degree " << nPoints << " in input data" );
    attrList = new coral::AttributeList(*spec);
    (*attrList)["version"].setValue(0U);
    coral::Blob& blobRamp=(*attrList)["RampVec"].data<coral::Blob>();
    (*attrList)["nPoints"].setValue(nPoints);
    blobRamp.resize(hashMax*sizeof(float)*nPoints);
    float* pblobRamp=static_cast<float*>(blobRamp.startingAddress());
 
    for (unsigned hs=0;hs<hashMax;++hs) {
      const HWIdentifier chid=m_sonline_idhelper->channel_Id(hs);
      std::vector<float> rampVec(ramp->ADC2DAC(chid,gain).asVector());
      
      if (rampVec.size()>=nPoints) {
    for (size_t i=0;i<nPoints;++i) {
      pblobRamp[nPoints*hs+i]=rampVec[i]*m_fixFactor;
    }
      }
      else {
    std::stringstream message;
    message <<"Polynom degree doesn't match. Expect " << nPoints << ", got " << rampVec.size() << "."; 
    for (size_t i=0;i<nPoints;++i) {
      pblobRamp[nPoints*hs+i]=defaultRamp[i];
    }
    ++nDefault;
      }
      ++n;
    }//end loop over hash ids
    coll->add(gain,*attrList);
  
  sc=detStore()->record(coll,"/LAR/ElecCalibMCSC/Ramp");
  if (sc.isFailure()) {
    ATH_MSG_ERROR( "Failed to record CondAttrListCollection for ramp " );
    return StatusCode::FAILURE;
  }
 
  ATH_MSG_INFO( "Converted Ramps to inline storage. Total number of channels " << n );
  ATH_MSG_INFO( "Number of channels filled with default ramp {0,1} " << nDefault << " (including disconnected)" );
 
   return StatusCode::SUCCESS;
}
 
StatusCode FixLArElecSCCalib::fix2(const LArOnOffIdMapping *cabling, const LArCalibLineMapping *cl) {
 
   ATH_MSG_INFO ( " in fix2() " );
 
   const uint32_t onlHashMax=m_sonline_idhelper->channelHashMax(); // all SC cells
 
   coral::AttributeListSpecification* spec_calib = new coral::AttributeListSpecification();
   spec_calib->extend("OnlineHashToCalibIds", "blob");
   spec_calib->extend<unsigned>("version");
   auto al_calib = std::make_unique<AthenaAttributeList>(*spec_calib);
   coral::Blob& blobCalib=(*al_calib)["OnlineHashToCalibIds"].data<coral::Blob>();
   (*al_calib)["version"].setValue(0U);
   blobCalib.resize(onlHashMax*sizeof(uint32_t)*5); //Bigger than necessary 
   ATH_MSG_DEBUG("blobCalib size: "<<onlHashMax<<"*5");
 
   spec_calib->release();
   spec_calib = nullptr;
 
   uint32_t* pBlobCalib=static_cast<uint32_t*>(blobCalib.startingAddress());
 
  size_t calibIndex=0;
 
  std::vector<unsigned> calibHist(17,0);
  unsigned calibHistMax=0;
 
  std::ofstream outfile("SCIdentifiers.txt");
 
  outfile << "hash id bec pn FT SL chan id calo pn sampl reg eta phi calib" << std::endl;
 
  for (uint32_t onlHash=0;onlHash<onlHashMax;++onlHash) {
    const HWIdentifier hwid=m_sonline_idhelper->channel_Id(onlHash);
    const std::vector<HWIdentifier>& calibIDs_tmp=cl->calibSlotLine(hwid);
    std::vector<HWIdentifier> calibIDs;
    for (unsigned i=0; i<calibIDs_tmp.size(); ++i) {
       if(std::find(calibIDs.begin(), calibIDs.end(), calibIDs_tmp[i]) == calibIDs.end()) calibIDs.push_back(calibIDs_tmp[i]);
    }
    // deduplicate
    Identifier id{};
    auto pLArIdBase = dynamic_cast<const LArOnlineID_Base*>(m_sonline_idhelper);
    auto pCaloCellBase = dynamic_cast<const CaloCell_Base_ID*>(m_scell_idhelper);
    if ((pLArIdBase == nullptr) or (pCaloCellBase==nullptr)){
      ATH_MSG_ERROR("dynamic_cast failed in FixLArElecSCCalib::fix2");
      return StatusCode::FAILURE;
    }
    if (cabling->isOnlineConnected(hwid)) {
       id=cabling->cnvToIdentifier(hwid);
       print(hwid,pLArIdBase, pCaloCellBase,&id,&calibIDs,outfile);
    } else {
       print(hwid,pLArIdBase, pCaloCellBase,nullptr,&calibIDs,outfile);
    }
    const size_t nCalibLines=calibIDs.size();
    if (nCalibLines > calibHistMax ) calibHistMax=nCalibLines;
    if(calibHistMax > 17) {
       ATH_MSG_ERROR( "Too much calib lines, adjust please !!!" << calibHistMax);
       return StatusCode::FAILURE;
    }
    (calibHist[nCalibLines])++;
    pBlobCalib[calibIndex++]=nCalibLines;
    for(uint32_t iCalib=0;iCalib<nCalibLines;++iCalib)
      pBlobCalib[calibIndex++]=calibIDs[iCalib].get_identifier32().get_compact();
  }
  blobCalib.resize(calibIndex*sizeof(uint32_t)); //Size down to actual size
  outfile.close();
  ATH_MSG_INFO( "calibHistMax: " << calibHistMax);
  ATH_MSG_INFO( "BlobSize CalibId:" << calibIndex);
  msg(MSG::INFO) << "nCalib[i] ";
  for (unsigned j=0;j<17;++j)
    msg() << calibHist[j] << "/";
  msg() << endmsg;
  ATH_CHECK(detStore()->record(std::move(al_calib),"/LAR/IdentifierOfl/CalibIdMap_SC"));
  return StatusCode::SUCCESS;
}
 
void FixLArElecSCCalib::print (const HWIdentifier& hwid, const LArOnlineID_Base* onlineID, const CaloCell_Base_ID* caloCellID,
                               const Identifier *id, std::vector<HWIdentifier>* calibIDs, std::ostream& out) {
  const IdentifierHash hwid_hash=onlineID->channel_Hash(hwid);
  out << hwid_hash << " " << std::hex << "0x" << hwid.get_identifier32().get_compact() << std::dec << " "
      << onlineID->barrel_ec(hwid) << " "
      << onlineID->pos_neg(hwid) << " "
      << onlineID->feedthrough(hwid) << " "
      << onlineID->slot(hwid) << " "
      << onlineID->channel(hwid) << " : ";
  if (id) {
    out   << std::hex << "0x" << id->get_identifier32().get_compact() << std::dec << " "
          << caloCellID->sub_calo(*id) << " "
          << caloCellID->pos_neg(*id) << " "
          << caloCellID->sampling(*id) << " "
          << caloCellID->region(*id) << " "
          << caloCellID->eta(*id) << " "
          << caloCellID->phi(*id) << " ";
  }
  else
    out << " disconnected ";
 
  if(calibIDs){
    for (size_t i=0;i<calibIDs->size();++i) {
      out << std::hex << "0x" << calibIDs->at(i).get_identifier32().get_compact() << " ";
    }
  } else 
    out << " no calib ";
 
  out << std::dec << std::endl;
}
 
StatusCode FixLArElecSCCalib::fix3(const LArOnOffIdMapping *cabling, const LArMCSym *sym) {
    
   ATH_MSG_INFO ( " in fix3() " );
 
   // Fix3 is for filling the MinBias blob  from ntuple
 
   auto spec = new coral::AttributeListSpecification();
   spec->extend("MinBias", "blob");
   spec->extend<unsigned>("version");
   auto coll=std::make_unique<CondAttrListCollection>(true);
 
   unsigned hashMax=m_sonline_idhelper->channelHashMax();
   coral::AttributeList attrList = coral::AttributeList(*spec);               
   attrList["version"].setValue(0U);                               
   coral::Blob& blob = attrList["MinBias"].data<coral::Blob>();
   blob.resize(hashMax*sizeof(float));
   float* pblob=static_cast<float*>(blob.startingAddress());
 
   std::unique_ptr<TFile> fin= std::make_unique<TFile>(m_infile.value().c_str());
   TTree *tin=dynamic_cast<TTree*>(fin->Get("m_tree"));
   if (not tin) return StatusCode::FAILURE;
   int           ncell{};
   int *         identifier = new int[2862];   
   int *         layer = new int[2862];   
   int *         region = new int[2862]; 
   int *         ieta = new int[2862];   
   float *       eta = new float[2862];   
   double *      average = new double[2862];   
   double *      rms = new double[2862];   
   TBranch        *b_ncell{};   
   TBranch        *b_identifier{};   
   TBranch        *b_layer{};   
   TBranch        *b_region{};   
   TBranch        *b_ieta{};   
   TBranch        *b_eta{};   
   TBranch        *b_average{};   
   TBranch        *b_rms{};   
   tin->SetMakeClass(1);
   tin->SetBranchAddress("ncell", &ncell, &b_ncell);
   tin->SetBranchAddress("identifier", identifier, &b_identifier);
   tin->SetBranchAddress("layer", layer, &b_layer);
   tin->SetBranchAddress("region", region, &b_region);
   tin->SetBranchAddress("ieta", ieta, &b_ieta);
   tin->SetBranchAddress("eta", eta, &b_eta);
   tin->SetBranchAddress("average", average, &b_average);
   tin->SetBranchAddress("rms", rms, &b_rms);
   tin->GetEntry(0);
 
 
   // read the ntuple (symmetrized)
   std::map<Identifier, float> vmap;
   for(int icell=0; icell<ncell; ++icell)  {
 
       Identifier32 id32(identifier[icell]); 
       Identifier id(id32);
       vmap[id] = average[icell];
 
   }
   // now fill all SC
   for (unsigned onlHash=0;onlHash<hashMax;++onlHash) {
      const HWIdentifier hwid=m_sonline_idhelper->channel_Id(onlHash);
      const Identifier id = cabling->cnvToIdentifier(hwid);
      const Identifier idsym = sym->ZPhiSymOfl(id);
      pblob[onlHash] = vmap[idsym];
   }
 
   coll->add(0,attrList);
   auto sz = coll->size();
   ATH_CHECK(detStore()->record(std::move(coll),"/LAR/ElecCalibMCSC/MinBias"));
   ATH_MSG_DEBUG("Stored coll with size "<<sz<<" into /LAR/ElecCalibMCSC/MinBias");
 
     return StatusCode::SUCCESS;
}