pyhton使用cffi传递numpy数据给C模块

pyhton使用cffi传递numpy数据给C模块

说明

• cffi是连接Python与c的桥梁，可实现在Python中调用c文件。cffi为c语言的外部接口，在Python中使用该接口可以实现在Python中使用外部c文件的数据结构及函数。
• cffi资料:https://cffi.readthedocs.io/en/release-1.12/ref.html
• 需要使用到ffi.cast

C代码 green.c

// gcc -shared -o libgreen.so -fPIC green.c
// https://github.com/boisgera/cffi-example
#include <stdio.h>

void greenify(unsigned char* data, unsigned int length) 
{
    unsigned int i;
    for(i=0; i<length; i++) 
    {
      data[3*i] = 255;
    }
}

python代码 example.py

#!/usr/bin/env python

from cffi import FFI
import numpy as np

# Array data for a 4x3 RGB (black) image.
dtype = [("r", np.uint8), ("g", np.uint8), ("b", np.uint8)]
array = np.zeros((3,4), dtype=dtype)

# declare data & function for cffi 
ffi = FFI()
data = ffi.cast("unsigned char*", array.__array_interface__["data"][0])
length = ffi.cast("unsigned int", array.size)
ffi.cdef("void greenify(unsigned char* data, unsigned int length);")

# open the C lib and call a C function
libgreen = ffi.dlopen("./libgreen.so")
libgreen.greenify(data, length)

# the array has been modified.
print(array)