impy-array 2.4.7__py3-none-any.whl → 2.4.9__py3-none-any.whl

impy/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "2.4.7"
+__version__ = "2.4.9"
 __author__ = "Hanjin Liu"
 __email__ = "liuhanjin-sc@g.ecc.u-tokyo.ac.jp"
 

impy/_const.py

@@ -8,13 +8,13 @@ MAX_GB_LIMIT = memory.total / 2 * 1e-9
 
 class GlobalConstant(MutableMapping[str, Any]):
     _const: dict[str, Any]
-    
+
     def __init__(self, **kwargs):
         object.__setattr__(self, "_const", dict(**kwargs))
-    
+
     def __len__(self) -> int:
         return len(self._const)
-    
+
     def __iter__(self):
         raise StopIteration
 
@@ -47,15 +47,15 @@ class GlobalConstant(MutableMapping[str, Any]):
                 dask.config.set(scheduler=v)
         else:
             raise RuntimeError("Cannot set new keys.")
-        
+
         self._const[k] = v
-    
+
     __getattr__ = __getitem__
     __setattr__ = __setitem__
 
     def __delitem__(self, v):
         raise RuntimeError("Cannot delete any items.")
-    
+
     def __repr__(self):
         return (
             f"""
@@ -65,12 +65,12 @@ class GlobalConstant(MutableMapping[str, Any]):
                 SCHEDULER   : {self['SCHEDULER']}
             """
         )
-    
+
     def asdict(self) -> dict[str, Any]:
         return self._const.copy()
 
 Const = GlobalConstant(
-    MAX_GB = MAX_GB_LIMIT/2,
+    MAX_GB = MAX_GB_LIMIT * 0.8,
     ID_AXIS = "N",
     RESOURCE = "numpy",
     SCHEDULER = "threads",
@@ -80,12 +80,12 @@ class SetConst:
     n_ongoing = 0
     _locked_keys: set[str] = set()
     _old_dict: dict[str, Any] = dict()
-    
+
     def __init__(self, dict_: dict[str, Any] | None =None, **kwargs):
         dict_ = dict_ or {}
         dict_.update(kwargs)
         self._kwargs = dict_
-    
+
     def __enter__(self):
         self.__class__.n_ongoing += 1
         if self.__class__.n_ongoing == 1:

impy/arrays/_utils/_filters.py

@@ -52,6 +52,8 @@ binary_closing = get_func("binary_closing")
 closing = get_func("grey_closing")
 gaussian_filter = get_func("gaussian_filter")
 median_filter = get_func("median_filter")
+min_filter = get_func("minimum_filter")
+max_filter = get_func("maximum_filter")
 spline_filter = get_func("spline_filter")
 convolve = get_func("convolve")
 white_tophat = get_func("white_tophat")
@@ -69,6 +71,8 @@ if TYPE_CHECKING:
     closing = scipy_ndi.grey_closing
     gaussian_filter = scipy_ndi.gaussian_filter
     median_filter = scipy_ndi.median_filter
+    min_filter = scipy_ndi.minimum_filter
+    max_filter = scipy_ndi.maximum_filter
     spline_filter = scipy_ndi.spline_filter
     convolve = scipy_ndi.convolve
     white_tophat = scipy_ndi.white_tophat

impy/arrays/imgarray.py

@@ -1,10 +1,12 @@
 from __future__ import annotations
 import warnings
+from functools import partial
+from itertools import product
+from typing import TYPE_CHECKING, Literal, Sequence, Iterable, Callable, overload, Any
+
 import numpy as np
 from numpy.typing import DTypeLike
-from functools import partial
 from scipy import ndimage as ndi
-from typing import TYPE_CHECKING, Literal, Sequence, Iterable, Callable, overload, Any
 
 from impy.arrays.labeledarray import LabeledArray
 from impy.arrays.label import Label
@@ -706,8 +708,7 @@ class ImgArray(LabeledArray):
     @dims_to_spatial_axes
     @check_input_and_output
     def structure_tensor_eig(self, sigma: nDFloat = 1, *, dims: Dims = None)-> tuple[ImgArray, ImgArray]:
-        """
-        Calculate structure tensor's eigenvalues and eigenvectors.
+        """Calculate structure tensor's eigenvalues and eigenvectors.
 
         Parameters
         ----------
@@ -747,8 +748,7 @@ class ImgArray(LabeledArray):
         dims: Dims = None,
         update: bool = False
     ) -> ImgArray:
-        """
-        Sobel filter. This filter is useful for edge detection.
+        """Edge detection such as Sobel filter.
 
         Parameters
         ----------
@@ -783,8 +783,7 @@ class ImgArray(LabeledArray):
         dims: Dims = None,
         update: bool = False,
     ) -> ImgArray:
-        """
-        Butterworth low-pass filter.
+        """Butterworth low-pass filter.
 
         Parameters
         ----------
@@ -829,9 +828,10 @@ class ImgArray(LabeledArray):
         dims: Dims = None,
         update: bool = False
     ) -> ImgArray:
-        """
-        Butterworth low-pass filter in real space. Butterworth kernel is created first using inverse
-        Fourier transform of weight function.
+        """Butterworth low-pass filter in real space.
+
+        Butterworth kernel is created first using inverse Fourier transform of weight
+        function.
 
         Parameters
         ----------
@@ -874,8 +874,7 @@ class ImgArray(LabeledArray):
         dims: Dims = None,
         update: bool = False
     ) -> ImgArray:
-        """
-        Butterworth high-pass filter.
+        """Butterworth high-pass filter.
 
         Parameters
         ----------
@@ -916,8 +915,7 @@ class ImgArray(LabeledArray):
         dims: Dims = None,
         update: bool = False
     ):
