如何使用Python的cffi和enum数组

时间:2018-10-25 22:46:33

标签: python c enums python-cffi

我有一个用C实现的函数,该函数将字符串转换为已定义的enum的数组。 C代码如下:

头文件(描述enum):

/* File `asdf.h` */
enum translation {
        ONE,
        ANOTHER,
        FINISH,
};

void translate(char *origin, enum translation *destination);

源文件(实现翻译功能):

/* File `asdf.c` */
#include "asdf.h"


void translate(char *origin, enum translation *destination)
{
    while (1) {
        if (*origin == 'F')
            break;
        if (*origin == 'O')
            *destination++ = ONE;
        if (*origin == 'A')
            *destination++ = ANOTHER;
        origin++;
    }
    *destination = FINISH;
}

现在,我希望能够使用CFFI从Python调用此函数,并能够检查输出数组的正确值(转换)。我该怎么办?

我尝试先编译模块,然后创建变量并以这种方式调用函数,但未能使其工作:

# File `cffienum.py`
from pathlib import Path
from cffi import FFI


ffibuilder = FFI()
ffibuilder.cdef(Path('asdf.h').read_text())
ffibuilder.set_source(
    '_asdf',  # name of the output C extension
    '#include "asdf.h"',
    sources=['asdf.c'])   # includes asdf.c as additional sources
ffibuilder.compile()


if __name__ == '__main__':
    from _asdf.lib import translate

    ffi = FFI()
    ffi.cdef(Path('asdf.h').read_text())
    c = ffi.dlopen('c')
    source = ffi.new('char source[]', b'OAF')
    destination = ffi.new('enum translation destination[10]')

    translate(source, destination)
    print(source, destination)

这是我得到的错误:

$ pipenv run python cffienum.py 
Traceback (most recent call last):
  File "cffienum.py", line 24, in <module>
    translate(source, destination)
TypeError: initializer for ctype 'enum translation *' must be a pointer to same type, not cdata 'enum translation[10]'

第一次接触CFFI而不是经验丰富的C程序员,所以我敢打赌我在这里做错了很多事情。

更新

请注意,如果不是:

void translate(char *origin, enum translation *destination)

我使用:

void translate(char *origin, char *destination)

或者:

void translate(char *origin, uint8_t *destination)

效果很好。因此,在使用enum translation * ...

时我必须缺少一些东西

1 个答案:

答案 0 :(得分:1)

您必须使用_adsl模块中的ffi对象:

from _asdf import ffi

而不是创建新的FFI实例并重新定义恰好相同的无关类型enum translation