这篇文章将为大家详细讲解有关使用用pybind11封装C++实现函数库的案例,小编觉得挺实用的,因此分享给大家做个参考,希望大家阅读完这篇文章后可以有所收获。
现有若干用C++编写的函数,为了能够用Python调用它们,可以通过pybind11将它们封装到一个Python模块中,在编写Python程序时只要导入该模块便能使用这些函数。
一.实现步骤
用C++函数实现函数功能;
编写头文件,将函数的声明包含在头文件中;
编写pybind11封装函数;
用Python编写setup脚本;
生成动态链接库pyd文件;
测试函数功能。
二.代码实现
1.用C++函数实现函数功能
//文件名:add.cpp
extern "C" double add(double in1, double in2)
{
return in1 + in2;
}
//文件名:sub.cpp
extern "C" double sub(double in1, double in2)
{
return in1 - in2;
}
//文件名:mul.cpp
extern "C" double mul(double in1, double in2)
{
return in1 * in2;
}
//文件名:div.cpp
extern "C" double div1(double in1, double in2)
{
return in1 / in2;
}
2.编写头文件,将函数的声明包含在头文件中
//文件名:functions.h
#ifndef FUNCTIONS_H
#define FUNCTIONS_H
#include <iostream>
using namespace std;
extern "C" double add(double, double);
extern "C" double sub(double, double);
extern "C" double mul(double, double);
extern "C" double div1(double, double);
#endif
3.编写pybind11封装函数
//文件名:functions_wrapper.cpp
#include <pybind11/pybind11.h>
#include "functions.h"
namespace py = pybind11;
PYBIND11_MODULE(functions, m){
m.doc() = "Simple Class";
m.def("add", &add);
m.def("sub", &sub);
m.def("mul", &mul);
m.def("div", &div1);
}
4.用Python编写setup脚本
#文件名:setup.py
from setuptools import setup, Extension
functions_module = Extension(
name = 'functions',
sources = ['add.cpp', 'sub.cpp', 'mul.cpp', 'div.cpp', 'functions_wrapper.cpp'],
include_dirs = [r'D:\software\pybind11-master\include',
r'D:\software\Anaconda\include']
)
setup(ext_modules = [functions_module])
5.生成动态链接库pyd文件
在命令行执行 python setup.py build_ext --inplace
,在当前路径下生成pyd动态库。
6.测试函数功能
#文件名:test.py
import functions
print(functions.add(0.9, 3))
print(functions.sub(0.9, 3))
print(functions.mul(0.9, 3))
print(functions.div(0.9, 3))
output:
3.9
-2.1
2.7
0.3
关于“使用用pybind11封装C++实现函数库的案例”这篇文章就分享到这里了,希望以上内容可以对大家有一定的帮助,使各位可以学到更多知识,如果觉得文章不错,请把它分享出去让更多的人看到。
亿速云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。