MsTrackSeed.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
#ifndef MUONTRACKEVENT_MSTRACKSEED_H
#define MUONTRACKEVENT_MSTRACKSEED_H
 
#include <vector>
 
#include "AthContainers/DataVector.h"
#include "MuonSpacePoint/SpacePointContainer.h"
#include "MuonTrackEvent/ExpandedSector.h"
#include "MuonPatternEvent/Segment.h"
 
#include "xAODMuon/MuonSegment.h"
 
#include "Acts/Utilities/OstreamFormatter.hpp"
namespace MuonR4{
 
    class MsTrackSeed {
        public:
            enum class Location: int8_t{
                Undefined =-1,
                Barrel,
                Endcap
            };
            static std::string toString(const Location loc);
            friend std::ostream& operator<<(std::ostream& ostr, const Location loc) {
                return ostr<<toString(loc);
            }
            friend std::ostream& operator<<(std::ostream& ostr, const MuonR4::MsTrackSeed& seed) {
                 return seed.print(ostr);
            }

            explicit MsTrackSeed(const Location loc,
                                 const ExpandedSector sector);
            std::span<const xAOD::MuonSegment* const> segments() const;
            std::span<const Muon::MuonStationIndex::StIndex> stations() const;
            std::vector<const SpacePointBucket*> buckets() const;
            void addSegment(const xAOD::MuonSegment* seg);
            void replaceSegment(const xAOD::MuonSegment* exist,
                                const xAOD::MuonSegment* updated);

            const Amg::Vector3D& position() const;
            void setPosition(Amg::Vector3D&& pos);
            Location location() const;
            ExpandedSector sector() const { return m_sector; }
            std::size_t nStations() const;
            std::size_t nMeasurements() const;
            void prepareOverlap(std::shared_ptr<std::uint8_t> marker);
            void triggerOverlapMarker();
            bool hasOverlap() const;
        private:
            std::ostream& print(std::ostream& ostr) const;
            float pathLength(const xAOD::MuonSegment& segment) const;
            Location m_loc{Location::Undefined};
            ExpandedSector m_sector{static_cast<std::int8_t>(0)};
            Amg::Vector3D m_pos{Amg::Vector3D::Zero()};
            std::vector<const xAOD::MuonSegment*> m_segments{};
            std::size_t m_nMeasurements{0ul};
            std::vector<Muon::MuonStationIndex::StIndex> m_stations{};
            std::vector<std::shared_ptr<std::uint8_t>> m_overlapMarker{};
    };
    using MsTrackSeedContainer = std::vector<MsTrackSeed>;
}
CLASS_DEF( MuonR4::MsTrackSeedContainer , 1290595104 , 1 )
ACTS_OSTREAM_FORMATTER (MuonR4::MsTrackSeed::Location);
#endif