Swift+Metal：如何从数组非起始索引更新MTLBuffer？

要从UInt16数组的非0索引（比如索引4）复制内容到MTLBuffer，核心是正确获取数组对应位置的指针，并计算准确的复制字节数。以下是两种可靠的实现方式：

方法一：复制从指定索引到数组末尾的所有元素 #

uint16Array.withUnsafeBufferPointer { bufferPtr in
    guard let baseAddr = bufferPtr.baseAddress else { return }
    // 偏移4个UInt16元素，得到起始指针
    let startPtr = baseAddr + 4
    // 计算总字节数：剩余元素个数 × UInt16的字节长度（2字节）
    let byteCount = (bufferPtr.count - 4) * MemoryLayout<UInt16>.stride
    // 复制到MTLBuffer
    mtlBuffer.contents().copyMemory(from: startPtr, byteCount: byteCount)
}

方法二：复制指定索引开始的固定个数元素 #

如果只需要复制从索引4开始的N个元素（比如4个），可以这样写：

let startIndex = 4
let copyElementCount = 4

uint16Array.withUnsafeBufferPointer { bufferPtr in
    // 先检查索引是否越界
    guard bufferPtr.count >= startIndex + copyElementCount else {
        print("索引越界")
        return
    }
    let startPtr = bufferPtr.baseAddress! + startIndex
    let byteCount = copyElementCount * MemoryLayout<UInt16>.stride
    mtlBuffer.contents().copyMemory(from: startPtr, byteCount: byteCount)
}

关键注意点 #

• 不能直接对Swift数组做指针加法（比如uint16Array + 4），必须通过withUnsafeBufferPointer获取数组的底层指针baseAddress后，再按元素个数偏移（因为baseAddress是UnsafePointer<UInt16>类型，+4表示跳过4个UInt16元素，对应8字节偏移）。
• 复制时的byteCount必须是元素个数乘以MemoryLayout<UInt16>.stride（固定为2），避免字节数计算错误导致内存越界。

内容的提问来源于stack exchange，提问作者SundanceK