Smoothing Endograft

endograft_generation.py

@@ -1,6 +1,7 @@
 import vtk
 import numpy as np
 import tempfile
+from scipy.interpolate import splprep, splev
 
 def arc_length(centerline):
     arc_lengths = np.zeros(len(centerline))
@@ -66,6 +67,15 @@ def project_point_to_surface(point, locator, polydata):
 
     return np.array(closest), normal
 
+def smooth_centerline(centerline, smoothing=0.5, samples=500):
+
+    centerline = np.asarray(centerline)
+    spline, _ = splprep(centerline.T, s=smoothing, k=3)
+    sample_positions = np.linspace(0, 1, samples)
+    smoothed_centerline = splev(sample_positions, spline)
+
+    return np.vstack(smoothed_centerline).T
+
 def sample_centerline_by_distance(centerline, arc_lengths, frames, query_distance):
 
     idx = np.searchsorted(arc_lengths, query_distance)
@@ -294,6 +304,8 @@ def merge_components(proximal, extension, distal):
     return clean
 
 def create_endograft(centerline, proximal_diameter, distal_diameter, proximal_length, extension_length, distal_length, proximal_gap, extension_gap, distal_gap, proximal_extension_overlap, distal_extension_overlap, proximal_distal_overlap, include_combined=False, phase_angle=0.0):
+    
+    centerline = smooth_centerline(centerline, smoothing=5, samples=500)
     frames = compute_parallel_frames(centerline)
 
     # PROXIMAL COMPONENT
@@ -303,7 +315,7 @@ def create_endograft(centerline, proximal_diameter, distal_diameter, proximal_le
 
     proximal_stents = create_stent_component(centerline, frames, proximal_start, proximal_length, proximal_diameter, proximal_fabric, peak_spacing=proximal_gap, component_type="proximal", phase_angle=phase_angle)
 
-    # # EXTENSION COMPONENT
+    # EXTENSION COMPONENT
     # if extension_length > 97:
     #     prox_extension_overlap = 3*peak_to_peak + 2*extension_gap
     # else: