napari-mlarray 0.0.1__py3-none-any.whl → 0.0.2__py3-none-any.whl

napari_mlarray/_reader.py

@@ -6,6 +6,8 @@ implement multiple readers or even other plugin contributions. see:
 https://napari.org/stable/plugins/building_a_plugin/guides.html#readers
 """
 from mlarray import MLArray
+from pathlib import Path
+import numpy as np
 
 
 def napari_get_reader(path):
@@ -69,9 +71,201 @@ def reader_function(path):
         layer. Both "meta", and "layer_type" are optional. napari will
         default to layer_type=="image" if not provided
     """
-    # handle both a string and a list of strings
     paths = [path] if isinstance(path, str) else path
-    # load all files into array
-    mlarrays = [MLArray.open(_path) for _path in paths]
-    layer_data = [(mlarray, {"affine": mlarray.affine, "metadata": mlarray.meta.to_dict()}, "labels" if mlarray.meta.is_seg.is_seg == True else "image") for mlarray in mlarrays]
+    layer_data = []
+    for path in paths:
+        name = Path(path).stem
+        mlarray = MLArray.open(path)
+        if mlarray.meta._has_array.has_array == True:
+            data = mlarray
+            metadata = {"name": f"{name}", "affine": mlarray.affine, "metadata": mlarray.meta.to_mapping()}
+            layer_type = "labels" if mlarray.meta.is_seg.is_seg == True else "image"
+            layer_data.append((data, metadata, layer_type))
+        if mlarray.meta.bbox.bboxes is not None:
+            data = bboxes_minmax_to_napari_rectangles_2d(mlarray.meta.bbox.bboxes)
+            edge_color = _napari_bbox_edge_colors(
+                data,
+                labels=getattr(mlarray.meta.bbox, "labels", None),
+            )
+            text = _napari_bbox_score_text(
+                scores=getattr(mlarray.meta.bbox, "scores", None),
+                labels=getattr(mlarray.meta.bbox, "labels", None),
+                count=len(data),
+                edge_color=edge_color,
+                rectangles=data,
+            )
+            metadata = {
+                "name": f"{name} (BBoxes)",
+                "shape_type": "rectangle",
+                "affine": mlarray.affine,
+                "metadata": mlarray.meta.to_mapping(),
+                "face_color": "transparent",
+                "edge_color": edge_color,
+            }
+            if text is not None:
+                metadata["text"] = text
+            layer_type = "shapes"
+            layer_data.append((data, metadata, layer_type))
     return layer_data
