Issue #93 try a 2 stage decimation

SciKit-Surgery · Mar 15, 2022 · 7424cf7 · 7424cf7
1 parent bee43a7
commit 7424cf7
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/sksurgerybard/algorithms/decimation.py b/sksurgerybard/algorithms/decimation.py
@@ -16,13 +16,24 @@ def decimate_actor(actor, target_vertices):
     start_points = polydata.GetNumberOfPoints()
     target_reduction = 1.0 - target_vertices/start_points
 
+    decimated = vtkPolyData()
+    #For big reductions let's try two stage, although I'm not
+    #sure this has any effect
+    if target_reduction > 0.9:
+        decimate = vtkDecimatePro()
+        decimate.SetInputData(polydata)
+        decimate.SetTargetReduction(0.9)
+        decimate.PreserveTopologyOn()
+        decimate.Update()
+        polydata.ShallowCopy(decimate.GetOutput())
+        start_points = polydata.GetNumberOfPoints()
+        target_reduction = 1.0 - target_vertices/start_points
+
     decimate = vtkDecimatePro()
     decimate.SetInputData(polydata)
     decimate.SetTargetReduction(target_reduction)
     decimate.PreserveTopologyOn()
     decimate.Update()
-
-    decimated = vtkPolyData()
     decimated.ShallowCopy(decimate.GetOutput())
 
     actor.GetMapper().SetInputData(decimated)