使用swig在python中调用c++代码

1. 使用sudo apt install swig来按照swig

2. 编写对应的.i文件

假设我们编写的c++文件和.h文件如下:

/* File: example.cpp */

#include "example.h"

int fact(int n) {
    if (n < 0){ /* This should probably return an error, but this is simpler */
        return 0;
    }
    if (n == 0) {
        return 1;
    }
    else {
        /* testing for overflow would be a good idea here */
        return n * fact(n-1);
    }
}


/* File: example.h */

int fact(int n);

则对应的.i文件如下:
/* File: example.i */
%module example

%{
#define SWIG_FILE_WITH_INIT
#include "example.h"
%}

int fact(int n);

3. 生成对应的.example_wrap.cxx和example.py.

运行命令:swig -c++ -python example.i

4. 使用 distutils工具编译

首先生成一个setup.py文件

#!/usr/bin/env python

"""
setup.py file for SWIG example
"""

from distutils.core import setup, Extension


example_module = Extension('_example',
                           sources=['example_wrap.cxx', 'example.cpp'], #一定要把cpp源文件包含,如果该源文件包含了别的依赖,则应该也包含进来
                           )

setup (name = 'example',
       version = '0.1',
       author      = "SWIG Docs",
       description = """Simple swig example from docs""",
       ext_modules = [example_module],
       py_modules = ["example"],
       )

然后使用命令python setup.py install编译生成即可

参考链接,第31.2 Preliminaries