小隐的博客

人生在世,笑饮一生
  博客园  :: 首页  :: 新随笔  :: 联系 :: 订阅 订阅  :: 管理

使用pybind11封装c++的dll,供python调用

Posted on 2024-07-26 10:36  隐客  阅读(35)  评论(0编辑  收藏  举报

用pip安装好pybind11  文件清单,都写在一个目录里

//文件名: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; 
}


//文件名: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


//文件名: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); 
}



#文件名: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])



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





注意替换setup.py中相关库的位置
在命令行执行 python setup.py build_ext --inplace
编译好之后会有一个pyd文件
然后测试test.py即可



另外,也可以直接在 vs中写好相关cpp的文件,配置好目录,然后生成dll,改成pyd即可。
QQ交流