Undo applying transform on displacement field, to reset field for nex…

…t use
SciKit-Surgery · Sep 11, 2020 · 22131db · 22131db
1 parent 8f68a61
commit 22131db
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/sksurgeryvtk/models/voxelise.py b/sksurgeryvtk/models/voxelise.py
@@ -659,6 +659,18 @@ def apply_displacement_to_mesh(mesh: Union[vtk.vtkDataObject, str],
         writer.SetFileName(save_mesh)
         writer.Update()
 
+    # Undo transformation so that field is 'reset' for future use
+    tf = loadTransformationMatrix(field)
+    LOGGER.debug("Reversing transform")
+    tfFilter = vtk.vtkTransformFilter()
+    tfFilter.SetTransform(tf)
+    tfFilter.SetInputData(field)
+    tfFilter.Update()
+    field = tfFilter.GetOutput()
+
+    # Apply transformation also to all vector fields:
+    applyTransformation(field, tf)
+
     return output
 
 # class NonRigidAlignment: