时间戳

首页 > 解决方案 > 如何使用pybind11将堆分配指针的`std::vector`传递给python？

python - 如何使用pybind11将堆分配指针的`std::vector`传递给python？

问题描述

假设我现在有一堆指针foo*，我想std::vector在 c++ 函数中使用它们将它们传递给 python get_foos，但我想在 c++ 中管理向量中那些堆分配指针的生命周期，即 python 应该只释放向量但不是当 c++ 向量foos超出 python 范围时的那些指针。

return_value_policy::reference似乎还不够，因为它也不会破坏向量，如果我错了，请纠正我。

struct foo {};

py::class_<foo>(m, "Foo");

auto get_foos() -> std::vector<foo*> {
  auto foos = std::vector<foo*> {};

  for(...) {
    auto p = get_foo_ptr(...);
    foos.emplace_back(p);
  }
  return foos;
}

标签： pythonc++pybind11

解决方案

您必须包含一个辅助标头<pybind11/stl.h>。

这是一个例子：


struct Foo {
    Foo(const char *name) : m_name(name) {}
    ~Foo() {}

    const char *get_name() const { return m_name.c_str(); }

private:
    string m_name;
};

vector<unique_ptr<Foo>> make_foos() {
    vector<unique_ptr<Foo>> foos;
    foos.push_back(make_unique<Foo>("Hello"));
    foos.push_back(make_unique<Foo>("World"));
    return foos;
}

vector<Foo *> get_foos() {
    static vector<unique_ptr<Foo>> all_foos = make_foos();

    vector<Foo *> foos;
    for (unique_ptr<Foo> &foo : all_foos) {
        foos.push_back(foo.get());
    }
    return foos;
}


PYBIND11_MODULE(example, m) {
    py::class_<Foo>(m, "Foo")
            .def("get_name", &Foo::get_name);

    m.def("get_foos", get_foos, py::return_value_policy::reference);
}

推荐阅读