I have a Metal texture, I want to access its data from Swift by making it a float4 array (so that I can access each pixel 4 color components).
I discovered this method of MTLTexture
:
getBytes(UnsafeMutableRawPointer, bytesPerRow: Int, bytesPerImage: Int, from: MTLRegion, mipmapLevel: Int, slice: Int)
I don't know at all how to use UnsafeMutableRawPointer, how it works, and how to get the data back into a simple Swift array.
My first attempt was to create a pointer and allocate enough space like that, but I don't even know if that's what I should do:
var pointer = UnsafeMutableRawPointer.allocate(bytes: myTextureSizeInBytes, alignedTo: 0)
Then I have no idea at all of how to get this data back into a standard Swift array...
Thank you.
First, let's assume you have a UnsafeRawPointer
and a length:
let ptr: UnsafeRawPointer = ...
let length: Int = ...
Now you want to convert that to an [float4]
. First, you can convert your UnsafeRawPointer
to a typed pointer by binding it to a type:
let float4Ptr = ptr.bindMemory(to: float4.self, capacity: length)
Now you can convert that to a typed buffer pointer:
let float4Buffer = UnsafeBufferPointer(start: float4Ptr, count: length)
And since a buffer is a collection, you can initialize an array with it:
let output = Array(float4Buffer)
For much more on working with UnsafeRawPointer
, see SE-0138, SE-0107, and the UnsafeRawPointer Migration Guide.
extension UnsafeMutableRawPointer {
func toArray<T>(to type: T.Type, capacity count: Int) -> [T]{
let pointer = bindMemory(to: type, capacity: count)
return Array(UnsafeBufferPointer(start: pointer, count: count))
}
}
var array = [1,2,3,4,5]
let ponter = UnsafeMutableRawPointer(mutating: array)
print(ponter.toArray(to: Int.self, capacity: array.count))
If you love us? You can donate to us via Paypal or buy me a coffee so we can maintain and grow! Thank you!
Donate Us With