Handle output filenames correctly, add undocumented options

Author: effigies

nipype/interfaces/freesurfer/tests/test_auto_MRIsExpand.py

@@ -10,6 +10,8 @@ def test_MRIsExpand_inputs():
     mandatory=True,
     position=-2,
     ),
+    dt=dict(argstr='-T %g',
+    ),
     environ=dict(nohash=True,
     usedefault=True,
     ),
@@ -20,16 +22,29 @@ def test_MRIsExpand_inputs():
     mandatory=True,
     position=-3,
     ),
-    out_file=dict(argstr='%s',
-    name_source='in_file',
-    name_template='%s.expanded',
+    navgs=dict(argstr='-navgs %d %d',
+    ),
+    nsurfaces=dict(argstr='-N %d',
+    ),
+    out_name=dict(argstr='%s',
     position=-1,
+    usedefault=True,
+    ),
+    pial=dict(argstr='-pial %s',
+    ),
+    smooth_averages=dict(argstr='-A %d',
+    ),
+    spring=dict(argstr='-S %g',
     ),
     subjects_dir=dict(),
     terminal_output=dict(nohash=True,
     ),
     thickness=dict(argstr='-thickness',
     ),
+    thickness_name=dict(argstr='-thickness_name %s',
+    ),
+    write_iterations=dict(argstr='-W %d',
+    ),
     )
     inputs = MRIsExpand.input_spec()
 

nipype/interfaces/freesurfer/utils.py

@@ -2882,19 +2882,47 @@ def _list_outputs(self):
 
 
 class MRIsExpandInputSpec(FSTraitedSpec):
+    # Input spec derived from
+    # https://github.com/freesurfer/freesurfer/blob/102e053/mris_expand/mris_expand.c
     in_file = File(
         exists=True, mandatory=True, argstr='%s', position=-3,
         desc='Surface to expand')
     distance = traits.Float(
         mandatory=True, argstr='%g', position=-2,
         desc='Distance in mm or fraction of cortical thickness')
-    out_file = File(
-        argstr='%s', position=-1,
-        name_template='%s.expanded', name_source='in_file',
-        desc='Output surface file')
+    out_name = traits.Str(
+        'expanded', argstr='%s', position=-1, usedefault=True,
+        desc=('Output surface file\n'
+              'If missing "lh." or "rh.", derive from `in_file`'))
     thickness = traits.Bool(
         argstr='-thickness',
         desc='Expand by fraction of cortical thickness, not mm')
+    thickness_name = traits.Str(
+        argstr="-thickness_name %s",
+        desc=('Name of thickness file (implicit: "thickness")\n'
+              'If no path, uses directory of `in_file`\n'
+              'If missing "lh." or "rh.", derive from `in_file`'))
+    navgs = traits.Tuple(
+        traits.Int, traits.Int,
+        argstr='-navgs %d %d',
+        desc=('Tuple of (n_averages, min_averages) parameters '
+              '(implicit: (16, 0))'))
+    pial = traits.Str(
+        argstr='-pial %s',
+        desc=('Name of pial file (implicit: "pial")\n'
+              'If no path, uses directory of `in_file`\n'
+              'If missing "lh." or "rh.", derive from `in_file`'))
+    spring = traits.Float(argstr='-S %g', desc="Spring term (implicit: 0.05)")
+    dt = traits.Float(argstr='-T %g', desc='dt (implicit: 0.25)')
+    write_iterations = traits.Int(
+        argstr='-W %d',
+        desc='Write snapshots of expansion every N iterations')
+    smooth_averages = traits.Int(
+        argstr='-A %d',
+        desc='Smooth surface with N iterations after expansion')
+    nsurfaces = traits.Int(
+        argstr='-N %d',
+        desc='Number of surfacces to write during expansion')
 
 
 class MRIsExpandOutputSpec(TraitedSpec):
@@ -2920,3 +2948,20 @@ class MRIsExpand(FSCommand):
     _cmd = 'mris_expand'
     input_spec = MRIsExpandInputSpec
     output_spec = MRIsExpandOutputSpec
+
+    def _format_arg(self, name, spec, value):
+        if name == 'out_name':
+            value = self._list_outputs()['out_file']
+        return super(MRIsExpand, self)._format_arg(name, spec, value)
+
+    def _list_outputs(self):
+        outputs = self._outputs().get()
+        # Mimic FreeSurfer output filename derivation, but in local directory
+        # if no path specified
+        out_file = self.inputs.out_name
+        path, base = os.path.split(out_file)
+        if all((path == '', not base.startswith('lh.'),
+                not base.startswith('rh.'))):
+                out_file = self.inputs.in_file[:3] + base
+        outputs["out_file"] = os.path.abspath(out_file)
+        return outputs