EnergyCorrelatorRatiosTool.h

Go to the documentation of this file.

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

 
#ifndef jetsubstructuremomenttools_energycorrelatorratiostool_header
#define jetsubstructuremomenttools_energycorrelatorratiostool_header
 
#include "JetSubStructureMomentTools/JetSubStructureMomentToolsBase.h"
#include "JetSubStructureMomentTools/ECFHelper.h"
 
class EnergyCorrelatorRatiosTool :
  public JetSubStructureMomentToolsBase {
    ASG_TOOL_CLASS(EnergyCorrelatorRatiosTool, IJetModifier)
 
    public:

      EnergyCorrelatorRatiosTool(const std::string& name);
     
      virtual StatusCode initialize() override;
 
      int modifyJet(xAOD::Jet &jet) const override;
 
    private:

      struct moments_t;

      bool m_doC3;
      bool m_doC4;
      std::vector<float> m_rawBetaVals; 
      bool m_doDichroic;

      std::map< float, moments_t > m_moments;
 
  };

 
struct EnergyCorrelatorRatiosTool::moments_t {

  std::string prefix;

  std::string suffix;

  float beta;

  std::unique_ptr< SG::AuxElement::ConstAccessor<float> > acc_ECF1;
  std::unique_ptr< SG::AuxElement::ConstAccessor<float> > acc_ECF2;
  std::unique_ptr< SG::AuxElement::ConstAccessor<float> > acc_ECF3;
  std::unique_ptr< SG::AuxElement::ConstAccessor<float> > acc_ECF4;
  std::unique_ptr< SG::AuxElement::ConstAccessor<float> > acc_ECF5;

  std::unique_ptr< SG::AuxElement::ConstAccessor<float> > acc_ECF1_ungroomed;
  std::unique_ptr< SG::AuxElement::ConstAccessor<float> > acc_ECF2_ungroomed;
  std::unique_ptr< SG::AuxElement::ConstAccessor<float> > acc_ECF3_ungroomed;

  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_C1;
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_C2;
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_C3;
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_C4;
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_D2;

  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_D2_dichroic;
 
  moments_t (float Beta, const std::string& Prefix)
    : prefix (Prefix),
      suffix (GetBetaSuffix(Beta)),
      beta (Beta),
 
      acc_ECF1 (std::make_unique< SG::AuxElement::ConstAccessor<float> >(prefix+"ECF1"+suffix)),
      acc_ECF2 (std::make_unique< SG::AuxElement::ConstAccessor<float> >(prefix+"ECF2"+suffix)),
      acc_ECF3 (std::make_unique< SG::AuxElement::ConstAccessor<float> >(prefix+"ECF3"+suffix)),
      acc_ECF4 (std::make_unique< SG::AuxElement::ConstAccessor<float> >(prefix+"ECF4"+suffix)),
      acc_ECF5 (std::make_unique< SG::AuxElement::ConstAccessor<float> >(prefix+"ECF5"+suffix)),
 
      acc_ECF1_ungroomed (std::make_unique< SG::AuxElement::ConstAccessor<float> >(prefix+"ECF1_ungroomed"+suffix)),
      acc_ECF2_ungroomed (std::make_unique< SG::AuxElement::ConstAccessor<float> >(prefix+"ECF2_ungroomed"+suffix)),
      acc_ECF3_ungroomed (std::make_unique< SG::AuxElement::ConstAccessor<float> >(prefix+"ECF3_ungroomed"+suffix)),
      
      dec_C1 (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"C1"+suffix)),
      dec_C2 (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"C2"+suffix)),
      dec_C3 (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"C3"+suffix)),
      dec_C4 (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"C4"+suffix)),
      dec_D2 (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"D2"+suffix)),
 
      dec_D2_dichroic (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"D2_dichroic"+suffix))
  {
  }
 
};
 
#endif