python--杂识--15--python调用c代码

两种方法：

• Python/C api
• ctypes

1 Python/C api

1.1编写代码

c_test.c

#include <Python.h>

// C的原生函数，实现两个整数的相加
int add(int a, int b)
{
    return a + b;
};

// compute_add【一般：模块名_函数名】，按照python规定的调用方式
/*
   1 定义一个新的静态函数，接收2个PyObject *参数，返回1个PyObject *值，args表示接收python的位置参数，kwargs表示接收python的关键字参数
   2 PyArg_ParseTuple方法将python输入的变量变成C的变量，即上述args→num; https://blog.csdn.net/james506/article/details/16824283
   3 紧接着调用C原生函数add_one，传入num
   4 最后将调用返回的C变量，转换为PyObject*或其子类，并通过Py_BuildValue方法，返回python值
*/
static PyObject* compute_add(PyObject* self, PyObject* args)   // (PyObject* self, PyObject* args, PyObject* kwargs)
{
    int a;
    int b;
    if (!PyArg_ParseTuple(args, "ii", &a, &b))  // ii 依次表示args中的参数类型，如果是python对象用O表示
        return NULL;
    return Py_BuildValue("i", add(a, b));
};

// compute_docs：python中add函数的功能说明
static char compute_docs[] =
"add(a, b): adding two integers\n";

// compute_funcs：函数集，创建一个数组，python可以调用的扩展函数集
/*
   1 add：表编译后python调用时使用的函数名；
   2 compute_add：表调用当前C代码中的哪一个函数，即static PyObject *compute_add；
   3 METH_VARARGS：表函数的参数传递形式，主要包括位置参数和关键字参数两种，关键字参数用METH_KEYWORDS；既有位置参数又有关键字参数时【METH_VARARGS | METH_KEYWORDS】
   4 compute_docs：表python中add函数的功能说明
   5 最后如果希望添加新的函数，则在最后的{NULL, NULL, 0, NULL}里按同样格式填写新的调用信息，比如加一些求和求乘积函数
*/
static PyMethodDef compute_funcs[] =
{
    {"add", (PyCFunction)compute_add, METH_VARARGS, compute_docs},
    {NULL, NULL, 0, NULL}
};

// compute_module：创建module的信息
static PyModuleDef compute_module =
{
    PyModuleDef_HEAD_INIT,
    "Compute",              // 模块名，即是生成模块后的名字，如numpy,opencv等等
    "",                     // 模块文档，可以为空NULL
    -1,                     // 模块中每个解释器的状态大小，模块在全局变量中保持状态，则为-1
    compute_funcs           // 即上一步定义的Methods，说明了可调用的函数集
};

// PyInit_Compute：module初始化
PyMODINIT_FUNC PyInit_Compute(void)
{
    return PyModule_Create(&compute_module);
}

setup.py

from distutils.core import setup, Extension  # 这里要用到distutils库


module1 = Extension('Compute', sources=['c_test.c'])  # 打包文件为c_test.c，这里没有设置路径，所有.c文件和setup.py放在同一目录下
setup(
    name='Compute',                                   # 打包名
    version='1.0',                                    # 版本
    description='This is a demo package',             # 说明文字
    ext_modules=[module1])

1.2 打包

python setup.py build
python setup.py install

1.3 查看是否安装成功

1.3.1 pip list

1.3.2 执行Compute.add

2 ctypes

2.1 编写代码

c_test.c

#include <stdio.h>
#include <stdlib.h>

// C的原生函数，实现两个整数的相加
int add(int a, int b)
{
    return a + b;
};

test.py

import ctypes

# 加载共享库
Compute = ctypes.CDLL('./c_test.so')

# 指定C函数参数和返回类型
Compute.add.argtypes = (ctypes.c_int, ctypes.c_int)
Compute.add.restype = ctypes.c_int

# 调用C函数
result = Compute.add(3, 4)
print(result)

2.2 编译c代码

gcc -fPIC -shared -o c_test.so c_test.c

2.3 执行 add

2.4 更详细的使用方式

Python基础库-ctypes