NSubjettinessRatiosTool.h

Go to the documentation of this file.

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

 
#ifndef jetsubstructuremomenttools_nsubjetinessratiostool_header
#define jetsubstructuremomenttools_nsubjetinessratiostool_header
 
#include "JetSubStructureMomentTools/JetSubStructureMomentToolsBase.h"
#include "JetSubStructureMomentTools/NSubjettinessHelper.h"
 
class NSubjettinessRatiosTool :
  public JetSubStructureMomentToolsBase {
    ASG_TOOL_CLASS(NSubjettinessRatiosTool, IJetModifier)
 
    public:

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

      struct moments_t;

      std::vector<float> m_rawAlphaVals; 
      bool m_doDichroic;
  
      std::map< float, moments_t > m_moments;
 
  };

 
struct NSubjettinessRatiosTool::moments_t {

  std::string prefix;

  std::string suffix;

  float alpha;

  std::unique_ptr< SG::AuxElement::ConstAccessor<float> > acc_Tau1;
  std::unique_ptr< SG::AuxElement::ConstAccessor<float> > acc_Tau2;
  std::unique_ptr< SG::AuxElement::ConstAccessor<float> > acc_Tau3;
  std::unique_ptr< SG::AuxElement::ConstAccessor<float> > acc_Tau4;

  std::unique_ptr< SG::AuxElement::ConstAccessor<float> > acc_Tau2_ungroomed;
  std::unique_ptr< SG::AuxElement::ConstAccessor<float> > acc_Tau3_ungroomed;
  std::unique_ptr< SG::AuxElement::ConstAccessor<float> > acc_Tau4_ungroomed;

  std::unique_ptr< SG::AuxElement::ConstAccessor<float> > acc_Tau1_wta;
  std::unique_ptr< SG::AuxElement::ConstAccessor<float> > acc_Tau2_wta;
  std::unique_ptr< SG::AuxElement::ConstAccessor<float> > acc_Tau3_wta;
  std::unique_ptr< SG::AuxElement::ConstAccessor<float> > acc_Tau4_wta;

  std::unique_ptr< SG::AuxElement::ConstAccessor<float> > acc_Tau2_wta_ungroomed;
  std::unique_ptr< SG::AuxElement::ConstAccessor<float> > acc_Tau3_wta_ungroomed;
  std::unique_ptr< SG::AuxElement::ConstAccessor<float> > acc_Tau4_wta_ungroomed;

  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_Tau21;
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_Tau32;
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_Tau42;

  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_Tau21_dichroic;
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_Tau32_dichroic;
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_Tau42_dichroic;

  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_Tau21_wta;
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_Tau32_wta;
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_Tau42_wta;

  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_Tau21_wta_dichroic;
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_Tau32_wta_dichroic;
  std::unique_ptr< SG::AuxElement::Decorator<float> > dec_Tau42_wta_dichroic;
 
  moments_t (float Alpha, const std::string& Prefix)
    : prefix (Prefix),
      suffix (GetAlphaSuffix(Alpha)),
      alpha (Alpha),
 
      acc_Tau1 (std::make_unique< SG::AuxElement::ConstAccessor<float> >(prefix+"Tau1"+suffix)),
      acc_Tau2 (std::make_unique< SG::AuxElement::ConstAccessor<float> >(prefix+"Tau2"+suffix)),
      acc_Tau3 (std::make_unique< SG::AuxElement::ConstAccessor<float> >(prefix+"Tau3"+suffix)),
      acc_Tau4 (std::make_unique< SG::AuxElement::ConstAccessor<float> >(prefix+"Tau4"+suffix)),
 
      acc_Tau2_ungroomed (std::make_unique< SG::AuxElement::ConstAccessor<float> >(prefix+"Tau2_ungroomed"+suffix)),
      acc_Tau3_ungroomed (std::make_unique< SG::AuxElement::ConstAccessor<float> >(prefix+"Tau3_ungroomed"+suffix)),
      acc_Tau4_ungroomed (std::make_unique< SG::AuxElement::ConstAccessor<float> >(prefix+"Tau4_ungroomed"+suffix)),
 
      acc_Tau1_wta (std::make_unique< SG::AuxElement::ConstAccessor<float> >(prefix+"Tau1_wta"+suffix)),
      acc_Tau2_wta (std::make_unique< SG::AuxElement::ConstAccessor<float> >(prefix+"Tau2_wta"+suffix)),
      acc_Tau3_wta (std::make_unique< SG::AuxElement::ConstAccessor<float> >(prefix+"Tau3_wta"+suffix)),
      acc_Tau4_wta (std::make_unique< SG::AuxElement::ConstAccessor<float> >(prefix+"Tau4_wta"+suffix)),
 
      acc_Tau2_wta_ungroomed (std::make_unique< SG::AuxElement::ConstAccessor<float> >(prefix+"Tau2_wta_ungroomed"+suffix)),
      acc_Tau3_wta_ungroomed (std::make_unique< SG::AuxElement::ConstAccessor<float> >(prefix+"Tau3_wta_ungroomed"+suffix)),
      acc_Tau4_wta_ungroomed (std::make_unique< SG::AuxElement::ConstAccessor<float> >(prefix+"Tau4_wta_ungroomed"+suffix)),
 
      dec_Tau21 (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"Tau21"+suffix)),
      dec_Tau32 (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"Tau32"+suffix)),
      dec_Tau42 (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"Tau42"+suffix)),
 
      dec_Tau21_dichroic (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"Tau21_dichroic"+suffix)),
      dec_Tau32_dichroic (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"Tau32_dichroic"+suffix)),
      dec_Tau42_dichroic (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"Tau42_dichroic"+suffix)),
 
      dec_Tau21_wta (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"Tau21_wta"+suffix)),
      dec_Tau32_wta (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"Tau32_wta"+suffix)),
      dec_Tau42_wta (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"Tau42_wta"+suffix)),
 
      dec_Tau21_wta_dichroic (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"Tau21_wta_dichroic"+suffix)),
      dec_Tau32_wta_dichroic (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"Tau32_wta_dichroic"+suffix)),
      dec_Tau42_wta_dichroic (std::make_unique< SG::AuxElement::Decorator<float> >(prefix+"Tau42_wta_dichroic"+suffix))
  {
  }
 
};
 
#endif