diff --git a/modules/python/src2/cv2.cpp b/modules/python/src2/cv2.cpp index 433f70af17..9369f297a2 100644 --- a/modules/python/src2/cv2.cpp +++ b/modules/python/src2/cv2.cpp @@ -91,6 +91,7 @@ typedef std::vector vector_float; typedef std::vector vector_double; typedef std::vector vector_Point; typedef std::vector vector_Point2f; +typedef std::vector vector_Point3f; typedef std::vector vector_Vec2f; typedef std::vector vector_Vec3f; typedef std::vector vector_Vec4f; @@ -733,6 +734,23 @@ bool pyopencv_to(PyObject* obj, Point2d& p, const char* name) return PyArg_ParseTuple(obj, "dd", &p.x, &p.y) > 0; } +template<> +bool pyopencv_to(PyObject* obj, Point3f& p, const char* name) +{ + (void)name; + if(!obj || obj == Py_None) + return true; + return PyArg_ParseTuple(obj, "fff", &p.x, &p.y, &p.z) > 0; +} + +template<> +bool pyopencv_to(PyObject* obj, Point3d& p, const char* name) +{ + (void)name; + if(!obj || obj == Py_None) + return true; + return PyArg_ParseTuple(obj, "ddd", &p.x, &p.y, &p.z) > 0; +} template<> PyObject* pyopencv_from(const Point& p) @@ -746,6 +764,12 @@ PyObject* pyopencv_from(const Point2f& p) return Py_BuildValue("(dd)", p.x, p.y); } +template<> +PyObject* pyopencv_from(const Point3f& p) +{ + return Py_BuildValue("(ddd)", p.x, p.y, p.z); +} + template<> bool pyopencv_to(PyObject* obj, Vec3d& v, const char* name) { @@ -773,6 +797,12 @@ PyObject* pyopencv_from(const Point2d& p) return Py_BuildValue("(dd)", p.x, p.y); } +template<> +PyObject* pyopencv_from(const Point3d& p) +{ + return Py_BuildValue("(ddd)", p.x, p.y, p.y); +} + template struct pyopencvVecConverter { static bool to(PyObject* obj, std::vector<_Tp>& value, const ArgInfo info)