fix(py): set initial buffered region with py constructed data

Author: thewtex

packages/core/python/itkwasm/itkwasm/image.py

@@ -34,11 +34,15 @@ class ImageRegion:
 @dataclass
 class Image:
     imageType: Union[ImageType, Dict] = field(default_factory=ImageType)
+
     name: str = "Image"
+
     origin: Sequence[float] = field(default_factory=list)
     spacing: Sequence[float] = field(default_factory=list)
     direction: ArrayLike = field(default_factory=_default_direction)
+
     size: Sequence[int] = field(default_factory=list)
+
     metadata: Dict = field(default_factory=dict)
     data: Optional[ArrayLike] = None
     bufferedRegion: Optional[ImageRegion] = None
@@ -67,10 +71,18 @@ def __post_init__(self):
             ] * dimension
 
         if self.bufferedRegion is None:
-            self.bufferedRegion = ImageRegion(
-                index=[
-                    0,
-                ]
-                * dimension,
-                size=self.size,
-            )
+            if self.data is not None:
+                self.bufferedRegion = ImageRegion(
+                    index=(0,) * dimension,
+                    size=self.data.shape[:dimension][::-1],
+                )
+            else:
+                self.bufferedRegion = ImageRegion(
+                    index=[
+                        0,
+                    ]
+                    * dimension,
+                    size=self.size,
+                )
+        elif isinstance(self.bufferedRegion, dict):
+            self.bufferedRegion = ImageRegion(**self.bufferedRegion)

packages/core/python/itkwasm/itkwasm/image_from_array.py

@@ -5,7 +5,7 @@
 except ImportError:
     from numpy import ndarray as ArrayLike
 
-from .image import Image, ImageType
+from .image import Image, ImageType, ImageRegion
 from .to_numpy_array import _dtype_to_component_type
 
 from .pixel_types import PixelTypes
@@ -42,5 +42,9 @@ def image_from_array(arr, is_vector: bool = False, image_type: Optional[ImageTyp
     image = Image(imageType=image_type)
     image.size = arr.shape[:dimension][::-1]
     image.data = arr
+    image.bufferedRegion = ImageRegion(
+        index=(0,) * dimension,
+        size=image.size,
+    )
 
     return image