9.5 使用 pybind11 构建 C++和 Python 项目
NOTE : 此示例代码可以在 https://github.com/dev-cafe/cmake-cookbook/tree/v1.0/chapter-9/recipe-05 中找到,其中有一个 C++示例。该示例在 CMake 3.11 版(或更高版本) 中是有效的,并且已经在 GNU/Linux、macOS 和 Windows 上进行过测试。
前面的示例中,我们使用 Boost.Python 与 C(C++) 接口。本示例中,我们将尝试使用 pybind11 将 Python 与 C++接口。其实现利用了 C++11 的特性,因此需要支持 C++11 的编译器。我们将演示在配置时如何获取 pybind11 依赖和构建我们的项目,包括一个使用 FetchContent 方法的 Python 接口,我们在第 4 章第 3 节和第 8 章第 4 节中有过讨论。在第 11 章第 2 节时,会通过 PyPI 发布一个用 CMake/pybind11 构建的 C++/Python 项目。届时将重新讨论这个例子,并展示如何打包它,使它可以用 pip 安装。
准备工作
我们将保持 account.cpp
不变,只修改 account.cpp
:
#pragma once #include <pybind11/pybind11.h> class Account { public: Account(); ~Account(); void deposit(const double amount); void withdraw(const double amount); double get_balance() const; private: double balance; }; namespace py = pybind11; PYBIND11_MODULE(account, m) { py::class_<Account>(m, "Account") .def(py::init()) .def("deposit", &Account::deposit) .def("withdraw", &Account::withdraw) .def("get_balance", &Account::get_balance); }
按照 pybind11 文档的方式,通过 CMake 构建( https://pybind11.readthedocs.io/en/stable/compile )。并使用 add_subdirectory
将 pybind11 导入项目。但是,不会将 pybind11 源代码显式地放到项目目录中,而是演示如何在配置时使用 FetchContent
( https://cmake.org/cmake/help/v3.11/module/FetchContent.html )。
为了在下一个示例中更好地重用代码,我们还将把所有源代码放到子目录中,并使用下面的项目布局:
. ├── account │ ├── account.cpp │ ├── account.hpp │ ├── CMakeLists.txt │ └── test.py └── CMakeLists.txt
具体实施
让我们详细分析一下这个项目中,各个 CMakeLists.txt
文件的内容:
- 主
CMakeLists.txt
文件:# define minimum cmake version cmake_minimum_required(VERSION 3.11 FATAL_ERROR) # project name and supported language project(recipe-05 LANGUAGES CXX) # require C++11 set(CMAKE_CXX_STANDARD 11) set(CMAKE_CXX_EXTENSIONS OFF) set(CMAKE_CXX_STANDARD_REQUIRED ON)
- 这个文件中,查询了用于测试的 Python 解释器:
find_package(PythonInterp REQUIRED)
- 然后,包含
account
子目录:add_subdirectory(account)
- 定义单元测试:
# turn on testing enable_testing() # define test add_test( NAME python_test COMMAND ${CMAKE_COMMAND} -E env ACCOUNT_MODULE_PATH=$<TARGET_FILE_DIR:account> ${PYTHON_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/account/test.py )
account/CMakeLists.txt
中,在配置时获取 pybind11 的源码:include(FetchContent) FetchContent_Declare( pybind11_sources GIT_REPOSITORY https://github.com/pybind/pybind11.git GIT_TAG v2.2 ) FetchContent_GetProperties(pybind11_sources) if(NOT pybind11_sources_POPULATED) FetchContent_Populate(pybind11_sources) add_subdirectory( ${pybind11_sources_SOURCE_DIR} ${pybind11_sources_BINARY_DIR} ) endif()
- 最后,定义 Python 模块。再次使用模块选项
add_library
。并将库目标的前缀和后缀属性设置为PYTHON_MODULE_PREFIX
和PYTHON_MODULE_EXTENSION
,这两个值由 pybind11 适当地推断出来:add_library(account MODULE account.cpp ) target_link_libraries(account PUBLIC pybind11::module ) set_target_properties(account PROPERTIES PREFIX "${PYTHON_MODULE_PREFIX}" SUFFIX "${PYTHON_MODULE_EXTENSION}" )
- 进行测试:
$ mkdir -p build $ cd build $ cmake .. $ cmake --build . $ ctest Start 1: python_test 1/1 Test #1: python_test ...................... Passed 0.04 sec 100% tests passed, 0 tests failed out of 1 Total Test time (real) = 0.04 sec
工作原理
pybind11 的功能和使用与 Boost.Python 非常类似。pybind11 是一个更轻量级的依赖 - 不过需要编译器支持 C++11。 account.hpp
中的接口定义与之前的示例非常类似:
#include <pybind11/pybind11.h> // ... namespace py = pybind11; PYBIND11_MODULE(account, m) { py::class_<Account>(m, "Account") .def(py::init()) .def("deposit", &Account::deposit) .def("withdraw", &Account::withdraw) .def("get_balance", &Account::get_balance); }
同样,我们可以了解到 Python 方法是如何映射到 C++函数的。解释 PYBIND11_MODULE
库是在导入的目标 pybind11::module
中定义,使用以下代码包括了这个模块:
add_subdirectory( ${pybind11_sources_SOURCE_DIR} ${pybind11_sources_BINARY_DIR} )
与之前的示例有两个不同之处:
- 不需要在系统上安装 pybind11
${pybind11_sources_SOURCE_DIR}
子目录,包含 pybind11 的CMakelist.txt
中,在我们开始构建项目时,这个目录并不存在
这个挑战的解决方案是用 FetchContent
,在配置时获取 pybind11 源代码和 CMake 模块,以便可以使用 add_subdirectory
引用。使用 FetchContent
模式,可以假设 pybind11 在构建树中可用,并允许构建和链接 Python 模块:
add_library(account MODULE account.cpp ) target_link_libraries(account PUBLIC pybind11::module )
使用下面的命令,确保 Python 模块库得到一个定义良好的前缀和后缀,并与 Python 环境兼容:
set_target_properties(account PROPERTIES PREFIX ${PYTHON_MODULE_PREFIX} SUFFIX ${PYTHON_MODULE_EXTENSION} )
主 CMakeLists.txt
文件的其余部分,都在执行测试(与前一个示例使用相同的 test.py
)。
更多信息
我们可以将 pybind11 源代码包含在项目源代码存储库中,这将简化 CMake 结构,并消除在编译时对 pybind11 源代码进行网络访问的要求。或者,我们可以将 pybind11 源路径定义为一个 Git 子模块( https://git-scm.com/book/en/v2/Git-Tools-Submodules ),以应对 pybind11 源依赖项的更新。
在示例中,我们使用 FetchContent
解决了这个问题,它提供了一种非常紧凑的方法来引用 CMake 子项目,而不是显式地跟踪它的源代码。同样,我们也可以使用超级构建的方法来解决这个问题(参见第 8 章)。
要查看如何简单函数、定义文档注释、映射内存缓冲区等进阶阅读,请参考 pybind11 文档: https://pybind11.readthedocs.io
绑定邮箱获取回复消息
由于您还没有绑定你的真实邮箱,如果其他用户或者作者回复了您的评论,将不能在第一时间通知您!
发布评论