Fixed rotation bug

brainglobe · Aug 2, 2024 · 303ddb3 · 303ddb3
1 parent 26d4a31
commit 303ddb3
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/brainglobe_registration/registration_widget.py b/brainglobe_registration/registration_widget.py
@@ -468,6 +468,14 @@ def _on_adjust_atlas_rotation(self, pitch: float, yaw: float, roll: float):
             output_chunks=(2, bounding_box[1], bounding_box[2]),
         )
 
+        self._atlas_annotations_layer.data = dask_affine_transform(
+            self._atlas.annotation,
+            np.linalg.inv(transform_matrix),
+            order=0,
+            output_shape=bounding_box,
+            output_chunks=(2, bounding_box[1], bounding_box[2]),
+        )
+
         # Resets the viewer grid to update the grid to the new atlas
         self._viewer.reset_view()
 

diff --git a/brainglobe_registration/sample_data.py b/brainglobe_registration/sample_data.py