Upload endograft_script.py

endograft_script.py

@@ -321,15 +321,10 @@ def create_endograft(centerline, diameter, prox_length,extension_length, distal_
     # -------------------------
     # EXTENSION COMPONENT
     # -------------------------
-    if extension_length > 0:
-        if extension_length > 97:
-            prox_extension_overlap = 3*peak_to_peak + 2*extension_gap
-        else:
-            prox_extension_overlap = 2*peak_to_peak + extension_gap
+    if extension_length > 97:
+        prox_extension_overlap = 3*peak_to_peak + 2*extension_gap
     else:
-        # Prevent VTK crash by returning empty polydata instead of a zero-length stack
-        extension_fabric = vtk.vtkPolyData()
-        extension_stents = vtk.vtkPolyData()
+        prox_extension_overlap = 2*peak_to_peak + extension_gap
 
     extension_start = prox_length - prox_extension_overlap
     extension_fabric = create_component_fabric(centerline, frames, extension_start, extension_length, diameter, start_amp_pp=0.0)
@@ -339,15 +334,10 @@ def create_endograft(centerline, diameter, prox_length,extension_length, distal_
     # -------------------------
     # DISTAL COMPONENT
     # -------------------------
-    if distal_length > 0:
-        distal_overlap = 3*peak_to_peak + 2*distal_gap
-        distal_start = prox_length + extension_length - prox_extension_overlap - distal_overlap
-        distal_fabric = create_component_fabric(centerline, frames, distal_start, distal_length, diameter, start_amp_pp=0.0)
-        distal_stents = create_component_stents(centerline, frames, distal_start, distal_length, diameter, distal_fabric, peak_spacing=distal_gap, component_type="distal")
-    else:
-        # Prevent VTK crash by returning empty polydata
-        distal_fabric = vtk.vtkPolyData()
-        distal_stents = vtk.vtkPolyData()
+    distal_overlap = 3*peak_to_peak + 2*distal_gap
+    distal_start = prox_length + extension_length - prox_extension_overlap - distal_overlap
+    distal_fabric = create_component_fabric(centerline, frames, distal_start, distal_length, diameter, start_amp_pp=0.0)
+    distal_stents = create_component_stents(centerline, frames, distal_start, distal_length, diameter, distal_fabric, peak_spacing=distal_gap, component_type="distal")
     # -------------------------
     # MERGE COMPONENTS
     # -------------------------