快捷導(dǎo)航

用pybind11封裝C++實現(xiàn)的函數(shù)庫的方法示例

更新時間：2020年02月24日 08:26:08 作者：追憶

這篇文章主要介紹了用pybind11封裝C++實現(xiàn)的函數(shù)庫，文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧

現(xiàn)有若干用C++編寫的函數(shù)，為了能夠用Python調(diào)用它們，可以通過pybind11將它們封裝到一個Python模塊中，在編寫Python程序時只要導(dǎo)入該模塊便能使用這些函數(shù)。

一.實現(xiàn)步驟

用C++函數(shù)實現(xiàn)函數(shù)功能；
編寫頭文件，將函數(shù)的聲明包含在頭文件中；
編寫pybind11封裝函數(shù)；
用Python編寫setup腳本；
生成動態(tài)鏈接庫pyd文件；
測試函數(shù)功能。

二.代碼實現(xiàn)

1.用C++函數(shù)實現(xiàn)函數(shù)功能

//文件名：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.編寫頭文件，將函數(shù)的聲明包含在頭文件中

//文件名：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封裝函數(shù)

//文件名：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.生成動態(tài)鏈接庫pyd文件

在命令行執(zhí)行 python setup.py build_ext --inplace ,在當前路徑下生成pyd動態(tài)庫。

6.測試函數(shù)功能

#文件名：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

到此這篇關(guān)于用pybind11封裝C++實現(xiàn)的函數(shù)庫的方法實現(xiàn)的文章就介紹到這了,更多相關(guān)pybind11封裝C++實現(xiàn)函數(shù)庫內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: