本文为李你干嘛原创,转载请注明出处:Pybind11绑定C++抽象类(DLL接口)摘要

假设我们将DLL中的接口封装成了C++抽象类,并将该类和DLL文件提供给用户,类似于抽象类导出DLL中描述的办法,如果这个时候我们想使用pybind11绑定这个C++抽象类,会遇到报错,如抽象类无法实例化等等,此时Pybind11给出了辅助类的办法overriding-virtual-functions-in-python,但是如果只想转换C++抽象类的一部分的话,这个方案是不适用的。Pybind11有个很强大的功能,如果我们将C++类使用py::class绑定后,那么C++暴露给python的这个类会自动转换成Python的类。如果我们要大量的在Python中使用到这个C++抽象类且接触不到其基类时,就没有办法完成这个抽象类的绑定。

在这里我们给出一个解决思路,即用Wrapper类将C++的抽象类封装,并对这个类使用pybind绑定,这样我们就有了一个Python端的Wrapper类。再根据官网给出的办法Custom Type Casters实现从Python端Wrapper类到C++抽象类和从C++抽象类到Python端Wrapper类的自动转换。这样当C++暴露给Python这个抽象类时,pybind会自动调用转换器将抽象类转换成Wrapper类的Python对象,当Python的Wrapper类传递给C++时,会将Wrapper类变成C++抽象类。

问题描述

假设我们将C++抽象类AbstractDLLInterface作为DLL接口,ConcreteDLLInterface1类作为具体实现,但是并不把它暴露给用户。

#ifdef MYDLL_EXPORTS#define MYDLL_API __declspec(dllexport)#else#define MYDLL_API __declspec(dllimport)#endif// 抽象类接口class MYDLL_API AbstractDLLInterface {public:    virtual ~AbstractDLLInterface() {}    virtual void dllFunction() = 0;    virtual AbstractDLLInterface* createInstance() const = 0;};// 具体的实现类1class ConcreteDLLInterface1 : public AbstractDLLInterface {public:    void dllFunction() override;    AbstractDLLInterface* createInstance() const override;};// 在实现文件中提供具体实现void ConcreteDLLInterface1::dllFunction() {    // DLL 接口函数的具体实现    // ...}AbstractDLLInterface* ConcreteDLLInterface1::createInstance() const {    return new ConcreteDLLInterface1();}

现在我们只有AbstractDLLInterface类的声明和一个DLL文件,我们的C++代码中需要经常使用AbstractDLLInterface类作为返回值或者函数参数,而我们需要把这一部分用Pybind绑定。下面给出解决方案。

解决方案创建Wrapper类

class Wrapper {public:    Wrapper(AbstractDLLInterface* instance) : instance_(instance) {}    void dllFunction() {        instance_->dllFunction();    }    AbstractDLLInterface* instance_;};

定义AbstractDLLInterface类的type_caster

namespace PYBIND11_NAMESPACE {    namespace detail {        template  struct type_caster {        public:            PYBIND11_TYPE_CASTER(AbstractDLLInterface, const_name("AbstractDLLInterface"));            /**             * Conversion part 1 (Python -> C++): convert a PyObject into an AbstractDLLInterface  */            bool load(handle src, bool) {                Wrapper wrapper = py::cast(src);                value = *(wrapper.instance_);                return true;            }            /**             * Conversion part 2 (C++ -> Python): convert an AbstractDLLInterface into a PyObject             */            static handle cast(AbstractDLLInterface src, return_value_policy policy, handle parent) {                std::shared_ptr wrapper_ptr = std::make_shared(&src);                return type_caster<std::shared_ptr>::cast(wrapper_ptr, py::return_value_policy::take_ownership, parent);            }        };    }} // namespace PYBIND11_NAMESPACE::detail

Pybind中的处理思路无非是在绑定好的类、函数上,在python遇到定义过的类或者类型等就将其从C++的类包装成python的类,在python端有参数要传递给C++就将参数从Python类转换成C++类。上面的代码就实现了这个过程,我们将Python中的Wrapper类与C++中的AbstractDLLInterface类视为等效的,那么如果有Python中的Wrapper类需要传入到C++中时,会调用load将AbstractDLLInterface类实例从Wrapper类中提取出来,如果C++中有AbstractDLLInterface类实例要传入到Python中时,会调用cast新建一个Wrapper实例,再将这个Wrapper实例转换成Python对象传入到Python空间中。

实际操作用大多数不会用到AbstractDLLInterface而是AbstractDLLInterface的智能指针,此处是对AbstractDLLInterface进行转换,但是处于安全性考虑最好对std::shared_ptr类型进行转换。

绑定Pybind

// 绑定代码PYBIND11_MODULE(my_module, m) {    py::class_<Wrapper, std::shared_ptr>(m, "Wrapper")        .def(py::init())        .def("dllFunction", &Wrapper::dllFunction);}

注意此时我们不需要绑定AbstractDLLInterface类,绑定AbstractDLLInterface类编译时会报错。py::class_传入参数Wrapper, std::shared_ptr可以保证

本文为李你干嘛原创,转载请注明出处:Pybind11绑定C++抽象类(DLL接口)