用C或C++为Python编写模块

Posted 2020-10-18 挖洞的土拨鼠

1、使用c或c++编写对应的函数例如：

1 //modtest.c
2 int abs(int number){
3     if(number<0){
4         return -number;
5     }
6     else{
7         return number;
8     }
9 }        

2、编写每个函数对应的包装函数：

1 static PyObject *modtest_abs(PyObject *self,Pyobject *args){
2     int num; //定义参数类型
3     if(!PyArg_ParseTuple(args,"i",&num)){     
4         //"i"表示数据类型 这里进行参数传递类型检查
5         return NULL;
6     }
7     return (Pyobject*)Py_BulidValue("i",abs(num));
8     //"i"返回值类型，第二个参数为c/c++函数的调用后的返回值
9 }    

3、编写每个模块的PyMethodDef MoudleMethod数组：

1 static PyMethodDef modtestMethods[] = {
2     {"abs",modtest_abs,METH_VARARGS},
3     {...},//几个函数就有几个数组项，当然最后还有一个空项
4     {NULL,NULL}
5 };

4、编写模块初始化函数：

1 void initmodtest(){
2     Py_InitMoudle("modtest","modtestMethods");
3 }

5、编译：
创建setup文件：

1 #！/usr/bin/env python
2 from distutils.core import setup,Extension
3 Mod = "modtest"
4 setup(name=MOD, ext_modules=[Extension(MOD, sources=[‘modtest.c‘])])
5 
6 #python setup.py build