Fixed tracking specs

chrisgorgo · chrisgorgo · commit aaa48beefd5d · 2014-03-06T15:47:25.000+01:00
diff --git a/CHANGES b/CHANGES
@@ -1,7 +1,11 @@
-Release 0.9.2 (January 31, 2014)
-===========
+Next Release
+============
 
 * ENH: New ANTs interface: ApplyTransformsToPoints
+* FIX: MRTrix tracking algorithms were ignoring mask parameters.
+
+Release 0.9.2 (January 31, 2014)
+============
 
 * FIX: DataFinder was broken due to a typo
 * FIX: Order of DataFinder outputs was not guaranteed, it's human sorted now
diff --git a/nipype/interfaces/mrtrix/tests/test_auto_DiffusionTensorStreamlineTrack.py b/nipype/interfaces/mrtrix/tests/test_auto_DiffusionTensorStreamlineTrack.py
@@ -16,12 +16,13 @@ def test_DiffusionTensorStreamlineTrack_inputs():
     usedefault=True,
     ),
     exclude_file=dict(argstr='-exclude %s',
-    position=2,
+    xor=['exclude_file', 'exclude_spec'],
     ),
-    exclude_spec=dict(argstr='-seed %s',
+    exclude_spec=dict(argstr='-exclude %s',
     position=2,
     sep=',',
     units='mm',
+    xor=['exclude_file', 'exclude_spec'],
     ),
     gradient_encoding_file=dict(argstr='-grad %s',
     mandatory=True,
@@ -35,12 +36,13 @@ def test_DiffusionTensorStreamlineTrack_inputs():
     position=-2,
     ),
     include_file=dict(argstr='-include %s',
-    position=2,
+    xor=['include_file', 'include_spec'],
     ),
-    include_spec=dict(argstr='-seed %s',
+    include_spec=dict(argstr='-include %s',
     position=2,
     sep=',',
     units='mm',
+    xor=['include_file', 'include_spec'],
     ),
     initial_cutoff_value=dict(argstr='-initcutoff %s',
     units='NA',
@@ -52,13 +54,14 @@ def test_DiffusionTensorStreamlineTrack_inputs():
     position=-3,
     usedefault=True,
     ),
-    mask_file=dict(argstr='-exclude %s',
-    position=2,
+    mask_file=dict(argstr='-mask %s',
+    xor=['mask_file', 'mask_spec'],
     ),
-    mask_spec=dict(argstr='-seed %s',
+    mask_spec=dict(argstr='-mask %s',
     position=2,
     sep=',',
     units='mm',
+    xor=['mask_file', 'mask_spec'],
     ),
     maximum_number_of_tracks=dict(argstr='-maxnum %d',
     ),
@@ -78,12 +81,13 @@ def test_DiffusionTensorStreamlineTrack_inputs():
     position=-1,
     ),
     seed_file=dict(argstr='-seed %s',
-    position=2,
+    xor=['seed_file', 'seed_spec'],
     ),
     seed_spec=dict(argstr='-seed %s',
     position=2,
     sep=',',
     units='mm',
+    xor=['seed_file', 'seed_spec'],
     ),
     step_size=dict(argstr='-step %s',
     units='mm',
diff --git a/nipype/interfaces/mrtrix/tests/test_auto_ProbabilisticSphericallyDeconvolutedStreamlineTrack.py b/nipype/interfaces/mrtrix/tests/test_auto_ProbabilisticSphericallyDeconvolutedStreamlineTrack.py
@@ -16,12 +16,13 @@ def test_ProbabilisticSphericallyDeconvolutedStreamlineTrack_inputs():
     usedefault=True,
     ),
     exclude_file=dict(argstr='-exclude %s',
-    position=2,
+    xor=['exclude_file', 'exclude_spec'],
     ),
-    exclude_spec=dict(argstr='-seed %s',
+    exclude_spec=dict(argstr='-exclude %s',
     position=2,
     sep=',',
     units='mm',
+    xor=['exclude_file', 'exclude_spec'],
     ),
     ignore_exception=dict(nohash=True,
     usedefault=True,
@@ -31,12 +32,13 @@ def test_ProbabilisticSphericallyDeconvolutedStreamlineTrack_inputs():
     position=-2,
     ),
     include_file=dict(argstr='-include %s',
-    position=2,
+    xor=['include_file', 'include_spec'],
     ),
-    include_spec=dict(argstr='-seed %s',
+    include_spec=dict(argstr='-include %s',
     position=2,
     sep=',',
     units='mm',
+    xor=['include_file', 'include_spec'],
     ),
     initial_cutoff_value=dict(argstr='-initcutoff %s',
     units='NA',
@@ -48,13 +50,14 @@ def test_ProbabilisticSphericallyDeconvolutedStreamlineTrack_inputs():
     position=-3,
     usedefault=True,
     ),
-    mask_file=dict(argstr='-exclude %s',
-    position=2,
+    mask_file=dict(argstr='-mask %s',
+    xor=['mask_file', 'mask_spec'],
     ),
-    mask_spec=dict(argstr='-seed %s',
+    mask_spec=dict(argstr='-mask %s',
     position=2,
     sep=',',
     units='mm',
+    xor=['mask_file', 'mask_spec'],
     ),
     maximum_number_of_tracks=dict(argstr='-maxnum %d',
     ),
@@ -76,12 +79,13 @@ def test_ProbabilisticSphericallyDeconvolutedStreamlineTrack_inputs():
     position=-1,
     ),
     seed_file=dict(argstr='-seed %s',
-    position=2,
+    xor=['seed_file', 'seed_spec'],
     ),
     seed_spec=dict(argstr='-seed %s',
     position=2,
     sep=',',
     units='mm',
+    xor=['seed_file', 'seed_spec'],
     ),
     step_size=dict(argstr='-step %s',
     units='mm',
diff --git a/nipype/interfaces/mrtrix/tests/test_auto_SphericallyDeconvolutedStreamlineTrack.py b/nipype/interfaces/mrtrix/tests/test_auto_SphericallyDeconvolutedStreamlineTrack.py
@@ -16,12 +16,13 @@ def test_SphericallyDeconvolutedStreamlineTrack_inputs():
     usedefault=True,
     ),
     exclude_file=dict(argstr='-exclude %s',
-    position=2,
+    xor=['exclude_file', 'exclude_spec'],
     ),
-    exclude_spec=dict(argstr='-seed %s',
+    exclude_spec=dict(argstr='-exclude %s',
     position=2,
     sep=',',
     units='mm',
+    xor=['exclude_file', 'exclude_spec'],
     ),
     ignore_exception=dict(nohash=True,
     usedefault=True,
@@ -31,12 +32,13 @@ def test_SphericallyDeconvolutedStreamlineTrack_inputs():
     position=-2,
     ),
     include_file=dict(argstr='-include %s',
-    position=2,
+    xor=['include_file', 'include_spec'],
     ),
-    include_spec=dict(argstr='-seed %s',
+    include_spec=dict(argstr='-include %s',
     position=2,
     sep=',',
     units='mm',
+    xor=['include_file', 'include_spec'],
     ),
     initial_cutoff_value=dict(argstr='-initcutoff %s',
     units='NA',
@@ -48,13 +50,14 @@ def test_SphericallyDeconvolutedStreamlineTrack_inputs():
     position=-3,
     usedefault=True,
     ),
-    mask_file=dict(argstr='-exclude %s',
-    position=2,
+    mask_file=dict(argstr='-mask %s',
+    xor=['mask_file', 'mask_spec'],
     ),
-    mask_spec=dict(argstr='-seed %s',
+    mask_spec=dict(argstr='-mask %s',
     position=2,
     sep=',',
     units='mm',
+    xor=['mask_file', 'mask_spec'],
     ),
     maximum_number_of_tracks=dict(argstr='-maxnum %d',
     ),
@@ -74,12 +77,13 @@ def test_SphericallyDeconvolutedStreamlineTrack_inputs():
     position=-1,
     ),
     seed_file=dict(argstr='-seed %s',
-    position=2,
+    xor=['seed_file', 'seed_spec'],
     ),
     seed_spec=dict(argstr='-seed %s',
     position=2,
     sep=',',
     units='mm',
+    xor=['seed_file', 'seed_spec'],
     ),
     step_size=dict(argstr='-step %s',
     units='mm',
diff --git a/nipype/interfaces/mrtrix/tests/test_auto_StreamlineTrack.py b/nipype/interfaces/mrtrix/tests/test_auto_StreamlineTrack.py
@@ -16,12 +16,13 @@ def test_StreamlineTrack_inputs():
     usedefault=True,
     ),
     exclude_file=dict(argstr='-exclude %s',
-    position=2,
+    xor=['exclude_file', 'exclude_spec'],
     ),
-    exclude_spec=dict(argstr='-seed %s',
+    exclude_spec=dict(argstr='-exclude %s',
     position=2,
     sep=',',
     units='mm',
+    xor=['exclude_file', 'exclude_spec'],
     ),
     ignore_exception=dict(nohash=True,
     usedefault=True,
@@ -31,12 +32,13 @@ def test_StreamlineTrack_inputs():
     position=-2,
     ),
     include_file=dict(argstr='-include %s',
-    position=2,
+    xor=['include_file', 'include_spec'],
     ),
-    include_spec=dict(argstr='-seed %s',
+    include_spec=dict(argstr='-include %s',
     position=2,
     sep=',',
     units='mm',
+    xor=['include_file', 'include_spec'],
     ),
     initial_cutoff_value=dict(argstr='-initcutoff %s',
     units='NA',
@@ -48,13 +50,14 @@ def test_StreamlineTrack_inputs():
     position=-3,
     usedefault=True,
     ),
-    mask_file=dict(argstr='-exclude %s',
-    position=2,
+    mask_file=dict(argstr='-mask %s',
+    xor=['mask_file', 'mask_spec'],
     ),
-    mask_spec=dict(argstr='-seed %s',
+    mask_spec=dict(argstr='-mask %s',
     position=2,
     sep=',',
     units='mm',
+    xor=['mask_file', 'mask_spec'],
     ),
     maximum_number_of_tracks=dict(argstr='-maxnum %d',
     ),
@@ -74,12 +77,13 @@ def test_StreamlineTrack_inputs():
     position=-1,
     ),
     seed_file=dict(argstr='-seed %s',
-    position=2,
+    xor=['seed_file', 'seed_spec'],
     ),
     seed_spec=dict(argstr='-seed %s',
     position=2,
     sep=',',
     units='mm',
+    xor=['seed_file', 'seed_spec'],
     ),
     step_size=dict(argstr='-step %s',
     units='mm',
diff --git a/nipype/interfaces/mrtrix/tracking.py b/nipype/interfaces/mrtrix/tracking.py
@@ -75,19 +75,26 @@ class StreamlineTrackInputSpec(CommandLineInputSpec):
     in_file = File(exists=True, argstr='%s', mandatory=True, position=-2, desc='the image containing the source data.' \
     'The type of data required depends on the type of tracking as set in the preceeding argument. For DT methods, ' \
     'the base DWI are needed. For SD methods, the SH harmonic coefficients of the FOD are needed.')
-
-    seed_file = File(exists=True, argstr='-seed %s', position=2, desc='seed file')
+    
+    seed_xor = ['seed_file', 'seed_spec']
+    seed_file = File(exists=True, argstr='-seed %s', desc='seed file', xor = seed_xor)
     seed_spec = traits.List(traits.Float, desc='seed specification in mm and radius (x y z r)', position=2,
-        argstr='-seed %s', minlen=4, maxlen=4, sep=',', units='mm')
-    include_file = File(exists=True, argstr='-include %s', position=2, desc='inclusion file')
+        argstr='-seed %s', minlen=4, maxlen=4, sep=',', units='mm', xor = seed_xor)
+    
+    include_xor = ['include_file', 'include_spec']
+    include_file = File(exists=True, argstr='-include %s', desc='inclusion file', xor = include_xor)
     include_spec = traits.List(traits.Float, desc='inclusion specification in mm and radius (x y z r)', position=2,
-        argstr='-seed %s', minlen=4, maxlen=4, sep=',', units='mm')
-    exclude_file = File(exists=True, argstr='-exclude %s', position=2, desc='exclusion file')
+        argstr='-include %s', minlen=4, maxlen=4, sep=',', units='mm', xor = include_xor)
+    
+    exclude_xor = ['exclude_file', 'exclude_spec']
+    exclude_file = File(exists=True, argstr='-exclude %s', desc='exclusion file', xor = exclude_xor)
     exclude_spec = traits.List(traits.Float, desc='exclusion specification in mm and radius (x y z r)', position=2,
-        argstr='-seed %s', minlen=4, maxlen=4, sep=',', units='mm')
-    mask_file = File(exists=True, argstr='-exclude %s', position=2, desc='mask file. Only tracks within mask.')
+        argstr='-exclude %s', minlen=4, maxlen=4, sep=',', units='mm', xor = exclude_xor)
+    
+    mask_xor = ['mask_file', 'mask_spec']
+    mask_file = File(exists=True, argstr='-mask %s', desc='mask file. Only tracks within mask.', xor = mask_xor)
     mask_spec = traits.List(traits.Float, desc='Mask specification in mm and radius (x y z r). Tracks will be terminated when they leave the ROI.', position=2,
-        argstr='-seed %s', minlen=4, maxlen=4, sep=',', units='mm')
+        argstr='-mask %s', minlen=4, maxlen=4, sep=',', units='mm', xor = mask_xor)
 
     inputmodel = traits.Enum('DT_STREAM', 'SD_PROB', 'SD_STREAM',
         argstr='%s', desc='input model type', usedefault=True, position=-3)
@@ -140,6 +147,9 @@ class StreamlineTrack(CommandLine):
     >>> strack.inputs.inputmodel = 'SD_PROB'
     >>> strack.inputs.in_file = 'data.Bfloat'
     >>> strack.inputs.seed_file = 'seed_mask.nii'
+    >>> strack.inputs.mask_file = 'mask.nii'
+    >>> strack.cmdline
+    'streamtrack -mask mask.nii -seed seed_mask.nii SD_PROB data.Bfloat data_tracked.tck'
     >>> strack.run()                                    # doctest: +SKIP
     """
     _cmd = 'streamtrack'
