Python 调用 C/C++ 程序的方法

Python 调用 C/C++ 程序的方法

最近写 BUG 的时候遇到 python 计算很慢的情况，于是调研了一波在 python 中嵌入 C++ 程序的方法，记录一下，便于查询。

一般来说在 python 调用 C/C++ 程序主要可以分为 3 步：

• 1、编写 C/C++ 实现程序。
• 2、将 C/C++ 程序编译成动态库。
• 3、在 Python 中调用编译生成的库。Python 在调用 C/C++ 程序时有一些不同，需要注意。

1、Python 调用 C 函数

Python 调用 C 语言程序比较简单，将 C 语言程序编译好，再使用 python 中的 ctypes 模块调用即可。

C 语言源文件：called_c.c

#include <stdio.h>
int foo(int a, int b)
{
	printf("a:%d, b:%d.", a, b);
	return 0;
}

在命令行或者终端输入：

gcc -o libpycall.so -shared -fPIC called_c.c  

生成 libpycall.so 动态库文件，之后就可以在 Python 中调用 foo 函数。

编译参数说明 -fPIC：生成位置无关目标代码，适用于动态连接；-L path：表示在 path 目录中搜索库文件，如 - L. 表示在当前目录；-I path：表示在 path 目录中搜索头文件；-o file：制定输出文件为 file；-shared：生成一个共享库文件；

Python 文件：py_call_c.py

import ctypes
dll=ctypes.cdll.LoadLibrary
lib=dll('./libpycall.so')
lib.foo(1,3)

运行 py_call_c.py 输出为：

a:1, b:3

2、Python 调用 C++ 类

由于 C++ 支持函数重载，在 g++ 以 C++ 方式编译时编译器会给函数的名称附加上额外的信息，这样 ctypes 模块就会找不到 g++ 编译生成的函数。因此，要让 g++ 按照 C 语言的方式编译才可以找到生成的函数名。让编译器以 C 语言的方式编译就要在代码中使用 extern 关键字将代码包裹起来。

C++ 源文件: cpp_called.cpp

#include <iostream>
using namespace std;
 
class TestLib
{
    public:
        void display();
        void display(int a);
};
void TestLib::display() {
    cout<<"First display"<<endl;
}
 
void TestLib::display(int a) {
    cout<<"Second display:"<<a<<endl;
}
extern "C" {
    TestLib obj;
    void display() {
        obj.display();
      }
    void display_int(int a) {
        obj.display(a);
      }
}

在命令行或者终端输入编译命令：

g++ -o libpycallcpp.so -shared -fPIC cpp_called.cpp  

生成 libpycallcpp.so，在 Python 中调用。
Python 文件：py_call_c.py

import ctypes   
dll=ctypes.cdll.LoadLibrary
lib=dll('./libpycallcpp.so')
lib.display()
lib.display_int(9)

输出为：

First display 
Second display:9

OK，基本功能实现。

参考