From a13a892cefc07d2d5e1478381e668518ba3ffb0b Mon Sep 17 00:00:00 2001 From: akeshavan Date: Wed, 5 Sep 2018 16:04:30 -0700 Subject: [PATCH] fix/enh: added the colorfa as an output in DTI --- nipype/interfaces/dipy/tensors.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/nipype/interfaces/dipy/tensors.py b/nipype/interfaces/dipy/tensors.py index c79e063683..33f5a73b08 100644 --- a/nipype/interfaces/dipy/tensors.py +++ b/nipype/interfaces/dipy/tensors.py @@ -21,6 +21,7 @@ class DTIOutputSpec(TraitedSpec): md_file = File(exists=True) rd_file = File(exists=True) ad_file = File(exists=True) + color_fa_file = File(exists=True) class DTI(DipyDiffusionInterface): @@ -62,7 +63,7 @@ def _run_interface(self, runtime): IFLOGGER.info('DTI parameters image saved as %s', out_file) # FA MD RD and AD - for metric in ["fa", "md", "rd", "ad"]: + for metric in ["fa", "md", "rd", "ad", "color_fa"]: data = getattr(ten_fit, metric).astype("float32") out_name = self._gen_filename(metric) nb.Nifti1Image(data, affine).to_filename(out_name) @@ -74,7 +75,7 @@ def _list_outputs(self): outputs = self._outputs().get() outputs['out_file'] = self._gen_filename('dti') - for metric in ["fa", "md", "rd", "ad"]: + for metric in ["fa", "md", "rd", "ad", "color_fa"]: outputs["{}_file".format(metric)] = self._gen_filename(metric) return outputs