Skip to content

Commit dfef177

Browse files
committed
add preprocessing cmd line
1 parent 61ebb29 commit dfef177

File tree

1 file changed

+16
-1
lines changed

1 file changed

+16
-1
lines changed

nipype/interfaces/dipy/preprocess.py

Lines changed: 16 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,15 +5,30 @@
55
import os.path as op
66
import nibabel as nb
77
import numpy as np
8+
from distutils.version import LooseVersion
89

910
from ...utils import NUMPY_MMAP
1011

1112
from ... import logging
1213
from ..base import (traits, TraitedSpec, File, isdefined)
13-
from .base import DipyBaseInterface
14+
from .base import (HAVE_DIPY, dipy_version, dipy_to_nipype_interface,
15+
get_dipy_workflows, DipyBaseInterface)
1416

1517
IFLOGGER = logging.getLogger('nipype.interface')
1618

19+
if HAVE_DIPY and LooseVersion(dipy_version()) >= LooseVersion('0.15'):
20+
from dipy.workflows import denoise, mask
21+
22+
l_wkflw = get_dipy_workflows(denoise) + get_dipy_workflows(mask)
23+
for name, obj in l_wkflw:
24+
new_name = name.replace('Flow', '')
25+
globals()[new_name] = dipy_to_nipype_interface(new_name, obj)
26+
del l_wkflw
27+
28+
else:
29+
IFLOGGER.info("We advise you to upgrade DIPY version. This upgrade will"
30+
" open access to more function")
31+
1732

1833
class ResampleInputSpec(TraitedSpec):
1934
in_file = File(

0 commit comments

Comments
 (0)