Add default timestamp behavior to write_geometry

Author: effigies

nibabel/freesurfer/io.py

@@ -1,7 +1,8 @@
 from __future__ import with_statement
 
 import numpy as np
-
+import os
+import time
 
 def _fread3(fobj):
     """Read a 3-byte int from an open binary file object
@@ -92,21 +93,25 @@ def read_geometry(filepath):
     coords = coords.astype(np.float)  # XXX: due to mayavi bug on mac 32bits
     return coords, faces
 
-def write_geometry(filepath, create_stamp, coords, faces):
+def write_geometry(filepath, coords, faces, create_stamp=None):
     """Write a triangular format Freesurfer surface mesh.
 
     Parameters
     ----------
     filepath : str
         Path to surface file
-    create_stamp : str
-        User/time stamp
     coords : numpy array
         nvtx x 3 array of vertex (x, y, z) coordinates
     faces : numpy array
         nfaces x 3 array of defining mesh triangles
+    create_stamp : str
+        User/time stamp (default: "created by <user> on <ctime>")
     """
     magic_bytes = np.array([255,255,254],dtype=np.uint8)
+
+    if create_stamp is None:
+        create_stamp = "created by %s on %s" % (os.getlogin(), time.ctime())
+
     with open(filepath, 'wb') as fobj:
         magic_bytes.tofile(fobj)
         fobj.write("%s\n\n" % create_stamp)

nibabel/freesurfer/tests/test_io.py

@@ -44,7 +44,6 @@ def test_geometry():
     ntf.close()
 
     surf_path = ntf.name
-    create_stamp = "created by %s on %s" % (os.getlogin(), time.ctime())
     write_geometry(surf_path, create_stamp, coords, faces)
 
     coords2, faces2 = read_geometry(surf_path)