Возможно ли в pybind11 использовать py :: cast для доступа к абстрактному базовому классу? - PullRequest
Новая
       36

Возможно ли в pybind11 использовать py :: cast для доступа к абстрактному базовому классу?

0 голосов
/ 08 ноября 2019

Ниже приведен минимальный рабочий пример - его можно скомпилировать, используя типичные инструкции pybind11 (я использую cmake).

У меня есть абстрактный базовый класс Abstract, который является чисто виртуальным. Я могу легко обернуть это в pybind11, используя «класс батута» (это хорошо задокументировано pybind11).

Далее, у меня есть конкретная реализация Abstract, ToBeWrapped, которая также упакована с использованием pybind11.

Моя проблема в том, что у меня есть некоторый клиентский код, который принимает произвольный PyObject* (или, в случае этого примера, оболочку pybind11 py::object) и ожидает приведения его к Abstract*.

Однако, как показано в моем примере, я не могу привести py::object к Abstract*.

У меня нет проблем с приведением к ToBeWrapped*, а затем сохранить его как Abstract*', however this would require my client code to know ahead of time what kind of Abstract * `отправляет интерпретатор Python, который отрицает назначение абстрактного базового класса.

TL; DR

Можно ли изменить этот код так, чтобыклиент accessMethod может произвольно обрабатывать Abstract*, переданный от интерпретатора Python? 

#include <pybind11/pybind11.h>
#include <iostream>

namespace py = pybind11;

// abstract base class - cannot be instantiated on its own
class Abstract
{
    public:
        virtual ~Abstract() = 0;
        virtual std::string print() const = 0;
};

Abstract::~Abstract(){}

// concrete implementation of Abstract
class ToBeWrapped : public Abstract
{
    public:
        ToBeWrapped(const std::string& msg = "heh?")
            : myMessage(msg){};
        std::string print() const override
        {
            return myMessage;
        }

    private:
        const std::string myMessage;
};

// We need a trampoline class in order to wrap this with pybind11
class AbstractPy : public Abstract
{
    public:
        using Abstract::Abstract;

        std::string print() const override
        {
            PYBIND11_OVERLOAD_PURE(
                std::string, // return type
                Abstract, // parent class
                print, // name of the function
                // arguments (if any)
            );
        }
};

// I have client code that accepts a raw PyObject* - this client code base implements its
// own python interpreter, and calls this "accessMethod" expecting to convert the python
// object to its c++ type.
//
// Rather than mocking up the raw PyObject* method (which would be trivial) I elected to
// keep this minimal example 100% pybind11
void accessMethod(py::object obj)
{
    // runtime error: py::cast_error
    //Abstract* casted = obj.cast<Abstract*>();

    // this works
    Abstract* casted = obj.cast<ToBeWrapped*>();
}

PYBIND11_MODULE(PyMod, m)
{
    m.doc() = R"pbdoc(
    This is a python module
    )pbdoc";

    py::class_<Abstract, AbstractPy>(m, "Abstract")
        .def("print", &Abstract::print)
    ;

    py::class_<ToBeWrapped>(m, "WrappedClass")
        .def(py::init<const std::string&>())
    ;
    m.def("access", &accessMethod, "This method will attempt to access the wrapped type");
}

1 Ответ

1 голос
/ 08 ноября 2019

Вам необходимо объявить иерархические отношения, так что: 

py::class_<ToBeWrapped>(m, "WrappedClass")

должно быть: 

py::class_<ToBeWrapped, Abstract>(m, "WrappedClass")
...