Вернуть массив строк C из расширения C - PullRequest
Новая
       13

Вернуть массив строк C из расширения C

0 голосов
/ 25 апреля 2019

У меня есть функция C, которая возвращает массив строк.Как я могу вызвать его в форме расширения Python C, которое вернет массив обратно в вызывающую функцию Python?(Я новичок в расширениях Python C и имею минимальный опыт работы с расширениями)

Это определение, которое я пробовал: 

static PyObject* _get_array(PyObject* self, PyObject* args)
{
    int64_t value;
    int init_level;
    int final_level;

    if(!PyArg_ParseTuple(args, "Lii", &value, &init_level, &final_level))
        return NULL;

    // returning the array as a Python object by o
    return Py_BuildValue("o", _get_array(value, init_level, final_level));
}

и метод def: 

static PyMethodDef array_methods[] = {
    { "get_array", _get_array, METH_VARARGS, "Returns a string array"},
    { NULL, NULL, 0, NULL }
};

Обновление

Функция get_array: 

#include <stdio.h>
#include <stdint.h>
#include <stdlib.h>
#include <Python.h>

char **get_array(int64_t value, int init_level, int final_level) {

  int SHIFTS []= {44, 40, 36, 32, 28, 24, 20, 16, 12, 8, 4, 0};

  long count =  1 << (4* (final_level - init_level));
  char** t_array;
  t_array = malloc(sizeof(char*)*count);

  int shift_coff = 11 -(final_level-init_level);
  int64_t base = (value << SHIFTS[shift_coff]);

  for (long i=0; i < count; i++){
    t_array[i] = malloc((4+final_level)*sizeof(char));
    sprintf(t_array[i], "%llX", (base + i));
  }

  return t_array;
}

1 Ответ

1 голос
/ 26 апреля 2019

Вы не можете вернуть свой char** напрямую Python, так как Python понимает только объекты типа PyObject* (поскольку он содержит информацию, необходимую для обработки подсчета ссылок и определения типа).Поэтому вы должны создать подходящий объект Python.Самым простым вариантом будет список строк.В следующем простом случае вы будете массивом с использованием строкового типа (вы можете сделать это легко, потому что все ваши строки имеют одинаковую длину).Ни у одного из них нет прямого Py_BuildValue преобразования, поэтому вы должны сами писать циклы.

Для списка строк вы просто создаете список с помощью PyList_New, а затем просматриваете элемент за элементом с PyList_SetItem: 

char** array = get_array(value, init_level, final_level);
PyObject* list = PyList_New(1 << (4* (final_level - init_level)));
if (!list) return NULL;

for (int i=0; i<(1 << (4* (final_level - init_level))); ++i) {
    PyObject* item = PyBytes_FromStringAndSize(array[i],(4+final_level));
    if (!item) goto failed;

    if (PyList_SetItem(list,i,item) != 0) {
        Py_DECREF(item);
        goto failed;
    }

    free(array[i]); // deallocate array as we go
}
free(array);

// returning the array as a Python object by o
return list;

failed:
Py_DECREF(list);
// also deallocate the rest of array?
return NULL;

Обратите внимание, что я не завершил управление памятью при сбое, поэтому у вас будет утечка array.

Для массива numpy вы выделяете массив с помощьюправильный тип строки, а затем скопируйте в нее данные 

char** array = get_array(value, init_level, final_level);

// create an "Sx" dtype, where x is a suitable number
PyArray_Descr *desc = PyArray_DescrNewFromType(NPY_STRING);
desc->elsize = (4+final_level);

npy_intp array_length[] = {1 << (4* (final_level - init_level))};
PyObject* nparray = PyArray_SimpleNewFromDescr(1,array_length,desc);
if (!nparray) return NULL; // clean up array too

for (int i=0; i<(1 << (4* (final_level - init_level))); ++i) {
    char* data = PyArray_GETPTR1((PyArrayObject*)nparray,i);

    // copy data
    for (int j=0; j<(4+final_level); ++j) {
        data[j] = array[i][j];
    }

    free(array[i]); // deallocate array as we go
}
free(array);

// returning the array as a Python object by o
return nparray;

Опять же, не вся обработка ошибок идеальна.Чтобы этот пример работал, вы должны вызвать import_array() в функции инициализации вашего модуля.

В обоих случаях вам лучше не выделять память в get_array, а писать напрямуюв ваши объекты Python.

...