Вызов функции numpy из C-кода

Question

Я пытаюсь перенести некоторый код MatLab с расширениями Mex в Python с numpy и scipy-библиотеками. Используя этот замечательный учебник http://www.scipy.org/Cookbook/C_Extensions/NumPy_arrays, я довольно быстро принимаю C-функции для вызова из Python. Но некоторые C-функции вызывают функции MatLab, поэтому я должен подставить этот код, вызвав функции numpy и scipy из C-кода.

Я попытался сделать что-то подобное Extending and Embedding the Python Interpreter. Однако я столкнулся с проблемой: как передать массив в аргумент функции. Более того, этот длинный путь найти функцию в модуле, а затем построить кортеж для аргументов не кажется очень элегантным.

Итак, как вызвать функцию sum в модуле numpy из C-кода, например?

Буду признателен за любые идеи или ссылки. Ruben

P.S. Вот пример:

PyObject *feedback(PyObject *self, PyObject *args){ 
    PyArrayObject *Vecin; 
    double mp=0,ret; 
    if(!PyArg_ParseTuple(args,"O!d",&PyArray_Type,&Vecin,&mp) 
     || Vecin == NULL) return NULL; 
    /* make python string with module name */ 
    PyObject *pName = PyString_FromString("numpy"); 
    if(pName == NULL){ 
     fprintf(stderr,"Couldn\'t setup string %s\n","numpy"); 
     return NULL; 
    } 
    /* import module */ 
    PyObject *pModule = PyImport_Import(pName); 
    Py_DECREF(pName); 
    if(pModule == NULL){ 
     fprintf(stderr,"Couldn\'t find module %s\n","numpy"); 
     return NULL; 
    } 
    /* get module dict */ 
    PyObject *dic = PyModule_GetDict(pModule); 
    if(dic == NULL){ 
     fprintf(stderr,"Couldn\'t find dic in module %s\n","numpy"); 
     return NULL; 
    } 
    /* find function */ 
    PyObject *pFunction = PyDict_GetItemString(dic, "sum"); 
    Py_DECREF(dic); 
    if(pFunction == NULL){ 
     fprintf(stderr,"Couldn\'t find function %s in dic\n","sum"); 
     return NULL; 
    } 
    Py_DECREF(pModule); 
    /* create typle for new function argument */ 
    PyObject *inarg = PyTuple_New(1); 
    if(inarg == NULL){ 
     fprintf(stderr,"Cannot convert to Build in Value\n"); 
     return NULL; 
    } 
    /* set one input paramter */ 
    PyTuple_SetItem(inarg, 0, (PyObject*)Vecin); 
    /* call sunction from module and get result*/ 
    PyObject *value = PyObject_CallObject(pFunction, inarg); 
    if(value == NULL){ 
     fprintf(stderr,"Function return NULL pointer\n"); 
     return NULL; 
    } 
    Py_DECREF(pFunction); 
    if(!PyArg_ParseTuple(value,"d",&ret)) return NULL; 
    return Py_BuildValue("d",ret*mp); 
} 

Результат

>>print mymod.FeedBack(np.array([1.,2.,3.,4.,5.]),2) 


Traceback (most recent call last): 
    File "mymodtest.py", line 10, in <module> 
    print mymod.FeedBack(np.array([1.,2.,3.,4.,5.]),2) 
SystemError: new style getargs format but argument is not a tuple 
Segmentation fault 

Answer 1

NumPy C API имеет PyArray_Sum:

PyObject *sum = PyArray_Sum(arr, NPY_MAXDIMS, PyArray_DESCR(arr)->type_num, NULL);