编写要嵌入的C、C++代码
cfib.h
// cfib.h
double cfib(int n); //定义一个函数声明
cfib.c
//cfib.c
double cfib(int n) {
int i;
double a = 0.0, b = 1.0, tmp;
for(i = 0; i < n; i++){
tmp = a;
a = a + b;
b = tmp;
}
return a;
}//函数体的实现
然后是pyx文件
fib_with_c.pyx
# fib_with_c.pyx
# 通过cdef extern from 导入头文件,写上里面的函数
cdef extern from "cfib.h":
double cfib(int n)
# 然后python可以直接调用
def fib_with_c(n):
"""调用c编写的斐波那契数列"""
return cfib(n)
最后是编译
setup.py
from distutils.core import setup, Extension
from Cython.Build import cythonize
# 我们看到之前是直接往cythonize里面传入一个文件名即可
# 但是现在我们传入了一个扩展对象,通过扩展对象的方式可以实现更多的功能
ext = Extension(name="wrapper_fib", sourse=["fib.pyx", "cfib.c"])
setup(ext_modules=cythonize(ext))
编译命令:
python setup.py build_ext --inplace
编译之后增加文件
wrapper_fib.cpython-36m-x86_64-linux-gnu.so
fib_with_c.c
调用一下
>>> import wrapper_fib
>>> wrapper_fib
<module 'wrapper_fib' from '/usr/local/lib/python3.6/dist-packages/wrapper_fib.cpython-36m-x86_64-linux-gnu.so'>
>>> wrapper_fib.fib_with_c(20)
6765.0
>>> wrapper_fib.fib_with_c.__doc__
'调用C编写的斐波那契数列'
>>>
我们看到成功调用C编写的斐波那契数列,这里我们使用了一种新的创建扩展模块的方法,我们来总结一下。
- 如果是单个pyx文件的话,那么直接通过cythonize(“xxx.pyx”)即可。
- 如果pyx文件还引入了C文件,那么通过
cythonize(Extension(name="xx", source=["", ""]))
的方式即可。name是编译之后的扩展模块的名字,sources是你要编译的源文件,我们这里是一个pyx文件一个C文件。建议后续都使用第二种方式,可定制性更强,而且我们之前使用的
cythonize("fib.pyx")
完全可以用cythonize(Extension("fib", ["fib.pyx"]))
进行替代。
移植
将.so导入其他Ubuntu系统可在python环境下直接导入使用。