mirror of
https://github.com/apple/swift.git
synced 2025-12-14 20:36:38 +01:00
771 lines
27 KiB
Swift
771 lines
27 KiB
Swift
//===--- MutableRawSpan.swift ---------------------------------------------===//
|
|
//
|
|
// This source file is part of the Swift.org open source project
|
|
//
|
|
// Copyright (c) 2024 - 2025 Apple Inc. and the Swift project authors
|
|
// Licensed under Apache License v2.0 with Runtime Library Exception
|
|
//
|
|
// See https://swift.org/LICENSE.txt for license information
|
|
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#if SPAN_COMPATIBILITY_STUB
|
|
import Swift
|
|
#endif
|
|
|
|
// A MutableRawSpan represents a span of memory which
|
|
// contains initialized `Element` instances.
|
|
@safe
|
|
@frozen
|
|
@available(SwiftCompatibilitySpan 5.0, *)
|
|
@_originallyDefinedIn(module: "Swift;CompatibilitySpan", SwiftCompatibilitySpan 6.2)
|
|
public struct MutableRawSpan: ~Copyable & ~Escapable {
|
|
@usableFromInline
|
|
internal let _pointer: UnsafeMutableRawPointer?
|
|
|
|
@usableFromInline
|
|
internal let _count: Int
|
|
|
|
@unsafe
|
|
@_alwaysEmitIntoClient
|
|
internal func _start() -> UnsafeMutableRawPointer {
|
|
unsafe _pointer._unsafelyUnwrappedUnchecked
|
|
}
|
|
|
|
@_alwaysEmitIntoClient
|
|
@inline(__always)
|
|
@lifetime(immortal)
|
|
public init() {
|
|
unsafe _pointer = nil
|
|
_count = 0
|
|
}
|
|
|
|
@unsafe
|
|
@_unsafeNonescapableResult
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(borrow pointer)
|
|
internal init(
|
|
_unchecked pointer: UnsafeMutableRawPointer?,
|
|
byteCount: Int
|
|
) {
|
|
unsafe _pointer = pointer
|
|
_count = byteCount
|
|
}
|
|
}
|
|
|
|
@available(SwiftCompatibilitySpan 5.0, *)
|
|
@_originallyDefinedIn(module: "Swift;CompatibilitySpan", SwiftCompatibilitySpan 6.2)
|
|
extension MutableRawSpan: @unchecked Sendable {}
|
|
|
|
@available(SwiftCompatibilitySpan 5.0, *)
|
|
@_originallyDefinedIn(module: "Swift;CompatibilitySpan", SwiftCompatibilitySpan 6.2)
|
|
extension MutableRawSpan {
|
|
|
|
@unsafe
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(borrow bytes)
|
|
public init(
|
|
_unsafeBytes bytes: UnsafeMutableRawBufferPointer
|
|
) {
|
|
let (baseAddress, count) = (bytes.baseAddress, bytes.count)
|
|
let span = unsafe MutableRawSpan(_unchecked: baseAddress, byteCount: count)
|
|
self = unsafe _overrideLifetime(span, borrowing: bytes)
|
|
}
|
|
|
|
@unsafe
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(borrow bytes)
|
|
public init(
|
|
_unsafeBytes bytes: borrowing Slice<UnsafeMutableRawBufferPointer>
|
|
) {
|
|
let rebased = unsafe UnsafeMutableRawBufferPointer(rebasing: bytes)
|
|
let span = unsafe MutableRawSpan(_unsafeBytes: rebased)
|
|
self = unsafe _overrideLifetime(span, borrowing: bytes)
|
|
}
|
|
|
|
@unsafe
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(borrow pointer)
|
|
public init(
|
|
_unsafeStart pointer: UnsafeMutableRawPointer,
|
|
byteCount: Int
|
|
) {
|
|
_precondition(byteCount >= 0, "Count must not be negative")
|
|
unsafe self.init(_unchecked: pointer, byteCount: byteCount)
|
|
}
|
|
|
|
@unsafe
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(borrow elements)
|
|
public init<Element: BitwiseCopyable>(
|
|
_unsafeElements elements: UnsafeMutableBufferPointer<Element>
|
|
) {
|
|
let bytes = UnsafeMutableRawBufferPointer(elements)
|
|
let span = unsafe MutableRawSpan(_unsafeBytes: bytes)
|
|
self = unsafe _overrideLifetime(span, borrowing: elements)
|
|
}
|
|
|
|
@unsafe
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(borrow elements)
|
|
public init<Element: BitwiseCopyable>(
|
|
_unsafeElements elements: borrowing Slice<UnsafeMutableBufferPointer<Element>>
|
|
) {
|
|
let rebased = unsafe UnsafeMutableBufferPointer(rebasing: elements)
|
|
let span = unsafe MutableRawSpan(_unsafeElements: rebased)
|
|
self = unsafe _overrideLifetime(span, borrowing: elements)
|
|
}
|
|
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(&elements)
|
|
public init<Element: BitwiseCopyable>(
|
|
_elements elements: inout MutableSpan<Element>
|
|
) {
|
|
let (start, count) = unsafe (elements._pointer, elements._count)
|
|
let span = unsafe MutableRawSpan(
|
|
_unchecked: start,
|
|
byteCount: count == 1 ? MemoryLayout<Element>.size
|
|
: count &* MemoryLayout<Element>.stride
|
|
)
|
|
self = unsafe _overrideLifetime(span, mutating: &elements)
|
|
}
|
|
}
|
|
|
|
@available(SwiftCompatibilitySpan 5.0, *)
|
|
@_originallyDefinedIn(module: "Swift;CompatibilitySpan", SwiftCompatibilitySpan 6.2)
|
|
extension MutableRawSpan {
|
|
@_alwaysEmitIntoClient
|
|
public var byteCount: Int { _count }
|
|
|
|
@_alwaysEmitIntoClient
|
|
public var isEmpty: Bool { byteCount == 0 }
|
|
|
|
@_alwaysEmitIntoClient
|
|
public var byteOffsets: Range<Int> {
|
|
unsafe Range(_uncheckedBounds: (0, byteCount))
|
|
}
|
|
}
|
|
|
|
@available(SwiftCompatibilitySpan 5.0, *)
|
|
@_originallyDefinedIn(module: "Swift;CompatibilitySpan", SwiftCompatibilitySpan 6.2)
|
|
extension MutableRawSpan {
|
|
|
|
@_alwaysEmitIntoClient
|
|
public func withUnsafeBytes<E: Error, Result: ~Copyable>(
|
|
_ body: (_ buffer: UnsafeRawBufferPointer) throws(E) -> Result
|
|
) throws(E) -> Result {
|
|
guard let pointer = unsafe _pointer, _count > 0 else {
|
|
return try unsafe body(.init(start: nil, count: 0))
|
|
}
|
|
return try unsafe body(.init(start: pointer, count: _count))
|
|
}
|
|
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(self: copy self)
|
|
public mutating func withUnsafeMutableBytes<E: Error, Result: ~Copyable>(
|
|
_ body: (UnsafeMutableRawBufferPointer) throws(E) -> Result
|
|
) throws(E) -> Result {
|
|
guard let pointer = unsafe _pointer, _count > 0 else {
|
|
return try unsafe body(.init(start: nil, count: 0))
|
|
}
|
|
return try unsafe body(.init(start: pointer, count: _count))
|
|
}
|
|
}
|
|
|
|
@available(SwiftCompatibilitySpan 5.0, *)
|
|
@_originallyDefinedIn(module: "Swift;CompatibilitySpan", SwiftCompatibilitySpan 6.2)
|
|
extension RawSpan {
|
|
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(borrow mutableRawSpan)
|
|
public init(_mutableRawSpan mutableRawSpan: borrowing MutableRawSpan) {
|
|
let (start, count) = unsafe (mutableRawSpan._pointer, mutableRawSpan._count)
|
|
let span = unsafe RawSpan(_unchecked: start, byteCount: count)
|
|
self = unsafe _overrideLifetime(span, borrowing: mutableRawSpan)
|
|
}
|
|
}
|
|
|
|
@available(SwiftCompatibilitySpan 5.0, *)
|
|
@_originallyDefinedIn(module: "Swift;CompatibilitySpan", SwiftCompatibilitySpan 6.2)
|
|
extension MutableRawSpan {
|
|
|
|
public var bytes: RawSpan {
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(borrow self)
|
|
borrowing get {
|
|
return RawSpan(_mutableRawSpan: self)
|
|
}
|
|
}
|
|
|
|
@unsafe
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(borrow self)
|
|
public borrowing func _unsafeView<T: BitwiseCopyable>(
|
|
as type: T.Type
|
|
) -> Span<T> {
|
|
let bytes = unsafe UnsafeRawBufferPointer(start: _pointer, count: _count)
|
|
let span = unsafe Span<T>(_unsafeBytes: bytes)
|
|
return unsafe _overrideLifetime(span, borrowing: self)
|
|
}
|
|
|
|
@unsafe
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(&self)
|
|
public mutating func _unsafeMutableView<T: BitwiseCopyable>(
|
|
as type: T.Type
|
|
) -> MutableSpan<T> {
|
|
let bytes = unsafe UnsafeMutableRawBufferPointer(
|
|
start: _pointer, count: _count
|
|
)
|
|
let span = unsafe MutableSpan<T>(_unsafeBytes: bytes)
|
|
return unsafe _overrideLifetime(span, mutating: &self)
|
|
}
|
|
}
|
|
|
|
@available(SwiftCompatibilitySpan 5.0, *)
|
|
@_originallyDefinedIn(module: "Swift;CompatibilitySpan", SwiftCompatibilitySpan 6.2)
|
|
extension MutableRawSpan {
|
|
|
|
/// Returns a new instance of the given type, constructed from the raw memory
|
|
/// at the specified offset.
|
|
///
|
|
/// The memory at this pointer plus `offset` must be properly aligned for
|
|
/// accessing `T` and initialized to `T` or another type that is layout
|
|
/// compatible with `T`.
|
|
///
|
|
/// This is an unsafe operation. Failure to meet the preconditions
|
|
/// above may produce an invalid value of `T`.
|
|
///
|
|
/// - Parameters:
|
|
/// - offset: The offset from this pointer, in bytes. `offset` must be
|
|
/// nonnegative. The default is zero.
|
|
/// - type: The type of the instance to create.
|
|
/// - Returns: A new instance of type `T`, read from the raw bytes at
|
|
/// `offset`. The returned instance is memory-managed and unassociated
|
|
/// with the value in the memory referenced by this pointer.
|
|
@unsafe
|
|
@_alwaysEmitIntoClient
|
|
public func unsafeLoad<T>(
|
|
fromByteOffset offset: Int = 0, as type: T.Type
|
|
) -> T {
|
|
_precondition(
|
|
UInt(bitPattern: offset) <= UInt(bitPattern: _count) &&
|
|
MemoryLayout<T>.size <= (_count &- offset),
|
|
"Byte offset range out of bounds"
|
|
)
|
|
return unsafe unsafeLoad(fromUncheckedByteOffset: offset, as: T.self)
|
|
}
|
|
|
|
/// Returns a new instance of the given type, constructed from the raw memory
|
|
/// at the specified offset.
|
|
///
|
|
/// The memory at this pointer plus `offset` must be properly aligned for
|
|
/// accessing `T` and initialized to `T` or another type that is layout
|
|
/// compatible with `T`.
|
|
///
|
|
/// This is an unsafe operation. This function does not validate the bounds
|
|
/// of the memory access, and failure to meet the preconditions
|
|
/// above may produce an invalid value of `T`.
|
|
///
|
|
/// - Parameters:
|
|
/// - offset: The offset from this pointer, in bytes. `offset` must be
|
|
/// nonnegative. The default is zero.
|
|
/// - type: The type of the instance to create.
|
|
/// - Returns: A new instance of type `T`, read from the raw bytes at
|
|
/// `offset`. The returned instance is memory-managed and unassociated
|
|
/// with the value in the memory referenced by this pointer.
|
|
@unsafe
|
|
@_alwaysEmitIntoClient
|
|
public func unsafeLoad<T>(
|
|
fromUncheckedByteOffset offset: Int, as type: T.Type
|
|
) -> T {
|
|
unsafe _start().load(fromByteOffset: offset, as: T.self)
|
|
}
|
|
|
|
/// Returns a new instance of the given type, constructed from the raw memory
|
|
/// at the specified offset.
|
|
///
|
|
/// The memory at this pointer plus `offset` must be initialized to `T`
|
|
/// or another type that is layout compatible with `T`.
|
|
///
|
|
/// This is an unsafe operation. Failure to meet the preconditions
|
|
/// above may produce an invalid value of `T`.
|
|
///
|
|
/// - Parameters:
|
|
/// - offset: The offset from this pointer, in bytes. `offset` must be
|
|
/// nonnegative. The default is zero.
|
|
/// - type: The type of the instance to create.
|
|
/// - Returns: A new instance of type `T`, read from the raw bytes at
|
|
/// `offset`. The returned instance isn't associated
|
|
/// with the value in the range of memory referenced by this pointer.
|
|
@unsafe
|
|
@_alwaysEmitIntoClient
|
|
public func unsafeLoadUnaligned<T: BitwiseCopyable>(
|
|
fromByteOffset offset: Int = 0, as type: T.Type
|
|
) -> T {
|
|
_precondition(
|
|
UInt(bitPattern: offset) <= UInt(bitPattern: _count) &&
|
|
MemoryLayout<T>.size <= (_count &- offset),
|
|
"Byte offset range out of bounds"
|
|
)
|
|
return unsafe unsafeLoadUnaligned(fromUncheckedByteOffset: offset, as: T.self)
|
|
}
|
|
|
|
/// Returns a new instance of the given type, constructed from the raw memory
|
|
/// at the specified offset.
|
|
///
|
|
/// The memory at this pointer plus `offset` must be initialized to `T`
|
|
/// or another type that is layout compatible with `T`.
|
|
///
|
|
/// This is an unsafe operation. This function does not validate the bounds
|
|
/// of the memory access, and failure to meet the preconditions
|
|
/// above may produce an invalid value of `T`.
|
|
///
|
|
/// - Parameters:
|
|
/// - offset: The offset from this pointer, in bytes. `offset` must be
|
|
/// nonnegative. The default is zero.
|
|
/// - type: The type of the instance to create.
|
|
/// - Returns: A new instance of type `T`, read from the raw bytes at
|
|
/// `offset`. The returned instance isn't associated
|
|
/// with the value in the range of memory referenced by this pointer.
|
|
@unsafe
|
|
@_alwaysEmitIntoClient
|
|
public func unsafeLoadUnaligned<T: BitwiseCopyable>(
|
|
fromUncheckedByteOffset offset: Int, as type: T.Type
|
|
) -> T {
|
|
unsafe _start().loadUnaligned(fromByteOffset: offset, as: T.self)
|
|
}
|
|
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(self: copy self)
|
|
public mutating func storeBytes<T: BitwiseCopyable>(
|
|
of value: T, toByteOffset offset: Int = 0, as type: T.Type
|
|
) {
|
|
_precondition(
|
|
UInt(bitPattern: offset) <= UInt(bitPattern: _count) &&
|
|
MemoryLayout<T>.size <= (_count &- offset),
|
|
"Byte offset range out of bounds"
|
|
)
|
|
unsafe storeBytes(of: value, toUncheckedByteOffset: offset, as: type)
|
|
}
|
|
|
|
@unsafe
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(self: copy self)
|
|
public mutating func storeBytes<T: BitwiseCopyable>(
|
|
of value: T, toUncheckedByteOffset offset: Int, as type: T.Type
|
|
) {
|
|
unsafe _start().storeBytes(of: value, toByteOffset: offset, as: type)
|
|
}
|
|
}
|
|
|
|
// MARK: sub-spans
|
|
@available(SwiftCompatibilitySpan 5.0, *)
|
|
@_originallyDefinedIn(module: "Swift;CompatibilitySpan", SwiftCompatibilitySpan 6.2)
|
|
extension MutableRawSpan {
|
|
|
|
/// Constructs a new span over the items within the supplied range of
|
|
/// positions within this span.
|
|
///
|
|
/// The returned span's first item is always at offset 0; unlike buffer
|
|
/// slices, extracted spans do not share their indices with the
|
|
/// span from which they are extracted.
|
|
///
|
|
/// - Parameter bounds: A valid range of positions. Every position in
|
|
/// this range must be within the bounds of this `MutableSpan`.
|
|
///
|
|
/// - Returns: A `MutableSpan` over the items within `bounds`
|
|
///
|
|
/// - Complexity: O(1)
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(&self)
|
|
mutating public func _mutatingExtracting(_ bounds: Range<Int>) -> Self {
|
|
_precondition(
|
|
UInt(bitPattern: bounds.lowerBound) <= UInt(bitPattern: _count) &&
|
|
UInt(bitPattern: bounds.upperBound) <= UInt(bitPattern: _count),
|
|
"Index range out of bounds"
|
|
)
|
|
return unsafe _mutatingExtracting(unchecked: bounds)
|
|
}
|
|
|
|
@available(*, deprecated, renamed: "_mutatingExtracting(_:)")
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(&self)
|
|
mutating public func extracting(_ bounds: Range<Int>) -> Self {
|
|
_mutatingExtracting(bounds)
|
|
}
|
|
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(copy self)
|
|
consuming public func _consumingExtracting(_ bounds: Range<Int>) -> Self {
|
|
_precondition(
|
|
UInt(bitPattern: bounds.lowerBound) <= UInt(bitPattern: _count) &&
|
|
UInt(bitPattern: bounds.upperBound) <= UInt(bitPattern: _count),
|
|
"Index range out of bounds"
|
|
)
|
|
return unsafe _consumingExtracting(unchecked: bounds)
|
|
}
|
|
|
|
/// Constructs a new span over the items within the supplied range of
|
|
/// positions within this span.
|
|
///
|
|
/// The returned span's first item is always at offset 0; unlike buffer
|
|
/// slices, extracted spans do not share their indices with the
|
|
/// span from which they are extracted.
|
|
///
|
|
/// This function does not validate `bounds`; this is an unsafe operation.
|
|
///
|
|
/// - Parameter bounds: A valid range of positions. Every position in
|
|
/// this range must be within the bounds of this `MutableSpan`.
|
|
///
|
|
/// - Returns: A `MutableSpan` over the items within `bounds`
|
|
///
|
|
/// - Complexity: O(1)
|
|
@unsafe
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(&self)
|
|
mutating public func _mutatingExtracting(unchecked bounds: Range<Int>) -> Self {
|
|
let newStart = unsafe _pointer?.advanced(by: bounds.lowerBound)
|
|
let newSpan = unsafe Self(_unchecked: newStart, byteCount: bounds.count)
|
|
return unsafe _overrideLifetime(newSpan, mutating: &self)
|
|
}
|
|
|
|
@unsafe
|
|
@available(*, deprecated, renamed: "_mutatingExtracting(unchecked:)")
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(&self)
|
|
mutating public func extracting(unchecked bounds: Range<Int>) -> Self {
|
|
unsafe _mutatingExtracting(unchecked: bounds)
|
|
}
|
|
|
|
@unsafe
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(copy self)
|
|
consuming public func _consumingExtracting(unchecked bounds: Range<Int>) -> Self {
|
|
let newStart = unsafe _pointer?.advanced(by: bounds.lowerBound)
|
|
let newSpan = unsafe Self(_unchecked: newStart, byteCount: bounds.count)
|
|
return unsafe _overrideLifetime(newSpan, copying: self)
|
|
}
|
|
|
|
/// Constructs a new span over the items within the supplied range of
|
|
/// positions within this span.
|
|
///
|
|
/// The returned span's first item is always at offset 0; unlike buffer
|
|
/// slices, extracted spans do not share their indices with the
|
|
/// span from which they are extracted.
|
|
///
|
|
/// - Parameter bounds: A valid range of positions. Every position in
|
|
/// this range must be within the bounds of this `MutableSpan`.
|
|
///
|
|
/// - Returns: A `MutableSpan` over the items within `bounds`
|
|
///
|
|
/// - Complexity: O(1)
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(&self)
|
|
mutating public func _mutatingExtracting(
|
|
_ bounds: some RangeExpression<Int>
|
|
) -> Self {
|
|
_mutatingExtracting(bounds.relative(to: byteOffsets))
|
|
}
|
|
|
|
@available(*, deprecated, renamed: "_mutatingExtracting(_:)")
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(&self)
|
|
mutating public func extracting(
|
|
_ bounds: some RangeExpression<Int>
|
|
) -> Self {
|
|
_mutatingExtracting(bounds)
|
|
}
|
|
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(copy self)
|
|
consuming public func _consumingExtracting(
|
|
_ bounds: some RangeExpression<Int>
|
|
) -> Self {
|
|
_consumingExtracting(bounds.relative(to: byteOffsets))
|
|
}
|
|
|
|
/// Constructs a new span over the items within the supplied range of
|
|
/// positions within this span.
|
|
///
|
|
/// The returned span's first item is always at offset 0; unlike buffer
|
|
/// slices, extracted spans do not share their indices with the
|
|
/// span from which they are extracted.
|
|
///
|
|
/// This function does not validate `bounds`; this is an unsafe operation.
|
|
///
|
|
/// - Parameter bounds: A valid range of positions. Every position in
|
|
/// this range must be within the bounds of this `MutableSpan`.
|
|
///
|
|
/// - Returns: A `MutableSpan` over the items within `bounds`
|
|
///
|
|
/// - Complexity: O(1)
|
|
@unsafe
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(&self)
|
|
mutating public func _mutatingExtracting(
|
|
unchecked bounds: ClosedRange<Int>
|
|
) -> Self {
|
|
let range = unsafe Range(
|
|
_uncheckedBounds: (bounds.lowerBound, bounds.upperBound + 1)
|
|
)
|
|
return unsafe _mutatingExtracting(unchecked: range)
|
|
}
|
|
|
|
@unsafe
|
|
@available(*, deprecated, renamed: "_mutatingExtracting(unchecked:)")
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(&self)
|
|
mutating public func extracting(
|
|
unchecked bounds: ClosedRange<Int>
|
|
) -> Self {
|
|
unsafe _mutatingExtracting(unchecked: bounds)
|
|
}
|
|
|
|
@unsafe
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(copy self)
|
|
consuming public func _consumingExtracting(
|
|
unchecked bounds: ClosedRange<Int>
|
|
) -> Self {
|
|
let range = unsafe Range(
|
|
_uncheckedBounds: (bounds.lowerBound, bounds.upperBound + 1)
|
|
)
|
|
return unsafe _consumingExtracting(unchecked: range)
|
|
}
|
|
|
|
/// Constructs a new span over all the items of this span.
|
|
///
|
|
/// The returned span's first item is always at offset 0; unlike buffer
|
|
/// slices, extracted spans do not share their indices with the
|
|
/// span from which they are extracted.
|
|
///
|
|
/// - Returns: A `MutableSpan` over all the items of this span.
|
|
///
|
|
/// - Complexity: O(1)
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(&self)
|
|
mutating public func _mutatingExtracting(_: UnboundedRange) -> Self {
|
|
let newSpan = unsafe Self(_unchecked: _pointer, byteCount: _count)
|
|
return unsafe _overrideLifetime(newSpan, mutating: &self)
|
|
}
|
|
|
|
@available(*, deprecated, renamed: "_mutatingExtracting(_:)")
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(&self)
|
|
mutating public func extracting(_: UnboundedRange) -> Self {
|
|
_mutatingExtracting(...)
|
|
}
|
|
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(copy self)
|
|
consuming public func _consumingExtracting(_: UnboundedRange) -> Self {
|
|
self
|
|
}
|
|
}
|
|
|
|
// MARK: prefixes and suffixes
|
|
@available(SwiftCompatibilitySpan 5.0, *)
|
|
@_originallyDefinedIn(module: "Swift;CompatibilitySpan", SwiftCompatibilitySpan 6.2)
|
|
extension MutableRawSpan {
|
|
|
|
/// Returns a span containing the initial elements of this span,
|
|
/// up to the specified maximum length.
|
|
///
|
|
/// If the maximum length exceeds the length of this span,
|
|
/// the result contains all the elements.
|
|
///
|
|
/// The returned span's first item is always at offset 0; unlike buffer
|
|
/// slices, extracted spans do not share their indices with the
|
|
/// span from which they are extracted.
|
|
///
|
|
/// - Parameter maxLength: The maximum number of elements to return.
|
|
/// `maxLength` must be greater than or equal to zero.
|
|
/// - Returns: A span with at most `maxLength` elements.
|
|
///
|
|
/// - Complexity: O(1)
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(&self)
|
|
mutating public func _mutatingExtracting(first maxLength: Int) -> Self {
|
|
#if compiler(>=5.3) && hasFeature(SendableCompletionHandlers)
|
|
_precondition(maxLength >= 0, "Can't have a prefix of negative length")
|
|
let newCount = min(maxLength, byteCount)
|
|
let newSpan = unsafe Self(_unchecked: _pointer, byteCount: newCount)
|
|
return unsafe _overrideLifetime(newSpan, mutating: &self)
|
|
#else
|
|
fatalError("Unsupported compiler")
|
|
#endif
|
|
}
|
|
|
|
@available(*, deprecated, renamed: "_mutatingExtracting(first:)")
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(&self)
|
|
mutating public func extracting(first maxLength: Int) -> Self {
|
|
_mutatingExtracting(first: maxLength)
|
|
}
|
|
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(copy self)
|
|
consuming public func _consumingExtracting(first maxLength: Int) -> Self {
|
|
#if compiler(>=5.3) && hasFeature(SendableCompletionHandlers)
|
|
_precondition(maxLength >= 0, "Can't have a prefix of negative length")
|
|
let newCount = min(maxLength, byteCount)
|
|
let newSpan = unsafe Self(_unchecked: _pointer, byteCount: newCount)
|
|
return unsafe _overrideLifetime(newSpan, copying: self)
|
|
#else
|
|
fatalError("Unsupported compiler")
|
|
#endif
|
|
}
|
|
|
|
/// Returns a span over all but the given number of trailing elements.
|
|
///
|
|
/// If the number of elements to drop exceeds the number of elements in
|
|
/// the span, the result is an empty span.
|
|
///
|
|
/// The returned span's first item is always at offset 0; unlike buffer
|
|
/// slices, extracted spans do not share their indices with the
|
|
/// span from which they are extracted.
|
|
///
|
|
/// - Parameter k: The number of elements to drop off the end of
|
|
/// the span. `k` must be greater than or equal to zero.
|
|
/// - Returns: A span leaving off the specified number of elements at the end.
|
|
///
|
|
/// - Complexity: O(1)
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(&self)
|
|
mutating public func _mutatingExtracting(droppingLast k: Int) -> Self {
|
|
#if compiler(>=5.3) && hasFeature(SendableCompletionHandlers)
|
|
_precondition(k >= 0, "Can't drop a negative number of elements")
|
|
let droppedCount = min(k, byteCount)
|
|
let newCount = byteCount &- droppedCount
|
|
let newSpan = unsafe Self(_unchecked: _pointer, byteCount: newCount)
|
|
return unsafe _overrideLifetime(newSpan, mutating: &self)
|
|
#else
|
|
fatalError("Unsupported compiler")
|
|
#endif
|
|
}
|
|
|
|
@available(*, deprecated, renamed: "_mutatingExtracting(droppingLast:)")
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(&self)
|
|
mutating public func extracting(droppingLast k: Int) -> Self {
|
|
_mutatingExtracting(droppingLast: k)
|
|
}
|
|
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(copy self)
|
|
consuming public func _consumingExtracting(droppingLast k: Int) -> Self {
|
|
#if compiler(>=5.3) && hasFeature(SendableCompletionHandlers)
|
|
_precondition(k >= 0, "Can't drop a negative number of elements")
|
|
let droppedCount = min(k, byteCount)
|
|
let newCount = byteCount &- droppedCount
|
|
let newSpan = unsafe Self(_unchecked: _pointer, byteCount: newCount)
|
|
return unsafe _overrideLifetime(newSpan, copying: self)
|
|
#else
|
|
fatalError("Unsupported compiler")
|
|
#endif
|
|
}
|
|
|
|
/// Returns a span containing the final elements of the span,
|
|
/// up to the given maximum length.
|
|
///
|
|
/// If the maximum length exceeds the length of this span,
|
|
/// the result contains all the elements.
|
|
///
|
|
/// The returned span's first item is always at offset 0; unlike buffer
|
|
/// slices, extracted spans do not share their indices with the
|
|
/// span from which they are extracted.
|
|
///
|
|
/// - Parameter maxLength: The maximum number of elements to return.
|
|
/// `maxLength` must be greater than or equal to zero.
|
|
/// - Returns: A span with at most `maxLength` elements.
|
|
///
|
|
/// - Complexity: O(1)
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(&self)
|
|
mutating public func _mutatingExtracting(last maxLength: Int) -> Self {
|
|
#if compiler(>=5.3) && hasFeature(SendableCompletionHandlers)
|
|
_precondition(maxLength >= 0, "Can't have a suffix of negative length")
|
|
let newCount = min(maxLength, byteCount)
|
|
let newStart = unsafe _pointer?.advanced(by: byteCount &- newCount)
|
|
let newSpan = unsafe Self(_unchecked: newStart, byteCount: newCount)
|
|
return unsafe _overrideLifetime(newSpan, mutating: &self)
|
|
#else
|
|
fatalError("Unsupported compiler")
|
|
#endif
|
|
}
|
|
|
|
@available(*, deprecated, renamed: "_mutatingExtracting(last:)")
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(&self)
|
|
mutating public func extracting(last maxLength: Int) -> Self {
|
|
_mutatingExtracting(last: maxLength)
|
|
}
|
|
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(copy self)
|
|
consuming public func _consumingExtracting(last maxLength: Int) -> Self {
|
|
#if compiler(>=5.3) && hasFeature(SendableCompletionHandlers)
|
|
_precondition(maxLength >= 0, "Can't have a suffix of negative length")
|
|
let newCount = min(maxLength, byteCount)
|
|
let newStart = unsafe _pointer?.advanced(by: byteCount &- newCount)
|
|
let newSpan = unsafe Self(_unchecked: newStart, byteCount: newCount)
|
|
return unsafe _overrideLifetime(newSpan, copying: self)
|
|
#else
|
|
fatalError("Unsupported compiler")
|
|
#endif
|
|
}
|
|
|
|
/// Returns a span over all but the given number of initial elements.
|
|
///
|
|
/// If the number of elements to drop exceeds the number of elements in
|
|
/// the span, the result is an empty span.
|
|
///
|
|
/// The returned span's first item is always at offset 0; unlike buffer
|
|
/// slices, extracted spans do not share their indices with the
|
|
/// span from which they are extracted.
|
|
///
|
|
/// - Parameter k: The number of elements to drop from the beginning of
|
|
/// the span. `k` must be greater than or equal to zero.
|
|
/// - Returns: A span starting after the specified number of elements.
|
|
///
|
|
/// - Complexity: O(1)
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(&self)
|
|
mutating public func _mutatingExtracting(droppingFirst k: Int) -> Self {
|
|
#if compiler(>=5.3) && hasFeature(SendableCompletionHandlers)
|
|
_precondition(k >= 0, "Can't drop a negative number of bytes")
|
|
let droppedCount = min(k, byteCount)
|
|
let newStart = unsafe _pointer?.advanced(by: droppedCount)
|
|
let newCount = byteCount &- droppedCount
|
|
let newSpan = unsafe Self(_unchecked: newStart, byteCount: newCount)
|
|
return unsafe _overrideLifetime(newSpan, mutating: &self)
|
|
#else
|
|
fatalError("Unsupported compiler")
|
|
#endif
|
|
}
|
|
|
|
@available(*, deprecated, renamed: "_mutatingExtracting(droppingFirst:)")
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(&self)
|
|
mutating public func extracting(droppingFirst k: Int) -> Self {
|
|
_mutatingExtracting(droppingFirst: k)
|
|
}
|
|
|
|
@_alwaysEmitIntoClient
|
|
@lifetime(copy self)
|
|
consuming public func _consumingExtracting(droppingFirst k: Int) -> Self {
|
|
#if compiler(>=5.3) && hasFeature(SendableCompletionHandlers)
|
|
_precondition(k >= 0, "Can't drop a negative number of bytes")
|
|
let droppedCount = min(k, byteCount)
|
|
let newStart = unsafe _pointer?.advanced(by: droppedCount)
|
|
let newCount = byteCount &- droppedCount
|
|
let newSpan = unsafe Self(_unchecked: newStart, byteCount: newCount)
|
|
return unsafe _overrideLifetime(newSpan, copying: self)
|
|
#else
|
|
fatalError("Unsupported compiler")
|
|
#endif
|
|
}
|
|
}
|