LArPhysWaves2Ntuple.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#include "LArCalibTools/LArPhysWaves2Ntuple.h"
#include "CaloIdentifier/CaloGain.h"
 
 
LArPhysWaves2Ntuple::LArPhysWaves2Ntuple(const std::string& name, ISvcLocator* pSvcLocator): LArWaves2Ntuple(name, pSvcLocator)
{ 
  m_keylist.clear() ;
  declareProperty("KeyList",        m_keylist);
  declareProperty("NtupleName",     m_ntName); 
  declareProperty("AddCorrUndo",   m_addCorrUndo=true);
  declareProperty("ApplyCorrection",m_applyCorr=false);
}
 
LArPhysWaves2Ntuple::~LArPhysWaves2Ntuple()
= default;
 
 
StatusCode LArPhysWaves2Ntuple::initialize() {
  m_ntTitle="Physics Wave";
  m_ntpath=std::string("/NTUPLES/FILE1/")+m_ntName;
  return LArWaves2Ntuple::initialize();
}
 
 
StatusCode LArPhysWaves2Ntuple::stop() 
{ 
  
  StatusCode sc;
 
  sc=m_nt->addItem("gain",m_gain,0,3);
  if (sc!=StatusCode::SUCCESS) {
    ATH_MSG_ERROR( "addItem 'gain' failed" );
    return StatusCode::FAILURE;
  }
 
  sc=m_nt->addItem("timeOffset",m_timeOffset,0.,100.);
  if (sc!=StatusCode::SUCCESS) {
    ATH_MSG_ERROR( "addItem 'timeOffset' failed" );
    return StatusCode::FAILURE;
  }
 
 
  if (m_addCorrUndo) {
    sc=m_nt->addItem("corrUndo",m_corrUndo,0,1);
    if (sc!=StatusCode::SUCCESS) {
      ATH_MSG_ERROR( "addItem 'corrUndo' failed" );
      return StatusCode::FAILURE;
    }
  }
 
  for ( unsigned k=0 ; k<m_keylist.size() ; k++ ) {
    const std::string& key = m_keylist[k] ;
 
 
    ATH_MSG_INFO( "Processing WaveContainer from StoreGate! key=" << m_keylist[k] ); 
    const LArPhysWaveContainer* physWaveContainer = nullptr;
    LArPhysWaveContainer* physWaveContainer_nc = nullptr;
    StatusCode sc;
    if (m_applyCorr) {
      sc = m_detStore->retrieve(physWaveContainer_nc,key);      
      if (sc.isFailure()) {
    ATH_MSG_ERROR( "Cannot read non-const LArPhysWaveContainer from StoreGate! key=" << key );
    ATH_MSG_ERROR( "The ApplyCorrection option works only for non-const containers!" );
    return sc;
      }
      else
    ATH_MSG_INFO( "Read non-const LArPhysWaveContainer from StoreGate! key= "  << key );
 
      if (!physWaveContainer_nc->correctionsApplied()) {
    sc=physWaveContainer_nc->applyCorrections();
    if (sc.isFailure()) {
      ATH_MSG_ERROR( "Failed to apply corrections to LArPhysWaveContainer!" );
    }
      }
      physWaveContainer=physWaveContainer_nc;
    }
    else {
      sc = m_detStore->retrieve(physWaveContainer,key);     
      if (sc.isFailure()) {
    ATH_MSG_ERROR( "Cannot read LArPhysWaveContainer from StoreGate! key=" << key );
    return StatusCode::FAILURE;
      } else 
    ATH_MSG_INFO( "Read const LArPhysWaveContainer from StoreGate! key= "  << key );
    }
    
    for (int igain=CaloGain::LARHIGHGAIN;igain<m_NGains.value() ;++igain){
      for (const HWIdentifier chid: m_onlineId->channel_range()) {
    m_gain=(long)igain;
    const LArPhysWave& wave=physWaveContainer->get(chid,igain);
    if (wave.isEmpty()) continue;
    if (m_addCorrUndo) m_corrUndo=0;
    fillWave(chid,wave); //Fill method from base-class
    m_timeOffset = wave.getTimeOffset();
    sc=ntupleSvc()->writeRecord(m_nt);
    if (sc!=StatusCode::SUCCESS) {
      ATH_MSG_ERROR( "writeRecord failed" );
      return sc;
    }
      }//end loop over identifiers
    }//end loop over gains
 
    if (m_addCorrUndo) {
      for (int igain=CaloGain::LARHIGHGAIN;igain<m_NGains ;++igain) {
    LArPhysWaveContainer::ConstCorrectionIt itUndo=physWaveContainer->undoCorrBegin(igain);
    LArPhysWaveContainer::ConstCorrectionIt itUndo_e=physWaveContainer->undoCorrEnd(igain);
    for(;itUndo!=itUndo_e;itUndo++) {
      const HWIdentifier chid(itUndo->first);
      const LArPhysWave& wave = itUndo->second;
      m_gain  = (long)igain;
      m_corrUndo = 1;
      fillWave(chid,wave); //Fill method from base-class
      sc=ntupleSvc()->writeRecord(m_nt);
      if (sc!=StatusCode::SUCCESS) {
        ATH_MSG_ERROR( "writeRecord failed" );
        return sc;
      }
    }//end loop over corrections
      }//end loop over gain
    }//end if addCorrUndo
  }//end loop over container keys
  ATH_MSG_INFO( "LArWave2Ntuple has finished." );
  return StatusCode::SUCCESS;
} // end finalize-method.