ref: Drop custom load_ants_h5 for fixed nt.manip.load

Author: effigies

fmriprep/utils/transforms.py

@@ -2,12 +2,7 @@
 
 from pathlib import Path
 
-import h5py
-import nibabel as nb
 import nitransforms as nt
-import numpy as np
-from nitransforms.io.itk import ITKCompositeH5
-from transforms3d.affines import compose as compose_affine
 
 
 def load_transforms(xfm_paths: list[Path], inverse: list[bool]) -> nt.base.TransformBase:
@@ -24,7 +19,8 @@ def load_transforms(xfm_paths: list[Path], inverse: list[bool]) -> nt.base.Trans
     for path, inv in zip(xfm_paths[::-1], inverse[::-1], strict=False):
         path = Path(path)
         if path.suffix == '.h5':
-            xfm = load_ants_h5(path)
+            # Load as a TransformChain
+            xfm = nt.manip.load(path)
         else:
             xfm = nt.linear.load(path)
         if inv:
@@ -36,70 +32,3 @@ def load_transforms(xfm_paths: list[Path], inverse: list[bool]) -> nt.base.Trans
     if chain is None:
         chain = nt.base.TransformBase()
     return chain
-
-
-def load_ants_h5(filename: Path) -> nt.base.TransformBase:
-    """Load ANTs H5 files as a nitransforms TransformChain"""
-    # Borrowed from https://github.com/feilong/process
-    # process.resample.parse_combined_hdf5()
-    #
-    # Changes:
-    #   * Tolerate a missing displacement field
-    #   * Return the original affine without a round-trip
-    #   * Always return a nitransforms TransformBase
-    #   * Construct warp affine from fixed parameters
-    #
-    # This should be upstreamed into nitransforms
-    h = h5py.File(filename)
-    xform = ITKCompositeH5.from_h5obj(h)
-
-    # nt.Affine
-    transforms = [nt.Affine(xform[0].to_ras())]
-
-    if '2' not in h['TransformGroup']:
-        return transforms[0]
-
-    transform2 = h['TransformGroup']['2']
-
-    # Confirm these transformations are applicable
-    if transform2['TransformType'][:][0] not in (
-        b'DisplacementFieldTransform_float_3_3',
-        b'DisplacementFieldTransform_double_3_3',
-    ):
-        msg = 'Unknown transform type [2]\n'
-        for i in h['TransformGroup'].keys():
-            msg += f'[{i}]: {h["TransformGroup"][i]["TransformType"][:][0]}\n'
-        raise ValueError(msg)
-
-    # Warp field fixed parameters as defined in
-    # https://itk.org/Doxygen/html/classitk_1_1DisplacementFieldTransform.html
-    shape = transform2['TransformFixedParameters'][:3]
-    origin = transform2['TransformFixedParameters'][3:6]
-    spacing = transform2['TransformFixedParameters'][6:9]
-    direction = transform2['TransformFixedParameters'][9:].reshape((3, 3))
-
-    # We are not yet confident that we handle non-unit spacing
-    # or direction cosine ordering correctly.
-    # If we confirm or fix, we can remove these checks.
-    if not np.allclose(spacing, 1):
-        raise ValueError(f'Unexpected spacing: {spacing}')
-    if not np.allclose(direction, direction.T):
-        raise ValueError(f'Asymmetric direction matrix: {direction}')
-
-    # ITK uses LPS affines
-    lps_affine = compose_affine(T=origin, R=direction, Z=spacing)
-    ras_affine = np.diag([-1, -1, 1, 1]) @ lps_affine
-
-    # ITK stores warps in Fortran-order, where the vector components change fastest
-    # Vectors are in mm LPS
-    itk_warp = np.reshape(
-        transform2['TransformParameters'],
-        (3, *shape.astype(int)),
-        order='F',
-    )
-
-    # Nitransforms warps are in RAS, with the vector components changing slowest
-    nt_warp = itk_warp.transpose(1, 2, 3, 0) * np.array([-1, -1, 1])
-
-    transforms.insert(0, nt.DenseFieldTransform(nb.Nifti1Image(nt_warp, ras_affine)))
-    return nt.TransformChain(transforms)

pyproject.toml

@@ -25,7 +25,7 @@ dependencies = [
     "nipype >= 1.8.5",
     "nireports >= 23.2.2",
     "nitime",
-    "nitransforms >= 21.0.0, != 24.0.0",
+    "nitransforms >= 24.0.2",
     "niworkflows >= 1.11.0",
     "numpy >= 1.22",
     "packaging",