EnergyCorrelatorTool.h

Go to the documentation of this file.

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

 
#ifndef jetsubstructuremomenttools_energycorrelatortool_header
#define jetsubstructuremomenttools_energycorrelatortool_header
 
#include "JetSubStructureMomentTools/JetSubStructureMomentToolsBase.h"
#include "JetSubStructureMomentTools/ECFHelper.h"
 
#include "AsgTools/PropertyWrapper.h"
#include "AsgDataHandles/WriteDecorHandleKeyArray.h"
#include "AsgDataHandles/WriteDecorHandle.h"
 
class EnergyCorrelatorTool :
  public JetSubStructureMomentToolsBase {
    ASG_TOOL_CLASS(EnergyCorrelatorTool, IJetModifier)
 
    public:
      
      EnergyCorrelatorTool(const std::string& name);
     
      virtual StatusCode initialize() override;
 
      StatusCode modify(xAOD::JetContainer& jets) const override;
 
    private:
      Gaudi::Property<std::string> m_jetContainerName{
    this, "JetContainer", "", "SG key for the input jet container"};


      struct moments_t {

    std::string prefix;

    std::string suffix;

    float beta;
 
    moments_t (float Beta, const std::string& Prefix)
      : prefix (Prefix),
        suffix (GetBetaSuffix(Beta)),
        beta (Beta) {}
      };

      float m_Beta;
      bool m_doC3;
      bool m_doC4;
      std::vector<float> m_rawBetaVals; 
      bool m_doDichroic;
      
      std::vector<std::pair< float, moments_t >> m_moments;
 
      SG::WriteDecorHandleKeyArray<xAOD::JetContainer> m_ECF1_Keys{
    this, "ECF1_Keys", {}};
      SG::WriteDecorHandleKeyArray<xAOD::JetContainer> m_ECF2_Keys{
    this, "ECF2_Keys", {}};
      SG::WriteDecorHandleKeyArray<xAOD::JetContainer> m_ECF3_Keys{
    this, "ECF3_Keys", {}};
      SG::WriteDecorHandleKeyArray<xAOD::JetContainer> m_ECF4_Keys{
    this, "ECF4_Keys", {}};
      SG::WriteDecorHandleKeyArray<xAOD::JetContainer> m_ECF5_Keys{
    this, "ECF5_Keys", {}};
 
      SG::WriteDecorHandleKeyArray<xAOD::JetContainer> m_ECF1_ungroomed_Keys{
    this, "ECF1_ungroomed_Keys", {}};
      SG::WriteDecorHandleKeyArray<xAOD::JetContainer> m_ECF2_ungroomed_Keys{
    this, "ECF2_ungroomed_Keys", {}};
      SG::WriteDecorHandleKeyArray<xAOD::JetContainer> m_ECF3_ungroomed_Keys{
    this, "ECF3_ungroomed_Keys", {}};
 
};
 
#endif