Skip to content

Commit 19599fb

Browse files
committed
Removed redundant _format_xarray from ANTSCommand and replaced other join functions with _format_xarray
1 parent 5e1161f commit 19599fb

File tree

3 files changed

+10
-23
lines changed

3 files changed

+10
-23
lines changed

nipype/interfaces/ants/base.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -83,9 +83,4 @@ def set_default_num_threads(cls, num_threads):
8383
type for any existing instances. For these, assign the
8484
<instance>.inputs.num_threads
8585
"""
86-
cls._num_threads = num_threads
87-
88-
def _format_xarray(self, val):
89-
""" Convienence method for converting [1,2,3] -> 1x2x3 """
90-
val = 'x'.join([str(x) for x in val])
91-
return val
86+
cls._num_threads = num_threads

nipype/interfaces/ants/registration.py

Lines changed: 6 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -173,17 +173,15 @@ def _regularization_constructor(self):
173173
self.inputs.regularization_deformation_field_sigma)
174174

175175
def _affine_gradient_descent_option_constructor(self):
176-
retval = ['--affine-gradient-descent-option']
177176
values = self.inputs.affine_gradient_descent_option
178177
defaults = [0.1, 0.5, 1.e-4, 1.e-4]
179178
for ii in range(len(defaults)):
180179
try:
181180
defaults[ii] = values[ii]
182181
except IndexError:
183182
break
184-
stringList = [('%g' % defaults[index]) for index in range(4)]
185-
parameters = 'x'.join(stringList)
186-
retval.append(parameters)
183+
parameters = self._format_xarray([('%g' % defaults[index]) for index in range(4)])
184+
retval = ['--affine-gradient-descent-option', parameters]
187185
return ' '.join(retval)
188186

189187
def _format_arg(self, opt, spec, val):
@@ -653,14 +651,13 @@ def _formatRegistration(self):
653651
retval.append('--convergence %s' % self._formatConvergence(ii))
654652
if isdefined(self.inputs.sigma_units):
655653
retval.append('--smoothing-sigmas %s%s' %
656-
(self._antsJoinList(self.inputs.smoothing_sigmas[
657-
ii]),
654+
(self._format_xarray(self.inputs.smoothing_sigmas[ii]),
658655
self.inputs.sigma_units[ii]))
659656
else:
660657
retval.append('--smoothing-sigmas %s' %
661-
self._antsJoinList(self.inputs.smoothing_sigmas[ii]))
658+
self._format_xarray(self.inputs.smoothing_sigmas[ii]))
662659
retval.append('--shrink-factors %s' %
663-
self._antsJoinList(self.inputs.shrink_factors[ii]))
660+
self._format_xarray(self.inputs.shrink_factors[ii]))
664661
if isdefined(self.inputs.use_estimate_learning_rate_once):
665662
retval.append('--use-estimate-learning-rate-once %d' %
666663
self.inputs.use_estimate_learning_rate_once[ii])
@@ -674,9 +671,6 @@ def _formatRegistration(self):
674671
retval.append('--use-histogram-matching %d' % histval)
675672
return " ".join(retval)
676673

677-
def _antsJoinList(self, antsList):
678-
return "x".join([str(i) for i in antsList])
679-
680674
def _get_outputfilenames(self, inverse=False):
681675
output_filename = None
682676
if not inverse:
@@ -699,8 +693,7 @@ def _get_outputfilenames(self, inverse=False):
699693
return inv_output_filename
700694

701695
def _formatConvergence(self, ii):
702-
convergence_iter = self._antsJoinList(
703-
self.inputs.number_of_iterations[ii])
696+
convergence_iter = self._format_xarray(self.inputs.number_of_iterations[ii])
704697
if len(self.inputs.convergence_threshold) > ii:
705698
convergence_value = self.inputs.convergence_threshold[ii]
706699
else:

nipype/interfaces/ants/segmentation.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -110,8 +110,7 @@ def _format_arg(self, opt, spec, val):
110110
if opt == 'mrf_smoothing_factor':
111111
retval = "--mrf [%g" % val
112112
if isdefined(self.inputs.mrf_radius):
113-
retval += ",%s" % 'x'.join([str(s)
114-
for s in self.inputs.mrf_radius])
113+
retval += ",%s" % self._format_xarray([str(s) for s in self.inputs.mrf_radius])
115114
return retval + "]"
116115
if opt == "icm_use_synchronous_update":
117116
retval = "--icm [%d" % val
@@ -354,10 +353,10 @@ def _format_arg(self, name, trait_spec, value):
354353

355354
if name == 'n_iterations':
356355
if isdefined(self.inputs.convergence_threshold):
357-
newval = '[ %s, %g ]' % ('x'.join([str(elt) for elt in value]),
356+
newval = '[ %s, %g ]' % (self._format_xarray([str(elt) for elt in value]),
358357
self.inputs.convergence_threshold)
359358
else:
360-
newval = '[ %s ]' % 'x'.join([str(elt) for elt in value])
359+
newval = '[ %s ]' % self._format_xarray([str(elt) for elt in value])
361360
return trait_spec.argstr % newval
362361

363362
return super(N4BiasFieldCorrection,

0 commit comments

Comments
 (0)