Swift数组类型之间的转换,无需复制或分配

时间:2019-04-11 20:24:14

标签: arrays swift swift4.2 unsafemutablepointer

我想访问UInt64的现有数组,就好像它是Int8的数组一样。关键要求是效率-我不想复制或重新分配数据,只能直接访问。我不需要副作用(例如,我希望能够在执行此代码块后继续使用uint64Array,因为它正在阅读有关具有未定义副作用的重新绑定的信息。)

我尝试使用Swift 4.2做到这一点:

var uint64Array = [UInt64](repeating: 0, count: 100)

uint64Array.withUnsafeMutableBufferPointer() {
    uint64Pointer in
    uint64Pointer.withMemoryRebound(to: Int8.self) {   // <- Error occurs here.
        int8Pointer in
        int8Pointer[0] = 1
        int8Pointer[1] = 2
        int8Pointer[2] = 3
        int8Pointer[3] = 4
    }
}

但是我在运行时在以下行中遇到致命错误:

    uint64Pointer.withMemoryRebound(to: Int8.self) {

这是正确的方法吗?如果是这样,为什么我会收到致命错误?

2 个答案:

答案 0 :(得分:1)

我认为问题在于您无法按照文档中的此注释直接绑定到其他类型:

  

仅使用此方法将缓冲区的内存重新绑定为与当前绑定的Element类型具有相同大小和跨度的类型。要将内存区域绑定到其他大小的类型,请将缓冲区转换为原始缓冲区,然后使用bindMemory(to :)方法。

如果要使用字节,那么最快的路线是:

var uint64Array = [UInt64](repeating: 0, count: 100)
uint64Array.withUnsafeMutableBytes { x in

    x[0] = 1
    x[1] = 2
    x[3] = 3
    x[4] = 4

}

如果您想使用其他类型,可以这样操作:

var uint64Array = [UInt64](repeating: 0, count: 100)

uint64Array.withUnsafeMutableBufferPointer() {
    uint64Pointer in

    let x = UnsafeMutableRawBufferPointer(uint64Pointer).bindMemory(to: Int32.self)
    x[0] = 1
    x[1] = 2
    x[3] = 3
    x[4] = 4

}

答案 1 :(得分:1)

感谢@brindy解决了这一问题。这是一个扩展实现,我可以尽其所能。

扩展名:

extension Array {
    mutating func bindMutableMemoryTo<T,R>(_ type: T.Type, _ closure: (UnsafeMutableBufferPointer<T>) throws -> R) rethrows -> R {
        return try self.withUnsafeMutableBytes() {
            return try closure($0.bindMemory(to: type))
        }
    }
}

用法:

   var uint64Array = [UInt64](repeating: 0, count: 100)
   uint64Array.bindMutableMemoryTo(Int8.self) {
        int8Pointer in
        int8Pointer[0] = 1 // LSB of uint64Array[0]
        int8Pointer[1] = 2
        int8Pointer[2] = 3
        int8Pointer[3] = 4 // MSB of uint64Array[0]
    }