Restore missed points affine setting in callback (#94)

In #58, we forgot to also propagate the affine transformation to the moving points layer. This PR fixes that.
jni · Feb 23, 2024 · ac135ce · ac135ce
1 parent 68eb9e9
commit ac135ce
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/affinder/affinder.py b/src/affinder/affinder.py
@@ -84,6 +84,9 @@ def next_layer_callback(
                 moving_image_layer.affine = convert_affine_to_ndims(
                         (ref_mat @ mat.params), moving_image_layer.ndim
                         )
+                moving_points_layer.affine = convert_affine_to_ndims(
+                        (ref_mat @ mat.params), moving_image_layer.ndim
+                        )
                 if output is not None:
                     np.savetxt(output, np.asarray(mat.params), delimiter=',')
             viewer.layers.selection.active = reference_points_layer