fix: also revise asaffine

Author: oesteban

nitransforms/manip.py

@@ -8,6 +8,7 @@
 ### ### ### ### ### ### ### ### ### ### ### ### ### ### ### ### ### ### ### ##
 """Common interface for transforms."""
 from collections.abc import Iterable
+import numpy as np
 
 from .base import (
     TransformBase,
@@ -139,10 +140,19 @@ def map(self, x, inverse=False):
 
         return x
 
-    def asaffine(self):
-        """Combine a succession of linear transforms into one."""
-        retval = self.transforms[-1]
-        for xfm in self.transforms[:-1][::-1]:
+    def asaffine(self, indices=None):
+        """
+        Combine a succession of linear transforms into one.
+
+        Parameters
+        ----------
+        indices : :obj:`numpy.array_like`
+            The indices of the values to extract.
+
+        """
+        affines = self.transforms if indices is None else np.take(self.transforms, indices)
+        retval = affines[0]
+        for xfm in affines[1:]:
             retval @= xfm
         return retval
 

nitransforms/tests/test_manip.py

@@ -68,16 +68,16 @@ def test_collapse_affines(tmp_path, data_path, ext0, ext1, ext2):
     """Check whether affines are correctly collapsed."""
     chain = TransformChain(
         [
+            Affine.from_filename(
+                data_path / "regressions" / f"from-scanner_to-bold_mode-image.{ext1}",
+                fmt=f"{FMT[ext1]}",
+            ),
             Affine.from_filename(
                 data_path
                 / "regressions"
                 / f"from-fsnative_to-scanner_mode-image.{ext0}",
                 fmt=f"{FMT[ext0]}",
             ),
-            Affine.from_filename(
-                data_path / "regressions" / f"from-scanner_to-bold_mode-image.{ext1}",
-                fmt=f"{FMT[ext1]}",
-            ),
         ]
     )
     assert np.allclose(