DRAFT: proof of principle bringing microvis model to ndv

pyproject.toml

@@ -158,6 +158,8 @@ disallow_any_generics = false
 disallow_subclassing_any = false
 show_error_codes = true
 pretty = true
+# sometimes experiencing a pydantic mypy plugin bug
+# dealing with recursive models, like ndv.models._scene.nodes.Node
 plugins = ["pydantic.mypy"]
 
 [[tool.mypy.overrides]]

src/ndv/_types.py

@@ -7,6 +7,7 @@
 from enum import Enum, IntFlag, auto
 from typing import TYPE_CHECKING, Annotated, Any, NamedTuple, cast
 
+import numpy.typing as npt
 from pydantic import PlainSerializer, PlainValidator
 from typing_extensions import TypeAlias
 
@@ -107,3 +108,27 @@
             CursorType.BDIAG_ARROW: Qt.CursorShape.SizeBDiagCursor,
             CursorType.FDIAG_ARROW: Qt.CursorShape.SizeFDiagCursor,
         }[self]
+
+
+class CameraType(str, Enum):
+    """Camera type."""
+
+    ARCBALL = "arcball"
+    PANZOOM = "panzoom"
+
+    def __str__(self) -> str:
+        return self.value
+
+
+ArrayLike: TypeAlias = npt.NDArray
+
+
+class ImageInterpolation(str, Enum):
+    """Image interpolation options."""
+
+    LINEAR = "linear"
+    NEAREST = "nearest"
+    BICUBIC = "bicubic"
+
+    def __str__(self) -> str:
+        return self.value

src/ndv/controllers/_array_viewer.py

@@ -66,7 +66,8 @@ def __init__(
                 stacklevel=2,
             )
         self._data_model = _ArrayDataDisplayModel(
-            data_wrapper=data, display=display_model or ArrayDisplayModel(**kwargs)
+            data_wrapper=data,
+            display=display_model or ArrayDisplayModel(**kwargs),
         )
 
         app = _app.gui_frontend()

src/ndv/models/_mapping.py

@@ -290,6 +290,8 @@ def _new(*args: Any, **kwargs: Any) -> ValidatedEventedDict[_KT, _VT]:
 
 def _get_schema(hint: Any, handler: GetCoreSchemaHandler) -> core_schema.CoreSchema:
     # check if the hint already has a core schema attached to it.
+    # this helps to avoid `Definitions error: definition `___` was never filled`
+    # for recursive types.
     if hasattr(hint, "__pydantic_core_schema__"):
         return cast("core_schema.CoreSchema", hint.__pydantic_core_schema__)
     # otherwise, call the handler to get the core schema.

src/ndv/models/_scene/__init__.py

@@ -0,0 +1,14 @@
+from ._transform import Transform
+from .canvas import Canvas
+from .nodes import Camera, Image, Node, Scene
+from .view import View
+
+__all__ = [
+    "Camera",
+    "Canvas",
+    "Image",
+    "Node",
+    "Scene",
+    "Transform",
+    "View",
+]