9

我在一个看似简单的问题上遇到了困难。我需要从 Python 代码中调用各种 C 函数。目前,我正在尝试通过 ctypes 来做到这一点。我在使用一个简单的示例实现时遇到问题,我使用它来确保在更改大量现有代码之前一切都按预期工作。C 函数接受一些参数,进行各种数学计算,然后返回一个双精度值。这是我用作测试的 C 文件。

#include "Python.h"

double eval(double* args, int length, double* rands, int randLength, int debug)
{
    double val1 = args[0];
    double val2 = rands[0];
    double ret = val1 + val2;
    return ret;
}

在我的实际测试功能中,我正在打印val1val2以确保它们接收到值。在那方面一切都很好。调用此函数的 Python 如下。

test = ctypes.CDLL("/home/mjjoyce/Dev/C2M2L/AquaticWavesModel.so")
rand = Random();

args = [2]
argsArr = (ctypes.c_double * len(args))()
argsArr[:] = args

argsRand = [rand.random()]
argsRandArr = (ctypes.c_double * len(argsRand))()
rgsRandArr[:] = argsRand

result = test.eval(argsArr, len(argsArr), argsRandArr, len(argsRandArr), 0)
print "result", result

当这打印返回的结果时,它通常会转储一个很大的负数。我的假设是我需要在 Python 处理该值之前进行一些转换,但我一生都找不到答案。我一直在搜索,但我不确定我是否遗漏了明显的信息,或者搜索了错误的信息。

此外,如果我将函数的返回类型更改为 int,则计算结果正确,并且返回并按预期打印一个 int。如果我将其设置为双倍,则 C 代码中的计算是正确的(我打印它们以检查),但 Python 不喜欢返回的结果。

有什么我想念的想法吗?

4

1 回答 1

16

argtypes您应该使用andrestype成员在 Python 中设置函数的参数类型和返回类型:

test.eval.argtypes = [ctypes.POINTER(ctypes.c_double),
                      ctypes.c_int,
                      ctypes.POINTER(ctypes.c_double),
                      ctypes.c_int,
                      ctypes.c_int]
test.eval.restype = ctypes.c_double

然后你可以像你正在做的那样调用它:构造(ctypes.c_double * length)用你的值初始化的类型数组,并且result将是一个double.

另请参阅此ctypes 教程,它有很多有用的信息。

于 2012-08-03T21:04:44.573 回答