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