Add parameter at given distance method for curves

FreeCAD · Jul 29, 2016 · 8fe7c94 · 8fe7c94
1 parent a14794b
commit 8fe7c94
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 3 deletions.
diff --git a/src/Mod/Part/App/GeometryCurvePy.xml b/src/Mod/Part/App/GeometryCurvePy.xml
@@ -63,6 +63,12 @@ Part.show(s)
 length([uMin,uMax,Tol]) -> Float</UserDocu>
 			</Documentation>
 		</Methode>
+		<Methode Name="parameterAtDistance">
+			<Documentation>
+				<UserDocu>Returns the parameter on the curve of a point at the given distance from a starting parameter. 
+parameterAtDistance([abscissa, startingParameter]) -> Float the</UserDocu>
+			</Documentation>
+		</Methode>
 		<Methode Name="value">
 			<Documentation>
 				<UserDocu>Computes the point of parameter u on this curve</UserDocu>

diff --git a/src/Mod/Part/App/GeometryCurvePyImp.cpp b/src/Mod/Part/App/GeometryCurvePyImp.cpp
@@ -322,6 +322,32 @@ PyObject* GeometryCurvePy::length(PyObject *args)
     return 0;
 }
 
+PyObject* GeometryCurvePy::parameterAtDistance(PyObject *args)
+{
+    Handle_Geom_Geometry g = getGeometryPtr()->handle();
+    Handle_Geom_Curve c = Handle_Geom_Curve::DownCast(g);
+    try {
+        if (!c.IsNull()) {
+            double abscissa;
+            double u = 0;
+            if (!PyArg_ParseTuple(args, "d|d", &abscissa,&u))
+                return 0;
+            GeomAdaptor_Curve adapt(c);
+            GCPnts_AbscissaPoint abscissaPoint(adapt,abscissa,u);
+            double parm = abscissaPoint.Parameter();
+            return PyFloat_FromDouble(parm);
+        }
+    }
+    catch (Standard_Failure) {
+        Handle_Standard_Failure e = Standard_Failure::Caught();
+        PyErr_SetString(PartExceptionOCCError, e->GetMessageString());
+        return 0;
+    }
+
+    PyErr_SetString(PartExceptionOCCError, "Geometry is not a curve");
+    return 0;
+}
+
 PyObject* GeometryCurvePy::value(PyObject *args)
 {
     Handle_Geom_Geometry g = getGeometryPtr()->handle();
@@ -418,12 +444,12 @@ PyObject* GeometryCurvePy::makeRuledSurface(PyObject *args)
         }
         // check the result surface type
         if (aSurf->IsKind(STANDARD_TYPE(Geom_RectangularTrimmedSurface))) {
-            Handle_Geom_RectangularTrimmedSurface aTSurf = 
+            Handle_Geom_RectangularTrimmedSurface aTSurf =
                 Handle_Geom_RectangularTrimmedSurface::DownCast(aSurf);
             return new RectangularTrimmedSurfacePy(new GeomTrimmedSurface(aTSurf));
         }
         else if (aSurf->IsKind(STANDARD_TYPE(Geom_BSplineSurface))) {
-            Handle_Geom_BSplineSurface aBSurf = 
+            Handle_Geom_BSplineSurface aBSurf =
                 Handle_Geom_BSplineSurface::DownCast(aSurf);
             return new BSplineSurfacePy(new GeomBSplineSurface(aBSurf));
         }
@@ -610,7 +636,7 @@ PyObject *GeometryCurvePy::getCustomAttributes(const char* /*attr*/) const
 
 int GeometryCurvePy::setCustomAttributes(const char* /*attr*/, PyObject* /*obj*/)
 {
-    return 0; 
+    return 0;
 }
 
 // Specialized intersection functions