Блог

Я хочу интегрировать python код с C , в котором c исходный код вызывает функцию / класс в python . Мне конкретно нужно передать массив из c кода в python код и получить другой массив из python в c исходном коде. Для иллюстрации, main.cc

 // call_function.c - A sample of calling 
// python functions from C code
// 

#define NPY_NO_DEPRECATED_API NPY_1_7_API_VERSION
#include <Python.h>
#include <iostream>
#include "numpy/arrayobject.h"


int main(int argc, char *argv[])
{
    PyObject *pName, *pModule, *pDict, *pFunc, *pValue, *pRet;

    const int SIZE{ 10 };
    npy_intp dims = SIZE;
    const int ND{ 1 };

    double* a = new double[10]();
    double* b = new double[10]();

    if (a == NULL)
    {
      exit(1);
    }
    std::cout << a[1] << std::endl;

    // Initialize the Python Interpreter
    Py_Initialize();
    import_array();
    PyRun_SimpleString("import sysn");
    PyRun_SimpleString("sys.path.append("/home/mehdi")");

    // Build the name object
    pName = PyUnicode_FromString("pytst");
    if (pName == NULL) exit(1);

    // Load the module object
    pModule = PyImport_Import(pName);
    if (pModule == NULL) exit(1);

    Py_DECREF(pName);
    if (!pModule){
        std::cout << "mymodule can not be imported" << std::endl;
        return 1;
    }

    // pDict is a borrowed reference 
    pDict = PyModule_GetDict(pModule);
    if (pDict == NULL) exit(1);

    PyObject *pArraya = PyArray_SimpleNewFromData(ND, amp;dims, NPY_DOUBLE, 
                                                  reinterpret_cast<void*>(a));

    import_array();

    PyArrayObject *np_arra = reinterpret_cast<PyArrayObject*>(pArraya);
    PyObject *pArrayb = PyArray_SimpleNewFromData(
        ND, amp;dims, NPY_LONGDOUBLE, reinterpret_cast<void*>(b));

    PyArrayObject *np_arrb = reinterpret_cast<PyArrayObject*>(pArrayb);
    if (!pArraya || !pArrayb)
        std::cout << "Error" << std::endl;


    // pFunc is also a borrowed reference 
    pFunc = PyDict_GetItemString(pDict, "multiply");

    if (PyCallable_Check(pFunc)) 
    {
        pRet = PyObject_CallObject(pFunc, pArraya);
    } else 
    {
        PyErr_Print();
    }

    if (!pRet)
        std::cout << "error" << std::endl;
    if (!PyArray_Check(pRet)) {
        std::cerr <<  " did not return an array." << std::endl;
    }


    if (pRet == NULL) exit(1);

    std::cout << 9 << std::endl;
    PyArrayObject *np_ret = reinterpret_cast<PyArrayObject*>(pRet);

    double* c_out;
    //int len = PyArray_SHAPE(np_ret)[0];
    std::cout << 10 << std::endl;

    c_out = reinterpret_cast<double*>(PyArray_DATA(np_ret));
    // Clean up
    Py_DECREF(pModule);
    Py_DECREF(pName);

    // Finish the Python Interpreter
    Py_Finalize();

    return 0;
}
}
  

pytst.py

 import numpy as np
def multiply(a, b):
    c = np.array([])
    for i in range(10):
      c = np.append(c, a[i]   b[i])
    return c
  

и для его компиляции я использую

g -Wall main.cc -o main -I/usr/include/python3.8 -I /usr/local/lib/python3.5/dist-packages/numpy/core/include .

Я преобразовал double* в PyObject* , чтобы перейти от python к c . но я получаю NULL for pRet (возвращает массив из кода python в c ). Я был бы очень признателен, если бы кто-нибудь дал мне знать, как происходит это взаимодействие между c и python кодом.

Python делает все, используя типы вариантов на основе PyObject; вам нужно будет создать экземпляры PyObject, представляющие списки, содержащие двойники, например

 PyObject* alist = PyList_New(10);
for (i= 0;  i < 10;  i  )
{
    PyList_SetItem(alist, 0, PyFloat_FromDouble(a[i]));
}
...etc...
  

[редактировать] при вызове метода Python вам необходимо передать параметры в виде кортежа, например:

 PyObject* parameters = PyTuple_New(2); 
PyTuple_SetItem(parameters, 0, pArraya);
PyTuple_SetItem(parameters, 1, pArrayb);
pRet = PyObject_CallObject(pFunc, parameters);
Py_DECREF(parameters);
  

Также посмотрите на некоторые онлайн-примеры, например
https://gist.github.com/nad2000/9f69c5096e10c34acddb

Я рекомендую использовать Cython, он намного удобнее в использовании, чем CPython, и в Интернете есть множество руководств.

Я решил использовать pybind11 для интеграции python и c . Простой и хороший справочник для обучения работе с библиотекой, вы можете использовать этот учебник

Глядя на прототип функции PyObject_CallObject, 2-й параметр должен иметь тип PyObject *, поэтому я не думаю, что double может быть напрямую передан PyObject_CallObject.

Вызовите вызываемый объект Python, с аргументами, заданными аргументами кортежа. Если аргументы не требуются, то аргументы могут быть нулевыми.

Смотрите полный документ наhttps://docs.python.org/3/c-api/object.html#c.PyObject_CallObject