使用 pybind11 混合类型转换和不透明类型

Question

我正在使用 pybind11 的不透明类型。例如，我为 stl 容器定义了一个std::vector<uint32_t>，它是方法的参数类型FromVec：

void FromVec(std::vector<uint32_t> vec);

PYBIND11_MAKE_OPAQUE(std::vector<uint32_t>);

PYBIND11_MODULE(tmap, m)
{
    py::bind_vector<std::vector<uint32_t>>(m, "VectorUint", 
      "Unsigned 32-bit int vector.");

    m.def("from_vec", &FromVec, py::arg("vec")
}

有了这个，我现在可以（在 Python 中）执行以下操作：

vec = VectorUint([2, 66, 262, 662, 26, 62])
from_vec(vec)

但是，类型转换不再起作用，因为该函数现在需要 aVectorUint并且不再接受list，例如：

l = [2, 66, 262, 662, 26, 62]
from_vec(l)

有没有一种方法可以同时允许倾斜类型和类型转换？也就是说，当list传递 a 而不是倾斜绑定时VectorUint，它会转换为 astd::vector<uint32_t>而不是 Python 抛出“不兼容的函数参数”？

Answer 1

只需在您的之后添加此py::bind_vector：

py::implicitly_convertible<py::list, std::vector<uint32_t>>();

它甚至可以与属性一起使用（例如使用创建def_readwrite）。您将获得两全其美的效果：向量是通过 python 的引用访问的，因此append、extend等可以按预期工作；你可以从 python 列表中分配（当然会被复制）。

编辑：不幸的是这个技巧不起作用std::map（不知道其他容器）。作为解决方法，您可以使用不同类型的中间映射，例如：

namespace pybind11::workaround
{
  template<typename K, typename V>
  class Map : public std::map<K, V> {};
}

namespace pybind11::detail
{
  template<typename K, typename V>
  struct type_caster<workaround::Map<K, V>> : map_caster<workaround::Map<K, V>, K, V> {};
}

inline void bindCommonTypes(pybind11::handle scope)
{
  using IntMap = std::map<int, int>;
  py::bind_map<IntMap>(scope, "IntMap")
  .def(py::init([](const py::dict& dict) {
    return dict.cast<pybind11::workaround::Map<int, int>();
  }));
  py::implicitly_convertible<py::dict, IntMap>();
}