swift - 如何在 Swift 3 中将 UnsafePointer<[Float]> 或 Float Array 数据转换为 Unsafe[Mutable]Pointer？

Question

我有一个来自 HDF5 文件的浮点值数组 (floatArray)，我想将此数据用作接受 Unsafe[Mutable]Pointer (inputForFunction).我是 UnsafePointer 的新手，所以我需要别人的帮助。

我想到的一种方法是，将数组存储到文件中，然后打开它并将文件描述符映射到函数输入的变量内存:

  // read the data from HDF5 file
  var floatArray: [Float] = try originalHDF5DataSet.read()

  // store the data
  let dataToStore = NSData(bytes: &floatArray, length: sizeOfArray * MemoryLayout<Float>.size)
  let tmpPath = URL(fileURLWithPath: NSTemporaryDirectory())
  let filePath = tmpPath.appendingPathComponent("floatArrayData").appendingPathExtension("dat")
  do{
    try data.write(to: filePath, options: .atomic)
    let fileManager : FileManager   = FileManager.default
    if fileManager.fileExists(atPath:filePath.path){
      print("Success")
    }else{
      print("Fail")
    }

    // open the data 
    let dataPath = filePath.path
    let fd = open(dataPath, O_RDONLY, S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH | S_IWOTH)
    assert(fd != -1, "Error: failed to open output file at \""+dataPath+"\"  errno = \(errno)\n")
    // memory map the parameters
    let hdr = mmap(nil, Int(sizeOfArray), PROT_READ, MAP_FILE | MAP_SHARED, fd, 0)
    // cast Void pointers to Float
    let inputForFunction = UnsafePointer<Float>(hdr!.assumingMemoryBound(to: Float.self))
  } catch{
    print("failed")
  }

但是，它可能不是最好的实现，因为存储数据的过程似乎很耗时。

这就是为什么我正在考虑使用 UnsafePointer 将指针从 floatArray 传递到 inputForFunction，但我不确定如何实现它。仅供引用，我想到的一种方法是直接使用 withUnsafePointer 方法:

var floatArray: [Float] = try originalHDF5DataSet.read()
var inputForFunction = UnsafeMutablePointer<Float>.allocate(capacity: Int(sizeOfArray))
withUnsafePointer(to: &floatArray[0]) { (ptr: UnsafePointer<Float>) in
  inputForFunction = UnsafeMutablePointer(mutating: ptr)
}

或者另一种方法是使用 withMemoryRebound(但以下方法无效):

var floatArray: [Float] = try originalHDF5DataSet.read()
var inputForFunction = UnsafeMutablePointer<Float>.allocate(capacity: Int(sizeOfArray))
withUnsafePointer(to: &floatArray) { (ptrDataArray: UnsafePointer<[Float]>) in
  inputForFunction = ptrDataArray.withMemoryRebound(to: inputForFunction.self, capacity: Int(sizeOfArray), { (ptr: UnsafePointer<[Float]>) -> UnsafeMutablePointer<Float> in
    return ptr[0]
  })
}

我的问题是

• 如何将floatArray的数据地址传递给inputForFunction？
• 如果我需要将 UnsafePointer<[Float]> 转换为 Unsafe[Mutable]Pointer，该怎么做？
• 如果需要，如何为此目的使用withMemoryRebound？我可以分配吗闭包内 inputForFunction 的地址/值？
• 当我检查floatArray 和floatArray[0] 的地址时，它们是不同的(这似乎与C/C++ 不同)。应将哪个地址传递给 inputForFunction？

Answer 1

您可以找到许多解释如何获得 Unsafe(Mutable)Pointer<Float> 的文章来自 [Float] .其中一些回答了您的一些问题。稍后尝试找到它们。

• 如何将floatArray的数据地址传给inputForFunction？

  您通常使用 Array 的方法 withUnsafeBufferPointer或 withUnsafeMutableBufferPointer .你可以获得 Unsafe(Mutable)BufferPointer<Element>在闭包参数中，它包含一个 Unsafe(Mutable)Pointer<Element>作为其 baseAddress属性(property)。

  var floatArray: [Float] = try originalHDF5DataSet.read()
  floatArray.withUnsafeBufferPointer {unsafeBufferPointer in
      let inputForFunction = unsafeBufferPointer.baseAddress
      //`unsafeBufferPointer` (including its `baseAddress`) is valid only in this closure.
      //So, do not pass `inputForFunction` outside of the closure, use it inside.
      //...
  }
  

• 如果我需要将 UnsafePointer<[Float]> 转换为 Unsafe[Mutable]Pointer，该怎么做？

  当您找到类似 UnsafePointer<[Float]> 的内容时，那你走错路了。在 Swift 中，Array是一个隐藏结构，可能包含对其元素的(多级)间接引用。您通常不需要使用 UnsafePointer<Array<Float>> .

• 如果需要，如何为此目的使用 withMemoryRebound？我可以将地址/值分配给闭包内的 inputForFunction 吗？

  当你有一个 AType 的数组时并希望将其作为 Unsafe(Mutable)Pointer<AnotherType> 传递, 你可以利用 withMemoryRebound .但就您而言，您只需要使用 Float , 所以 withMemoryRebound您的目的可能不需要。

• 当我检查 floatArray 和 floatArray[0] 的地址时，它们是不同的(这似乎与 C/C++ 不同)。应该将哪个地址传递给 inputForFunction？

  没有。正如我在上面所写的，floatArray 的地址 只是指向某个包含隐藏结构的地方，因此它对您的目的完全没有用。而address of floatArray[0] 也不能保证是整个内容的第一个元素的地址。 Swift 可能会创建一个只包含一个元素的时间区域，并传递时间区域的地址。

---

顺便说一句，你真的需要转换你的[Float]吗？至 Unsafe(Mutable)Pointer<Float> ？

如果你的函数签名是这样的:

func takingUnsafePointer(_ pointer: UnsafePointer<Float>)

你可以这样调用它:

takingUnsafePointer(floatArray)
//In this case `floatArray` can be a `let` constant.

否则，如果函数签名是:

func takingUnsafeMutablePointer(_ pointer: UnsafeMutablePointer<Float>)

称它为:

takingUnsafeMutablePointer(&floatArray)
//In this case `floatArray` needs to be a `var`, not `let`.

请记住，在这两种情况下，传递的指针仅在函数调用时有效。您不应该将指针导出到其他地方。