使用boost.python封装C++库

C++以高性能著称，但是编写较为复杂．而简洁是Python的强项．如果能珠联璧合，就能发挥两家之长．本文尝试用boost库的python模块封装Ｃ++

前期准备

需要安装Python和boost库，特别注意的是，在安装boost时，要指明with-python的版本

sudo apt install python3-dev build-essential

git clone https://github.com/boostorg/boost.git

cd boost

./bootstrap.sh --with-python=python3.5 # python3.6的话就写３.6

./b2

sudo ./b2 install

项目结构

➜  py3 tree

.

├── CMakeLists.txt

├── demo.cpp

├── demo.h

└── pymodule.cpp

1. CMakeLists.txt

#CMakeLists.txt

cmake_minimum_required(VERSION 3.5) #CMake的最低版本要求

project(demo)

set(CMAKE_CXX_STANDARD 11)

# 添加Python包

find_package(PythonLibs 3.5 REQUIRED)

include_directories(${PYTHON_INCLUDE_DIR})

# 添加Boost包

find_package(Boost COMPONENTS python3 REQUIRED)

include_directories(${Boost_INCLUDE_DIR})

# 生成动态链接库

set(PYMODULE "cpp_py_module")

set(SOURCE_FILES pymodule.cpp demo.cpp demo.h)

add_library(${PYMODULE} SHARED ${SOURCE_FILES})

target_link_libraries(${PYMODULE} ${Boost_LIBRARIES})

set_target_properties(${PYMODULE} PROPERTIES PREFIX "") # 去掉前缀，否则会生成lib***.so

2. demo.h

//demo.h

#ifndef DEMO_H_

#define DEMO_H_

#include <string>

class Person

{

  public:

    Person(std::string name, int age);

    std::string getName();

    int getAge();

  private:

    std::string name;

    int age;

};

double add(double a, double b);

#endif // DEMO_H_

3. demo.cpp

//demo.cpp

#include "demo.h"

Person::Person(std::string name, int age)

{

    this->name = name;

    this->age = age;

}

std::string Person::getName()

{

    return name;

}

int Person::getAge()

{

    return age;

}

double add(double a, double b)

{

    return a + b;

}

4. pymodule.cpp

#include "demo.h"

#include <boost/python.hpp>

// module名要和CMakeLists.txt中生成的so目标名一致

BOOST_PYTHON_MODULE(cpp_py_module)

{

    using namespace boost::python;

    class_<Person>("Person",

                   init<std::string, int>()) // 构造函数

        .def("getName", &Person::getName)  // 成员函数

        .def("getAge", &Person::getAge);

    def("add", add); // 普通函数

}

测试

编译生成动态链接库

cmake .

make

会在当前目录下生成cpp_py_module.so

在当前目录下用python3测试模块

In [1]: import cpp_py_module

In [2]: cpp_py_module.add(4, 5)

Out[2]: 9.0

In [3]: p = cpp_py_module.Person("Jack", 25)

In [4]: p.getName()

Out[4]: 'Jack'

In [5]: p.getAge()

Out[5]: 25

小结

Python作为胶水语言真是名副其实，这样就可以用Python解决输入格式之类的dirty work，然后C++解决密集计算．

使用Boost还有一个好处就是不用修改原来的Ｃ++代码，只要外面加一层pymodule.cpp的wrapper就好了，简洁明了．