python与C++交互(自动绑定)

前言


上一篇研究了通过Python的内置库ctypes实现与C库的交互。详情参考python与C库交互。但是这种方法存在一个问题就是C的类型转换必须手动去映射,非常容易出错,而且会增加工作量。

本文通过一种新的思路实现类型的自动绑定,那就是pybind11。

python与C++交互(自动绑定)_第1张图片

pybind11简介


pybind11可以实现C++11和Python之间的无缝连接。pybind11是一个轻量级的头文件库,它在 Python 中公开 C++ 类型,反之亦然,主要用于创建现有 C++ 代码的 Python 绑定。

pybind11 可以将以下核心 C++ 功能映射到 Python:

  • 函数接受和返回每个值、引用或指针的自定义数据结构

  • 实例方法和静态方法

  • 重载函数

  • 实例属性和静态属性

  • 任意异常类型

  • 枚举

  • 回调

  • 迭代器和范围

  • 自定义运算符

  • 单继承和多继承

  • STL 数据结构

  • 具有引用计数的智能指针,例如std::shared_ptr

  • 具有正确引用计数的内部引用

  • 可以在 Python 中扩展具有虚拟(和纯虚拟)方法的 C++ 类

github地址:

https://github.com/pybind/pybind11

python与C++交互(自动绑定)_第2张图片

接下来通过几个示例,来介绍强大的pybind11

示例一:加法


C++代码通过cmake形式进行组织。

cmake文件:

cmake_minimum_required(VERSION 3.0.0)
project(MyPyBind VERSION 0.1.0)


add_subdirectory(extern/pybind11-2.9.2)
pybind11_add_module(MyPyBind MyPyBind.cpp)

C++代码

#include 
int add(int i, int j)
{
  return i + j;
}


PYBIND11_MODULE(MyPyBind, m)
{
  m.def("add", &add);
}

python代码

import MyPyBind
ret = MyPyBind.add(1, 2)
print(ret)

说明:

1、cmake文件需要做一些修改,即将之前的add_library方法换成pybind11_add_module方法,另外还需要引入pybind11的目录,通过add_subdirectory。

2、在C++代码中,需要包含pybind11的头文件,之后便可以进行绑定。

3、C++业务代码和之前一模一样。

4、需要手动编写需要映射的函数,PYBIND11_MODULE(MyPyBind, m)中的MyPyBind需要和cmake中的pybind11_add_module中的库名一致

5、绑定函数非常简单,固定格式m.def("add", &add);

6、调用cmake编译之后,会生成一个MyPyBind.pyd的文件,将文件拷贝到python目录中,即可调用

7、python调用非常简单,直接import之后,就可以使用。

示例二:结构体绑定


C++代码

struct ST
{
  std::string str;
  uint64_t i;
};


void say_st(ST &st)
{
  cout << st.str << endl;
  cout << st.i << endl;
}

C++绑定代码

pybind11::class_(m, "ST")
      .def(pybind11::init())
      .def_readwrite("str", &ST::str)
      .def_readwrite("i", &ST::i);


  m.def("say_st", &say_st);

python调用

st = MyPyBind.ST()
st.str = '9527'
st.i = 850
MyPyBind.say_st(st)

说明:

1、结构体绑定格式是固定的,分为两步:

.def(pybind11::init())和 .def_readwrite("str", &ST::str)。第一步是绑定结构体,第二步是绑定结构体的成员。

示例三:类绑定


C++代码

class CS
{
public:
  CS(int i, int j)
      : mI(i), mJ(j)
  {
  }


  void Print()
  {
    std::cout << "i= " << mI << " j= " << mJ << std::endl;
  }


private:
  int mI;
  int mJ;
};

绑定代码

pybind11::class_(m, "CS")
      .def(pybind11::init())
      .def("print", &CS::Print);

python调用

cs = MyPyBind.CS(1, 2)
cs.print()

说明:

1、类的绑定其实和结构体绑定格式是一致的,还可以设置带初始化的类。

2、类的函数也可以绑定

示例四:枚举绑定


C++代码

enum class Animal
{
  dog,
  cat
};


void say_Animal(Animal animal)
{
  cout << static_cast(animal) << endl;
}

绑定代码

pybind11::enum_(m, "Animal")
      .value("dog", Animal::dog)
      .value("cat", Animal::cat)
      .export_values();
      
  m.def("say_Animal", &say_Animal);

pyhon调用

MyPyBind.say_Animal(MyPyBind.Animal.cat)

说明:

1、通过pybind11::enum_来实现枚举的绑定,通过.value实现值的绑定

示例五:STL绑定


C++代码

#include 
class CSSTL
{
public:
  void Set(std::vector v)
{
    mv = v;
  }
  void Print()
{
    for (auto &item : mv)
    {
      std::cout << item << "  ";
    }
    std::cout << std::endl;
  }


private:
  std::vector mv;
};

绑定代码

pybind11::class_(m, "CSSTL")
      .def(pybind11::init())
      .def("Set", &CSSTL::Set)
      .def("Print", &CSSTL::Print);

python调用

csStl = MyPyBind.CSSTL()
csStl.Set([4, 5, 6])
csStl.Print()

说明:

1、stl的绑定,基本上是自动实现,和之前类的绑定一致。

pybin11提供的自动转换包括std::vector<>/std::list<>/std::array<> 转换成 Python list ;std::set<>/std::unordered_set<> 转换成 python set ; andstd::map<>/std::unordered_map<> z转换成dict 几种

2、C++代码中需要引入一个头文件#include 

示例五:虚函数绑定


C++代码

class Room
{
public:
  virtual void RoomName() = 0;
};


void CallRoom(Room &room)
{
  room.RoomName();
}
class PyRoom : public Room
{
public:
  using Room::Room;
  void RoomName() override
{
    PYBIND11_OVERRIDE_PURE(void, Room, RoomName);
  }
};

绑定代码

pybind11::class_(m, "Room")
      .def(pybind11::init<>())
      .def("RoomName", &Room::RoomName);


  m.def("CallRoom", &CallRoom);

python调用

class PyRoom(MyPyBind.Room):
    def RoomName(self):
        print("RoomName")




room = PyRoom()
MyPyBind.CallRoom(room)

说明:

1、虚函数最大的作用可以用来实现回调,即C++调用Python的代码。

2、需要额外引入一个类,并通过固定格式PYBIND11_OVERRIDE_PURE来实现转换。

3、绑定代码,也需要引入额外引入的类,其他和之前类的绑定一致。pybind11::class_

总结


本文主要介绍了如何通过pybind11实现C++和python数据类型的自动转换。并列举了加法、结构体、类、枚枚举、虚函数5个典型示例。

获取示例代码途径:

关注公众号,加技术交流群即可获取本工程实例代码。

python与C++交互(自动绑定)_第3张图片

 

你可能感兴趣的:(c++,python,java,机器学习,cmake)