TEST: Add FreeSurferSubject geometry collection, test loading Cifti2 as cimg

Author: effigies

nibabel/arrayproxy.py

@@ -173,6 +173,11 @@ def __init__(self, file_like, spec, *, mmap=True, order=None, keep_file_open=Non
             self._should_keep_file_open(file_like, keep_file_open)
         self._lock = RLock()
 
+    def copy(self):
+        spec = self._shape, self._dtype, self._offset, self._slope, self._inter
+        return ArrayProxy(self.file_like, spec,
+                          mmap=self._mmap, keep_file_open=self._keep_file_open)
+
     def __del__(self):
         """If this ``ArrayProxy`` was created with ``keep_file_open=True``,
         the open file object is closed if necessary.

nibabel/tests/test_coordimage.py

@@ -0,0 +1,51 @@
+import os
+from pathlib import Path
+
+import nibabel as nb
+from nibabel import pointset as ps
+from nibabel import coordimage as ci
+from .test_pointset import FreeSurferHemisphere
+from nibabel.tests.nibabel_data import get_nibabel_data
+
+CIFTI2_DATA = Path(get_nibabel_data()) / 'nitest-cifti2'
+
+
+class FreeSurferSubject(ci.GeometryCollection):
+    @classmethod
+    def from_subject(klass, subject_id, subjects_dir=None):
+        """ Load a FreeSurfer subject by ID """
+        if subjects_dir is None:
+            subjects_dir = os.environ["SUBJECTS_DIR"]
+        return klass.from_spec(Path(subjects_dir) / subject_id)
+
+    @classmethod
+    def from_spec(klass, pathlike):
+        """ Load a FreeSurfer subject from its directory structure """
+        subject_dir = Path(pathlike)
+        surfs = subject_dir / "surf"
+        structures = {
+            "lh": FreeSurferHemisphere.from_filename(surfs / "lh.white"),
+            "rh": FreeSurferHemisphere.from_filename(surfs / "rh.white"),
+            }
+        subject = super().__init__(structures)
+        subject._subject_dir = subject_dir
+        return subject
+
+
+def test_Cifti2Image_as_CoordImage():
+    ones = nb.load(CIFTI2_DATA / "ones.dscalar.nii")
+    axes = [ones.header.get_axis(i) for i in range(ones.ndim)]
+
+    parcels = []
+    for name, slicer, bma in axes[1].iter_structures():
+        if bma.volume_shape:
+            substruct = ps.NdGrid(bma.volume_shape, bma.affine)
+            indices = bma.voxel
+        else:
+            substruct = None
+            indices = bma.vertex
+        parcels.append(ci.Parcel(name, None, indices))
+    caxis = ci.CoordinateAxis(parcels)
+    dobj = ones.dataobj.copy()
+    dobj.order = 'C'  # Hack for image with BMA as the last axis
+    cimg = ci.CoordinateImage(dobj, caxis, ones.header)