added node maskfilter

Author: Lionel Butry

nipype/interfaces/mrtrix3/__init__.py

@@ -30,4 +30,5 @@
     SHConv,
     TensorMetrics,
     TransformFSLConvert,
+    MaskFilter,
 )

nipype/interfaces/mrtrix3/utils.py

@@ -1186,3 +1186,64 @@ def _list_outputs(self):
         outputs = self.output_spec().get()
         outputs["out_file"] = op.abspath(self.inputs.out_file)
         return outputs
+
+class MaskFilterInputSpec(CommandLineInputSpec):
+    in_file = File(
+        exists=True,
+        mandatory=True,
+        argstr="%s",
+        position=-3,
+        desc="Input mask",
+    )
+    filter = traits.Str(
+        mandatory=True, 
+        argstr="%s",
+        position=-2,
+        desc="Filter to perform (e.g. dilate, erode)"
+    )
+    out_file = File(
+        name_source=["input_image"],
+        mandatory=True,
+        argstr="%s",
+        position=-1
+        desc="Output mask"
+    )
+    npass = traits.Int(
+        argstr="-npass %d",
+        position=1,
+        desc="Number of passes"
+    )
+
+class MaskFilterOutputSpec(TraitedSpec):
+    out_file = File(exists=True, desc="the filtered output mask")
+
+class MaskFilter(CommandLine):
+    """
+    Perform filtering operations on 3D / 4D mask images. 
+    Only supports dilate / erode filters at the moment.
+    
+
+    Example
+    -------
+
+    >>> import nipype.interfaces.mrtrix3 as mrt
+    >>> mf = mrt.MaskFilter()
+    >>> mf.inputs.in_file = 'mask.mif'
+    >>> mf.inputs.filter = 'dilate'
+    >>> mf.inputs.npass = 2
+    >>> mf.out_file = 'mask_filtered.mif'
+    >>> mf.cmdline
+    'maskfilter -npass 2 mask.mif dilate mask_filtered.mif'
+    >>> mf.run()
+    """
+
+    _cmd = "maskfilter"
+    input_spec = MaskFilterInputSpec
+    output_spec = MaskFilterOutputSpec
+        
+    def _list_outputs(self):
+        outputs = self.output_spec().get()
+        outputs = self.output_spec().get()
+        inputs = self.input_spec().get()
+        outputs["output_image"] = self._gen_filename(inputs["input_image"], suffix="_filtered")
+        return outputs