-        """
-        Butterworth band-pass filter.
+        """Butterworth band-pass filter.
 
         Parameters
         ----------
@@ -961,8 +959,7 @@ class ImgArray(LabeledArray):
         dims: Dims = None,
         update: bool = False,
     ) -> ImgArray:
-        """
-        General linear convolution by running kernel filtering.
+        """General linear convolution by running kernel filtering.
 
         Parameters
         ----------
@@ -996,8 +993,7 @@ class ImgArray(LabeledArray):
         dims: Dims = None,
         update: bool = False,
     ) -> ImgArray:
-        """
-        Morphological erosion.
+        """Morphological erosion.
 
         If input is binary image, the running function will automatically switched to
         ``binary_erosion`` to speed up calculation.
@@ -1037,8 +1033,9 @@ class ImgArray(LabeledArray):
         dims: Dims = None,
         update: bool = False,
     ) -> ImgArray:
-        """
-        Morphological dilation. If input is binary image, the running function will automatically switched to
+        """Morphological dilation.
+
+        If input is binary image, the running function will automatically switched to
         ``binary_dilation`` to speed up calculation.
 
         Parameters
@@ -1076,8 +1073,9 @@ class ImgArray(LabeledArray):
         dims: Dims = None,
         update: bool = False,
     ) -> ImgArray:
-        """
-        Morphological opening. If input is binary image, the running function will automatically switched to
+        """Morphological opening.
+
+        If input is binary image, the running function will automatically switched to
         ``binary_opening`` to speed up calculation.
 
         Parameters
@@ -1115,8 +1113,7 @@ class ImgArray(LabeledArray):
         dims: Dims = None,
         update: bool = False,
     ) -> ImgArray:
-        """
-        Morphological closing.
+        """Morphological closing.
 
         If input is binary image, the running function will automatically switched to
         ``binary_closing`` to speed up calculation.
@@ -1156,8 +1153,7 @@ class ImgArray(LabeledArray):
         dims: Dims = None,
         update: bool = False,
     ) -> ImgArray:
-        """
-        Tophat morphological image processing. This is useful for background subtraction.
+        """Tophat morphological image processing. This is useful for background subtraction.
 
         Parameters
         ----------
@@ -1188,8 +1184,7 @@ class ImgArray(LabeledArray):
         *,
         dims: Dims = None
     ) -> ImgArray:
-        """
-        Smoothen binary mask image at its edges. This is useful to make a "soft mask".
+        """Smoothen binary mask image at its edges. This is useful to make a "soft mask".
 
         This method applies erosion/dilation to a binary image and then smooth its edges by Gaussian.
         The total value is always larger after Gaussian smoothing.
@@ -1241,8 +1236,7 @@ class ImgArray(LabeledArray):
         cval: float = 0.0,
         dims: Dims = None,
         update: bool = False) -> ImgArray:
-        """
-        Mean filter. Kernel is filled with same values.
+        """Mean filter. Kernel is filled with same values.
 
         Parameters
         ----------
@@ -1262,6 +1256,65 @@ class ImgArray(LabeledArray):
             kwargs=dict(mode=mode, cval=cval),
         )
 
+
+    @_docs.write_docs
+    @dims_to_spatial_axes
+    @check_input_and_output
+    def min_filter(
+        self,
+        radius: float = 1,
+        *,
+        mode: PaddingMode = "reflect",
+        cval: float = 0.0,
+        dims: Dims = None,
+    ) -> ImgArray:
+        """Minimum filter.
+
+        Parameters
+        ----------
+        {radius}{mode}{cval}{dims}
+
+        Returns
+        -------
+        ImgArray
+            Filtered image
+        """
+        disk = _structures.ball_like(radius, len(dims))
+        return self.as_float()._apply_dask(
+            _filters.min_filter,
+            c_axes=complement_axes(dims, self.axes),
+            kwargs=dict(footprint=disk, mode=mode, cval=cval),
+        )
+
+    @_docs.write_docs
+    @dims_to_spatial_axes
+    @check_input_and_output
+    def max_filter(
+        self,
+        radius: float = 1,
+        *,
+        mode: PaddingMode = "reflect",
+        cval: float = 0.0,
+        dims: Dims = None,
+    ) -> ImgArray:
+        """Maximum filter.
+
+        Parameters
+        ----------
+        {radius}{mode}{cval}{dims}
+
+        Returns
+        -------
+        ImgArray
+            Filtered image
+        """
+        disk = _structures.ball_like(radius, len(dims))
+        return self.as_float()._apply_dask(
+            _filters.max_filter,
+            c_axes=complement_axes(dims, self.axes),
+            kwargs=dict(footprint=disk, mode=mode, cval=cval),
+        )
+
     @_docs.write_docs
     @dims_to_spatial_axes
     @check_input_and_output
@@ -3987,8 +4040,7 @@ class ImgArray(LabeledArray):
         upsample_factor: int = 10,
         max_shift: nDFloat | None = None,
     ) -> MarkerFrame:
-        """
-        Calculate yx-directional drift using the method equivalent to
+        """Calculate yx-directional drift using the method equivalent to
         ``skimage.registration.phase_cross_correlation``.
 
         Parameters
@@ -4063,9 +4115,10 @@ class ImgArray(LabeledArray):
         dims: Dims = 2,
         update: bool = False,
     ) -> ImgArray:
-        """
-        Drift correction using iterative Affine translation. If translation vectors ``shift``
-        is not given, then it will be determined using ``track_drift`` method of ImgArray.
+        """Drift correction using iterative Affine translation.
+
+        If translation vectors  ``shift`` is not given, then it will be determined using
+        ``track_drift`` method of ImgArray.
 
         Parameters
         ----------
