pybind11：将c ++类（带有现有的python绑定）返回给python

Question

我正在尝试为c ++方法创建一个包装器，它返回一个来自外部c ++库（vtk）的c ++类（vtkPolyData）。同一个库有python绑定可用，它已经安装在我的python环境中。你怎么告诉pybind c ++类（vtkPolydata）和它的python变种是一样的？

我试着用这个custom type caster macro。但我得到TypeError：Unable to convert function return value to a Python type! The signature was : (self: Versa3dLib.skeletonizer, offset distance: float) -> vtkPolyData

这是令人困惑的，因为看起来转换映射到正确的类型但python无法解释它。所以我不确定是什么问题，因为我也没看到宏有任何问题。我注意到在python vtkPolyData中有类型vtkCommonDataModelPython.vtkPolyData。是为什么转换没有正确完成？

#include "skeletonizer.h"
#include <pybind11/pybind11.h>
#include <pybind11/stl.h>
#include "PybindVTKTypeCaster.h"
#include <vtkSmartPointer.h>

namespace py = pybind11;

PYBIND11_VTK_TYPECASTER(vtkPolyData)
PYBIND11_DECLARE_HOLDER_TYPE(T, vtkSmartPointer<T>);

namespace pybind11 { namespace detail {
    template <typename T>
    struct holder_helper<vtkSmartPointer<T>> { // <-- specialization
        static const T *get(const vtkSmartPointer<T> &p) { return p.GetPointer(); }
    };
}}

PYBIND11_MODULE(Versa3dLib, m)
{
    py::class_<skeletonizer>(m, "skeletonizer")
        .def(py::init<vtkPolyData *>())
        .def("get_offset", &skeletonizer::get_offset,
             "return vtkPolyData offset",
             py::arg("offset distance"));
}

Skeletonizer

#ifndef SKELETONIZER_H
#define SKELETONIZER_H

#include <vtkPolyData.h>
#include <vector>
#include <vtkSmartPointer.h>


using namespace std;

class skeletonizer
{
    public:
        skeletonizer(vtkPolyData* data);
        vtkSmartPointer<vtkPolyData> get_offset(double dist);
};

#endif

捣碎器cpp

#include "skeletonizer.h"


skeletonizer::skeletonizer(vtkPolyData* data)
{
};

vtkSmartPointer<vtkPolyData> skeletonizer::get_offset(double dist)
{
    vtkSmartPointer<vtkPolyData> offsets = vtkSmartPointer<vtkPolyData>::New();

    return offsets;
};

Answer 1

我认为这应该是一个更通用的解决方案（希望更容易使用？）：

• vtk_pybind (README)
• vtk_pybind.h
• example binding code
• example python test using bindings

我相信这应该是对VTK代码的改进：

• 使用SFINAE推广类型脚轮（而不是要求显式实例化......）。
• 允许直接投射vtkSmartPointer和vtkNew（假设其中的类型是VTK类型）。

使代码类型跟随Drake's C++ + Python binding conventions。

对于上面的解决方案，我认为它是利用SMTK代码，但持有人类型的瞬时是不正确的 - 你需要智能指针的type_caster专业化（我发布的vtk_pybind代码将提供）。

我会看看我是否可以在SMTK上发布一个问题，看看他们是否想要改进/简化他们的绑定代码（特别是如果人们引用它的话！）。

编辑：在这里发布的问题：https://gitlab.kitware.com/cmb/smtk/issues/228