如何用pybind11包装模板化的类

时间:2017-07-05 11:32:42

标签: python c++ templates pybind11

我有一个类似于下面的类结构,其中我有两个类型A和B,它们具有相似的签名,只在参数/返回类型上有所不同。然后我使用类模板来处理这两个类,在Python中实现了ducktyping的静态变体。现在我想用pybind11将这段代码包装到Python中,我希望得到一个使用标准的动态ducktyping来获取两种类型的类。我该怎么做?

我基本上是在寻找一种方法来禁用pybind11中的严格类型检查,或指定多种类型,以便接受TypeA和TypeB。定义下面示例的方式,只有TypeA通过检查。由于功能签名不同,我也无法将A和B统一为基类。

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

class TypeA {
public:
    typedef double InType;
    typedef std::string OutType;
    const OutType operator()(const InType arg)
    {
        return std::to_string(arg);
    }
};

class TypeB {
public:
    typedef std::string InType;
    typedef double OutType;
    const OutType operator()(const InType arg)
    {
        return std::stod(arg);
    }
};

template<typename T>
class DuckType {
public:
    void runType(const typename T::InType arg)
    {
        T x;
        const typename T::OutType y = x(arg);
        std::cout << y << std::endl;
    }
};

namespace py = pybind11;

PYBIND11_PLUGIN(ducktyping) {

    pybind11::module m("ducktyping", "Testing ducktyping with templates");

    typedef DuckType<TypeA> Type;
    py::class_<Type>(m, "DuckType")
    .def("run_type", &Type::runType);
    ;
    return m.ptr();
}

2 个答案:

答案 0 :(得分:0)

如果要绑定的函数具有可以在C ++中“驱动”推理的参数,则可以潜在地使用它们来驱动pybind11中的重载分辨率: https://pybind11.readthedocs.io/en/stable/advanced/functions.html#overload-resolution-order

例如,您可以将实例化绑定为重载,然后让pybind11进行计算:

// Let's say `DuckType::runType` is static.
m.def("runType", &DuckType<TypeA>::runType, py::arg("arg"));
m.def("runType", &DuckType<TypeB>::runType, py::arg("arg"));

然后,假设用户可以实例化这些类型,则调度将在Python中工作,并且InType被定义为不会偶然拦截另一个类的类型:

runType(arg=1.2)  # TypeA
runType(arg="hello world")  # TypeB

答案 1 :(得分:0)

可以在以下 github 问题中找到示例

https://github.com/pybind/pybind11/issues/199#issuecomment-323995589

我会复制这里的例子来完成

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

namespace py = pybind11;

template <typename T>
class MyClass : public SuperClass
{
 MyClass(int foo)   {}
 void func (int baa)
}

template <typename T>
void MyClass<T> :: func(int baa) {}

PYBIND11_MODULE(example, m) {
py::class_< MyClass<double>, shared_ptr< MyClass<double> >, SuperClass>(m, "MyClass")
 .def(py::init<int>())
 .def("func", &MyClass<double>::func);
}