繁体   English   中英

使用python驱动程序在C ++代码之间来回传递数据

[英]Passing data to and from C++ code using a python driver

我正在尝试将数组从python驱动程序传递给c ++函数。 我传递给c ++函数的数组之一是结果数组。 我知道数据正确传递给了代码,但似乎在返回的途中被截断了。 小数点后的精度下降了。 有什么想法我做错了吗? 下面是该代码的简化版本以及结果输出。

python代码:

import ctypes
from ctypes import *

def get_calculate_windows():
    dll = ctypes.CDLL('./cppFunctions.so', mode=ctypes.RTLD_GLOBAL)
    func = dll.calculate_windows
    func.argtypes = [POINTER(c_float), POINTER(c_float), c_size_t]
    return func

def calculate_windows(data, answer, size):
    data_p = data.ctypes.data_as(POINTER(c_float))
    answer_p = answer.ctypes.data_as(POINTER(c_float))

    __calculate_windows(data_p, answer_p, size)


###### MAIN FUNCTION #####

data = np.array(myData, dtype=np.float32)
ans = np.array(myAnswer, dtype=np.float32)

print ans[:10]
calculate_windows(data, ans, myLength)
print ans[:10]

C ++代码:

extern "C" {

    void calculate_windows(float *h_data, float *result, size_t size ) {

        for (int i=0; i<size; i++ ) {
            result[i]=i/10;
        }

    }
}

输出:

[ 0.  0.  0.  0.  0.  0.  0.  0.  0.  0.]
[ 0.  0.  0.  0.  0.  0.  0.  0.  0.  0.]

输出应为:

[ 0.  0.  0.  0.  0.  0.  0.  0.  0.  0.]
[ 0.0  0.1  0.2  0.3  0.4  0.5  0.6  0.7  0.8  0.9]

我知道数据正确传递给了代码,但似乎在返回的途中被截断了。 小数点后我失去了精度。 有什么想法我做错了吗?

您正在使用整数除法而不是浮点除法。

您可以使用以下方法解决此问题:

result[i] = i / 10.;

代替:

result[i] = i / 10;

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM