LArReadHadDMCoeffFile.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#include "LArCalibTest/LArReadHadDMCoeffFile.h"
#include "GaudiKernel/MsgStream.h"
#include "CaloIdentifier/CaloCell_ID.h"
#include "StoreGate/StoreGateSvc.h"
#include "PathResolver/PathResolver.h"
 
 
LArReadHadDMCoeffFile::LArReadHadDMCoeffFile(const std::string & name, 
                         ISvcLocator * pSvcLocator) : 
  AthAlgorithm(name,pSvcLocator) {
  declareProperty("HadDMCoeffFileName",m_hadDMCoeffFileName);
  declareProperty("CorrectionKey",m_key="HadDMCoeff");
 
  m_data = 0;
}
 
 
LArReadHadDMCoeffFile::~LArReadHadDMCoeffFile() {}
 
                            
StatusCode LArReadHadDMCoeffFile::initialize() {
   ATH_MSG_INFO ( "=== LArReadHadDMCoeffFile::initialize() ===" );
   ATH_CHECK(initDataFromFile(m_hadDMCoeffFileName));
 
   if(msgLvl (MSG::DEBUG)) m_data->PrintData();
 
   ATH_CHECK( detStore()->record(m_data,m_key) );
   ATH_CHECK( detStore()->setConst(m_data) );
   return StatusCode::SUCCESS;
}
 
 
StatusCode  LArReadHadDMCoeffFile::execute() 
{ 
  return StatusCode::SUCCESS;
}
 
StatusCode  LArReadHadDMCoeffFile::finalize() 
{ 
  return StatusCode::SUCCESS;
}
 
 
StatusCode LArReadHadDMCoeffFile::initDataFromFile(const std::string& hadDMCoeffFileName)
{
   m_data = new CaloHadDMCoeff();
 
   // Find the full path to filename
   std::string file = PathResolver::find_file (hadDMCoeffFileName, "DATAPATH");
   ATH_MSG_INFO ( "Reading file  " << file );
 
   FILE *fin = fopen(file.c_str(),"r");
   if ( fin == NULL) {
      ATH_MSG_ERROR ( "Can't open file " << file );
      return StatusCode::FAILURE;
   }
 
   char line[1024], title[128];
   int offset = 0;
   int indx_zone = 0;
   int nFrac, nEner, nEta, nPars;
   while(fgets(line,1024,fin)) {
      if(line[0] == '#' || line[0] == '\n' ) continue;
      int izone, is_on;
      if(sscanf(line,"%80d %80d %127s\n",&izone, &is_on, title)!=3 || indx_zone != izone) {
         ATH_MSG_ERROR ( "Format error #1." );
         fclose (fin);
         return StatusCode::FAILURE;         
      }
      CaloHadDMCoeff::HadDMArea dmArea;
      dmArea.m_is_on = is_on;
      dmArea.m_indx = indx_zone;
      dmArea.m_title = title;
 
      fgets(line,1024,fin);
      if(sscanf(line,"%80d %80f %80f\n", &nFrac, &dmArea.m_MinFrac, &dmArea.m_MaxFrac)!=3 ||
         nFrac<0 || nFrac>1000)
      {
         ATH_MSG_ERROR ( "Format error #2." );
         fclose (fin);
         return StatusCode::FAILURE;
      }
      fgets(line,1024,fin);
      if(sscanf(line,"%80d %80f %80f\n", &nEner, &dmArea.m_MinEner, &dmArea.m_MaxEner)!=3 ||
         nEner < 0 || nEner > 1000)
      {
         ATH_MSG_ERROR ( "Format error #3." );
         fclose (fin);
         return StatusCode::FAILURE;         
      }
      fgets(line,1024,fin);
      if(sscanf(line,"%80d %80f %80f\n", &nEta, &dmArea.m_MinEta, &dmArea.m_MaxEta)!=3 ||
         nEta < 0 || nEta > 1000)
      {
         ATH_MSG_ERROR ( "Format error #4." );
         fclose (fin);
         return StatusCode::FAILURE;         
      }
      fgets(line,1024,fin);
      if(sscanf(line,"%80d \n", &nPars)!=1 || nPars < 0 || nPars> 1000) {
         ATH_MSG_ERROR ( "Format error #5." );
         fclose (fin);
         return StatusCode::FAILURE;         
      }
      dmArea.m_nPars = nPars;
      
      for(int i_frac=0; i_frac<nFrac; i_frac++){
         int iparset = 0, ifrac = 0, iener = 0, ieta = 0;
         float frac = 0, ener = 0, eta = 0;
         for(int i_ener=0; i_ener<nEner; i_ener++){
            for(int i_eta=0; i_eta<nEta; i_eta++){
               fgets(line,1024,fin);
               std::string sLine;
               sLine = std::string(line);
               std::istringstream es( sLine.c_str() );
 
               if ( es >> iparset >> ifrac >> iener >> ieta >> frac >> ener >> eta) {
                  CaloHadDMCoeff::HadDMCoeff pars;
                  pars.resize(nPars);
                  for(int i=0; i<nPars; i++) {
                     if(!(es >> pars[i]) ) {
                        ATH_MSG_ERROR ( "Format error #6." );
                        fclose (fin);
                        return StatusCode::FAILURE;         
                     }
                  }
                  if(ifrac != i_frac || iener != i_ener || ieta != i_eta || (int)pars.size() != nPars){
                     ATH_MSG_ERROR ( "Format error #7." );
                     fclose (fin);
                     return StatusCode::FAILURE;
                  }
 
                  if(i_frac==0 && i_ener==0) dmArea.m_EtaBins.push_back(eta);
                  m_data->addHadDMCoeff(pars);
                  pars.clear();
               } else {
                 ATH_MSG_ERROR ( "Format error #8." );
               }
            } // i_eta
            if(i_frac == 0) dmArea.m_EnerBins.push_back(ener);
         } // i_ener
         dmArea.m_FracBins.push_back(frac);
      }
      dmArea.m_offset = offset;
//      std::cout << "indx_zone: " << indx_zone << " offset:" << offset << std::endl;
      m_data->addHadDMArea(dmArea);
      offset += nFrac*nEner*nEta; // ofset for next DM area
      indx_zone++;
   }
   fclose(fin);
 
  return StatusCode::SUCCESS;
}