Added check for symmetrical atlas

brainglobe · Aug 15, 2024 · 89ffcaf · 89ffcaf
1 parent 1537b8e
commit 89ffcaf
Showing 1 changed file with 10 additions and 7 deletions.
diff --git a/brainrender/scene.py b/brainrender/scene.py
@@ -287,14 +287,17 @@ def add_brain_region(
         actors = self.add(*regions)
 
         # slice to keep only one hemisphere
-        if hemisphere == "right":
-            mesh_center = self.root._mesh.bounds().reshape((3, 2)).mean(axis=1)
-            plane = self.atlas.get_plane(pos=mesh_center, norm=(0, 0, 1))
-        elif hemisphere == "left":
-            mesh_center = self.root._mesh.bounds().reshape((3, 2)).mean(axis=1)
-            plane = self.atlas.get_plane(pos=mesh_center, norm=(0, 0, -1))
-
         if hemisphere in ("left", "right"):
+            if self.atlas.metadata["symmetric"]:
+                mesh_center = (
+                    self.root._mesh.bounds().reshape((3, 2)).mean(axis=1)
+                )
+            else:
+                mesh_center = self.root._mesh.center_of_mass()
+
+            normal = (0, 0, 1) if hemisphere == "right" else (0, 0, -1)
+            plane = self.atlas.get_plane(pos=mesh_center, norm=normal)
+
             if not isinstance(actors, list):
                 actors._mesh.cut_with_plane(
                     origin=plane.center,