快捷導(dǎo)航

Python中CFFI簡介和使用實際示例

更新時間：2025年09月09日 11:42:46 作者：東北豆子哥

CFFI是Python調(diào)用C代碼的接口庫,支持ABI和API兩種模式,提供自動內(nèi)存管理與NumPy集成能力,相比ctypes,其接口更友好,PyPy性能更優(yōu),適合高性能計算和C庫集成場景,本文給大家介紹Python中CFFI簡介和使用實際示例,感興趣的朋友跟隨小編一起看看吧

Python中CFFI介紹和使用

CFFI (C Foreign Function Interface) 是Python的一個外部函數(shù)接口庫，用于調(diào)用C代碼。它提供了與C語言交互的簡單方式，是替代ctypes的另一種選擇。

CFFI的特點

支持兩種模式：ABI模式和API模式
自動生成綁定代碼：可以自動生成Python與C之間的橋梁代碼
支持CPython和PyPy：在PyPy上性能表現(xiàn)優(yōu)異
內(nèi)存管理安全：自動管理內(nèi)存，減少內(nèi)存泄漏風(fēng)險
支持C語言標(biāo)準(zhǔn)：支持C99和部分C11特性

安裝CFFI

pip install cffi

使用模式

1. ABI模式 (應(yīng)用程序二進制接口)

ABI模式不需要編譯，直接加載動態(tài)庫：

from cffi import FFI
ffi = FFI()
ffi.cdef("""
    int printf(const char *format, ...);
""")
C = ffi.dlopen(None)  # 加載標(biāo)準(zhǔn)C庫
C.printf(b"Hello, %s!\n", b"World")

2. API模式 (應(yīng)用程序接口)

API模式需要編譯，性能更好：

from cffi import FFI
ffi = FFI()
ffi.cdef("""
    int add(int a, int b);
""")
# 內(nèi)聯(lián)C源代碼
ffi.set_source("_example",
"""
    int add(int a, int b) {
        return a + b;
    }
""")
if __name__ == "__main__":
    ffi.compile()

使用編譯后的模塊：

from _example import ffi, lib
result = lib.add(2, 3)
print(result)  # 輸出: 5

實際示例

示例1：調(diào)用標(biāo)準(zhǔn)C庫函數(shù)

from cffi import FFI
ffi = FFI()
ffi.cdef("""
    double sin(double x);
    double cos(double x);
""")
C = ffi.dlopen(None)  # 加載標(biāo)準(zhǔn)C庫
x = 3.1415926535 / 4
print("sin:", C.sin(x))
print("cos:", C.cos(x))

示例2：與自定義C代碼交互

example.h

#ifndef EXAMPLE_H
#define EXAMPLE_H
int add(int a, int b);
void print_message(const char *message);
#endif

example.c

#include <stdio.h>
#include "example.h"
int add(int a, int b) {
    return a + b;
}
void print_message(const char *message) {
    printf("Message: %s\n", message);
}

編譯為共享庫：

gcc -shared -o libexample.so -fPIC example.c

Python代碼：

from cffi import FFI
ffi = FFI()
ffi.cdef("""
    int add(int a, int b);
    void print_message(const char *message);
""")
lib = ffi.dlopen("./libexample.so")
print("Add result:", lib.add(5, 7))
lib.print_message(b"Hello from Python!")

示例3：處理復(fù)雜數(shù)據(jù)結(jié)構(gòu)

from cffi import FFI
ffi = FFI()
ffi.cdef("""
    typedef struct {
        int x;
        int y;
    } Point;
    double distance(Point p1, Point p2);
""")
ffi.set_source("_geometry",
"""
    #include <math.h>
    typedef struct {
        int x;
        int y;
    } Point;
    double distance(Point p1, Point p2) {
        int dx = p1.x - p2.x;
        int dy = p1.y - p2.y;
        return sqrt(dx*dx + dy*dy);
    }
""")
if __name__ == "__main__":
    ffi.compile()

使用編譯后的模塊：

from _geometry import ffi, lib
p1 = ffi.new("Point *", [1, 2])
p2 = ffi.new("Point *", [4, 6])
distance = lib.distance(p1[0], p2[0])
print("Distance:", distance)  # 輸出: 5.0

內(nèi)存管理

CFFI提供了幾種內(nèi)存管理方式：

ffi.new()：分配內(nèi)存并在Python對象被垃圾回收時自動釋放
ffi.gc()：顯式指定垃圾回收時的釋放函數(shù)
ffi.release()：手動釋放內(nèi)存

from cffi import FFI
ffi = FFI()
ffi.cdef("""
    typedef struct {
        char *name;
        int age;
    } Person;
    Person *create_person(char *name, int age);
    void free_person(Person *p);
""")
# 假設(shè)這是與C庫交互的部分
ffi.set_source("_person", "")
if __name__ == "__main__":
    ffi.compile()
from _person import ffi, lib
# 自動內(nèi)存管理
person = ffi.new("Person *")
person.name = ffi.new("char[]", b"Alice")
person.age = 30
# 手動內(nèi)存管理
p = lib.create_person(b"Bob", 25)
try:
    print(ffi.string(p.name), p.age)
finally:
    lib.free_person(p)

與NumPy集成

CFFI可以與NumPy數(shù)組高效交互：

import numpy as np
from cffi import FFI
ffi = FFI()
ffi.cdef("""
    void square_array(double *array, int length);
""")
ffi.set_source("_numpy_example",
"""
    void square_array(double *array, int length) {
        for (int i = 0; i < length; i++) {
            array[i] = array[i] * array[i];
        }
    }
""")
if __name__ == "__main__":
    ffi.compile()
from _numpy_example import ffi, lib
arr = np.array([1.0, 2.0, 3.0, 4.0], dtype=np.float64)
ptr = ffi.cast("double *", arr.ctypes.data)
lib.square_array(ptr, len(arr))
print(arr)  # 輸出: [ 1.  4.  9. 16.]