当前位置:首页 > 通信资讯 > 正文

现有若干用C++编写的函数,为了能够用Python调用它们,可以通过pybind11将它们封装到一个Python模块中,在编写Python程序时只要导入该模块便能使用这些函数。

一.实现步骤

  • 用C++函数实现函数功能;
  • 编写头文件,将函数的声明包含在头文件中;
  • 编写pybind11封装函数;
  • 用Python编写setup脚本;
  • 生成动态链接库pyd文件;
  • 测试函数功能。

用pybind11封装c++实现的函数库的方法示例(用pybind11封装c++实现的函数库的方法示例)

二.代码实现

1.用C++函数实现函数功能

  1. //文件名:add.cpp
  2. extern "C" double add(double in1, double in2)
  3. {
  4. return in1 + in2;
  5. }
  1. //文件名:sub.cpp
  2. extern "C" double sub(double in1, double in2)
  3. {
  4. return in1 - in2;
  5. }
  1. //文件名:mul.cpp
  2. extern "C" double mul(double in1, double in2)
  3. {
  4. return in1 * in2;
  5. }
  1. //文件名:div.cpp
  2. extern "C" double div1(double in1, double in2)
  3. {
  4. return in1 / in2;
  5. }

2.编写头文件,将函数的声明包含在头文件中

  1. //文件名:functions.h
  2. #ifndef FUNCTIONS_H
  3. #define FUNCTIONS_H
  4. #include <iostream>
  5. using namespace std;
  6. extern "C" double add(double, double);
  7. extern "C" double sub(double, double);
  8. extern "C" double mul(double, double);
  9. extern "C" double div1(double, double);
  10. #endif

3.编写pybind11封装函数

  1. //文件名:functions_wrapper.cpp
  2. #include <pybind11/pybind11.h>
  3. #include "functions.h"
  4. namespace py = pybind11;
  5. PYBIND11_MODULE(functions, m){
  6. m.doc() = "Simple Class";
  7. m.def("add", &add);
  8. m.def("sub", &sub);
  9. m.def("mul", &mul);
  10. m.def("div", &div1);
  11. }

4.用Python编写setup脚本

  1. #文件名:setup.py
  2. from setuptools import setup, Extension
  3. functions_module = Extension(
  4. name = 'functions',
  5. sources = ['add.cpp', 'sub.cpp', 'mul.cpp', 'div.cpp', 'functions_wrapper.cpp'],
  6. include_dirs = [r'D:\software\pybind11-master\include',
  7. r'D:\software\Anaconda\include']
  8. )
  9. setup(ext_modules = [functions_module])

5.生成动态链接库pyd文件

在命令行执行 python setup.py build_ext --inplace ,在当前路径下生成pyd动态库。

用pybind11封装c++实现的函数库的方法示例(用pybind11封装c++实现的函数库的方法示例)

用pybind11封装c++实现的函数库的方法示例(用pybind11封装c++实现的函数库的方法示例)

6.测试函数功能

  1. #文件名:test.py
  2. import functions
  3. print(functions.add(0.9, 3))
  4. print(functions.sub(0.9, 3))
  5. print(functions.mul(0.9, 3))
  6. print(functions.div(0.9, 3))

output:
3.9
-2.1
2.7
0.3

到此这篇关于用pybind11封装C++实现的函数库的方法实现的文章就介绍到这了,更多相关pybind11封装C++实现函数库内容请搜索服务器之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持服务器之家!

原文链接:https://segmentfault.com/a/1190000021820775

如果您对该产品感兴趣,请填写办理(客服微信:xiaoxiongyidong)

为您推荐:

发表评论

◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。