扩展Python模块系列(二)----一个简单的例子

本节使用一个简单的例子引出Python C/C++ API的详细使用方法。针对的是CPython的解释器。

目标:创建一个Python内建模块test,提供一个功能函数distance, 计算空间中两个点之间的距离。

可以在Python代码这样使用test模块:

import test
s = test.distance((0, 0, 0), (1, 1, 1))

先上代码:

[test.c]

#include <Python.h>
#include <math.h> static PyObject* distance(PyObject* self, PyObject* args)
{
double x0, y0, z0, x1, y1, z1; if (!PyArg_ParseTuple(args, "(ddd)(ddd)", &x0, &y0, &z0, &x1, &y1, &z1))
{
return NULL;
}
return PyFloat_FromDouble(sqrt((x0 - x1) * (x0 - x1) + (y0 - y1) * (y0 - y1) + (z0 - z1) * (z0 - z1)));
} static PyMethodDef cformula_methods[] =
{
{ "distance", distance, METH_VARARGS, "Return the 2D Distance of 2 Points." },
{ NULL, NULL, , NULL },
}; PyMODINIT_FUNC inittest(void)
{
Py_InitModule3("test", cformula_methods, "Common test Written in C.");
}

[Source.cpp]

#include <Python.h>

PyMODINIT_FUNC inittest();

int main()
{
//PyImport_AppendInittab("test", inittest);
Py_Initialize();
inittest();
PyRun_SimpleString("import test");
PyRun_SimpleString("s = test.distance((0, 0, 0), (1, 1, 1))");
PyRun_SimpleString("print s");
return ;
}

编译运行后,结果如下:

扩展Python模块系列(二)----一个简单的例子

如果希望将test模块打包为一个动态链接库,供其他Python程序使用,即打包为test.pyd,在其他Python程序中可以直接import test,就和正常的Python内建模块一样。

步骤如下:

1) 在test.c同级目录下,新建一个python文件setup.py

2)setup.py:

from distutils.core import setup, Extension
setup(ext_modules=[Extension('test', sources = ['test.c'])])

3) 执行python命令: python setup.py build

扩展Python模块系列(二)----一个简单的例子

这种情况是因为没有指定C的编译器,本文使用VS2015提供的编译器,所以首先执行:SET VS90COMNTOOLS=%VS140COMNTOOLS%

扩展Python模块系列(二)----一个简单的例子

会在该目录下发现build/lib.win32-2.7中有一个test.pyd,这就是编译后的动态库,可以直接import

扩展Python模块系列(二)----一个简单的例子

实现细节:

1. 任何扩展Python模块的C程序,一般只需要包含<Python.h>头文件即可,文档中这样描述:

All function, type and macro definitions needed to use the Python/C API are included in your code by the following line:

#include "Python.h"

在包含了Python.h之后,隐含地会自动包含C语言的标准头文件<stdio.h>, <string.h>, <errno.h>, <limits.h>, <assert.h> and <stdlib.h>

2. 本质上Python C API提供了对C函数的wrapper。假设有一个现成的C函数, int add(int a, int b), 想把它实现在Python的模块里,需要实现一个wrapper函数 static PyObject* addAB(PyObject* self, PyObject* args), 将args解析为两个整数a、b,然后调用add(a,b),将返回值打包为一个Python整型对象返回。

static PyObject* distance(PyObject* self, PyObject* args)

函数一定要声明为static,将其限定在此文件范围;参数self是Python内部使用的,遵循范式即可;参数args是函数的参数列表,是一个tuple。

PyArg_ParseTuple(args, "(ddd)(ddd)", &x0, &y0, &z0, &x1, &y1, &z1)

此函数将参数列表args解析为两个tuple, 每个tuple是三个double类型的元素。如果参数列表不符合"(ddd)(ddd)"这种形式,直接返回NULL。

PyFloat_FromDouble

此函数将一个C原生的double,封装Python 的PyFloatObject。

3. 定义该模块对外提供的函数接口

static PyMethodDef cformula_methods[]

在此数据结构中定义模块test对外提供的函数接口,第一个参数"distance"是Python内部记录的,就是test.distance调用时,python查找的函数名;第二个参数distance是函数具体实现,本质是一个函数指针;第三个参数是METH_VARARGS告诉Python此函数一个典型的PyCFunction,参数为两个PyOBject*,参数2使用PyArg_ParseTuple来解析;最后一个是函数说明。

4.定义模块初始化函数

PyMODINIT_FUNC inittest(void)
{
Py_InitModule3("test", cformula_methods, "Common test Written in C.");
}

函数inittest必须这样定义,如果模块名为example,那么模块初始化函数为initexample。宏定义PyMODINIT_FUNC定义如下:

#       if defined(__cplusplus)
# define PyMODINIT_FUNC extern "C" void
# else /* __cplusplus */
# define PyMODINIT_FUNC void
# endif /* __cplusplus */

针对cpp的实现加了修饰extern "C"。

Py_InitModule3创建模块test, 当import test时,Python解释器会找到inittest创建的模块test。

5. 测试代码Source.cpp

在Python虚拟机环境初始化 Py_Initialize()之后,调用inittest(),则会创建新模块test。

如果希望在import test时才初始化模块test,那么在Py_Initialize()之前调用PyImport_AppendInittab("test", inittest); 然后再注释掉initest()。

PyImport_AppendInittab:Add a single module to the existing table of built-in modules.The new module can be imported by the namename, and uses the function initfunc as the initialization function called on the first attempted import. This should be called before Py_Initialize().

本节通过一个简单的例子来展示了如何用C语言扩展Python模块,并简单解释了用到的Python C/C++ API。在下一节中,将更加深入地了解Python C/C++ API,以及在写扩展程序时,会遇到的坑,比如:异常处理、引用计数等等。

上一篇:一个简单的例子了解states


下一篇:用一个简单的例子来理解python高阶函数