eFexTOBDecorator.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
//***************************************************************************
//                           eFexTOBDecorator  -  description
//                              -------------------
//      This algorithm decorates the eFEX TOBs with jet descriminant variables and, for the EM TOBs, with the cluster layer supercell Et sums
//      recalculated from input data using the eFEXTOBEtTool
//      All values are in 25 MeV counts
//      
//     begin                : 03 02 2022
//     email                : paul.daniel.thompson@cern.ch
//***************************************************************************/
#include "eFexTOBDecorator.h"
#include "L1CaloFEXSim/eFEXegTOB.h"
#include "L1CaloFEXSim/eFEXtauAlgo.h"
 
namespace LVL1 {
 
  eFexTOBDecorator::eFexTOBDecorator(const std::string& name, ISvcLocator* svc) : AthAlgorithm(name, svc){}
 
  StatusCode eFexTOBDecorator::initialize() {
    ATH_MSG_INFO( "L1CaloFEXTools/eFexTOBDecorator::initialize()");
 
    // initialise read and decorator write handles
    ATH_CHECK( m_eFEXegEDMContainerKey.initialize() );
    ATH_CHECK( m_eFEXtauEDMContainerKey.initialize() );
 
    ATH_CHECK( m_RetaCoreDec.initialize() );
    ATH_CHECK( m_RetaEnvDec.initialize() );
    ATH_CHECK( m_RhadEMDec.initialize() );
    ATH_CHECK( m_RhadHadDec.initialize() );
    ATH_CHECK( m_WstotDenDec.initialize() );    
    ATH_CHECK( m_WstotNumDec.initialize() );    
 
    ATH_CHECK( m_ClusterEtSumPSDec.initialize() );
    ATH_CHECK( m_ClusterEtSumL1Dec.initialize() );
    ATH_CHECK( m_ClusterEtSumL2Dec.initialize() );
    ATH_CHECK( m_ClusterEtSumL3Dec.initialize() );
 
    ATH_CHECK( m_RCoreDec.initialize() );      
    ATH_CHECK( m_REnvDec.initialize() );       
    ATH_CHECK( m_REMCoreDec.initialize() );    
    ATH_CHECK( m_REMHadDec.initialize() );     
 
    // Retrieve the TOB ET sum tool
    ATH_CHECK( m_eFEXTOBEtTool.retrieve() );
 
    return StatusCode::SUCCESS;
  }
 
