EnergyCorrelatorGeneralizedRatiosTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef jetsubstructuremomenttools_energycorrelatorgeneralizedratiostool_header
#define jetsubstructuremomenttools_energycorrelatorgeneralizedratiostool_header
 
#include "JetSubStructureMomentTools/JetSubStructureMomentToolsBase.h"
#include "JetSubStructureMomentTools/ECFHelper.h"
 
class EnergyCorrelatorGeneralizedRatiosTool :
  public JetSubStructureMomentToolsBase {
    ASG_TOOL_CLASS(EnergyCorrelatorGeneralizedRatiosTool, IJetModifier)
    
    public:
      
      EnergyCorrelatorGeneralizedRatiosTool(const std::string& name);
 
      virtual StatusCode initialize() override;
 
      int modifyJet(xAOD::Jet &jet) const override;
 
    private:
 
      struct moments_t;
      
      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::ConstAccessor<float> > m_acc_ECFG_2_1_2;
      std::unique_ptr< SG::AuxElement::ConstAccessor<float> > m_acc_ECFG_3_1_1;
      std::unique_ptr< SG::AuxElement::ConstAccessor<float> > m_acc_ECFG_3_2_1;
      std::unique_ptr< SG::AuxElement::ConstAccessor<float> > m_acc_ECFG_3_2_2;
      std::unique_ptr< SG::AuxElement::ConstAccessor<float> > m_acc_ECFG_3_3_1;
      std::unique_ptr< SG::AuxElement::ConstAccessor<float> > m_acc_ECFG_4_2_2;
      std::unique_ptr< SG::AuxElement::ConstAccessor<float> > m_acc_ECFG_4_4_1;
      
      std::unique_ptr< SG::AuxElement::Decorator<float> > m_dec_L1;
      std::unique_ptr< SG::AuxElement::Decorator<float> > m_dec_L2;
      std::unique_ptr< SG::AuxElement::Decorator<float> > m_dec_L3;
      std::unique_ptr< SG::AuxElement::Decorator<float> > m_dec_L4;
      std::unique_ptr< SG::AuxElement::Decorator<float> > m_dec_L5;
 
};
 
struct EnergyCorrelatorGeneralizedRatiosTool::moments_t {
 
  std::string prefix;
 
  std::string suffix;
 
  float beta;
 
  std::unique_ptr< SG::AuxElement::ConstAccessor<float> > acc_ECFG_2_1;
  std::unique_ptr< SG::AuxElement::ConstAccessor<float> > acc_ECFG_3_1;
  std::unique_ptr< SG::AuxElement::ConstAccessor<float> > acc_ECFG_3_2;
  std::unique_ptr< SG::AuxElement::ConstAccessor<float> > acc_ECFG_4_1;
  std::unique_ptr< SG::AuxElement::ConstAccessor<float> > acc_ECFG_4_2;
 
  std::unique_ptr< SG::AuxElement::ConstAccessor<float> > acc_ECFG_2_1_ungroomed;
  std::unique_ptr< SG::AuxElement::ConstAccessor<float> > acc_ECFG_3_1_ungroomed;
  std::unique_ptr< SG::AuxElement::ConstAccessor<float> > acc_ECFG_3_2_ungroomed;
  
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_M2;
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_M3;
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_N2;
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_N3;
 
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_M2_dichroic;
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_N2_dichroic;
 
  moments_t (float Beta, const std::string& Prefix)
    : prefix (Prefix),
      suffix (GetBetaSuffix(Beta)),
      beta (Beta),
 
      acc_ECFG_2_1 (std::make_unique< SG::AuxElement::ConstAccessor<float> >(prefix+"ECFG_2_1"+suffix)),
      acc_ECFG_3_1 (std::make_unique< SG::AuxElement::ConstAccessor<float> >(prefix+"ECFG_3_1"+suffix)),
      acc_ECFG_3_2 (std::make_unique< SG::AuxElement::ConstAccessor<float> >(prefix+"ECFG_3_2"+suffix)),
      acc_ECFG_4_1 (std::make_unique< SG::AuxElement::ConstAccessor<float> >(prefix+"ECFG_4_1"+suffix)),
      acc_ECFG_4_2 (std::make_unique< SG::AuxElement::ConstAccessor<float> >(prefix+"ECFG_4_2"+suffix)),
 
      acc_ECFG_2_1_ungroomed (std::make_unique< SG::AuxElement::ConstAccessor<float> >(prefix+"ECFG_2_1_ungroomed"+suffix)),
      acc_ECFG_3_1_ungroomed (std::make_unique< SG::AuxElement::ConstAccessor<float> >(prefix+"ECFG_3_1_ungroomed"+suffix)),
      acc_ECFG_3_2_ungroomed (std::make_unique< SG::AuxElement::ConstAccessor<float> >(prefix+"ECFG_3_2_ungroomed"+suffix)),
 
      dec_M2 (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"M2"+suffix)),
      dec_M3 (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"M3"+suffix)),
      dec_N2 (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"N2"+suffix)),
      dec_N3 (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"N3"+suffix)),
 
      dec_M2_dichroic (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"M2_dichroic"+suffix)),
      dec_N2_dichroic (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"N2_dichroic"+suffix))
  {
  }
 
};
 
#endif