ClusterAtMLScaleTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef JETRECTOOLS_CLUSTERATMLSCALETOOL_H
#define JETRECTOOLS_CLUSTERATMLSCALETOOL_H

 
 
#include "JetRecTools/JetConstituentModifierBase.h"
#include "xAODBase/IParticleContainer.h"
#include "xAODCaloEvent/CaloClusterContainer.h"
#include "xAODCaloEvent/CaloCluster.h"
#include "AsgTools/AsgTool.h" 
#include <string>
 
class ClusterAtMLScaleTool : public JetConstituentModifierBase{
  ASG_TOOL_CLASS(ClusterAtMLScaleTool, IJetConstituentModifier)
 
  public:
 
  ClusterAtMLScaleTool(const std::string& name);
  ~ClusterAtMLScaleTool();
 
  // Check that the configuration is reasonable
  virtual StatusCode initialize() override;
  
  private:
  // Property to configure the ML energy decoration key
  Gaudi::Property<std::string> m_clusterMLCorrectedEnergyKey{this, "ClusterMLCorrectedEnergyDecorationKey", "clusterE_ML", 
    "Name of the decoration storing the ML-corrected cluster energy"};
  // Implement the correction
  virtual StatusCode process_impl(xAOD::IParticleContainer* cont) const override; 
  StatusCode setClustersToMLScale(xAOD::CaloClusterContainer& cont) const;
 
 
        
};
 
 
#endif