如何使用UnsafeMutableRawPointer来填充数组？

我有一个Metal纹理,我想通过使它成为一个float4数组从Swift访问它的数据(这样我就可以访问每个像素的4个颜色组件).

我发现了这种方法MTLTexture:

getBytes(UnsafeMutableRawPointer, bytesPerRow: Int, bytesPerImage: Int, from: MTLRegion, mipmapLevel: Int, slice: Int)

我根本不知道如何使用UnsafeMutableRawPointer,它是如何工作的,以及如何将数据恢复到一个简单的Swift数组中.

我的第一次尝试是创建一个指针,并分配足够的空间,但我甚至不知道这是不是我应该做的:

var pointer = UnsafeMutableRawPointer.allocate(bytes: myTextureSizeInBytes, alignedTo: 0)  

然后我根本不知道如何将这些数据恢复到标准的Swift数组......

谢谢.

首先,让我们假设你有一个UnsafeRawPointer和一个长度:

let ptr: UnsafeRawPointer = ...
let length: Int = ...

现在你要将其转换为[float4].首先,您可以UnsafeRawPointer通过将其绑定到类型来将您转换为类型指针:

let float4Ptr = ptr.bindMemory(to: float4.self, capacity: length)

现在您可以将其转换为类型化的缓冲区指针:

let float4Buffer = UnsafeBufferPointer(start: float4Ptr, count: length)

由于缓冲区是一个集合,您可以使用它初始化一个数组:

let output = Array(float4Buffer)

有关使用的更多信息UnsafeRawPointer,请参阅SE-0138,SE-0107和UnsafeRawPointer迁移指南.