MsTrackSeed.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 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);
            }

            MsTrackSeed(const Location loc,
                        const ExpandedSector sector);
            std::span<const xAOD::MuonSegment* const> segments() 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; }
        private:
            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{};          
    };
    using MsTrackSeedContainer = std::vector<MsTrackSeed>;
    std::ostream& operator<<(std::ostream& ostr, const MuonR4::MsTrackSeed& seed);
 
}
CLASS_DEF( MuonR4::MsTrackSeedContainer , 1290595104 , 1 )
ACTS_OSTREAM_FORMATTER (MuonR4::MsTrackSeed::Location);
#endif