@@ -4092,7 +4145,7 @@ class ImgArray(LabeledArray):
         Examples
         --------
         Drift correction of multichannel image using the first channel as the reference.
-            >>> img.drift_correction(ref=img["c=0"])
+        >>> img.drift_correction(ref=img["c=0"])
         """
 
         if along is None:
@@ -4107,7 +4160,6 @@ class ImgArray(LabeledArray):
             elif not isinstance(ref, ImgArray):
                 ref = self[ref]
             if ref.axes != [along] + dims:
-                from itertools import product
                 _c_axes = complement_axes([along] + dims, str(ref.axes))
                 fmt = slicer.get_formatter(_c_axes)
                 out = np.empty_like(self)

impy/arrays/labeledarray.py

@@ -3,7 +3,7 @@ import numpy as np
 from pathlib import Path
 import itertools
 from functools import partial
-from typing import TYPE_CHECKING, Any, Sequence, MutableMapping, Protocol
+from typing import TYPE_CHECKING, Any, Literal, Sequence, MutableMapping, Protocol
 from warnings import warn
 from scipy import ndimage as ndi
 
@@ -23,7 +23,7 @@ from impy.array_api import xp
 
 if TYPE_CHECKING:
     from typing_extensions import Self
-    from numpy.typing import ArrayLike, DTypeLike
+    from numpy.typing import ArrayLike, DTypeLike, NDArray
     from impy.frame import PathFrame
     from impy.roi import RoiList
 
@@ -401,10 +401,18 @@ class LabeledArray(MetaArray):
         out._set_info(self)
         return out
 
-    def as_float(self) -> Self:
-        if self.dtype == np.float32:
+    def as_float(self, *, depth: int = 32) -> Self:
+        if depth == 16:
+            dtype = np.float16
+        elif depth == 32:
+            dtype = np.float32
+        elif depth == 64:
+            dtype = np.float64
+        else:
+            raise ValueError(f"depth must be 16, 32, or 64, but got {depth}")
+        if self.dtype == dtype:
             return self
-        out = self.value.astype(np.float32).view(self.__class__)
+        out = self.value.astype(dtype).view(self.__class__)
         out._set_info(self)
         return out
 
@@ -417,8 +425,12 @@ class LabeledArray(MetaArray):
             return self.as_uint16()
         elif dtype == "uint8":
             return self.as_uint8()
+        elif dtype == "float16":
+            return self.as_float(depth=16)
         elif dtype == "float32":
-            return self.as_float()
+            return self.as_float(depth=32)
+        elif dtype == "float64":
+            return self.as_float(depth=64)
         elif dtype == "bool":
             return self.astype("bool")
         elif dtype == "float64":
@@ -576,6 +588,7 @@ class LabeledArray(MetaArray):
         *,
         order: int = 3,
         prefilter: bool | None = None,
+        how: Literal["pack", "equal"] = "pack",
     ) -> PropArray:
         """
         Measure line profile (kymograph) iteratively for every slice of image. This
@@ -590,7 +603,12 @@ class LabeledArray(MetaArray):
             `a = [[y0, x0], [y1, x1], ..., [yn, xn]]`
         b : array-like, optional
             Destination coordinate. If specified, `a` must be the source coordinate.
-        {order}
+        {order}{prefilter}
+        how : "pack" or "equal", default "pack"
+            How to sample points on the line. "pack" means that the points are sampled
+            at 1-px interval from the start point. "equal" means that the points are
+            sampled equally at the interval larger but neareset to 1-px so that the
+            points covers the entire line.
 
         Returns
         -------
@@ -616,7 +634,12 @@ class LabeledArray(MetaArray):
         a = np.asarray(a, dtype=np.float32)
         _, ndim = a.shape
         seg = SegmentedLine(a)
-        coords = seg.sample_points().T
+        if how == "pack":
+            coords = np.stack(seg.arange(), axis=0)
+        elif how == "equal":
+            coords = np.stack(seg.linspace(int(seg.length())), axis=0)
+        else:
+            raise ValueError(f"No option matches how={how}")
 
         if ndim == self.ndim:
             dims = self.axes
@@ -632,7 +655,11 @@ class LabeledArray(MetaArray):
                         axes=c_axes+[new_axis], propname="reslice")
 
         out.set_scale(self)
-        out.set_scale({new_axis: self.scale[dims[-1]] * seg.interv})
+        axis = self.axes[dims[-1]]
+        if how == "equal":
+            out.set_scale({new_axis: axis.scale * seg.interv}, unit=axis.unit)
+        else:
+            out.set_scale({new_axis: axis.scale}, unit=axis.unit)
         return out
 
     @_docs.write_docs
@@ -1576,31 +1603,61 @@ class SegmentedLine:
         dist = np.sqrt(np.sum(vec**2, axis=1))
         dist_sum = np.sum(dist)
         npoints = int(dist_sum)
-        interv = dist_sum / npoints
+        self.interv = dist_sum / npoints
 
-        self.length = dist_sum
         self.vec = vec
         self.dist = dist
         self.nodes = nodes
-        self.interv = interv
 
     def sample_points(self) -> np.ndarray:
         res = 0
         out = [self.nodes[0:1]]
         npoints = 1
+        interv = self.length() / int(self.length())
         for v, d, p in zip(self.vec, self.dist, self.nodes[:-1]):
             res0 = res
-            d_int, res = divmod(d + res, self.interv)
-            idx = self.interv*(np.arange(d_int) + 1) - res0
+            d_int, res = divmod(d + res, interv)
+            idx = interv * (np.arange(d_int) + 1) - res0
             xs = idx[:, np.newaxis] * v[np.newaxis]/d + p
             out.append(xs)
             npoints += xs.shape[0]
 
