MuonCombinedInDetCandidateAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MUONCOMBINEDALGS_MUONCOMBINEDINDETCANDIDATEALG_H
#define MUONCOMBINEDALGS_MUONCOMBINEDINDETCANDIDATEALG_H
 
 
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "MuonCombinedEvent/InDetCandidateCollection.h"
#include "MuonRecToolInterfaces/IMuonSystemExtensionTool.h"
#include "MuonStationIndex/MuonStationIndex.h"
 
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/WriteHandleKey.h"
#include "StoreGate/ReadDecorHandleKeyArray.h"
 
#include "TrkToolInterfaces/ITrackSelectorTool.h"
#include "xAODTracking/TrackParticleContainer.h"
 
class MuonCombinedInDetCandidateAlg : public AthReentrantAlgorithm {
public:
    using AthReentrantAlgorithm::AthReentrantAlgorithm;
    ~MuonCombinedInDetCandidateAlg() = default;
 
    virtual StatusCode initialize() override final;
    virtual StatusCode execute(const EventContext& ctx) const override final;
 
private:
    SG::ReadHandleKeyArray<xAOD::TrackParticleContainer> m_indetTrackParticleLocation{
        this, "TrackParticleLocation", {"InDetTrackParticles"}};
    SG::ReadHandleKeyArray<CaloExtensionCollection> m_caloExtensionLocation{this, "CaloExtensionLocation", {} };
    SG::ReadHandleKey<xAOD::TrackParticleContainer> m_indetForwardTrackParticleLocation{this, "ForwardParticleLocation",
                                                                                        "InDetForwardTrackParticles"};
 
    SG::ReadDecorHandleKeyArray<xAOD::TrackParticleContainer> m_trkLinkKey{this, "TrackLinkKeys", {}};
    SG::ReadHandleKey<CaloExtensionCollection> m_caloFwdExtensionLocation{this, "CaloFwdExtensionLocation", ""};
 
    SG::WriteHandleKey<InDetCandidateCollection> m_candidateCollectionName{this, "InDetCandidateLocation", "InDetCandidates"};
 
    ToolHandle<Trk::ITrackSelectorTool> m_trackSelector{
        this, "TrackSelector", "InDet::InDetDetailedTrackSelectorTool/MuonCombinedInDetDetailedTrackSelectorTool", "Track selector tool"};
    ToolHandle<Trk::ITrackSelectorTool> m_forwardTrackSelector{
        this, "InDetForwardTrackSelector", "InDet::InDetDetailedTrackSelectorTool/MuonCombinedInDetDetailedForwardTrackSelectorTool",
        "Forward track selector tool"};
    ToolHandle<Muon::IMuonSystemExtensionTool> m_muonSystemExtensionTool{this, "MuonSystemExtensionTool", "", "Muon system extension tool"};
 
    struct InDetCandidateCache {
        const xAOD::TrackParticleContainer* inDetContainer{nullptr};
        const CaloExtensionCollection* extensionContainer{nullptr};
        bool flagAsSAF{false};
        std::unique_ptr<InDetCandidateCollection> outputContainer = std::make_unique<InDetCandidateCollection>(SG::OWN_ELEMENTS);
        const Trk::ITrackSelectorTool* trackSelector{nullptr};
    };
 
    StatusCode create(const EventContext& ctx, InDetCandidateCache& output_cache) const;
 
    bool isValidTrackParticle(const Trk::ITrackSelectorTool* currentTrackSelector, const xAOD::TrackParticle* const tp) const;
    void printTrackParticleInfo(const xAOD::TrackParticle* const tp, const std::string& what) const;
 
    static int getCount(const xAOD::TrackParticle& tp, const xAOD::SummaryType type);
 
    Gaudi::Property<bool> m_doSiliconForwardMuons{this, "DoSiliconAssocForwardMuons", false};
    Gaudi::Property<float> m_extThreshold{this, "ExtensionPtThreshold", 2500};
    Gaudi::Property<bool> m_extendSAF{this, "ExtendSAF", false};
    Gaudi::Property<bool> m_extendBulk{this, "ExtendBulk", true};
    Gaudi::Property<bool> m_requireExtension{this, "RequireExtension", true};
    Gaudi::Property<bool> m_waitForTrackLink{this, "WaitForTrackLink", true};
};
 
#endif