@@ -1129,11 +1129,28 @@ def _format_arg(self, opt, spec, val):
1129
1129
retval = ''
1130
1130
if not isdefined (self .inputs .out_label_fusion ):
1131
1131
retval = '-o {0}' .format (self .inputs .out_intensity_fusion_name_format )
1132
+ elif opt == 'atlas_image' :
1133
+ atlas_image_cmd = " " .join (
1134
+ [ '-g [{0}]' .format ("," .join (
1135
+ fn for fn in ai ) ) for ai in self .inputs .atlas_image ]
1136
+ )
1137
+ retval = atlas_image_cmd
1138
+ elif opt == 'target_image' :
1139
+ target_image_cmd = " " .join (
1140
+ [ '-t [{0}]' .format ("," .join (
1141
+ fn for fn in ai ) ) for ai in self .inputs .target_image ]
1142
+ )
1143
+ retval = target_image_cmd
1144
+ elif opt == 'atlas_segmentation_image' :
1145
+ assert len (val ) == len (self .inputs .atlas_image ), "Number of specified " \
1146
+ "segmentations should be identical to the number of atlas image " \
1147
+ "sets {0}!={1}" .format (len (val ), len (self .inputs .atlas_image ))
1148
+ atlas_segmentation_image_cmd = " " .join (
1149
+ [ '-t [{0}]' .format ( fn ) for fn in self .inputs .atlas_segmentation_image ]
1150
+ )
1151
+ retval = atlas_segmentation_image_cmd
1132
1152
else :
1133
- if opt == 'atlas_segmentation_image' :
1134
- assert len (val ) == len (self .inputs .atlas_image ), "Number of specified " \
1135
- "segmentations should be identical to the number of atlas image " \
1136
- "sets {0}!={1}" .format (len (val ), len (self .inputs .atlas_image ))
1153
+
1137
1154
return super (ANTSCommand , self )._format_arg (opt , spec , val )
1138
1155
return retval
1139
1156
0 commit comments