maint: restructure submodule io to contain submodules

Author: oesteban

nitransforms/io/__init__.py

@@ -0,0 +1,22 @@
+# emacs: -*- mode: python-mode; py-indent-offset: 4; indent-tabs-mode: nil -*-
+# vi: set ft=python sts=4 ts=4 sw=4 et:
+### ### ### ### ### ### ### ### ### ### ### ### ### ### ### ### ### ### ### ##
+#
+#   See COPYING file distributed along with the NiBabel package for the
+#   copyright and license terms.
+#
+### ### ### ### ### ### ### ### ### ### ### ### ### ### ### ### ### ### ### ##
+"""
+Read and write transforms.
+
+.. currentmodule:: nitransforms
+
+.. autosummary::
+   :toctree: ../generated
+
+   transform
+"""
+from .lta import LinearTransform, LinearTransformArray, VolumeGeometry
+
+
+__all__ = ['LinearTransform', 'LinearTransformArray', 'VolumeGeometry']

nitransforms/io/base.py

@@ -0,0 +1,23 @@
+"""Read/write linear transforms."""
+import numpy as np
+from nibabel.wrapstruct import LabeledWrapStruct as LWS
+
+
+class LabeledWrapStruct(LWS):
+    def __setitem__(self, item, value):
+        self._structarr[item] = np.asanyarray(value)
+
+
+class StringBasedStruct(LabeledWrapStruct):
+    def __init__(self,
+                 binaryblock=None,
+                 endianness=None,
+                 check=True):
+        if binaryblock is not None and getattr(binaryblock, 'dtype',
+                                               None) == self.dtype:
+            self._structarr = binaryblock.copy()
+            return
+        super(StringBasedStruct, self).__init__(binaryblock, endianness, check)
+
+    def __array__(self):
+        return self._structarr

nitransforms/io.py renamed to nitransforms/io/lta.py

@@ -1,8 +1,10 @@
+"""Read/write linear transforms."""
 import numpy as np
 from nibabel.volumeutils import Recoder
 from nibabel.affines import voxel_sizes
 
-from .patched import LabeledWrapStruct
+from .base import StringBasedStruct
+
 
 transform_codes = Recoder((
     (0, 'LINEAR_VOX_TO_VOX'),
@@ -13,21 +15,6 @@
     fields=('code', 'label'))
 
 
-class StringBasedStruct(LabeledWrapStruct):
-    def __init__(self,
-                 binaryblock=None,
-                 endianness=None,
-                 check=True):
-        if binaryblock is not None and getattr(binaryblock, 'dtype',
-                                               None) == self.dtype:
-            self._structarr = binaryblock.copy()
-            return
-        super(StringBasedStruct, self).__init__(binaryblock, endianness, check)
-
-    def __array__(self):
-        return self._structarr
-
-
 class VolumeGeometry(StringBasedStruct):
     template_dtype = np.dtype([
         ('valid', 'i4'),              # Valid values: 0, 1

nitransforms/patched.py

@@ -1,5 +1,4 @@
 import numpy as np
-from nibabel.wrapstruct import LabeledWrapStruct as LWS
 
 
 def shape_zoom_affine(shape, zooms, x_flip=True, y_flip=False):
@@ -64,8 +63,3 @@ 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):
-        self._structarr[item] = np.asanyarray(value)