  StatusCode eFexTOBDecorator::execute() {
    
    // read the TOB containers
    SG::ReadHandle<xAOD::eFexEMRoIContainer> eFEXegEDMContainerObj{m_eFEXegEDMContainerKey};
    if (!eFEXegEDMContainerObj.isValid()) {
      ATH_MSG_ERROR("Failed to retrieve EDM collection");
      return StatusCode::SUCCESS; 
    }
    const xAOD::eFexEMRoIContainer* emEDMConstPtr = eFEXegEDMContainerObj.cptr();
 
    SG::ReadHandle<xAOD::eFexTauRoIContainer> eFEXtauEDMContainerObj{m_eFEXtauEDMContainerKey};
    if (!eFEXtauEDMContainerObj.isValid()) {
      ATH_MSG_ERROR("Failed to retrieve tau EDM collection");
      return StatusCode::SUCCESS; 
    }
    const xAOD::eFexTauRoIContainer* tauEDMConstPtr = eFEXtauEDMContainerObj.cptr();
    
    //Setup EM Decorator Handlers
    SG::WriteDecorHandle<xAOD::eFexEMRoIContainer, unsigned int >   RetaCoreDec  (m_RetaCoreDec );
    SG::WriteDecorHandle<xAOD::eFexEMRoIContainer, unsigned int >   RetaEnvDec   (m_RetaEnvDec  );
    SG::WriteDecorHandle<xAOD::eFexEMRoIContainer, unsigned int >   RhadEMDec    (m_RhadEMDec   );
    SG::WriteDecorHandle<xAOD::eFexEMRoIContainer, unsigned int >   RhadHadDec   (m_RhadHadDec  );
    SG::WriteDecorHandle<xAOD::eFexEMRoIContainer, unsigned int >   WstotDenDec  (m_WstotDenDec );
    SG::WriteDecorHandle<xAOD::eFexEMRoIContainer, unsigned int >   WstotNumDec  (m_WstotNumDec );
 
    SG::WriteDecorHandle<xAOD::eFexEMRoIContainer, unsigned int >   ClusterEtSumPSDec  (m_ClusterEtSumPSDec );
    SG::WriteDecorHandle<xAOD::eFexEMRoIContainer, unsigned int >   ClusterEtSumL1Dec  (m_ClusterEtSumL1Dec );
    SG::WriteDecorHandle<xAOD::eFexEMRoIContainer, unsigned int >   ClusterEtSumL2Dec  (m_ClusterEtSumL2Dec );
    SG::WriteDecorHandle<xAOD::eFexEMRoIContainer, unsigned int >   ClusterEtSumL3Dec  (m_ClusterEtSumL3Dec );
 
    //looping over EM TOB to decorate them
    for ( const xAOD::eFexEMRoI* emRoI : *emEDMConstPtr ){
              
      float eta = emRoI->eta();
      float phi = emRoI->phi();
      int seed = emRoI->seed();
      int UnD = emRoI->UpNotDown();
      std::vector<unsigned int> ClusterCellETs;
      std::vector<unsigned int> RetaSums;
      std::vector<unsigned int> RhadSums;
      std::vector<unsigned int> WstotSums;
 
      ATH_CHECK( m_eFEXTOBEtTool->getegSums(eta, phi, seed, UnD, ClusterCellETs, RetaSums, RhadSums, WstotSums) );
 
      RetaCoreDec (*emRoI) = RetaSums[0];
      RetaEnvDec  (*emRoI) = RetaSums[1];
      RhadEMDec   (*emRoI) = RhadSums[0];
      RhadHadDec  (*emRoI) = RhadSums[1];
      WstotDenDec (*emRoI) = WstotSums[0];
      WstotNumDec (*emRoI) = WstotSums[1];
 
      // cluster et per layer summed from SCells
      unsigned int ps(0); // 0-1 - pre-sampler
      unsigned int l1(0); // 2-7 - layer 1
      unsigned int l2(0); // 8-13 - layer 2
      unsigned int l3(0); // 14-15 - layer 3
      for(std::size_t i = 0; i < ClusterCellETs.size(); ++i) {
    if (i<2)
      ps += ClusterCellETs[i];
    else if (i < 8 )
      l1 += ClusterCellETs[i];
    else if (i < 14 )
      l2 += ClusterCellETs[i];
    else
      l3 += ClusterCellETs[i];
      }
 
      ClusterEtSumPSDec (*emRoI) = ps;
      ClusterEtSumL1Dec (*emRoI) = l1;
      ClusterEtSumL2Dec (*emRoI) = l2;
      ClusterEtSumL3Dec (*emRoI) = l3;
    }
 
    //Setup Tau Decorator Handlers
    SG::WriteDecorHandle<xAOD::eFexTauRoIContainer, unsigned int >   RCoreDec   (m_RCoreDec   );
    SG::WriteDecorHandle<xAOD::eFexTauRoIContainer, unsigned int >   REnvDec    (m_REnvDec    );
    SG::WriteDecorHandle<xAOD::eFexTauRoIContainer, unsigned int >   REMCoreDec (m_REMCoreDec );
    SG::WriteDecorHandle<xAOD::eFexTauRoIContainer, unsigned int >   REMHadDec  (m_REMHadDec  );
 
    //looping over Tau TOB to decorate them
    for ( const xAOD::eFexTauRoI* tauRoI : *tauEDMConstPtr ){
              
      float eta = tauRoI->eta();
      float phi = tauRoI->phi();
      int seed = tauRoI->seed();
      int UnD = tauRoI->upNotDown();
      std::vector<unsigned int> RcoreSums;
      std::vector<unsigned int> RemSums;
 
      ATH_CHECK( m_eFEXTOBEtTool->gettauSums(eta, phi, seed, UnD, RcoreSums, RemSums) );
 
      RCoreDec   (*tauRoI) = RcoreSums[0];
      REnvDec    (*tauRoI) = RcoreSums[1];
      REMCoreDec (*tauRoI) = RemSums[0];
      REMHadDec  (*tauRoI) = RemSums[1];
    }
    
    // Return gracefully
    return StatusCode::SUCCESS;
  }
 
}