使用 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)

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

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

有没有办法同时允许倾斜类型和类型转换？也就是说，当传递list而不是倾斜绑定VectorUint时，它被转换为std::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>();
}

Answer 2

如果需要从list自动转换为std::vector，则可以将此类重载添加到绑定代码中：

m.def("from_vec", &FromVec, py::arg("vec")); // accepts VectorUnit 
m.def("from_vec", [](py::list& list){ /* do conversion manually and call FromVec */ } ); // accepts python list