Add image_like function for generic SpatialImage

Author: effigies

nibabel/__init__.py

@@ -59,6 +59,7 @@
 from .freesurfer import MGHImage
 from .funcs import (squeeze_image, concat_images, four_to_three,
                     as_closest_canonical)
+from .spatialimages import image_like
 from .orientations import (io_orientation, orientation_affine,
                            flip_axis, OrientationError,
                            apply_orientation, aff2axcodes)

nibabel/spatialimages.py

@@ -606,7 +606,6 @@ def orthoview(self):
         return OrthoSlicer3D(self.dataobj, self.affine,
                              title=self.get_filename())
 
-
     def as_reoriented(self, ornt):
         """Apply an orientation change and return a new image
 
@@ -636,3 +635,11 @@ def as_reoriented(self, ornt):
         new_aff = self.affine.dot(inv_ornt_aff(ornt, self.shape))
 
         return self.__class__(t_arr, new_aff, self.header)
+
+
+def image_like(img, data):
+    ''' Create new SpatialImage with metadata of `img`, and data
+    contained in `data`.
+    '''
+    return img.__class__(data, img.affine, img.header.copy(),
+                         extra=img.extra.copy())

nibabel/tests/test_spatialimages.py

@@ -16,7 +16,7 @@
 
 from io import BytesIO
 from ..spatialimages import (SpatialHeader, SpatialImage, HeaderDataError,
-                             Header, ImageDataError)
+                             Header, ImageDataError, image_like)
 from ..imageclasses import spatial_axes_first
 
 from unittest import TestCase
@@ -659,3 +659,13 @@ class MyHeader(Header):
 
         MyHeader()
         assert_equal(len(w), 1)
+
+
+def test_image_like():
+    zeros = SpatialImage(np.zeros((2, 3, 4)), np.eye(4))
+    ones = image_like(zeros, np.ones((2, 3, 4)))
+
+    assert np.all(ones.dataobj != zeros.dataobj)
+    assert np.all(ones.affine == zeros.affine)
+    assert ones.header == zeros.header
+    assert ones.extra == zeros.extra