Skip to content

Commit 9ae8759

Browse files
smoiaoesteban
authored andcommitted
Added nonlinear tests for FSL
1 parent 95b0692 commit 9ae8759

File tree

2 files changed

+13
-8
lines changed

2 files changed

+13
-8
lines changed

nitransforms/nonlinear.py

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -77,14 +77,15 @@ def map(self, x, inverse=False):
7777

7878
@classmethod
7979
def from_filename(cls, filename, fmt="X5"):
80-
if fmt == "afni":
81-
_factory = io.afni.AFNIDisplacementsField
82-
elif fmt == "itk":
83-
_factory = io.itk.ITKDisplacementsField
84-
else:
85-
raise NotImplementedError
86-
87-
return cls(_factory.from_filename(filename))
80+
_factory = {
81+
"afni": io.afni.AFNIDisplacementsField,
82+
"itk": io.itk.ITKDisplacementsField,
83+
"fsl": io.fsl.FSLDisplacementsField,
84+
}
85+
if fmt not in _factory:
86+
raise NotImplementedError(f"Unsupported format <{fmt}>")
87+
88+
return cls(_factory[fmt].from_filename(filename))
8889

8990

9091
load = DisplacementsFieldTransform.from_filename

nitransforms/tests/test_nonlinear.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212
from ..nonlinear import DisplacementsFieldTransform, load as nlload
1313
from ..io.itk import ITKDisplacementsField
1414

15+
1516
TESTS_BORDER_TOLERANCE = 0.05
1617
APPLY_NONLINEAR_CMD = {
1718
"itk": """\
@@ -22,6 +23,9 @@
2223
3dNwarpApply -nwarp {transform} -source {moving} \
2324
-master {reference} -interp NN -prefix resampled.nii.gz
2425
""".format,
26+
'fsl': """\
27+
applywarp -i {moving} -r {reference} -o resampled.nii.gz \
28+
-w {transform} --interp=nn""".format,
2529
}
2630

2731

0 commit comments

Comments
 (0)