Reconstruction/MuonIdentification/MuonCombinedEvent/MuonCombinedEvent/MuonCandidate.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MUONCOMBINEDEVENT_MUONCANDIDATE_H
#define MUONCOMBINEDEVENT_MUONCANDIDATE_H
 
#include <memory>
#include <vector>
 
#include "AthLinks/ElementLink.h"
#include "xAODMuon/MuonSegmentContainer.h"
#include "xAODTracking/TrackParticle.h"
#include "xAODTracking/TrackParticleContainer.h"
 
namespace Trk {
    class Track;
}
namespace Muon {
    class MuonSegment;
}
 
namespace MuonCombined {
 
    class MuonCandidate {
    public:
        MuonCandidate(const ElementLink<xAOD::TrackParticleContainer>& msTrackLink);
 
        /*Constructor taking two ElementLinks, and the index location of the extrapolated track in the container*/
        MuonCandidate(const ElementLink<xAOD::TrackParticleContainer>& msTrackLink, const ElementLink<TrackCollection>& saTrackLink,
                      size_t container_idx);
 
        /* Update constructor to insert the element link to the extrapolated MSOE xAOD TrackParticle */
        MuonCandidate(const MuonCandidate& oldCandidate, const ElementLink<xAOD::TrackParticleContainer>& msoeTrackLink);

        ~MuonCandidate();

        const Trk::Track& muonSpectrometerTrack() const;

        const ElementLink<xAOD::TrackParticleContainer>& muonSpectrometerTrackLink() const;

        const Trk::Track* extrapolatedTrack() const;

        const Trk::Track* primaryTrack() const;

        const ElementLink<TrackCollection>& extrapolatedTrackLink() const;
        size_t extrapolatedElementID() const;
        const ElementLink<xAOD::TrackParticleContainer>& extrapolatedParticleLink() const;

        std::string toString() const;

        void setSegments(std::vector<const Muon::MuonSegment*>&& segments);
        const std::vector<const Muon::MuonSegment*>& getSegments() const;
 
    private:
        const ElementLink<xAOD::TrackParticleContainer> m_muonSpectrometerTrackLink{};
        const ElementLink<xAOD::TrackParticleContainer> m_extrapolatedParticleLink{};
        const ElementLink<TrackCollection> m_extrapolatedTrackLink{};

        const Trk::Track* m_muonSpectrometerTrack{nullptr};

        std::vector<const Muon::MuonSegment*> m_assoc_segments;

        size_t m_extContIdx{0};
    };
 
}  // namespace MuonCombined
 
#endif