Upload centerline_extraction.py

centerline_extraction.py

@@ -137,12 +137,12 @@ def compute_centerlines_network(
     aorta_iliac_endpoints_ras = lps_endpoints_to_ras(aorta_iliac_endpoints)
 
     # Decimate the surface by reducing surface mesh points before voronoi centerline extraction as it scales badly with large surfaces
-    surface_for_centerline = decimate_surface_for_voronoi(surface_polydata_ras, target_points=50000)
+    #surface_for_centerline = decimate_surface_for_voronoi(surface_polydata_ras, target_points=50000)
 
     # Extract main aorta→iliacs centerline via Voronoi method
     try:
         print("Extracting main centerline...")
-        centerline_polydata = extract_centerline_from_surface(surface_for_centerline, aorta_iliac_endpoints_ras, curve_sampling_distance, smooth_iterations=0)
+        centerline_polydata = extract_centerline_from_surface(surface_polydata_ras, aorta_iliac_endpoints_ras, curve_sampling_distance, smooth_iterations=0)
         print(f"Main centerline: {centerline_polydata.GetNumberOfCells()} cells, {centerline_polydata.GetNumberOfPoints()} points")
     except Exception as e:
         print(f"Main centerline extraction failed: {type(e).__name__}: {e}")
@@ -153,7 +153,7 @@ def compute_centerlines_network(
     branch_bifurcations = []
     if centerline_polydata and branch_endpoints:
         try:
-            branch_centerlines, branch_bifurcations = process_all_branches(branch_endpoints, centerline_polydata, surface_for_centerline, curve_sampling_distance)
+            branch_centerlines, branch_bifurcations = process_all_branches(branch_endpoints, centerline_polydata, surface_polydata_ras, curve_sampling_distance)
         except Exception as e:
             print(f"Branch extraction failed: {type(e).__name__}: {e}")