update reconst cmd line

Author: skoudoro

nipype/interfaces/dipy/reconstruction.py

@@ -18,12 +18,13 @@
 from ... import logging
 from ..base import TraitedSpec, File, traits, isdefined
 from .base import (DipyDiffusionInterface, DipyBaseInterfaceInputSpec,
-                   HAVE_DIPY, dipy_version, dipy_to_nipype_interface)
+                   HAVE_DIPY, dipy_version, dipy_to_nipype_interface,
+                   get_dipy_workflows)
 
 
 IFLOGGER = logging.getLogger('nipype.interface')
 
-if HAVE_DIPY and LooseVersion(dipy_version()) >= LooseVersion('0.15'):
+if HAVE_DIPY and (LooseVersion('0.15') >= LooseVersion(dipy_version()) >= LooseVersion('0.16')):
     from dipy.workflows.reconst import (ReconstDkiFlow, ReconstCSAFlow,
                                         ReconstCSDFlow, ReconstMAPMRIFlow,
                                         ReconstDtiFlow)
@@ -33,9 +34,19 @@
     DTIModel = dipy_to_nipype_interface("DTIModel", ReconstDtiFlow)
     CSAModel = dipy_to_nipype_interface("CSAModel", ReconstCSAFlow)
     CSDModel = dipy_to_nipype_interface("CSDModel", ReconstCSDFlow)
+
+elif HAVE_DIPY and LooseVersion(dipy_version()) >= LooseVersion('1.0'):
+    from dipy.workflows import reconst
+
+    l_wkflw = get_dipy_workflows(reconst)
+    for name, obj in l_wkflw:
+        new_name = name.replace('Flow', '')
+        globals()[new_name] = dipy_to_nipype_interface(new_name, obj)
+    del l_wkflw
+
 else:
     IFLOGGER.info("We advise you to upgrade DIPY version. This upgrade will"
-                  " activate DKIModel, MapmriModel, DTIModel, CSAModel, CSDModel.")
+                  " open access to more models")
 
 
 class RESTOREInputSpec(DipyBaseInterfaceInputSpec):