PFMomentCalculatorTool.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#include "PFMomentCalculatorTool.h"
#include "eflowCaloObject.h"
#include "eflowRecCluster.h"
 
#include "xAODCaloEvent/CaloClusterKineHelper.h"
 
PFMomentCalculatorTool::PFMomentCalculatorTool(const std::string& type,const std::string& name,const IInterface* parent) :
  base_class( type, name, parent)
{
}
 
StatusCode PFMomentCalculatorTool::initialize(){
 
  /* Retrieve the cluster collection tool */
  ATH_CHECK(m_clusterCollectionTool.retrieve());
 
  /* Retrieve the cluster moments maker */
  ATH_CHECK(m_clusterMomentsMaker.retrieve());
 
  /* Retrieve the cluster calib hit moments maker */
  if (m_useCalibHitTruth) ATH_CHECK(m_clusterCalibHitMomentsMaker2.retrieve());
  else m_clusterCalibHitMomentsMaker2.disable();
 
  return StatusCode::SUCCESS;
}
 
StatusCode PFMomentCalculatorTool::execute(const EventContext& ctx, eflowCaloObjectContainer& theEflowCaloObjectContainer) {
 
  /* Collect all the clusters in a temporary container (with VIEW_ELEMENTS!) */
  bool useNonModifiedClusters = true;
  if (true == m_LCMode) useNonModifiedClusters = false;
  std::unique_ptr<xAOD::CaloClusterContainer> tempClusterContainer = m_clusterCollectionTool->execute(theEflowCaloObjectContainer, useNonModifiedClusters);
 
  /* Set the layer energies */
  /* This must be set before the cluster moment calculations, which use the layer energies */
  for (auto cluster : *tempClusterContainer) CaloClusterKineHelper::calculateKine(cluster, true, true);
 
  /* Remake the cluster moments */
  ATH_CHECK(m_clusterMomentsMaker->execute(ctx, tempClusterContainer.get()));
 
  if (m_useCalibHitTruth){
    ATH_CHECK(m_clusterCalibHitMomentsMaker2->execute(ctx, tempClusterContainer.get()));
  }
 
  return StatusCode::SUCCESS;
}
 
StatusCode PFMomentCalculatorTool::finalize(){ return StatusCode::SUCCESS; }