+
+
+def bboxes_minmax_to_napari_rectangles_2d(
+    bboxes,
+    *,
+    dtype=np.float32,
+    validate: bool = True,
+) -> np.ndarray:
+    """
+    Convert 2D axis-aligned bounding boxes from min/max format to napari Shapes rectangles.
+
+    Accepted input formats (both mean the same thing):
+      1) (N, 2, 2): [[min_dim0, max_dim0], [min_dim1, max_dim1]]
+         Example (dim order is whatever you use, e.g. (y, x)):
+           [[[ymin, ymax], [xmin, xmax]], ...]
+
+      2) (N, 4): [min_dim0, min_dim1, max_dim0, max_dim1]
+         Example:
+           [[ymin, xmin, ymax, xmax], ...]
+
+    Output format (napari Shapes rectangle vertices):
+      (N, 4, 2) with vertices in non-twisting cyclic order:
+        (min0, min1) -> (min0, max1) -> (max0, max1) -> (max0, min1)
+
+    Raises:
+      ValueError if bboxes are not 2D (i.e., D != 2) or shapes are invalid.
+    """
+    arr = np.asarray(bboxes)
+
+    # Normalize input to shape (N, 2, 2)
+    if arr.ndim == 2 and arr.shape[1] == 4:
+        # (N, 4) -> (N, 2, 2)
+        arr = np.stack(
+            [
+                arr[:, [0, 2]],  # dim0: [min0, max0]
+                arr[:, [1, 3]],  # dim1: [min1, max1]
+            ],
+            axis=1,
+        )
+    elif arr.ndim == 3 and arr.shape[1:] == (2, 2):
+        pass
+    else:
+        raise ValueError(
+            f"Expected bboxes of shape (N, 2, 2) or (N, 4). Got {arr.shape}."
+        )
+
+    N, D, two = arr.shape
+    if D != 2 or two != 2:
+        # Defensive; should never hit because of checks above.
+        raise ValueError(f"Only 2D bboxes are supported. Got (N, {D}, {two}).")
+
+    mins = arr[:, :, 0]
+    maxs = arr[:, :, 1]
+
+    if validate and np.any(maxs < mins):
+        bad = np.argwhere(maxs < mins)
+        raise ValueError(
+            "Found bbox with max < min at indices (bbox_index, dim): "
+            f"{bad[:10].tolist()}" + (" ..." if len(bad) > 10 else "")
+        )
+
+    min0, min1 = mins[:, 0], mins[:, 1]
+    max0, max1 = maxs[:, 0], maxs[:, 1]
+
+    # Cyclic order (no twisting):
+    rects = np.stack(
+        [
+            np.stack([min0, min1], axis=1),
+            np.stack([min0, max1], axis=1),
+            np.stack([max0, max1], axis=1),
+            np.stack([max0, min1], axis=1),
+        ],
+        axis=1,
+    ).astype(dtype, copy=False)
+
+    return rects
+
+
+def _napari_bbox_edge_colors(rectangles, labels):
+    """Return RGBA edge colors for each bbox."""
+    count = len(rectangles)
+    if count == 0:
+        return np.empty((0, 4), dtype=np.float32)
+
+    if labels is not None and len(labels) == count:
+        unique_labels = list(dict.fromkeys(labels))
+        label_to_color = {
+            label: _palette_rgba(idx) for idx, label in enumerate(unique_labels)
+        }
+        colors = np.array([label_to_color[label] for label in labels], dtype=np.float32)
+    else:
+        colors = np.array([_palette_rgba(idx) for idx in range(count)], dtype=np.float32)
+
+    return colors
+
+
+def _napari_bbox_score_text(scores, labels, count, edge_color, rectangles):
+    """Return napari Shapes text metadata if scores are provided."""
+    have_scores = scores is not None and len(scores) == count
+    have_labels = labels is not None and len(labels) == count
+    if not have_scores and not have_labels:
+        return None
+
+    # Place text at the top-left corner of each rectangle.
+    top_left = rectangles[:, 0, :]
+    top_left = np.maximum(top_left - np.array([4.0, 0.0], dtype=top_left.dtype), 0)
+
+    strings = []
+    for idx in range(count):
+        parts = []
+        if have_labels:
+            parts.append(f"Label: {labels[idx]}")
+        if have_scores:
+            parts.append(f"Score: {scores[idx]:.3f}")
+        # Add a trailing empty line to create spacing below the score.
+        parts.append("\n")
+        strings.append("\n".join(parts))
+
+    return {
+        "string": strings,
+        "color": edge_color,
+        "size": 12,
+        "anchor": "upper_left",
+        "position": top_left,
+    }
+
+
+def _palette_rgba(index):
+    """Simple, distinct-ish palette; returns RGBA in 0..1."""
+    palette = [
+        (0.90, 0.10, 0.12, 1.0),
+        (0.00, 0.48, 1.00, 1.0),
+        (0.20, 0.80, 0.20, 1.0),
+        (0.98, 0.60, 0.00, 1.0),
+        (0.60, 0.20, 0.80, 1.0),
+        (0.10, 0.75, 0.80, 1.0),
+        (0.80, 0.80, 0.00, 1.0),
+        (0.95, 0.40, 0.60, 1.0),
+        (0.90, 0.30, 0.00, 1.0),
+        (0.00, 0.70, 0.40, 1.0),
+        (0.40, 0.80, 1.00, 1.0),
+        (1.00, 0.20, 0.70, 1.0),
+        (0.50, 0.90, 0.20, 1.0),
+        (0.20, 0.90, 0.70, 1.0),
+        (0.70, 0.50, 1.00, 1.0),
+        (1.00, 0.50, 0.20, 1.0),
+        (0.20, 0.60, 1.00, 1.0),
+        (1.00, 0.70, 0.20, 1.0),
+        (0.60, 1.00, 0.20, 1.0),
+        (0.20, 1.00, 0.40, 1.0),
+        (0.20, 1.00, 0.90, 1.0),
+        (0.20, 0.90, 1.00, 1.0),
+        (0.40, 0.60, 1.00, 1.0),
+        (0.80, 0.20, 1.00, 1.0),
+        (1.00, 0.20, 0.30, 1.0),
+        (1.00, 0.30, 0.50, 1.0),
+        (1.00, 0.60, 0.60, 1.0),
+        (1.00, 0.90, 0.30, 1.0),
+        (0.60, 1.00, 0.60, 1.0),
+        (0.60, 0.90, 1.00, 1.0),
+    ]
+    return palette[index % len(palette)]

napari_mlarray/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.0.1'
-__version_tuple__ = version_tuple = (0, 0, 1)
+__version__ = version = '0.0.2'
+__version_tuple__ = version_tuple = (0, 0, 2)
 
 __commit_id__ = commit_id = None

napari_mlarray/_writer.py

