ARK-OS/arkos/arkrt/ark.ui.basic/Sources/Values/Axis.swift

/// An axis in a 2D coordinate system.
public enum Axis: Sendable, CaseIterable {
    /// The horizontal axis.
    case horizontal
    /// The vertical axis.
    case vertical

    /// Gets the orientation with this axis as its main axis.
    var orientation: Orientation {
        switch self {
            case .horizontal:
                .horizontal
            case .vertical:
                .vertical
        }
    }

    /// A set of axes represented as an efficient bit field.
    public struct Set: OptionSet, Sendable {
        // Required to satisfy older compilers (5.10) due to our custom
        // `contains(_:)` overload below which confuses the inference
        // of SetAlgebra's Element associated type.
        public typealias Element = Self

        /// The horizontal axis.
        public static let horizontal = Set(rawValue: 1)
        /// The vertical axis.
        public static let vertical = Set(rawValue: 2)

        public var rawValue: UInt8

        public init(rawValue: UInt8) {
            self.rawValue = rawValue
        }

        /// Gets whether a given member is a member of the option set.
        public func contains(_ member: Axis) -> Bool {
            switch member {
                case .horizontal:
                    contains(Axis.Set.horizontal)
                case .vertical:
                    contains(Axis.Set.vertical)
            }
        }
    }
}