Merge pull request #2396 from VisionandCognition/afni_volreg_inweightvol

ENH Allow input weight for AFNI's volreg.

Author: effigies

nipype/interfaces/afni/preprocess.py

@@ -2483,6 +2483,12 @@ class VolregInputSpec(AFNICommandInputSpec):
         mandatory=True,
         exists=True,
         copyfile=False)
+    in_weight_volume = traits.Either(
+        traits.Tuple(File(exists=True), traits.Int),
+        File(exists=True),
+        desc='weights for each voxel specified by a file with an '
+             'optional volume number (defaults to 0)',
+        argstr="-weight '%s[%d]'")
     out_file = File(
         name_template='%s_volreg',
         desc='output image file name',
@@ -2575,6 +2581,11 @@ class Volreg(AFNICommand):
     input_spec = VolregInputSpec
     output_spec = VolregOutputSpec
 
+    def _format_arg(self, name, trait_spec, value):
+        if name == 'in_weight_volume' and not isinstance(value, tuple):
+            value = (value, 0)
+        return super(Volreg, self)._format_arg(name, trait_spec, value)
+
 
 class WarpInputSpec(AFNICommandInputSpec):
     in_file = File(

nipype/interfaces/afni/tests/test_auto_Volreg.py

@@ -26,6 +26,7 @@ def test_Volreg_inputs():
             mandatory=True,
             position=-1,
         ),
+        in_weight_volume=dict(argstr="-weight '%s[%d]'", ),
         interp=dict(argstr='-%s', ),
         md1d_file=dict(
             argstr='-maxdisp1D %s',