From bd61641b9080559fc45a328b2e7b95ac9a616cb6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Al=C3=A1n=20Mu=C3=B1oz?= <alan.munoz@ed.ac.uk>
Date: Thu, 5 Jan 2023 23:29:12 +0000
Subject: [PATCH] feat(image): add name, dimorder to ImageDir

---
 src/aliby/io/image.py | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

diff --git a/src/aliby/io/image.py b/src/aliby/io/image.py
index fca64f5d..816c9e13 100644
--- a/src/aliby/io/image.py
+++ b/src/aliby/io/image.py
@@ -224,16 +224,28 @@ class ImageDir(BaseLocalImage):
 
         # If extra channels, pick the first stack of the last dimensions
 
-        pixels = img
         while len(img.shape) > 3:
             img = img[..., 0]
+
         if self._meta:
             self._meta["size_x"], self._meta["size_y"] = img.shape[-2:]
 
-            img = da.reshape(img, (*self._dim_values(), *img.shape[1:]))
+            # img = da.reshape(img, (*self._meta, *img.shape[1:]))
+            img = da.reshape(img, self._meta.values())
             pixels = self.format_data(img)
         return pixels
 
+    @property
+    def name(self):
+        return self.path.stem
+
+    @property
+    def dimorder(self):
+        # Assumes only dimensions start with "size"
+        return [
+            k.split("_")[-1] for k in self._meta.keys() if k.startswith("size")
+        ]
+
 
 class Image(BridgeOmero):
     """
-- 
GitLab