正确使用 ctypes 传递数组

各位小伙伴们，大家好呀！看看今天我又给各位带来了什么文章？本文标题是《正确使用 ctypes 传递数组》，很明显是关于Golang的文章哈哈哈，其中内容主要会涉及到等等，如果能帮到你，觉得很不错的话，欢迎各位多多点评和分享！

我有一个 go 函数，它获取一个数组作为输入参数，然后将它们简单地打印到控制台：

package main
import "c"
func testarray(xs []float64) {
for arg := range xs {
            fmt.println(arg)
        }
}

我想从我的 python 代码中调用该函数，我正在尝试使用 ctypes，但我没有找到用于传递数组的正确 ctypes。这是我的python代码：

from ctypes import *
lib = cdll.LoadLibrary("./main.so")
print ("Loaded go generated SO library")
lib.testArray.argtypes = [?]
arr =[1, 3, .5, 2, 1, 1, 2, 3, .5, 3, 1, 1, 3, 1, 1, 3, 2, 1]
lib.testArray(arr)

我将找到合适的 ctypes 来代替“问号”，以使我的代码正常工作。

解决方案

您可以将 go 代码挂钩到 python 代码，但要使其可从 python 访问，您必须 //export go 函数。

您不能将切片直接传递给 go 代码。您必须使用 go 数组来伪造此内容，并使用从 c 或 python 代码调用的导出函数中的转换向该数组添加切片标头。 peterSO's answer 到相关问题 Accessing C array in golang 显示了如何使用 unsafe.pointer 来执行此操作（尽管对于 int 而不是 float64）。我有点怀疑这里的硬编码数组大小会破坏未来的某些 go 实现，但它现在确实有效。 （这里的 c 数组也可能有 8 gb 大小限制。）

因此（现已测试）：

package main

import "c"

import (
    "fmt"
    "unsafe"
)

//export ctestarray
func ctestarray(xsbase *c.double, n c.int) {
    xs := (*[1 << 30]float64)(unsafe.pointer(xsbase))[:n:n]
    testarray(xs)
}

func testarray(xs []float64) {
    for _, arg := range xs {
        fmt.println(arg)
    }
}

func main() {}

（从技术上讲，您不需要中间函数：testarray 代码可以在创建切片变量后出现。但我建议使用包装器构造，以便您的 go-only 代码不使用 unsafe：所有潜在的可怕错误仅限于包装器。）

剩下的棘手部分是在 python 中构建数组并传递它。您必须对实际大小进行编码。还好，继续重置argtypes似乎就可以了：

from ctypes import *
lib = cdll.loadlibrary("./main.so")
print ("loaded go generated so library")
lib.ctestarray.restype = none
l = [1, 3, .5, 2, 1, 1, 2, 3, .5, 3, 1, 1, 3, 1, 1, 3, 2, 1]
arr = (c_double * len(l))(*l)
lib.ctestarray.argtypes = [type(arr), c_int]
lib.ctestarray(arr, len(arr))
print("done 1")

l = [42, 3.1415]
arr = (c_double * len(l))(*l)
lib.ctestarray.argtypes = [type(arr), c_int]
lib.ctestarray(arr, len(arr))
print("done 2")

l = [i/2 for i in range(40)]
arr = (c_double * len(l))(*l)
lib.ctestarray.argtypes = [type(arr), c_int]
lib.ctestarray(arr, len(arr))

定义一个小函数来完成这项工作可能是最明智的，例如：

import ctypes

lib = ctypes.cdll.LoadLibrary("./main.so")
lib.CTestArray.restype = None

def CTestArray(l):
    arr = (ctypes.c_double * len(l))(*l)
    lib.CTestArray.argtypes = [type(arr), ctypes.c_int]
    lib.CTestArray(arr, len(arr))

CTestArray([1, 3, .5, 2, 1, 1, 2, 3, .5, 3, 1, 1, 3, 1, 1, 3, 2, 1])
CTestArray([42, 3.1415])
CTestArray([i/2 for i in range(40)])

当然，此时设置参数类型有点愚蠢。

以上就是《正确使用 ctypes 传递数组》的详细内容，更多关于的资料请关注golang学习网公众号！