Вызов функции C в Python и возврат 2 значений - PullRequest
Новая
       29

Вызов функции C в Python и возврат 2 значений

0 голосов
/ 05 июля 2018

Я пытаюсь выяснить, как вернуть 2 вейла из функции C, которую я вызвал в python. Я прочитал материал онлайн и использую struct для вывода двух переменных. Я могу вывести переменные, когда я вызываю эту функцию в том же файле C. Однако, когда я пытаюсь вызвать его в python, он все равно возвращает только одно значение.

Это мой код C: 

struct re_val {

    double predict_label;
    double prob_estimates;  

};

struct re_val c_func(const char* dir, double a, double b, double c, double d )
{
    double x[] = {a,b,c,d};

    printf ("x[0].index: %d \n", 1);
    printf ("x[0].value: %f \n", x[0]);

    printf ("x[1].index: %d \n", 2);
    printf ("x[1].value: %f \n", x[1]);

    printf ("x[2].index: %d \n", 3);
    printf ("x[2].value: %f \n", x[2]);

    printf ("x[3].index: %d \n", 4);
    printf ("x[3].value: %f \n", x[3]);

    printf ("\nThis is the Directory: %s \n", dir); 

    struct re_val r;
    r.predict_label = 5.0;
    r.prob_estimates = 8.0;   

    return r;

}

Это мой код Python: 

calling_function = ctypes.CDLL("/home/ruven/Documents/Sonar/C interface/Interface.so")
calling_function.c_func.argtypes = [ctypes.c_char_p, ctypes.c_double, ctypes.c_double, ctypes.c_double, ctypes.c_double]
calling_function.c_func.restype =  ctypes.c_double
q = calling_function.c_func("hello",1.3256, 2.45, 3.1248, 4.215440)
print q

В настоящее время, когда я запускаю свой файл python в терминале, он выводит следующее: 

x[0].index: 1 
x[0].value: 1.325600 

x[1].index: 2 
x[1].value: 2.450000 

x[2].index: 3 
x[2].value: 3.124800 

x[3].index: 4 
x[3].value: 4.215440 

This is the Directory: hello 

5.0

Вместо этого я хотел бы вывести это: 

x[0].index: 1 
x[0].value: 1.325600 

x[1].index: 2 
x[1].value: 2.450000 

x[2].index: 3 
x[2].value: 3.124800 

x[3].index: 4 
x[3].value: 4.215440 

This is the Directory: hello 

5.0
8.0

1 Ответ

0 голосов
/ 05 июля 2018

Ваш код на C в порядке, проблема в том, как вы используете Python ctypes. Вы должны сказать, что функция возвращает struct re_val, а не двойное число: 

calling_function.c_func.restype =  ctypes.c_double

Вышеприведенное заставляет функцию возвращать одно двойное значение в глазах ctypes. Вы должны сказать python, что функция возвращает структуру: 

import ctypes as ct

# Python representation of the C struct re_val
class ReVal(ct.Structure):
    _fields_ = [("predict_label", ct.c_double),("prob_estimates", ct.c_double)]

calling_function = ctypes.CDLL("/home/ruven/Documents/Sonar/C interface/Interface.so")
calling_function.c_func.argtypes = [ctypes.c_char_p, ctypes.c_double, ctypes.c_double, ctypes.c_double, ctypes.c_double]
# and instead of c_double use:
calling_function.c_func.restype = ReVal

Таким образом, вы указываете типу Python ctypes, что функция возвращает агрегатный объект, который является подклассом ctypes.Structure, который соответствует struct re_val из библиотеки c.

ПРИМЕЧАНИЕ Будьте очень осторожны с аргументами argtypes и restype, если вы используете их неправильно, вы легко можете потерпеть крах интерпретатора Python. Тогда вы получите сигфоут вместо приятного отслеживания.

...