EnergyCorrelatorTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef jetsubstructuremomenttools_energycorrelatortool_header
#define jetsubstructuremomenttools_energycorrelatortool_header
 
#include "JetSubStructureMomentTools/JetSubStructureMomentToolsBase.h"
#include "JetSubStructureMomentTools/ECFHelper.h"
 
class EnergyCorrelatorTool :
  public JetSubStructureMomentToolsBase {
    ASG_TOOL_CLASS(EnergyCorrelatorTool, IJetModifier)
 
    public:
      
      EnergyCorrelatorTool(const std::string& name);
     
      virtual StatusCode initialize() override;
 
      int modifyJet(xAOD::Jet &injet) const override;
 
    private:
 
      struct moments_t;
 
      float m_Beta;
      bool m_doC3;
      bool m_doC4;
      std::vector<float> m_rawBetaVals; 
      bool m_doDichroic;
      
      std::map< float, moments_t > m_moments;
 
  };
 
struct EnergyCorrelatorTool::moments_t {
 
  std::string prefix;
 
  std::string suffix;
 
  float beta;
 
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_ECF1;
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_ECF2;
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_ECF3;
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_ECF4;
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_ECF5;
 
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_ECF1_ungroomed;
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_ECF2_ungroomed;
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_ECF3_ungroomed;
 
  moments_t (float Beta, const std::string& Prefix)
    : prefix (Prefix),
      suffix (GetBetaSuffix(Beta)),
      beta (Beta),
 
      dec_ECF1 (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"ECF1"+suffix)),
      dec_ECF2 (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"ECF2"+suffix)),
      dec_ECF3 (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"ECF3"+suffix)),
      dec_ECF4 (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"ECF4"+suffix)),
      dec_ECF5 (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"ECF5"+suffix)),
 
      dec_ECF1_ungroomed (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"ECF1_ungroomed"+suffix)),
      dec_ECF2_ungroomed (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"ECF2_ungroomed"+suffix)),
      dec_ECF3_ungroomed (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"ECF3_ungroomed"+suffix))
  {
  }
 
};
 
#endif