pybind11自定义类型分段错误

2024-09-30 02:35:59 发布

您现在位置:Python中文网/ 问答频道 /正文

我尝试用pybind11编写python绑定。因为我还需要非简单的python类型,所以我必须创建自定义类型caster。我的问题是每当我打电话 get_point从python返回LPoint2d,但以下命令将导致分段错误

据我所知,python引用计数可能会导致该错误。但由于我对pybind11还很陌生,所以我不确定。我如何避免这个错误

在下面的示例中,我使用了来自panda3dLPoint2d作为示例类型。但其他类型(如cv2.KeyPoint)也会发生错误

Python:

>>> from panda3d.core import LPoint2d
>>> import test_module as tm
>>> foo = tm.Test()
>>> foo.get_point()
LPoint2d(0, 0)
>>> bar = 42
Segmentation fault (core dumped)

C++代码:

#include <pybind11/pybind11.h>

#include <panda3d/lpoint2.h>

class Test
{
public:
    Test(){}

    LPoint2d getPoint(){
        return m_point;
    }

    void setPoint(LPoint2d &p){
        m_point = p;
    }

private:
    LPoint2d m_point;
};

namespace py = pybind11;

namespace pybind11 { namespace detail {

template <> struct type_caster<LPoint2d> {
    public:

    PYBIND11_TYPE_CASTER(LPoint2d, _("LPoint2d"));

        bool load(handle src, bool) {
            value = LPoint2d(4,2);
            return true;
        }

        static handle cast(const LPoint2d &lp, return_value_policy, handle defval) {
            py::object p3d = py::module::import("panda3d.core");
            py::object plp = p3d.attr("LPoint2d")(lp.get_x(), lp.get_y());

            return {plp.ptr()};
        }
};

}}

PYBIND11_MODULE(test_module, m) {

  py::class_<Test>(m, "Test")
    .def(py::init<>())
    .def("get_point", &Test::getPoint)
    .def("set_point", &Test::setPoint)
    ;

}

Tags: pycoretestimport类型getreturn错误
1条回答
网友
1楼 · 发布于 2024-09-30 02:35:59

似乎只要增加参考计数器就可以解决这个问题

static handle cast(const LPoint2d &src, return_value_policy policy, handle parent) {
        py::object p3d = py::module::import("panda3d.core");
        py::object plp = p3d.attr("LPoint2d")(src.get_x(), src.get_y());
        plp.inc_ref();

        return plp.ptr();
    }

这可能不是一个好的解决方案,但目前看来似乎有效

相关问题 更多 >

    热门问题