pybind11:Python至C ++数据类型转换不起作用

时间:2019-03-17 18:18:27

标签: python c++ python-c-api pybind11

问题

我正在尝试转换列表列表,该列表由C ++代码内部调用的python函数返回。尽管 pybind11 库允许将类型从python数据类型转换为C ++数据类型,但是我尝试将python返回的列表列表转换为std::list的字符串std::list C ++,每次都会失败。

代码

这是python函数(该函数返回包含字符串值的列表的列表):

def return_sheet(self):

     """Returns the sheet in a list of lists

     """

     dataTable = []

     for r in range(self._isheet.nrows):

         datalist = []

         for c in range(self._isheet.ncols):

             datalist.append(self._isheet.cell_value(r,c))

         dataTable.append(datalist)

 return dataTable

在这里,我在C ++中使用 pybind11 来调用它:

py::list obj = _tool.attr("return_sheet")();

data = py::cast<SheetData>(obj); // This is where the problem lies, This cast crashes the program

其中SheetDatatypedef的地方:

typedef std::list<std::list<std::string> > SheetData;

在调试时,我发现程序实际上在此行崩溃:

py::object dataTable = _tool.attr("return_sheet")(); // Where _tool.attr("return_sheet")() gives an py::object which is a list of list of str

有人知道,如何将python列表的列表成功转换为C ++的std::list的{​​{1}}?

编辑

这是我正在c ++中嵌入的python程序文件[xlanalyser.py]:https://pastebin.com/gARnkMTv

这是c ++代码[main.cpp]:https://pastebin.com/wDDUB1s4

注意:xlanalyser.py中的所有其他函数在嵌入c ++时不会导致崩溃[只有return_sheet()函数会导致崩溃]

1 个答案:

答案 0 :(得分:0)

您可以将Python / C API用作解决方法(检查功能CastToSheetData)。我在下面提供了完整的示例:

program.py

def return_matrix():

    dataTable = []
    for r in range(0,2):
        datalist = []

        for c in range(0,2):
            datalist.append(str(r+c))

        dataTable.append(datalist)
    return dataTable

main.cpp

#include <pybind11/embed.h>
#include <iostream>
#include <list>
#include <string>

typedef std::list<std::list<std::string> > SheetData;

namespace py = pybind11;

SheetData CastToSheetData(PyObject *obj)
{
    SheetData data;
    PyObject *iter = PyObject_GetIter(obj);

    if (!iter)
        return data;
    while (true) {
        std::list<std::string> aux_list;
        PyObject *next = PyIter_Next(iter);
        if (!next) {
            // nothing left in the iterator
            break;
        }
        PyObject *iter2 = PyObject_GetIter(next);
        if (!iter2)
            continue;
        while(true) {
            PyObject *next2 = PyIter_Next(iter2);
            if (!next2) {
                // nothing left in the iterator
                break;
            }
            PyObject* pyStrObj = PyUnicode_AsUTF8String(next2); 
            char* zStr = PyBytes_AsString(pyStrObj); 
            std::string foo(strdup(zStr));
            aux_list.push_back(foo);
            Py_DECREF(pyStrObj);
        }
        data.push_back(aux_list);
    }

    return data;
}


int main()
{
    py::scoped_interpreter guard{};
    py::module calc = py::module::import("program");
    py::object result = calc.attr("return_matrix")();

    SheetData data = CastToSheetData(result.ptr());

    for (auto l : data)
    {
        std::cout << "[ ";
        for(auto s : l)
            std::cout << s <<  " ";
        std::cout << "]" << std::endl;
    }

    return 0;
}

输出:

[ 0 1 ]
[ 1 2 ]

可能最好的方法是使用CastToSheetData方法中load函数中的代码来创建自定义type_caster