NSubjettinessTool.h

Go to the documentation of this file.

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

 
#ifndef jetsubstructuremomenttools_nsubjetinesstool_header
#define jetsubstructuremomenttools_nsubjetinesstool_header
 
#include "JetSubStructureMomentTools/JetSubStructureMomentToolsBase.h"
#include "JetSubStructureMomentTools/NSubjettinessHelper.h"
 
#include "AsgTools/PropertyWrapper.h"
#include "AsgDataHandles/WriteDecorHandleKeyArray.h"
 
class NSubjettinessTool :
  public JetSubStructureMomentToolsBase {
    ASG_TOOL_CLASS(NSubjettinessTool, IJetModifier)
 
    public:
      // Constructor and destructor
      NSubjettinessTool(const std::string& name);
 
      StatusCode initialize() override;
 
      StatusCode modify(xAOD::JetContainer& jets) const override;
 
    private:
      Gaudi::Property<std::string> m_jetContainerName{
    this, "JetContainer", "", "SG key for the input jet container"};

  
      struct moments_t{
    std::string prefix;

    std::string suffix;

    float alpha;
 
    moments_t (float Alpha, const std::string& Prefix)
      : prefix (Prefix),
        suffix (GetAlphaSuffix(Alpha)),
        alpha (Alpha) {}
      };

      float m_Alpha;
      std::vector<float> m_rawAlphaVals; 
      bool m_doDichroic;

      std::vector<std::pair< float, moments_t >> m_moments;
 
      SG::WriteDecorHandleKeyArray<xAOD::JetContainer> m_Tau1_Keys{
    this, "Tau1_Keys", {}};
      SG::WriteDecorHandleKeyArray<xAOD::JetContainer> m_Tau2_Keys{
    this, "Tau2_Keys", {}};
      SG::WriteDecorHandleKeyArray<xAOD::JetContainer> m_Tau3_Keys{
    this, "Tau3_Keys", {}};
      SG::WriteDecorHandleKeyArray<xAOD::JetContainer> m_Tau4_Keys{
    this, "Tau4_Keys", {}};
 
      SG::WriteDecorHandleKeyArray<xAOD::JetContainer> m_Tau2_ungroomed_Keys{
    this, "Tau2_ungroomed_Keys", {}};
      SG::WriteDecorHandleKeyArray<xAOD::JetContainer> m_Tau3_ungroomed_Keys{
    this, "Tau3_ungroomed_Keys", {}};
      SG::WriteDecorHandleKeyArray<xAOD::JetContainer> m_Tau4_ungroomed_Keys{
    this, "Tau4_ungroomed_Keys", {}};
 
      SG::WriteDecorHandleKeyArray<xAOD::JetContainer> m_Tau1_wta_Keys{
    this, "Tau1_wta_Keys", {}};
      SG::WriteDecorHandleKeyArray<xAOD::JetContainer> m_Tau2_wta_Keys{
    this, "Tau2_wta_Keys", {}};
      SG::WriteDecorHandleKeyArray<xAOD::JetContainer> m_Tau3_wta_Keys{
    this, "Tau3_wta_Keys", {}};
      SG::WriteDecorHandleKeyArray<xAOD::JetContainer> m_Tau4_wta_Keys{
    this, "Tau4_wta_Keys", {}};
 
      SG::WriteDecorHandleKeyArray<xAOD::JetContainer> m_Tau2_wta_ungroomed_Keys{
    this, "Tau2_wta_ungroomed_Keys", {}};
      SG::WriteDecorHandleKeyArray<xAOD::JetContainer> m_Tau3_wta_ungroomed_Keys{
    this, "Tau3_wta_ungroomed_Keys", {}};
      SG::WriteDecorHandleKeyArray<xAOD::JetContainer> m_Tau4_wta_ungroomed_Keys{
    this, "Tau4_wta_ungroomed_Keys", {}};
 
};
 
#endif