-        if npoints <= int(self.length):
+        if npoints <= int(self.length()):
             out.append(self.nodes[-1:])
 
         return np.concatenate(out, axis=0)
 
+    def length(self) -> np.float64:
+        return np.sum(self.lengths())
+
+    def lengths(self) -> NDArray[np.float64]:
+        return np.sqrt(np.sum(np.diff(self.nodes, axis=0) ** 2, axis=1))
+
+    def linspace(self, num: int) -> tuple[NDArray[np.float64], ...]:
+        """Return a tuple of x and y coordinates of np.linspace along the line."""
+        tnots = np.cumsum(
+            np.concatenate([[0], self.lengths()], dtype=np.float64)
+        )
+        teval = np.linspace(0, tnots[-1], num)
+        return tuple(
+            np.interp(teval, tnots, self.nodes[:, i])
+            for i in range(self.nodes.shape[1])
+        )
+
+    def arange(
+        self, step: float = 1.0
+    ) -> tuple[NDArray[np.float64], NDArray[np.float64]]:
+        tnots = np.cumsum(
+            np.concatenate([[0], self.lengths()], dtype=np.float64)
+        )
+        length = tnots[-1]
+        num, rem = divmod(length, step)
+        teval = np.linspace(0, length - rem, int(num + 1))
+        return tuple(
+            np.interp(teval, tnots, self.nodes[:, i])
+            for i in range(self.nodes.shape[1])
+        )
+
 def _count_list_depth(x) -> int:
     n = 0
     out = x

impy/arrays/lazy.py

@@ -890,6 +890,47 @@ class LazyImgArray(AxesMixin):
             kwargs=dict(footprint=disk, mode=mode, cval=cval),
         )
 
+    @_docs.copy_docs(ImgArray.min_filter)
+    @dims_to_spatial_axes
+    @same_dtype
+    @check_input_and_output_lazy
+    def min_filter(
+        self,
+        radius: float = 1,
+        *,
+        mode: PaddingMode = "reflect",
+        cval: float = 0,
+        dims: Dims = None,
+        update: bool = False
+    ) -> LazyImgArray:
+        disk = _structures.ball_like(radius, len(dims))
+        return self._apply_map_overlap(
+            xp.ndi.minimum_filter,
+            depth=_ceilint(radius),
+            c_axes=complement_axes(dims, self.axes),
+            kwargs=dict(footprint=disk, mode=mode, cval=cval),
+        )
+
+    @_docs.copy_docs(ImgArray.max_filter)
+    @dims_to_spatial_axes
+    @same_dtype
+    @check_input_and_output_lazy
+    def max_filter(
+        self,
+        radius: float = 1,
+        *,
+        mode: PaddingMode = "reflect",
+        cval: float = 0,
+        dims: Dims = None,
+        update: bool = False
+    ) -> LazyImgArray:
+        disk = _structures.ball_like(radius, len(dims))
+        return self._apply_map_overlap(
+            xp.ndi.maximum_filter,
+            depth=_ceilint(radius),
+            c_axes=complement_axes(dims, self.axes),
+            kwargs=dict(footprint=disk, mode=mode, cval=cval),
+        )
 
     @_docs.copy_docs(ImgArray.std_filter)
     @dims_to_spatial_axes
@@ -1612,10 +1653,18 @@ class LazyImgArray(AxesMixin):
         out._set_info(self)
         return out
 
-    def as_float(self) -> LazyImgArray:
-        if self.dtype == np.float32:
+    def as_float(self,  *, depth: int = 32) -> LazyImgArray:
+        if depth == 16:
+            dtype = np.float16
+        elif depth == 32:
+            dtype = np.float32
+        elif depth == 64:
+            dtype = np.float64
+        else:
+            raise ValueError(f"depth must be 16, 32, or 64, but got {depth}")
+        if self.dtype == dtype:
             return self
-        out = self.value.astype(np.float32)
+        out = self.value.astype(dtype)
         out = self.__class__(out)
         out._set_info(self)
         return out
@@ -1628,12 +1677,12 @@ class LazyImgArray(AxesMixin):
             return self.as_uint16()
         elif dtype == "uint8":
             return self.as_uint8()
+        elif dtype == "float16":
+            return self.as_float(depth=16)
         elif dtype == "float32":
-            return self.as_float()
+            return self.as_float(depth=32)
         elif dtype == "float64":
-            warn("Data type float64 is not valid for images. It was converted to float32 instead",
-                 UserWarning)
-            return self.as_float()
+            return self.as_float(depth=64)
         elif dtype == "complex64":
             out = self.value.astype(np.complex64)
             out = self.__class__(out)

impy/core.py

@@ -99,8 +99,7 @@ def array(
     axes: str = None,
     like: MetaArray | None = None,
 ) -> ImgArray:
-    """
-    make an ImgArray object, like ``np.array(x)``
+    """Make an ImgArray object, like ``np.array(x)``
 
     Parameters
     ----------
