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"
 
class EnergyCorrelatorGeneralizedTool :
  public JetSubStructureMomentToolsBase {
    ASG_TOOL_CLASS(EnergyCorrelatorGeneralizedTool, IJetModifier)
 
    public:
 
      EnergyCorrelatorGeneralizedTool(const std::string& name);
 
      virtual StatusCode initialize() override;
 
      int modifyJet(xAOD::Jet &injet) const override;
 
    private:
 
      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;
 
      std::unique_ptr< SG::AuxElement::Decorator<float> > m_dec_ECFG_2_1_2;
      std::unique_ptr< SG::AuxElement::Decorator<float> > m_dec_ECFG_3_1_1;
      std::unique_ptr< SG::AuxElement::Decorator<float> > m_dec_ECFG_3_2_1;
      std::unique_ptr< SG::AuxElement::Decorator<float> > m_dec_ECFG_3_2_2;
      std::unique_ptr< SG::AuxElement::Decorator<float> > m_dec_ECFG_3_3_1;
      std::unique_ptr< SG::AuxElement::Decorator<float> > m_dec_ECFG_4_2_2;
      std::unique_ptr< SG::AuxElement::Decorator<float> > m_dec_ECFG_4_4_1;
 
      std::unique_ptr< SG::AuxElement::Decorator<float> > m_dec_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