interpolate_nurbs_curve_with_tangents: add cyclic option.

nortikin · Dec 24, 2022 · 12ea912 · 12ea912
1 parent fe9ea68
commit 12ea912
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/utils/curve/nurbs_solver_applications.py b/utils/curve/nurbs_solver_applications.py
@@ -188,6 +188,7 @@ def knotvector_with_tangents_from_tknots(degree, u):
 
 def interpolate_nurbs_curve_with_tangents(degree, points, tangents,
             metric='DISTANCE', tknots=None,
+            cyclic = False,
             implementation = SvNurbsMaths.NATIVE,
             logger = None):
 
@@ -200,6 +201,10 @@ def interpolate_nurbs_curve_with_tangents(degree, points, tangents,
     if ndim not in {3,4}:
         raise Exception(f"Points must be 3 or 4 dimensional, not {ndim}")
 
+    if cyclic:
+        points = np.append(points, [points[0]], axis=0)
+        tangents = np.append(tangents, [tangents[0]], axis=0)
+
     if tknots is None:
         tknots = Spline.create_knots(points, metric=metric)