ENH: Added GenerateEdgeMapImage to semtools

Author: aghayoor

nipype/interfaces/semtools/brains/__init__.py

@@ -1,4 +1,4 @@
 from __future__ import absolute_import
 from .segmentation import SimilarityIndex, BRAINSTalairach, BRAINSTalairachMask
-from .utilities import HistogramMatchingFilter
+from .utilities import HistogramMatchingFilter, GenerateEdgeMapImage
 from .classify import BRAINSPosteriorToContinuousClass

nipype/interfaces/semtools/brains/utilities.py

@@ -45,3 +45,40 @@ class HistogramMatchingFilter(SEMLikeCommandLine):
     _cmd = " HistogramMatchingFilter "
     _outputs_filenames = {'outputVolume': 'outputVolume.nii'}
     _redirect_x = False
+
+
+class GenerateEdgeMapImageInputSpec(CommandLineInputSpec):
+    inputMRVolumes = InputMultiPath(File(exists=True), desc="List of input structural MR volumes to create the maximum edgemap", argstr="--inputMRVolumes %s...")
+    inputMask = File(desc="Input mask file name. If set, image histogram percentiles will be calculated within the mask", exists=True, argstr="--inputMask %s")
+    minimumOutputRange = traits.Int(desc="Map lower quantile and below to minimum output range. It should be a small number greater than zero. Default is 1", argstr="--minimumOutputRange %d")
+    maximumOutputRange = traits.Int(desc="Map upper quantile and above to maximum output range. Default is 255 that is the maximum range of unsigned char", argstr="--maximumOutputRange %d")
+    lowerPercentileMatching = traits.Float(desc="Map lower quantile and below to minOutputRange. It should be a value between zero and one", argstr="--lowerPercentileMatching %f")
+    upperPercentileMatching = traits.Float(desc="Map upper quantile and above to maxOutputRange. It should be a value between zero and one", argstr="--upperPercentileMatching %f")
+    outputEdgeMap = traits.Either(traits.Bool, File(), hash_files=False, desc="output edgemap file name", argstr="--outputEdgeMap %s")
+    outputMaximumGradientImage = traits.Either(traits.Bool, File(), hash_files=False, desc="output gradient image file name", argstr="--outputMaximumGradientImage %s")
+    numberOfThreads = traits.Int(desc="Explicitly specify the maximum number of threads to use.", argstr="--numberOfThreads %d")
+
+
+class GenerateEdgeMapImageOutputSpec(TraitedSpec):
+    outputEdgeMap = File(desc="(required) output file name", exists=True)
+    outputMaximumGradientImage = File(desc="output gradient image file name", exists=True)
+
+class GenerateEdgeMapImage(SEMLikeCommandLine):
+
+    """title: GenerateEdgeMapImage
+
+category: BRAINS.Utilities
+
+description: Automatic edgemap generation for edge-guided super-resolution reconstruction
+
+version: 1.0
+
+contributor: Ali Ghayoor
+
+"""
+
+    input_spec = GenerateEdgeMapImageInputSpec
+    output_spec = GenerateEdgeMapImageOutputSpec
+    _cmd = " GenerateEdgeMapImage "
+    _outputs_filenames = {'outputEdgeMap': 'outputEdgeMap', 'outputMaximumGradientImage': 'outputMaximumGradientImage'}
+    _redirect_x = False