将Python Python返回值提升为C ++程序

时间:2017-05-04 04:59:56

标签: python c++ boost boost-python

我在C ++中有一个python嵌入,其中C ++在python中调用一个函数,应该返回一个值。这是我的python代码:

def predict(window, g_slope, g_compliance):
    ps = model.predict_on_batch(
        x=np.asarray([window])
    )

    p_slopes = ps[0]
    p_compliances = ps[1]
    p_slopes = np.reshape(p_slopes, (np.shape(p_slopes)[1],))
    p_compliances = np.reshape(p_compliances, (np.shape(p_compliances)[1],))
    p_slope = p_slopes[-1]
    p_compliance = p_compliances[-1]

    n_slope = standardize_data(g_slope, means['GROUND_SLOPE'], variances['GROUND_SLOPE'])
    n_compliance = standardize_data(g_compliance, means['GROUND_STIFFNESS'], variances['GROUND_STIFFNESS'])

    #print('p_compliance: {0}, n_compliance: {1}, p_slope: {2}, n_slope: {3} '.format(str(p_compliance), str(n_compliance), str(n_slope), str(p_slope)))

    return(p_slope, n_slope, p_compliance, n_compliance)

它返回我希望在C ++程序中收到的元组。 我的C ++程序调用这样的函数:

void ContactLearningApp::PythonWorkerThread() {

    bp::object f = m_interface.attr("predict_on_data");

    printf("Start Python thread. \n");

    while (true) {
        //printf("Inside while loop and waiting. \n");
        std::unique_lock<std::mutex> ul(m_mutex);
        while (m_data.size() <= 0) {
            m_NotEmptyCV.wait(ul);
        }
        Data dat = m_data.back();
        m_data.pop_back();

        ul.unlock();

        m_pydata_ptr py_ptr(new PyData);

        py_ptr->InitWithData(dat);
        try {
            bp::tuple results = bp::extract<bp::tuple>(f(bp::ptr(py_ptr.get())));
            printf("p_slope: %f, n_slope: %f, p_compliance: %f, n_compliance %f \n", results[0], results[1], results[2], results[3]);
        }
        catch (const bp::error_already_set &import) {
            // Parse and output the exception
            std::string perror_str = parse_python_exception();
            std::cout << "Error in Python: " << perror_str << std::endl;
        }
    }

}

我试图提取从Python返回的元组但是我收到以下错误消息:

Expecting an object of type tuple; got an object of type NoneType instead. 

我知道它不是没有,因为我能够正确访问元组中的每个值。提取元组的正确方法是什么?

1 个答案:

答案 0 :(得分:0)

boost::python::object支持operator[],它允许按索引访问序列(列表,元组等)元素。这意味着您无需将函数返回的object转换为tuple,您可以直接访问其元素:

bp::object func = bp::import("my_module").attr("my_func");
bp::object foo = func();
bp::object bar = foo[0];

如果返回的python对象不支持索引访问,则会抛出异常。请注意,您可能仍需要将索引提取的序列元素转换为相应的C ++类型。

通常你可以“按原样”从C ++向函数对象传递参数,而不会弄乱原始的PyObject*指针。在大多数情况下,Boost.Python会进行必要的转换。