2015-09-12 122 views
0

我正在为C++函数repeating_count::count(string str)编写Python包装。这个函数的算法并不重要,但它总是返回int。 我的包装获取list[str]作为输入并返回list[int]。我的包装的代码如下:扩展C++函数的奇怪行为

PyObject* count_rep_list(PyObject *mod, PyObject *args){ 
    PyObject* inputList = PyList_GetItem(args, 0); 
    PyObject* outputList = PyList_New(0); 
    cout << PyList_Size(inputList); 
    char* str; 
    for(size_t i = 0; i < PyList_Size(inputList); ++i){ 
     PyObject* list_item = PyList_GetItem(inputList, i); 
     if(!PyArg_Parse(list_item, "s", &str)){ 
      return NULL; 
     } 
     PyList_Append(outputList, PyLong_FromSize_t(repeating_count::count(string(str)))); 
    } 
    return outputList; 
} 

然后我做了这个功能的Python模块:

PyMODINIT_FUNC PyInit_repeating_count() { 
    static PyMethodDef ModuleMethods[] = { 
      {"count_rep_list", count_rep_list, METH_VARARGS, "counting repeatings for list of strings"}, 
      { NULL, NULL, 0, NULL} 
    }; 
    static PyModuleDef ModuleDef = { 
      PyModuleDef_HEAD_INIT, 
      "repeating_count", 
      "counting repeatings", 
      -1, ModuleMethods, 
      NULL, NULL, NULL, NULL 
    }; 
    PyObject * module = PyModule_Create(&ModuleDef); 
    return module; 
} 

我成功编译和.so文件链接这个模块。但是,当我想从Python 3中调用该函数,我抓住

所以,我在做什么错

分割故障(核心转储)?

的Python代码片段调用count()

from repeating_count import * 
print(count_rep_list(["sdfsf", "sf", "sdfvgsdgsd"])) 
+1

str是一个单元化的指针,你传递给PyArg_Parse –

+0

@KennyOstrom,非常感谢。我已更正了代码。 – VeLKerr

回答

0

我解决我的问题。代码中只有一个错误。我不得不使用PyTuble_GetItem(args, 0)而不是PyList_GetItem(args, 0)因为args是元组。