NSubjettinessTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef jetsubstructuremomenttools_nsubjetinesstool_header
#define jetsubstructuremomenttools_nsubjetinesstool_header
 
#include "JetSubStructureMomentTools/JetSubStructureMomentToolsBase.h"
#include "JetSubStructureMomentTools/NSubjettinessHelper.h"
 
class NSubjettinessTool :
  public JetSubStructureMomentToolsBase {
    ASG_TOOL_CLASS(NSubjettinessTool, IJetModifier)
 
    public:
      // Constructor and destructor
      NSubjettinessTool(const std::string& name);
 
      StatusCode initialize();
 
      int modifyJet(xAOD::Jet &injet) const;
 
    private:
 
      struct moments_t;
 
      float m_Alpha;
      std::vector<float> m_rawAlphaVals; 
      bool m_doDichroic;
 
      std::map< float, moments_t > m_moments;
 
  };
 
struct NSubjettinessTool::moments_t {
 
  std::string prefix;
 
  float alpha;
 
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_Tau1;
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_Tau2;
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_Tau3;
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_Tau4;
 
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_Tau2_ungroomed;
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_Tau3_ungroomed;
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_Tau4_ungroomed;
 
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_Tau1_wta;
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_Tau2_wta;
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_Tau3_wta;
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_Tau4_wta;
 
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_Tau2_wta_ungroomed;
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_Tau3_wta_ungroomed;
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_Tau4_wta_ungroomed;
 
  moments_t (float Alpha, const std::string& Prefix)
    : prefix (Prefix),
      alpha (Alpha)
  {
    std::string suffix = GetAlphaSuffix(alpha);
 
    dec_Tau1 = std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"Tau1"+suffix);
    dec_Tau2 = std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"Tau2"+suffix);
    dec_Tau3 = std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"Tau3"+suffix);
    dec_Tau4 = std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"Tau4"+suffix);
 
    dec_Tau2_ungroomed = std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"Tau2_ungroomed"+suffix);
    dec_Tau3_ungroomed = std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"Tau3_ungroomed"+suffix);
    dec_Tau4_ungroomed = std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"Tau4_ungroomed"+suffix);
 
    dec_Tau1_wta = std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"Tau1_wta"+suffix);
    dec_Tau2_wta = std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"Tau2_wta"+suffix);
    dec_Tau3_wta = std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"Tau3_wta"+suffix);
    dec_Tau4_wta = std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"Tau4_wta"+suffix);
 
    dec_Tau2_wta_ungroomed = std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"Tau2_wta_ungroomed"+suffix);
    dec_Tau3_wta_ungroomed = std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"Tau3_wta_ungroomed"+suffix);
    dec_Tau4_wta_ungroomed = std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"Tau4_wta_ungroomed"+suffix);
 
  }
 
};
 
#endif