PFClusterSelectorTool.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
#include "CaloIdentifier/CaloCell_ID.h"
#include "eflowRec/eflowRecCluster.h"
#include "PFClusterSelectorTool.h"
#include "xAODCaloEvent/CaloCluster.h"
#include <exception>
 
PFClusterSelectorTool::PFClusterSelectorTool(const std::string& type,const std::string& name,const IInterface* parent):
  base_class(type, name, parent)
{
}
 
StatusCode PFClusterSelectorTool::initialize(){
 
  ATH_CHECK(m_caloClustersReadHandleKey.initialize());
  if(!m_caloCalClustersReadHandleKey.key().empty()) {
    ATH_CHECK(m_caloCalClustersReadHandleKey.initialize());
  }
 
  if (!m_caloClusterReadDecorHandleKeyNLeadingTruthParticles.empty()){
    ATH_CHECK(m_caloClusterReadDecorHandleKeyNLeadingTruthParticles.initialize());
  }
 
  ATH_CHECK(m_caloMgrKey.initialize());
 
  return StatusCode::SUCCESS;
}
 
StatusCode
PFClusterSelectorTool::execute(
  eflowRecClusterContainer& theEFlowRecClusterContainer,
  xAOD::CaloClusterContainer& theCaloClusterContainer) const
{
 
  SG::ReadHandle<xAOD::CaloClusterContainer> caloClustersReadHandle(m_caloClustersReadHandleKey);
  
  /* Verify the read handle has a valid pointer, and if not return */
  if (!caloClustersReadHandle.isValid()){
    ATH_MSG_WARNING(" Invalid ReadHandle for xAOD::CaloCluster with key: " <<  caloClustersReadHandle.key());
    return StatusCode::SUCCESS;
  }
 
  const xAOD::CaloClusterContainer* calclusters = nullptr;
  if (!m_caloCalClustersReadHandleKey.key().empty()) {
    SG::ReadHandle<xAOD::CaloClusterContainer> caloCalClustersReadHandle(m_caloCalClustersReadHandleKey);
    calclusters = caloCalClustersReadHandle.get();
  }
 
  SG::ReadCondHandle<CaloDetDescrManager> caloMgrHandle{m_caloMgrKey};
  ATH_CHECK(caloMgrHandle.isValid());
  
  /* Fill the vector of eflowRecClusters */
  unsigned int nClusters = caloClustersReadHandle->size();
  for (unsigned int iCluster = 0; iCluster < nClusters; ++iCluster) {
    /* Create the eflowRecCluster and put it in the container */
    std::unique_ptr<eflowRecCluster> thisEFRecCluster  = std::make_unique<eflowRecCluster>(ElementLink<xAOD::CaloClusterContainer>(*caloClustersReadHandle, iCluster), theCaloClusterContainer);
    
    if (calclusters){
      std::map<IdentifierHash,double> cellsWeightMap;
      retrieveLCCalCellWeight(caloClustersReadHandle->at(iCluster)->e(), iCluster, cellsWeightMap, *calclusters,**caloMgrHandle);
 
      thisEFRecCluster->setCellsWeight(std::move(cellsWeightMap));
    }
 
    if (!m_caloClusterReadDecorHandleKeyNLeadingTruthParticles.empty()){
 
      std::string decorHandleName = m_caloClusterReadDecorHandleKeyNLeadingTruthParticles.key();
      std::string::size_type pos = decorHandleName.find(".");
      std::string decorName = decorHandleName.substr(pos+1);
 
      SG::AuxElement::Accessor< std::vector< std::pair<unsigned int, double> > > accessor(decorName);
      accessor(*theCaloClusterContainer[iCluster]) = accessor(*caloClustersReadHandle->at(iCluster));
    }
 
    thisEFRecCluster->setClusterId(iCluster);
    theEFlowRecClusterContainer.push_back(std::move(thisEFRecCluster));
  }
  
  return StatusCode::SUCCESS;
}
 
StatusCode PFClusterSelectorTool::finalize(){
  return StatusCode::SUCCESS;
}
 
void
PFClusterSelectorTool::retrieveLCCalCellWeight(
  const double& energy,
  const unsigned& index,
  std::map<IdentifierHash, double>& cellsWeight,
  const xAOD::CaloClusterContainer& caloCalClustersContainer,
  const CaloDetDescrManager& calo_dd_man) const
{
  /* match CaloCluster with CaloCalCluster to obtain cell weight */
  /* first try the position at 'index'. If we are lucky, the loop can be
   * avoided. */
  /* Note the read handle has been tested to be valid prior to the call of this
   * function */
  const xAOD::CaloCluster* matchedCalCluster = caloCalClustersContainer.at(index);
  if (matchedCalCluster){
    if (!(fabs(energy - matchedCalCluster->rawE()) < 0.001)) {
      matchedCalCluster = nullptr;
      for (const auto *iCalCalCluster : caloCalClustersContainer) {
        matchedCalCluster = iCalCalCluster;
        if (fabs(energy - matchedCalCluster->rawE()) < 0.001) {
            break;
          }
      }
      if (!matchedCalCluster) ATH_MSG_WARNING("Invalid pointer to matched cluster - failed to find cluster match");
    }
    if (not matchedCalCluster){ 
      throw std::runtime_error("matchedCluster is a null pointer in PFClusterSelectorTool::retrieveLCCalCellWeight");
    }
    /* obtain cell index and cell weight */    
    const CaloCell_ID*               calo_id  = calo_dd_man.getCaloCell_ID();
    xAOD::CaloCluster::const_cell_iterator itCell = matchedCalCluster->cell_begin();
    xAOD::CaloCluster::const_cell_iterator endCell = matchedCalCluster->cell_end();
    for (; itCell != endCell; ++itCell) {
      const CaloCell* pCell = *itCell;
      Identifier myId = pCell->ID();
      IdentifierHash myHashId = calo_id->calo_cell_hash(myId);
      cellsWeight[myHashId] = itCell.weight();
    }
  } else ATH_MSG_WARNING("Invalid pointer to matched cluster - could not look up local hadron cell weights");
}