@@ -34,7 +34,7 @@ def write_single_image(path: str, data: Any, meta: dict) -> list[str]:
     -------
     [path] : A list containing the string path to the saved file.
     """
-    mlarray = MLArray(data, meta=Meta.from_dict(meta["metadata"]))
+    mlarray = MLArray(data, meta=Meta.from_mapping(meta["metadata"]))
     mlarray.save(path)
 
     # return path to any file(s) that were successfully written

{napari_mlarray-0.0.1.dist-info → napari_mlarray-0.0.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: napari-mlarray
-Version: 0.0.1
+Version: 0.0.2
 Summary: A reader/writer Napari plugin for MLArray images.
 Author: Karol-G
 Author-email: karol.gotkowski@dkfz.de
@@ -44,6 +44,7 @@ Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: mlarray
+Requires-Dist: numpy
 Provides-Extra: all
 Requires-Dist: napari[all]; extra == "all"
 Dynamic: license-file

napari_mlarray-0.0.2.dist-info/RECORD

@@ -0,0 +1,12 @@
+napari_mlarray/__init__.py,sha256=c2b0_m5sORovlHyyozJS1l00lXjFM7ULIkrHqojq5N4,249
+napari_mlarray/_reader.py,sha256=IvFdBZnA3AaTt_wqh9B_q_7tXVV2N3nr722thUCAncY,9373
+napari_mlarray/_version.py,sha256=huLsL1iGeXWQKZ8bjwDdIWC7JOkj3wnzBh-HFMZl1PY,704
+napari_mlarray/_widget.py,sha256=K6MYwgFiQg7-dORp-kC_VivrfppiUwMmevbPMhwMT9c,4810
+napari_mlarray/_writer.py,sha256=EG-013mGR14L5V7tygzoZg2EPyIwnZAP8zG8flg58hU,1209
+napari_mlarray/napari.yaml,sha256=HILzIxDmFpRPgCmM7uVzzZ86Ayp2ecWiSXHTCC1qjx4,796
+napari_mlarray-0.0.2.dist-info/licenses/LICENSE,sha256=LKlNG6Bx5z0YnzAp9GjCxCR0aypSPO_JcUHmuVGtwds,1162
+napari_mlarray-0.0.2.dist-info/METADATA,sha256=LYZJf7Fan_qTRKnxY_KOxAJnBkfbpt8pTYn-_xa58Wk,5562
+napari_mlarray-0.0.2.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+napari_mlarray-0.0.2.dist-info/entry_points.txt,sha256=ibu_ymiLzJPNpL0x0Fdendi5fQiq_baxx224E5pAW0c,62
+napari_mlarray-0.0.2.dist-info/top_level.txt,sha256=ZfJPiLTSmZ9eakEU1J6znzWi6dS_OTGAXQSI08HOgQg,15
+napari_mlarray-0.0.2.dist-info/RECORD,,

napari_mlarray-0.0.1.dist-info/RECORD

@@ -1,12 +0,0 @@
-napari_mlarray/__init__.py,sha256=c2b0_m5sORovlHyyozJS1l00lXjFM7ULIkrHqojq5N4,249
-napari_mlarray/_reader.py,sha256=TT6-L8t7wlYRA5laoVH-frf-ogBxXlEQWE7y70IUHHs,3045
-napari_mlarray/_version.py,sha256=qf6R-J7-UyuABBo8c0HgaquJ8bejVbf07HodXgwAwgQ,704
-napari_mlarray/_widget.py,sha256=K6MYwgFiQg7-dORp-kC_VivrfppiUwMmevbPMhwMT9c,4810
-napari_mlarray/_writer.py,sha256=tAFCX1-LnmDqeDFpO2H22Xrju58-3D-d0ZrEvdAXMbE,1206
-napari_mlarray/napari.yaml,sha256=HILzIxDmFpRPgCmM7uVzzZ86Ayp2ecWiSXHTCC1qjx4,796
-napari_mlarray-0.0.1.dist-info/licenses/LICENSE,sha256=LKlNG6Bx5z0YnzAp9GjCxCR0aypSPO_JcUHmuVGtwds,1162
-napari_mlarray-0.0.1.dist-info/METADATA,sha256=tHD_AJuOHEEWg3OU6FepCJSMQDbVPHtCGoVa1y6ZBHg,5541
-napari_mlarray-0.0.1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-napari_mlarray-0.0.1.dist-info/entry_points.txt,sha256=ibu_ymiLzJPNpL0x0Fdendi5fQiq_baxx224E5pAW0c,62
-napari_mlarray-0.0.1.dist-info/top_level.txt,sha256=ZfJPiLTSmZ9eakEU1J6znzWi6dS_OTGAXQSI08HOgQg,15
-napari_mlarray-0.0.1.dist-info/RECORD,,