RCJetSubstructureAug.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
// RCJetSubstructureAug.cxx, (c) ATLAS Detector software
// Author: G. Albouy (galbouy@lpsc.in2p3.fr)
// This tool computes substructure variables for ReClustered jets
// from LCTopo clusters ghost associated to RC jets 
// by constructing cluster jets 
 
#ifndef DERIVATIONFRAMEWORK_RCJetSubstructureAug_H
#define DERIVATIONFRAMEWORK_RCJetSubstructureAug_H
 
#include <string>
#include "fastjet/tools/Filter.hh"
#include "fastjet/contrib/SoftDrop.hh"
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "DerivationFrameworkInterfaces/IAugmentationTool.h"
#include "xAODJet/JetContainer.h"
#include "StoreGate/ReadHandleKey.h"
 
#include "ExpressionEvaluation/ExpressionParserUser.h"
 
namespace DerivationFramework {
    enum EJetParser { kRCJetSelection , kNumJetParser };
    class RCJetSubstructureAug :  public extends<ExpressionParserUser<AthAlgTool, kNumJetParser>, IAugmentationTool> {
        public: 
        RCJetSubstructureAug(const std::string& t, const std::string& n, const IInterface* p);
        virtual ~RCJetSubstructureAug();
        virtual StatusCode initialize() override;
        virtual StatusCode finalize() override;
        virtual StatusCode addBranches() const override;
 
        private:
        
        StringProperty m_streamName
            { this, "StreamName", "", "Name of the stream" };
        Gaudi::Property< std::vector<std::string> > m_ghostNames
            { this, "GhostConstitNames", {"GhostLCTopo"}, "Names of the ghost constituents for substructure computation"};
        StringProperty m_selectionString 
            { this, "SelectionString", "", "Selection to apply to jet"};
        StringProperty m_suffix
            { this, "Suffix", "", "Suffix for variables"};
        StringProperty m_grooming
            { this, "Grooming", "", "Name of gromming technic to apply (Trimming or SofDrop)"};
        
        Gaudi::Property<float> m_rclus  
            {this, "RClusTrim", 0.3 , "R for reclustering (0 for none)"};
        Gaudi::Property<float> m_ptfrac      
            {this, "PtFracTrim", 0.03, "pT fraction for retaining subjets"};
        Gaudi::Property<float> m_beta 
            {this, "BetaSoft", 1. , "How much to consider angular dependence"};
        Gaudi::Property<float> m_zcut 
            {this, "ZcutSoft", 1. , "pT fraction for retaining subjets"};
        Gaudi::Property<float> m_R0 
            {this, "R0Soft", 1. , "Normalization of angular distance, usually the characteristic jet radius (default R0 = 1)"};
        
        SG::ReadHandleKey< xAOD::JetContainer > m_jetKey{ this, "JetContainerKey", ""};
 
        // Struct to hold all decorators
        struct moments_t;
        moments_t* m_moments;
 
        // The filter object that will apply the grooming
        std::unique_ptr<fastjet::Filter> m_trimmer;
        std::unique_ptr<fastjet::contrib::SoftDrop> m_softdropper;
 
    };
 
    struct RCJetSubstructureAug::moments_t {
        SG::AuxElement::Decorator<float> dec_Qw;
 
        SG::AuxElement::Decorator<float> dec_Tau1;
        SG::AuxElement::Decorator<float> dec_Tau2;
        SG::AuxElement::Decorator<float> dec_Tau3;
        SG::AuxElement::Decorator<float> dec_Tau4;
        SG::AuxElement::Decorator<float> dec_Tau21;
        SG::AuxElement::Decorator<float> dec_Tau32;
 
        SG::AuxElement::Decorator<float> dec_Split12;
        SG::AuxElement::Decorator<float> dec_Split23;
        SG::AuxElement::Decorator<float> dec_Split34;
 
        SG::AuxElement::Decorator<float> dec_ECF1;
        SG::AuxElement::Decorator<float> dec_ECF2;
        SG::AuxElement::Decorator<float> dec_ECF3;
        SG::AuxElement::Decorator<float> dec_ECF4;
        SG::AuxElement::Decorator<float> dec_C2;
        SG::AuxElement::Decorator<float> dec_D2;
 
        SG::AuxElement::Decorator<float> dec_pT;
        SG::AuxElement::Decorator<float> dec_m;
        SG::AuxElement::Decorator<float> dec_NConstits;
        SG::AuxElement::Decorator<float> dec_eta;
        SG::AuxElement::Decorator<float> dec_phi;
 
        SG::AuxElement::Decorator<float> dec_timing;
        
        moments_t (const std::string& suffix):
        
            dec_Qw("Qw_" + suffix),
 
            dec_Tau1("Tau1_" + suffix), 
            dec_Tau2("Tau2_" + suffix),
            dec_Tau3("Tau3_" + suffix),
            dec_Tau4("Tau4_" + suffix),
            dec_Tau21("Tau21_" + suffix),
            dec_Tau32("Tau32_" + suffix),
 
            dec_Split12("Split12_" + suffix),
            dec_Split23("Split23_" + suffix),
            dec_Split34("Split34_" + suffix),
 
            dec_ECF1("ECF1_" + suffix),
            dec_ECF2("ECF2_" + suffix),
            dec_ECF3("ECF3_" + suffix),
            dec_ECF4("ECF4_" + suffix),
            dec_C2("C2_" + suffix),
            dec_D2("D2_" + suffix),
 
            dec_pT("pT_" + suffix),
            dec_m("m_" + suffix),
            dec_NConstits("NConstits_" + suffix),
            dec_eta("eta_" + suffix),
            dec_phi("phi_" + suffix),
 
            dec_timing("timing_" + suffix){}
 
    };
 
}
 
#endif // DERIVATIONFRAMEWORK_RCJetSubstructureAug_H