pybind11-回傳std::vector的shared

std::shared_ptr我有一個存盤 a of的成員變數std::vector<uint32_t>。我想創建一個 Python 系結，test_func2()以便無需任何額外副本即可訪問該向量。這是一個骨架代碼。

#include <vector>
#include <memory>
#include <pybind11/pybind11.h>
#include <pybind11/stl.h>
#include <pybind11/numpy.h>

namespace py = pybind11;



class TestLoader
{
private:
    std::shared_ptr<std::vector<uint32_t>>  tileData;

public:
    TestLoader();
    ~TestLoader();
    void test_func1();
    std::shared_ptr<std::vector<uint32_t>> test_func2()  const;
};

void  TestLoader::test_func1() {
    tileData = std::make_shared<std::vector<uint32_t>>(100000000);
    for(auto &x: *tileData){ x = 1;}
}
std::shared_ptr<std::vector<uint32_t>>  TestLoader::test_func2() const{

    return tileData;
}

介面代碼如下：

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

namespace py = pybind11;
PYBIND11_MODULE(fltest_lib, m) {
    py::class_<TestLoader,  std::shared_ptr<TestLoader>>(m, "TestLoader")
    .def(py::init<const std::string &>())
    .def("test_func1", &TestLoader::test_func1)
    .def("test_func2", &TestLoader::test_func2, py::return_value_policy::reference_internal);
}

但是，這無法編譯，并且我收到一條很長的錯誤訊息。一個特定的行如下：

/home/samee/fl_test/lib/pybind11/include/pybind11/cast.h:653:61: error: static assertion failed: Holder classes are only supported for custom types
  653 |     static_assert(std::is_base_of<base, type_caster<type>>::value,
      |                                                             ^~~~~

任何可以避免這種情況的幫助都將非常有幫助。

uj5u.com熱心網友回復：

根據這個問題，它不起作用，因為std::vector<uint32_t>沒有轉換為python型別。因此，您將不得不回傳取消參考的向量。為避免復制，您可以使用PYBIND11_MAKE_OPAQUE

#include <pybind11/pybind11.h>
#include <pybind11/stl.h>
#include <pybind11/stl_bind.h>

#include "test_loader.h"

namespace py = pybind11;

PYBIND11_MAKE_OPAQUE(std::vector<uint32_t>);

PYBIND11_MODULE(fltest_lib, m) {
  py::bind_vector<std::vector<uint32_t>>(m, "VectorUInt32");
  py::class_<TestLoader, std::shared_ptr<TestLoader>>(m, "TestLoader")
      .def(py::init())
      .def("test_func1", &TestLoader::test_func1)
      .def("test_func2",
           [](const TestLoader& tl) -> const std::vector<uint32_t>& {
             return *tl.test_func2();
           }, py::return_value_policy::reference_internal);
}

轉載請註明出處，本文鏈接：https://www.uj5u.com/qukuanlian/425971.html

標籤：C c 11 pybind11

上一篇：用于同步訪問向量單元的互斥體向量

下一篇：使std::ostream遇到特殊字符時自動識別

pybind11-回傳std::vector的shared_ptr