EnergyCorrelatorGeneralizedTool.h

Go to the documentation of this file.

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

 
#ifndef jetsubstructuremomenttools_energycorrelatorgeneralizedtool_header
#define jetsubstructuremomenttools_energycorrelatorgeneralizedtool_header
 
#include "JetSubStructureMomentTools/JetSubStructureMomentToolsBase.h"
#include "JetSubStructureMomentTools/ECFHelper.h"
 
#include "AsgTools/PropertyWrapper.h"
#include "AsgDataHandles/WriteDecorHandleKey.h"
 
class EnergyCorrelatorGeneralizedTool :
  public JetSubStructureMomentToolsBase {
    ASG_TOOL_CLASS(EnergyCorrelatorGeneralizedTool, IJetModifier)
 
    public:

      EnergyCorrelatorGeneralizedTool(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;

      float m_Beta;
      bool m_doM3;
      bool m_doN3;
      bool m_doLSeries;
      std::vector<float> m_rawBetaVals; 
      bool m_doDichroic;

      std::map< float, moments_t > m_moments;

      SG::WriteDecorHandleKey<xAOD::JetContainer> m_ECFG_2_1_2_Key{
    this, "ECFG_2_1_2_Key", "ECFG_2_1_2"};
      SG::WriteDecorHandleKey<xAOD::JetContainer> m_ECFG_3_1_1_Key{
    this, "ECFG_3_1_1_Key", "ECFG_3_1_1"};
      SG::WriteDecorHandleKey<xAOD::JetContainer> m_ECFG_3_2_1_Key{
    this, "ECFG_3_2_1_Key", "ECFG_3_2_1"};
      SG::WriteDecorHandleKey<xAOD::JetContainer> m_ECFG_3_2_2_Key{
    this, "ECFG_3_2_2_Key", "ECFG_3_2_2"};
      SG::WriteDecorHandleKey<xAOD::JetContainer> m_ECFG_3_3_1_Key{
        this, "ECFG_3_3_1_Key", "ECFG_3_3_1"};
      SG::WriteDecorHandleKey<xAOD::JetContainer> m_ECFG_4_2_2_Key{
    this, "ECFG_4_2_2_Key", "ECFG_4_2_2"};
      SG::WriteDecorHandleKey<xAOD::JetContainer> m_ECFG_4_4_1_Key{
    this, "ECFG_4_4_1_Key", "ECFG_4_4_1"};

      SG::WriteDecorHandleKey<xAOD::JetContainer> m_ECFG_3_3_2_Key{
    this, "ECFG_3_3_2_Key", "ECFG_3_3_2"};
 
  };

 
struct EnergyCorrelatorGeneralizedTool::moments_t {

  std::string prefix;

  std::string suffix;

  float beta;

  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_ECFG_2_1;
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_ECFG_3_1;
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_ECFG_3_2;
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_ECFG_4_1;
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_ECFG_4_2;
  
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_ECFG_2_1_ungroomed;
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_ECFG_3_1_ungroomed;
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_ECFG_3_2_ungroomed;
  
  moments_t (float Beta, const std::string& Prefix)
    : prefix (Prefix),
      suffix (GetBetaSuffix(Beta)),
      beta (Beta),
 
      dec_ECFG_2_1 (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"ECFG_2_1"+suffix)),
      dec_ECFG_3_1 (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"ECFG_3_1"+suffix)),
      dec_ECFG_3_2 (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"ECFG_3_2"+suffix)),
      dec_ECFG_4_1 (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"ECFG_4_1"+suffix)),
      dec_ECFG_4_2 (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"ECFG_4_2"+suffix)),
    
      dec_ECFG_2_1_ungroomed (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"ECFG_2_1_ungroomed"+suffix)),
      dec_ECFG_3_1_ungroomed (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"ECFG_3_1_ungroomed"+suffix)),
      dec_ECFG_3_2_ungroomed (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"ECFG_3_2_ungroomed"+suffix))
  {
  }
 
};
 
#endif