From d6f2446e57a902dc822d22cadfae50d45ec2c504 Mon Sep 17 00:00:00 2001
From: tomate44 <cg@grellier.fr>
Date: Sun, 16 Oct 2016 17:33:27 +0200
Subject: [PATCH] add uIso and vIso methods to SurfaceOfExtrusion

---
 src/Mod/Part/App/SurfaceOfExtrusionPy.xml    | 10 +++
 src/Mod/Part/App/SurfaceOfExtrusionPyImp.cpp | 93 ++++++++++++++++++++
 2 files changed, 103 insertions(+)

diff --git a/src/Mod/Part/App/SurfaceOfExtrusionPy.xml b/src/Mod/Part/App/SurfaceOfExtrusionPy.xml
index 60d0b7075..86008f9c4 100644
--- a/src/Mod/Part/App/SurfaceOfExtrusionPy.xml
+++ b/src/Mod/Part/App/SurfaceOfExtrusionPy.xml
@@ -30,5 +30,15 @@
 			</Documentation>
 			<Parameter Name="BasisCurve" Type="Object"/>
 		</Attribute>
+		<Methode Name="uIso">
+			<Documentation>
+				<UserDocu>Builds the U isoparametric curve</UserDocu>
+			</Documentation>
+		</Methode>
+		<Methode Name="vIso">
+			<Documentation>
+				<UserDocu>Builds the V isoparametric curve</UserDocu>
+			</Documentation>
+		</Methode>
 	</PythonExport>
 </GenerateModel>
diff --git a/src/Mod/Part/App/SurfaceOfExtrusionPyImp.cpp b/src/Mod/Part/App/SurfaceOfExtrusionPyImp.cpp
index 71ab883c9..99f78b103 100644
--- a/src/Mod/Part/App/SurfaceOfExtrusionPyImp.cpp
+++ b/src/Mod/Part/App/SurfaceOfExtrusionPyImp.cpp
@@ -24,11 +24,18 @@
 #include "PreCompiled.h"
 #ifndef _PreComp_
 # include <Geom_SurfaceOfLinearExtrusion.hxx>
+# include <Geom_Curve.hxx>
+# include <Geom_Line.hxx>
+# include <gp_Lin.hxx>
 #endif
 
 #include "Geometry.h"
 #include "SurfaceOfExtrusionPy.h"
 #include "SurfaceOfExtrusionPy.cpp"
+#include "GeometryCurvePy.h"
+#include "LinePy.h"
+#include "BezierCurvePy.h"
+#include "BSplineCurvePy.h"
 
 #include <Base/GeometryPyCXX.h>
 #include <Base/VectorPy.h>
@@ -138,6 +145,92 @@ void  SurfaceOfExtrusionPy::setBasisCurve(Py::Object arg)
     }
 }
 
+PyObject* SurfaceOfExtrusionPy::uIso(PyObject * args)
+{
+    double v;
+    if (!PyArg_ParseTuple(args, "d", &v))
+        return 0;
+
+    try {
+        Handle_Geom_Surface surf = Handle_Geom_Surface::DownCast
+            (getGeometryPtr()->handle());
+        Handle_Geom_Curve c = surf->UIso(v);
+        if (c->IsKind(STANDARD_TYPE(Geom_TrimmedCurve))) {
+            Handle_Geom_TrimmedCurve aCurve = Handle_Geom_TrimmedCurve::DownCast(c);
+            return new GeometryCurvePy(new GeomTrimmedCurve(aCurve));
+        }
+        if (c->IsKind(STANDARD_TYPE(Geom_BezierCurve))) {
+            Handle_Geom_BezierCurve aCurve = Handle_Geom_BezierCurve::DownCast(c);
+            return new BezierCurvePy(new GeomBezierCurve(aCurve));
+        }
+        if (c->IsKind(STANDARD_TYPE(Geom_BSplineCurve))) {
+            Handle_Geom_BSplineCurve aCurve = Handle_Geom_BSplineCurve::DownCast(c);
+            return new BSplineCurvePy(new GeomBSplineCurve(aCurve));
+        }
+        if (c->IsKind(STANDARD_TYPE(Geom_Line))) {
+            Handle_Geom_Line aLine = Handle_Geom_Line::DownCast(c);
+            GeomLineSegment* line = new GeomLineSegment();
+            Handle_Geom_TrimmedCurve this_curv = Handle_Geom_TrimmedCurve::DownCast
+                (line->handle());
+            Handle_Geom_Line this_line = Handle_Geom_Line::DownCast
+                (this_curv->BasisCurve());
+            this_line->SetLin(aLine->Lin());
+            return new LinePy(line);
+        }
+        PyErr_Format(PyExc_NotImplementedError, "Iso curve is of type '%s'",
+            c->DynamicType()->Name());
+        return 0;
+    }
+    catch (Standard_Failure) {
+        Handle_Standard_Failure e = Standard_Failure::Caught();
+        PyErr_SetString(PartExceptionOCCError, e->GetMessageString());
+        return 0;
+    }
+}
+
+PyObject* SurfaceOfExtrusionPy::vIso(PyObject * args)
+{
+    double v;
+    if (!PyArg_ParseTuple(args, "d", &v))
+        return 0;
+
+    try {
+        Handle_Geom_Surface surf = Handle_Geom_Surface::DownCast
+            (getGeometryPtr()->handle());
+        Handle_Geom_Curve c = surf->VIso(v);
+        if (c->IsKind(STANDARD_TYPE(Geom_TrimmedCurve))) {
+            Handle_Geom_TrimmedCurve aCurve = Handle_Geom_TrimmedCurve::DownCast(c);
+            return new GeometryCurvePy(new GeomTrimmedCurve(aCurve));
+        }
+        if (c->IsKind(STANDARD_TYPE(Geom_BezierCurve))) {
+            Handle_Geom_BezierCurve aCurve = Handle_Geom_BezierCurve::DownCast(c);
+            return new BezierCurvePy(new GeomBezierCurve(aCurve));
+        }
+        if (c->IsKind(STANDARD_TYPE(Geom_BSplineCurve))) {
+            Handle_Geom_BSplineCurve aCurve = Handle_Geom_BSplineCurve::DownCast(c);
+            return new BSplineCurvePy(new GeomBSplineCurve(aCurve));
+        }
+        if (c->IsKind(STANDARD_TYPE(Geom_Line))) {
+            Handle_Geom_Line aLine = Handle_Geom_Line::DownCast(c);
+            GeomLineSegment* line = new GeomLineSegment();
+            Handle_Geom_TrimmedCurve this_curv = Handle_Geom_TrimmedCurve::DownCast
+                (line->handle());
+            Handle_Geom_Line this_line = Handle_Geom_Line::DownCast
+                (this_curv->BasisCurve());
+            this_line->SetLin(aLine->Lin());
+            return new LinePy(line);
+        }
+        PyErr_Format(PyExc_NotImplementedError, "Iso curve is of type '%s'",
+            c->DynamicType()->Name());
+        return 0;
+    }
+    catch (Standard_Failure) {
+        Handle_Standard_Failure e = Standard_Failure::Caught();
+        PyErr_SetString(PartExceptionOCCError, e->GetMessageString());
+        return 0;
+    }
+}
+
 PyObject *SurfaceOfExtrusionPy::getCustomAttributes(const char* /*attr*/) const
 {
     return 0;