Add a write_morph_data function.

Author: Ben Cipollini

nibabel/freesurfer/__init__.py

@@ -1,6 +1,6 @@
 """Reading functions for freesurfer files
 """
 
-from .io import read_geometry, read_morph_data, \
+from .io import read_geometry, read_morph_data, write_morph_data, \
     read_annot, read_label, write_geometry, write_annot
 from .mghformat import load, save, MGHImage

nibabel/freesurfer/io.py

@@ -160,6 +160,25 @@ def read_morph_data(filepath):
     return curv
 
 
+def write_morph_data(filename, values):
+    '''
+    '''
+    with open(filename, 'wb') as f:
+
+      # magic number
+      np.array([255], dtype='>u1').tofile(f)
+      np.array([255], dtype='>u1').tofile(f)
+      np.array([255], dtype='>u1').tofile(f)
+
+      # vertices number and two un-used int4
+      np.array([len(values)], dtype='>i4').tofile(f)
+      np.array([0], dtype='>i4').tofile(f)
+      np.array([1], dtype='>i4').tofile(f)
+
+      # now the data
+      np.array(values, dtype='>f4').tofile(f)
+
+
 def read_annot(filepath, orig_ids=False):
     """Read in a Freesurfer annotation from a .annot file.