ENH: Add omero metadata read support

Author: thewtex

ngff_zarr/from_ngff_zarr.py

@@ -16,7 +16,15 @@
 
 from .ngff_image import NgffImage
 from .to_multiscales import Multiscales
-from .v04.zarr_metadata import Axis, Dataset, Scale, Translation
+from .v04.zarr_metadata import (
+    Axis,
+    Dataset,
+    Scale,
+    Translation,
+    Omero,
+    OmeroChannel,
+    OmeroWindow,
+)
 from .validate import validate as validate_ngff
 
 zarr_version = packaging.version.parse(zarr.__version__)
@@ -145,9 +153,29 @@ def from_ngff_zarr(
             Axis(name="y", type="space"),
             Axis(name="x", type="space"),
         ]
+
     coordinateTransformations = None
     if "coordinateTransformations" in metadata:
         coordinateTransformations = metadata["coordinateTransformations"]
+
+    omero = None
+    if "omero" in root.attrs:
+        omero_data = root.attrs["omero"]
+        omero = Omero(
+            channels=[
+                OmeroChannel(
+                    color=channel["color"],
+                    window=OmeroWindow(
+                        min=channel["window"]["min"],
+                        max=channel["window"]["max"],
+                        start=channel["window"]["start"],
+                        end=channel["window"]["end"],
+                    ),
+                )
+                for channel in omero_data["channels"]
+            ]
+        )
+
     if version == "0.5":
         from .v05.zarr_metadata import Metadata
 
@@ -156,6 +184,7 @@ def from_ngff_zarr(
             datasets=datasets,
             name=name,
             coordinateTransformations=coordinateTransformations,
+            omero=omero,
         )
     else:
         from .v04.zarr_metadata import Metadata
@@ -166,6 +195,7 @@ def from_ngff_zarr(
             name=name,
             version=metadata["version"],
             coordinateTransformations=coordinateTransformations,
+            omero=omero,
         )
 
     return Multiscales(images, metadata)

ngff_zarr/v04/zarr_metadata.py

@@ -2,6 +2,7 @@
 from typing import List, Optional, Union
 
 from typing_extensions import Literal
+import re
 
 SupportedDims = Union[
     Literal["c"], Literal["x"], Literal["y"], Literal["z"], Literal["t"]
@@ -165,10 +166,34 @@ class Dataset:
     coordinateTransformations: List[Transform]
 
 
+@dataclass
+class OmeroWindow:
+    min: float
+    max: float
+    start: float
+    end: float
+
+
+@dataclass
+class OmeroChannel:
+    color: str
+    window: OmeroWindow
+
+    def validate_color(self):
+        if not re.fullmatch(r"[0-9A-Fa-f]{6}", self.color):
+            raise ValueError(f"Invalid color '{self.color}'. Must be 6 hex digits.")
+
+
+@dataclass
+class Omero:
+    channels: List[OmeroChannel]
+
+
 @dataclass
 class Metadata:
     axes: List[Axis]
     datasets: List[Dataset]
     coordinateTransformations: Optional[List[Transform]]
+    omero: Optional[Omero] = None
     name: str = "image"
     version: str = "0.4"

ngff_zarr/v05/zarr_metadata.py

@@ -1,12 +1,13 @@
 from typing import List, Optional
 from dataclasses import dataclass
 
-from ..v04.zarr_metadata import Axis, Transform, Dataset
+from ..v04.zarr_metadata import Axis, Transform, Dataset, Omero
 
 
 @dataclass
 class Metadata:
     axes: List[Axis]
     datasets: List[Dataset]
     coordinateTransformations: Optional[List[Transform]]
+    omero: Optional[Omero] = None
     name: str = "image"

test/test_omero.py

@@ -0,0 +1,55 @@
+from ngff_zarr import from_ngff_zarr
+
+from ._data import test_data_dir
+
+
+def test_read_omero(input_images):  # noqa: ARG001
+    dataset_name = "13457537"
+    store_path = test_data_dir / "input" / f"{dataset_name}.zarr"
+    multiscales = from_ngff_zarr(store_path, validate=True)
+
+    omero = multiscales.metadata.omero
+    assert omero is not None
+    assert len(omero.channels) == 6
+
+    # Channel 0
+    assert omero.channels[0].color == "FFFFFF"
+    assert omero.channels[0].window.min == 0.0
+    assert omero.channels[0].window.max == 65535.0
+    assert omero.channels[0].window.start == 0.0
+    assert omero.channels[0].window.end == 1200.0
+
+    # Channel 1
+    assert omero.channels[1].color == "FFFFFF"
+    assert omero.channels[1].window.min == 0.0
+    assert omero.channels[1].window.max == 65535.0
+    assert omero.channels[1].window.start == 0.0
+    assert omero.channels[1].window.end == 1200.0
+
+    # Channel 2
+    assert omero.channels[2].color == "FFFFFF"
+    assert omero.channels[2].window.min == 0.0
+    assert omero.channels[2].window.max == 65535.0
+    assert omero.channels[2].window.start == 0.0
+    assert omero.channels[2].window.end == 1200.0
+
+    # Channel 3
+    assert omero.channels[3].color == "FFFFFF"
+    assert omero.channels[3].window.min == 0.0
+    assert omero.channels[3].window.max == 65535.0
+    assert omero.channels[3].window.start == 0.0
+    assert omero.channels[3].window.end == 1200.0
+
+    # Channel 4
+    assert omero.channels[4].color == "0000FF"
+    assert omero.channels[4].window.min == 0.0
+    assert omero.channels[4].window.max == 65535.0
+    assert omero.channels[4].window.start == 0.0
+    assert omero.channels[4].window.end == 5000.0
+
+    # Channel 5
+    assert omero.channels[5].color == "FF0000"
+    assert omero.channels[5].window.min == 0.0
+    assert omero.channels[5].window.max == 65535.0
+    assert omero.channels[5].window.start == 0.0
+    assert omero.channels[5].window.end == 100.0