diff --git a/fsl/wrappers/eddy.py b/fsl/wrappers/eddy.py index ae5c7bb4f0075c65fa363970237b26baf7797bd1..3419c99d133045a4b34762e85de4d1e5f24528d1 100644 --- a/fsl/wrappers/eddy.py +++ b/fsl/wrappers/eddy.py @@ -61,8 +61,8 @@ def eddy_cuda(imain, mask, index, acqp, bvecs, bvals, out, **kwargs): return cmd -@wutils.fileOrImage('imain', 'fout', 'iout') -@wutils.fileOrArray('datain') +@wutils.fileOrImage('imain', 'fout', 'iout', outprefix='out') +@wutils.fileOrArray('datain', outprefix='out') @wutils.fslwrapper def topup(imain, datain, **kwargs): """Wrapper for the ``topup`` command.""" @@ -78,3 +78,25 @@ def topup(imain, datain, **kwargs): cmd += wutils.applyArgStyle('--=', valmap=valmap, **kwargs) return cmd + +@wutils.fileOrImage('imain', 'out') +@wutils.fileOrArray('datain') +@wutils.fslwrapper +def applytopup(imain, datain, index, **kwargs): + """Wrapper for the ``applytopup`` command.""" + + valmap = { + 'verbose' : wutils.SHOW_IF_TRUE + } + + asrt.assertFileExists(datain) + asrt.assertIsNifti(imain) + + cmd = [ + 'applytopup', '--imain={}'.format(imain), + '--inindex={}'.format(index), + '--datain={}'.format(datain), + ] + cmd += wutils.applyArgStyle('--=', valmap=valmap, **kwargs) + + return cmd