CaloEMFrac.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
*/
 
// $Id: CaloEMFrac.cxx,v 1.2 2007-10-26 01:47:23 ssnyder Exp $
#include "CaloConditions/CaloEMFrac.h"
 
 
CaloEMFrac::CaloEMFrac()
  : m_nBinEta(0),
    m_nBinLogEClus(0),
    m_nBinLogEDens(0),
    m_nBinLogDepth(0),
    m_MinEta(0),
    m_MinLogEClus(0),
    m_MinLogEDens(0),
    m_MinLogDepth(0),
    m_MaxEta(0),
    m_MaxLogEClus(0),
    m_MaxLogEDens(0),
    m_MaxLogDepth(0)
{
}
 
 
CaloEMFrac::CaloEMFrac(
            int   nBinEta,     
            float MinEta,      
            float MaxEta,      
            int   nBinLogEClus,
            float MinLogEClus, 
            float MaxLogEClus, 
            int   nBinLogEDens,
            float MinLogEDens, 
            float MaxLogEDens, 
            int   nBinLogDepth,
            float MinLogDepth, 
            float MaxLogDepth  
            ) : m_nBinEta       (nBinEta),      
                m_nBinLogEClus  (nBinLogEClus),    
                m_nBinLogEDens  (nBinLogEDens),
                m_nBinLogDepth  (nBinLogDepth),
                m_MinEta        (MinEta),       
                m_MinLogEClus   (MinLogEClus),      
                m_MinLogEDens   (MinLogEDens), 
                m_MinLogDepth   (MinLogDepth), 
                m_MaxEta        (MaxEta),       
                m_MaxLogEClus   (MaxLogEClus),      
                m_MaxLogEDens   (MaxLogEDens), 
                m_MaxLogDepth   (MaxLogDepth) 
{
} 
 
 
const struct CaloEMFrac::EMFracData *
CaloEMFrac::getEMFracData(float eta,
                          float logEClus,
                          float logEDens,
                          float logDepth) const
{
  int iBin = getBin(eta,logEClus,logEDens,logDepth);
  if ( iBin > -1 && iBin < (int)m_data.size() ) 
    return (& m_data[iBin]);
  else
    return nullptr;
}
 
 
 
void CaloEMFrac::setEMFracData(float eta,
                               float logEClus,
                               float logEDens,
                               float logDepth,
                               const struct EMFracData & theData)
{
  const int iEtaLogEClus = getEtaLogEClusBin(eta, logEClus);
  if ( iEtaLogEClus > -1 ) {
    const int iFilledEntries = m_data.size()/(m_nBinLogEDens*m_nBinLogDepth);
    if ( iEtaLogEClus >= iFilledEntries ) 
      m_data.resize((iFilledEntries+1)*(m_nBinLogEDens*m_nBinLogDepth));
    const int iBin = getBin(eta,logEClus,logEDens,logDepth);
    if ( iBin > -1 ) {
      m_data[iBin] = theData;
    }
  }
}
 
 
int CaloEMFrac::getEtaLogEClusBin(float eta, float logEClus) const
{
  int iEta,iLogEClus;
 
  iEta = (int)((eta-m_MinEta)/(m_MaxEta-m_MinEta)*m_nBinEta);
  if ( iEta >= 0 && iEta < m_nBinEta ) {
    iLogEClus = (int)((logEClus-m_MinLogEClus)/(m_MaxLogEClus-m_MinLogEClus)*m_nBinLogEClus);
    if ( iLogEClus >= 0 && iLogEClus < m_nBinLogEClus ) {
      return (iEta*m_nBinLogEClus+iLogEClus);
    }
  }
  return (-1);
 
}
 
 
int CaloEMFrac::getBin(float eta,
                       float logEClus,
                       float logEDens,
                       float logDepth) const
{
  int iLogEDens,iLogDepth;
 
  const int iEtaLogEClus = getEtaLogEClusBin(eta, logEClus);
  if ( iEtaLogEClus > -1 ) {
    iLogEDens = (int)((logEDens-m_MinLogEDens)/(m_MaxLogEDens-m_MinLogEDens)*m_nBinLogEDens);
    if ( iLogEDens >= 0 && iLogEDens < m_nBinLogEDens ) {
      iLogDepth = (int)((logDepth-m_MinLogDepth)/(m_MaxLogDepth-m_MinLogDepth)*m_nBinLogDepth);
      if ( iLogDepth >= 0 && iLogDepth < m_nBinLogDepth ) {
    return (iLogDepth + m_nBinLogDepth * 
        (iLogEDens + m_nBinLogEDens * 
         iEtaLogEClus));
      }
    }
  }
  return -1;
}