时间戳

首页 > 解决方案 > Pybind - 使用指向派生类的共享指针调用函数

python - Pybind - 使用指向派生类的共享指针调用函数

问题描述

我有以下设置（1个基类，1个派生类，1个容器）。容器将 ashared_ptr<Base>作为输入。

#include <pybind11/pybind11.h>
namespace py = pybind11;

struct Base { };
struct Derived : public Base { };

struct Container { void input(const std::shared_ptr<Base>& ptr) { } };

PYBIND11_MODULE(PybindTest, m)
{
    py::class_<Base,    std::shared_ptr<Base>>(m, "Base").def(py::init<>());
    py::class_<Derived, std::shared_ptr<Derived>>(m, "Derived").def(py::init<>());

    py::class_<Container, std::shared_ptr<Container>>(m, "Container")
        .def(py::init<>())
        .def("input", &Container::input);
}

在 C++ 中，我可以传递shared_ptr<Base>或传递shared_ptr<Derived>给input函数。但是在 Python 中我得到一个错误：

import PybindTest as p
p.Container().input(p.Base())      # All good
p.Container().input(p.Derived())   # Throws Error

# TypeError                                 Traceback (most recent call last)
# <ipython-input-10-70fe5b9f3a41> in <module>
#       1 import PybindTest as p
#       2 p.Container().input(p.Base())
# ----> 3 p.Container().input(p.Derived())
# 
# TypeError: input(): incompatible function arguments. The following argument types are supported:
#     1. (self: PybindTest.Container, arg0: PybindTest.Base) -> None
# 
# Invoked with: <PybindTest.Container object at 0x0000022378B4FF80>, <PybindTest.Derived object at 0x0000022378B4FCE0>

我试过玩弄类似的东西

.def("input", py::overload_cast<const std::shared_ptr<Derived>&> (&Container::input))
.def("input", [](const std::shared_ptr<Derived> & ptr) { this->input(ptr); })

但是这两个都没有编译。有什么建议吗？

我使用带有 Python 3.6 x64 的 Windows 10 并使用 VS 2019 编译所有内容。

标签： pythonc++python-3.xc++17pybind11

解决方案

正如@nm 建议的那样：

py::class_<Derived, std::shared_ptr<Derived>, Base>(m, "Derived")
    .def(py::init<>());

来自pybind11 文档（方法一：模板参数）

推荐阅读