fix: patch warp struct

Author: mgxd

nitransforms/io.py

@@ -1,8 +1,9 @@
 import numpy as np
-from nibabel.wrapstruct import LabeledWrapStruct
 from nibabel.volumeutils import Recoder
 from nibabel.affines import voxel_sizes
 
+from .patched import LabeledWrapStruct
+
 transform_codes = Recoder((
     (0, 'LINEAR_VOX_TO_VOX'),
     (1, 'LINEAR_RAS_TO_RAS'),
@@ -150,7 +151,7 @@ def from_string(klass, string):
             val = np.genfromtxt([valstring.encode()],
                                 dtype=klass.dtype[key])
             sa[key] = val.reshape(sa[key].shape)
-        assert lines.pop(0) == '1 4 4'
+        assert lines.pop(0) == '1 4 4'  # xforms, shape + 1, shape + 1
         val = np.genfromtxt([valstring.encode() for valstring in lines[:4]],
                             dtype='f4')
         sa['m_L'] = val

nitransforms/patched.py

@@ -1,4 +1,5 @@
 import numpy as np
+from nibabel.wrapstruct import LabeledWrapStruct as LWS
 
 
 def shape_zoom_affine(shape, zooms, x_flip=True, y_flip=False):
@@ -63,3 +64,16 @@ def shape_zoom_affine(shape, zooms, x_flip=True, y_flip=False):
     aff[:3, :3] = np.diag(zooms)
     aff[:3, -1] = -origin * zooms
     return aff
+
+
+class LabeledWrapStruct(LWS):
+    def __setitem__(self, item, value):
+        ''' Set values in structured data
+        Examples
+        --------
+        >>> wstr = WrapStruct()
+        >>> wstr['integer'] = 3
+        >>> wstr['integer']
+        array(3, dtype=int16)
+        '''
+        self._structarr[item] = np.asanyarray(value)