I’m using buffer
in many other vDSP operations before the code below. To those operations it is naturally a contiguous piece of memory, and that’s handy as there are vastly reduced number of calls to vDSP.
// […] many operations with buffer here
var buffer = UnsafeMutableBufferPointer<Double>.allocate(capacity: 10000000)
var sortedIndices: UnsafeMutablePointer<UInt> = UnsafeMutablePointer<UInt>.allocate(capacity: 10000000)
for (indexIndices, indexC) in buffer.indices.enumerated() {
sortedIndices[indexIndices] = UInt(indexC)
}
vDSP_vsortiD(
UnsafePointer<Double>(buffer.baseAddress)!,
sortedIndices,
nil,
UInt(buffer.count),
vDSP.SortOrder.descending.rawValue
)
I need to sort buffer
sliced up. Also, I’d like not to copy its contents before every sort into smaller buffers as this is performance critical (as many times as possible per second).
vDSP_vsortiD()
and vDSP_vsorti()
do not support Slice
.
For example this code is not possible since .baseAddress
doesn’t exist on a Slice
and I guess UnsafePointer
doesn’t understand Slice
either.
var sliceOfBuffer: Slice<UnsafeMutableBufferPointer<Double>> = buffer[0...30000]
vDSP_vsortiD(
UnsafePointer<Double>(sliceOfBuffer.baseAddress)!, // This fails
sortedIndices[0...30000], // This also fails
nil,
UInt(30000),
vDSP.SortOrder.descending.rawValue
)
// Is there another sort indices function which accepts those types as params?
Many operations under vDSP
accept Slice
, however the above sort functions don’t.
Swift v5.6
You can use withUnsafeBufferPointer()
to pass the address of the slice's element storage to the vDSP function:
sliceOfBuffer.withUnsafeBufferPointer { bp in
vDSP_vsortiD(
bp.baseAddress!,
sortedIndices,
nil,
UInt(sliceOfBuffer.count),
vDSP.SortOrder.descending.rawValue
)
}
With respect to slicing the indices: It seems that one can pass sortedIndices + sliceOfBuffer.startIndex
to the vDSP function. However, the documentation states that the indices must be initialized starting at zero, so I would not rely on that.