PflowIsolationDecorAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
#ifndef DERIVATIONFRAMEWORKMUONS_PFLOWISOLATIONDECORALG_H
#define DERIVATIONFRAMEWORKMUONS_PFLOWISOLATIONDECORALG_H
 
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/WriteDecorHandleKey.h"
#include "xAODPrimitives/IsolationType.h"
#include "xAODTracking/TrackParticleContainer.h"
#include "StoreGate/ReadDecorHandleKeyArray.h"
#include "RecoToolInterfaces/INeutralEFlowIsolationTool.h"

 
namespace DerivationFramework {
    class PflowIsolationDecorAlg : public AthReentrantAlgorithm {
    public:
        using AthReentrantAlgorithm::AthReentrantAlgorithm;

        ~PflowIsolationDecorAlg() = default;

        StatusCode initialize() override;
        StatusCode execute(const EventContext& ctx) const override;
 
    private:
        ToolHandle<xAOD::INeutralEFlowIsolationTool> m_isoTool{this, "IsolationTool", ""};
        xAOD::CaloCorrection m_calo_corr;

        SG::ReadHandleKey<xAOD::TrackParticleContainer> m_trk_key{this, "TrackCollection", "InDetTrackParticles"};

        Gaudi::Property<float> m_pt_min{this, "PtMin", 3. * Gaudi::Units::GeV};
 
        Gaudi::Property<std::string> m_customName{this, "customName", ""};

        Gaudi::Property<std::vector<std::string>> m_trkSel_Decors{this, "TrackSelections", {},
                                                  "List of decorator names of which one needs to be true to run the isolation" };
        SG::ReadDecorHandleKeyArray<xAOD::TrackParticleContainer> m_trkSel_keys{this, "SelectionKeys", {},
                                                                                "Will be overwritten in initialize"};
 
        SG::WriteDecorHandleKey<xAOD::TrackParticleContainer> m_neflowCone20_key{this, "NeflowCone20Key", m_trk_key, "",
                                                                               "Will be overwritten in initialize"};
        SG::WriteDecorHandleKey<xAOD::TrackParticleContainer> m_neflowCone30_key{this, "NeflowCone30Key", m_trk_key, "",
                                                                               "Will be overwritten in initialize"};
        SG::WriteDecorHandleKey<xAOD::TrackParticleContainer> m_neflowCone40_key{this, "NeflowCone40Key", m_trk_key, "",
                                                                               "Will be overwritten in initialize"};
       
        std::vector<xAOD::Iso::IsolationType> m_pflow_isos{xAOD::Iso::neflowisol40, xAOD::Iso::neflowisol30, xAOD::Iso::neflowisol20};
    };
}
#endif