diff --git a/src/vbl_aquarium/models/pinpoint.py b/src/vbl_aquarium/models/pinpoint.py index bb725b6..b0c6154 100644 --- a/src/vbl_aquarium/models/pinpoint.py +++ b/src/vbl_aquarium/models/pinpoint.py @@ -2,7 +2,7 @@ from pydantic import Field -from vbl_aquarium.models.unity import Vector2, Vector3, Color +from vbl_aquarium.models.unity import Color, Vector2, Vector3 from vbl_aquarium.utils.vbl_base_model import VBLBaseModel # CRANIOTOMY @@ -13,6 +13,7 @@ class CraniotomyModel(VBLBaseModel): size: Vector2 position: Vector3 + # TRANSFORM @@ -25,6 +26,7 @@ class AffineTransformModel(VBLBaseModel): # RIG + class RigModel(VBLBaseModel): name: str image: str @@ -35,6 +37,7 @@ class RigModel(VBLBaseModel): # Probes and insertions + class InsertionModel(VBLBaseModel): position: Vector3 angles: Vector3 @@ -42,6 +45,7 @@ class InsertionModel(VBLBaseModel): transform_name: str reference_coord: Vector3 + class ProbeModel(VBLBaseModel): insertion: InsertionModel @@ -53,6 +57,7 @@ class ProbeModel(VBLBaseModel): # Full scene + class SceneModel(VBLBaseModel): atlas_name: str transform_name: str