@@ -116,12 +115,7 @@ def array(
 
     """
     if isinstance(arr, np.ndarray) and dtype is None:
-        if arr.dtype in (np.uint8, np.uint16, np.float32):
-            dtype = arr.dtype
-        elif arr.dtype.kind == "f":
-            dtype = np.float32
-        else:
-            dtype = arr.dtype
+        dtype = arr.dtype
 
     axes, name = _normalize_params(axes, name, like)
 
@@ -150,8 +144,7 @@ def asarray(
     axes: str | None = None,
     like: MetaArray | None = None,
 ) -> ImgArray:
-    """
-    make an ImgArray object, like ``np.asarray(x)``
+    """Make an ImgArray object, like ``np.asarray(x)``
 
     Parameters
     ----------
@@ -296,9 +289,8 @@ def _inject_numpy_function(func: Callable[_P, Any | None]) -> Callable[_P, ImgAr
         return asarray(npfunc(*args, **kwargs), name=name, axes=axes, like=like)
 
     _func.__doc__ = (
-        f"""
-        impy version of numpy.{func.__name__}. This function has additional parameters ``axes``
-        and ``name``. Original docstring follows.
+        f"""impy version of numpy.{func.__name__}. This function has additional
+        parameters ``axes`` and ``name``. Original docstring follows.
 
         Additional Parameters
         ---------------------
@@ -396,7 +388,7 @@ def gaussian_kernel(
     """
     if np.isscalar(sigma):
         sigma = (sigma,)*len(shape)
-    g = gauss.GaussianParticle([(np.array(shape)-1)/2, sigma, peak, 0])
+    g = gauss.GaussianParticle([(np.array(shape) - 1) / 2, sigma, peak, 0])
     ker = g.generate(shape)
     ker = array(ker, name=name or "Gaussian-Kernel", axes=axes)
     if axes is None and ker.ndim == 3:
@@ -551,12 +543,12 @@ def imread(
         return _imread_stack(path, dtype=dtype, key=key, squeeze=squeeze)
     elif "*" in path:
         return _imread_glob(path, dtype=dtype, key=key, squeeze=squeeze)
-    elif not os.path.exists(path):
+    elif not Path(path).exists():
         raise FileNotFoundError(f"No such file or directory: {path}")
 
     # read tif metadata
     if not is_memmap:
-        size = os.path.getsize(path) / 2**30
+        size = Path(path).stat().st_size / 2**30
         if size > Const["MAX_GB"]:
             raise MemoryError(f"Too large {size:.2f} GB")
 

impy/io/_mrc.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+from pathlib import Path
 from typing import TYPE_CHECKING
 import warnings
 import os
@@ -59,7 +60,13 @@ def _(path: str, img: ImpyArray, lazy: bool = False):
 
     if img.dtype == "bool":
         img = img.astype(np.int8)
-
+    elif img.dtype == "float64":
+        warnings.warn(
+            "MRC format does not support float64. Converting to float32.",
+            UserWarning,
+            stacklevel=2,
+        )
+        img = img.astype(np.float32)
     if lazy:
         from dask import array as da
 
@@ -75,9 +82,10 @@ def _(path: str, img: ImpyArray, lazy: bool = False):
     # get voxel_size
     if img.axes not in ("zyx", "yx"):
         raise ImageAxesError(
-            f"Can only save zyx- or yx- image as a mrc file, but image has {img.axes} axes."
-            )
-    if os.path.exists(path):
+            f"Can only save zyx- or yx- image as a mrc file, but image has {img.axes} "
+            "axes."
+        )
+    if Path(path).exists():
         with mrcfile.open(path, mode="r+") as mrc:
             mrc.set_data(img.value)
             mrc.voxel_size = voxel_size
@@ -98,13 +106,12 @@ _MRC_MODE = {
 }
 
 def _parse_mrcfile(mrc: MrcObject) -> ImageMetadata:
-    ndim = len(mrc.voxel_size.item())
-    if ndim == 3:
-        axes = "zyx"
-    elif ndim == 2:
+    if mrc.is_single_image():
         axes = "yx"
+    elif mrc.is_volume_stack():
+        axes = "tzyx"
     else:
-        raise RuntimeError(f"ndim = {ndim} not supported")
+        axes = "zyx"
 
     scale = dict.fromkeys(axes, 1.0)
     for a in axes:

impy/io/_tiff.py

@@ -14,7 +14,7 @@ from impy.utils.axesop import complement_axes
 if TYPE_CHECKING:
     from tifffile import TiffFile
 
-@IO.mark_reader(".tif", ".tiff")
+@IO.mark_reader(".tif", ".tiff", ".lsm")
 def _(path: str, memmap: bool = False) -> ImageData:
     """The tif file reader."""
     from tifffile import TiffFile
@@ -28,7 +28,7 @@ def _(path: str, memmap: bool = False) -> ImageData:
 
     return ImageData.from_metadata(image, meta)
 
-@IO.mark_header_reader(".tif", ".tiff")
+@IO.mark_header_reader(".tif", ".tiff", ".lsm")
 def _(path: str) -> ImageMetadata:
     """The tif header reader."""
     from tifffile import TiffFile
@@ -39,7 +39,7 @@ def _(path: str) -> ImageMetadata:
     return meta
 
 
-@IO.mark_writer(".tif", ".tiff")
+@IO.mark_writer(".tif", ".tiff", ".lsm")
 def _(path: str, img: ImpyArray, lazy: bool = False):
     """The TIFF writer."""
     from tifffile import imwrite, memmap
@@ -103,7 +103,7 @@ def _parse_tifffile(tif: TiffFile) -> ImageMetadata:
     tags = {v.name: v.value for v in pagetag.values()}
 
     labels = ijmeta.get("Labels")
-    scale = dict()
+    scale: dict[str, float] = {}
     dz = ijmeta.get("spacing", 1.0)
     scale_unit: dict[str, float] = {}
     try:
@@ -133,9 +133,9 @@ def _parse_tifffile(tif: TiffFile) -> ImageMetadata:
     _unit = ijmeta.pop("unit", None)
     for k in scale.keys():
         scale_unit[k] = _unit
-    if tif.is_ome:
+    if ome_meta := tif.ome_metadata:
         try:
-            xml = xml2dict(tif.ome_metadata)
+            xml = xml2dict(ome_meta)
             meta_data: dict[str, Any] = xml["OME"]["Image"]
             if isinstance(meta_data, list):
                 meta_data = meta_data[0]
@@ -156,6 +156,14 @@ def _parse_tifffile(tif: TiffFile) -> ImageMetadata:
 
         except Exception:
             pass
+    if lsm_meta := tif.lsm_metadata:
+        scale_unit = "μm"
+        if voxel_size_x := lsm_meta.get("VoxelSizeX"):
+            scale["x"] = voxel_size_x * 1e6
+        if voxel_size_y := lsm_meta.get("VoxelSizeY"):
+            scale["y"] = voxel_size_y * 1e6
+        if voxel_size_z := lsm_meta.get("VoxelSizeZ"):
+            scale["z"] = voxel_size_z * 1e6
 
     return ImageMetadata(
         axes=axes,
@@ -188,7 +196,7 @@ def _get_ijmeta_from_img(img: ImpyArray, update_lut=True):
         info = {}
     metadata["Info"] = str(info)
     scale_unit = img.scale_unit
-    if scale_unit[0] in ("μ", "\xb5", "\xc2"):
+    if scale_unit and scale_unit[0] in ("μ", "\xb5", "\xc2"):
         scale_unit = "\\u00B5" + scale_unit[1:]
     metadata["unit"] = scale_unit
 

impy/lazy/core.py

@@ -108,8 +108,7 @@ def array(
     axes: str = None,
     like: MetaArray | None = None,
 ) -> LazyImgArray:
-    """
-    make an LazyImgArray object, like ``np.array(x)``
+    """Make an LazyImgArray object, like ``np.array(x)``
 
     Parameters
     ----------
@@ -144,12 +143,7 @@ def array(
         arr = da.asarray(arr)
 
     if isinstance(arr, np.ndarray) and dtype is None:
-        if arr.dtype in (np.uint8, np.uint16, np.float32):
-            dtype = arr.dtype
-        elif arr.dtype.kind == "f":
-            dtype = np.float32
-        else:
-            dtype = arr.dtype
+        dtype = arr.dtype
 
     axes, name = _normalize_params(axes, name, like)
 
@@ -172,8 +166,7 @@ def asarray(
     like: MetaArray | None = None,
     chunks="auto",
 ) -> LazyImgArray:
-    """
-    make an LazyImgArray object, like ``np.asarray(x)``
+    """Make an LazyImgArray object, like ``np.asarray(x)``
 
     Parameters
     ----------

impy/random.py

@@ -127,7 +127,7 @@ class ImageGenerator:
     def standard_normal(
         self,
         size: int | tuple[int, ...] | None = None,
-        dtype = None,
+        dtype=np.float32,
         *,
         axes: AxesLike | None = None,
         name: str | None = None,
@@ -142,7 +142,7 @@ class ImageGenerator:
     def standard_exponential(
         self,
         size: int | tuple[int, ...] | None = None,
-        dtype = None,
+        dtype=np.float32,
         method: Literal["zig", "inv"] = None,
         *,
         axes: AxesLike | None = None,
@@ -159,7 +159,7 @@ class ImageGenerator:
     def random(
         self,
         size: int | tuple[int, ...] | None = None,
-        dtype = None,
+        dtype=np.float32,
         *,
         axes: AxesLike | None = None,
         name: str | None = None,
@@ -187,7 +187,7 @@ class ImageGenerator:
         arr = self._rng.normal(loc=loc, scale=scale, size=size)
         if np.isscalar(arr):
             return arr
-        return asarray(xp.asnumpy(arr), axes=axes, name=name)
+        return asarray(xp.asnumpy(arr), axes=axes, name=name).astype(np.float32)
 
     def poisson(
         self,

impy/utils/gauss.py

@@ -19,7 +19,10 @@ def diagonal_gaussian(r, *params):
     ndim = len(params[:-2])//2
     mu = params[:ndim]
     sg = params[ndim:-2]
-    z_value = np.array([(x0 - mu0)/sg0 for x0, mu0, sg0 in zip(r, mu, sg)])
+    z_value = np.array(
+        [(x0 - mu0) / sg0 for x0, mu0, sg0 in zip(r, mu, sg)],
+        dtype=np.float32
+    )
     return a * np.exp(-np.sum(z_value**2, axis=0) / 2) + b
 
 class Gaussian:

{impy_array-2.4.7.dist-info → impy_array-2.4.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: impy-array
-Version: 2.4.7
+Version: 2.4.9
 Summary: Speed up coding/extending image analysis in Python.
 Project-URL: Download, https://github.com/hanjinliu/impy
 Author-email: Hanjin Liu <liuhanjin-sc@g.ecc.u-tokyo.ac.jp>
@@ -131,6 +131,7 @@ img_prj.imsave("image_max.tif")        # Save in the same place. Don't spend tim
 `impy` automatically chooses the proper reader/writer according to the extension.
 
 - Tiff file (".tif", ".tiff")
+- LSM file (".lsm")
 - MRC file (".mrc", ".rec", ".st", ".map", ".map.gz")
 - Zarr file (".zarr")
 - ND2 file (".nd2")

{impy_array-2.4.7.dist-info → impy_array-2.4.9.dist-info}/RECORD

@@ -1,22 +1,22 @@
-impy/__init__.py,sha256=ElW_q_sd5NE1DWtX7zciQLcgl47n3a55WpQwq2ChVLk,1638
+impy/__init__.py,sha256=hoRBdnPuuB2oQ4-ECrHuGN3vObmgt0R40T_uxML6VwQ,1638
 impy/__main__.py,sha256=BAQkQ4X8EFo42pbNx__EqpZ5KL4MTfTUWsVbJIQF0O0,4974
-impy/_const.py,sha256=XFkgkyqs_clQJDqr2vdh5rCyfM83iMFChOLJjwNWHtM,3736
+impy/_const.py,sha256=uNHxxY5ltZuOsJWpwlBWFmUM0QfMgIk5WODqlHuBomo,3700
 impy/_types.py,sha256=gO4l0hx9qxe6eI0PyC4ukW3k4IYxCEZ9gT-rQ3jj3zU,808
 impy/array_api.py,sha256=9suP_Wck9maSiEa5qK6aZrCfaky6xhCMfIRcc9WuIs8,5585
 impy/binder.py,sha256=CTMb11g6uYDVVQV5Wtef865IO0G2V3vGXc8SiGYRD5k,9651
 impy/collections.py,sha256=2e55bLjZN7W4LFUCkyxekcDfLwZodMaU4JG5NQ3RS54,10946
-impy/core.py,sha256=UcMY7vQpp9l1CW_hukexqx92RZcEBWR48Ce_bVrQv0U,25809
+impy/core.py,sha256=itTwflodeyUJf5A4mK-XyrrQ1TGiFz0QU4w8Eel5iFg,25615
 impy/correlation.py,sha256=Yw1T_0eL9FmsMcnCysc49oRJQMn8ho5z9bAzNg5i59Y,28232
-impy/random.py,sha256=RJZAgXp3gkOcCsKbAbuMI1pDdA4_atY3mausFxPNWpk,6973
+impy/random.py,sha256=MQl0ViN6A7xDtM8BS4gbv1atgPKsrmp0TXzV1OtsDlU,7004
 impy/roi.py,sha256=p9ttjBl5SGHHXj0xg5UjPlg_V9cY3MnNZxFfYIjTYsg,14164
 impy/arrays/__init__.py,sha256=IHpTBIJysomS64rOZsv960YBmM1vc4M-asTSs1OJBWg,454
 impy/arrays/_imshow.py,sha256=6mJ_PDqmE4UZO9e9SUgyY_YrzQdR2QKlTG7KF_aC_tI,6949
 impy/arrays/axesmixin.py,sha256=9PK6kMZAfNVipTL65aF5-yR3qt2fy4ATl1OACXLYrAE,10596
 impy/arrays/big.py,sha256=tHNijyPgyIfTltLP3hQwZdHOpSX3IwG5DHPamajZ8Kg,2745
-impy/arrays/imgarray.py,sha256=MxPVVdxnsJDqNVawSykNkFr2y6Cb1ESVYeCzAQXzrN4,159129
+impy/arrays/imgarray.py,sha256=jJQTAWxua0-MmQNIzfL1ducm6VxQeZQEeuSTSMYFNkY,160457
 impy/arrays/label.py,sha256=l0SZXcZxU8qwUgj0QZPZ32XsNaoBtenlrn6hNR03SPI,6406
-impy/arrays/labeledarray.py,sha256=ZbwgP1Fk-lzFckQ12aQO7QnjNMxpmn_y6E3j3DMLCl0,57081
-impy/arrays/lazy.py,sha256=hYTZeaW6HEq0CgPcpfqLcLWSpBrJli6ZTN_ddH_56ro,59902
+impy/arrays/labeledarray.py,sha256=vbWUONnMBOXwLZBHlO2fjl9zqF1nljVdq1y0EFc2ias,59456
+impy/arrays/lazy.py,sha256=XrslWD3Eg0QSD3pKkbvuGDyHJpL-gGsNxbJwTIByuYU,61437
 impy/arrays/phasearray.py,sha256=IojWaNs5sEujn6Lhgqe00H0xF-4CZEFWt5cVW6U_DWQ,13981
 impy/arrays/specials.py,sha256=4sNg_zeJUce_rwnasPoyL2eZ0-gx_4Z-KwitmLB4a5k,7896
 impy/arrays/tiled.py,sha256=T44sS6v1Ovln1z1g61Eg8irgZR1Jmpb6BCY3nS6TmDE,9400
@@ -24,7 +24,7 @@ impy/arrays/_utils/__init__.py,sha256=M5HogfyUBvV8kp-X1ZthS8vDLUP7-O2f9Gvsc1ZXdT
 impy/arrays/_utils/_corr.py,sha256=ThGQV3KSv8ir8BTWDoV83ysalPB2p--LmSCv-J9qq2w,10047
 impy/arrays/_utils/_deconv.py,sha256=5UM7CoRpWphEYuuiq89bmFCKU-n1G_ArMCkDq7tamdQ,2754
 impy/arrays/_utils/_docs.py,sha256=AT6tRxXcX433FCfPXhE9wsaeduy_Um44KOqrYHYTFZE,4968
-impy/arrays/_utils/_filters.py,sha256=aigQnaURbUnmOXGDxkWksISWQbKAUdwtS8kzbaMzgoQ,7226
+impy/arrays/_utils/_filters.py,sha256=pVkL4tuH8u51sc4fMS9U1edne4GdWYB9_l31ZtXkrtQ,7394
 impy/arrays/_utils/_glcm.py,sha256=AoUP-p1MPtAzP0Eyt_DEE_wZYwezNICzdkXMBY-OEBo,4150
 impy/arrays/_utils/_linalg.py,sha256=06jnv-P6nwUjYVTHX2f9i-KDSYZ0bxwNfWWP8A8e_Vw,3233
 impy/arrays/_utils/_misc.py,sha256=vyyw-G4iu7RiNRAJWyvWWGRalWlcMy5Kg_0VvLBBPq8,2953
@@ -44,28 +44,28 @@ impy/axes/_slicer.py,sha256=Ts4m9V3PBlEb3xeHd6j1XdMEb1SwYGbBF5rnC9WWYZE,2657
 impy/frame/__init__.py,sha256=Z4XfXzFSDq49tMrVoeYx1ThnHyw43iLVSVcte6b6-as,65
 impy/frame/frames.py,sha256=7NxDeuaG8CxzMnzczarmLz6ANBpl1tZIhn8KwdvBYg8,12450
 impy/io/__init__.py,sha256=wp1EOl0O6anbkxpsh_lW2V898l-UPIKiOEkLw1q5r-I,303
-impy/io/_mrc.py,sha256=DTWVpRiG2aUd-nlheyAOl5uILfrC0nYnX8bdVRyeUZ8,3392
+impy/io/_mrc.py,sha256=JiPIKar62GhMlZ8SRVbSPcLBbamdQ_2WfuG1_abI7Xg,3611
 impy/io/_nd2.py,sha256=Squ8-wbCquJ8WfROEUZka26jrCkNNE24OTg_ottY7IY,1380
 impy/io/_npy.py,sha256=G6fnjA2RkV7ohFV-R1-TA6o9FjdfcTBwLiccfYj-iVo,2222
 impy/io/_registry.py,sha256=xAD6M46uJhEm9aFO_B0ZFsvRAMNnfXPRd4fPz_awaJI,10251
-impy/io/_tiff.py,sha256=p2u1c0qx0wwcWQQhPXHy6fWpLkPHNeZYn3aMvCcyg5Y,6096
+impy/io/_tiff.py,sha256=kfRB6H-fKBk-lrv_y6tY2fNbG_BEcPm2Ja41-0EtZ_8,6528
 impy/io/_utils.py,sha256=HHyuVqds52pod4Y-CyRG4FVBmonQVBm7OAFj4QqhRCw,3604
 impy/io/_zarr.py,sha256=Nfr5VJ3DzKN5d7nYoq_ZG9AG2GIDWHYJ33LpG9SwGcs,1336
 impy/lazy/__init__.py,sha256=hRjpE2zCy5qDAr_27imwRM9J0C7q-a_b5wfC3ux8ZLE,59
-impy/lazy/core.py,sha256=_B-ELfmy8dMoMNFiAyb7vtmravJEjtAHF-AxtwPCnzs,9828
+impy/lazy/core.py,sha256=8pU0_T6IAN1rUNc7EbdORz2vqof0bFuUHQ7aJGztW7s,9637
 impy/lazy/random.py,sha256=2rwHRsHPOvi8tI9kTtGuZ5SzN0MA7yn-30bZSvHeLVc,7713
 impy/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 impy/utils/axesop.py,sha256=c86XdNlUBFLrwHhAT2j0eELxopNe2QEVaNeOQhXr2do,3076
 impy/utils/deco.py,sha256=lItN0UdadZsjI9eW3fI402k7DNR9jcnD6ZgjdkYsDdg,5946
-impy/utils/gauss.py,sha256=_jvmaUYSRB8KvX9Qp36rhNaNw9lllf68g_43wshpAMo,4713
+impy/utils/gauss.py,sha256=a5C-Ig3aX8SyjqLuQ8VwdjQHdbkNC0LAiKSikTBXZtU,4758
 impy/utils/misc.py,sha256=zV_NeqEaiTW4Rki5fqWfspQjczRaoTJ5MxTOUx6qstY,622
 impy/utils/slicer.py,sha256=otDrizWdzoUjFUMlsha0zNFSS4TkgAmLPOAotaTVaUA,4316
 impy/utils/utilcls.py,sha256=3Td9G2ZkU4C0BiW2LH0sq7RBjdco_BtyxNr-s-seLhs,437
 impy/viewer/__init__.py,sha256=MH-jhpFxP8JVJwUzbKMLRhhWSEYavVHuik9uOS3dDNU,292
 impy/viewer/utils.py,sha256=_RWV9LePbmYeC8qRBUBf3qG70BaLOndlzRLcvUYgTtE,12248
 impy/viewer/viewer.py,sha256=Sfq-obSLoHkBkIW0NDWCxEEaX7-YsC4ifAJ-nxpvJws,17632
-impy_array-2.4.7.dist-info/METADATA,sha256=pQoD1_kxoBzLFncPKdriWz4-mLvI2DUt_jo3u7Pl0CA,7547
-impy_array-2.4.7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-impy_array-2.4.7.dist-info/entry_points.txt,sha256=reyrgy4odXvi1CcPOIIyRYBmsNGG2tO7ZkFOdG5CBkU,44
-impy_array-2.4.7.dist-info/licenses/LICENSE,sha256=rpxKafH9r42sxLLYYB5VbZBHOdtYeLuYJjVU1LBwrxk,1546
-impy_array-2.4.7.dist-info/RECORD,,
+impy_array-2.4.9.dist-info/METADATA,sha256=2bE_GlsO3-oxky5vfOAddrBg5-uDzoIYYjihUqzzEiQ,7567
+impy_array-2.4.9.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+impy_array-2.4.9.dist-info/entry_points.txt,sha256=reyrgy4odXvi1CcPOIIyRYBmsNGG2tO7ZkFOdG5CBkU,44
+impy_array-2.4.9.dist-info/licenses/LICENSE,sha256=rpxKafH9r42sxLLYYB5VbZBHOdtYeLuYJjVU1LBwrxk,1546
+impy_array-2.4.9.dist-info/RECORD,,