Skip to content

Commit fe15295

Browse files
authored
Merge pull request #3192 from sulantha2006/master
ENH: Add reverse-ordered transform lists to ants.Registration outputs
2 parents f847fa7 + f6ba809 commit fe15295

File tree

2 files changed

+17
-0
lines changed

2 files changed

+17
-0
lines changed

nipype/interfaces/ants/registration.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -591,12 +591,18 @@ class RegistrationOutputSpec(TraitedSpec):
591591
forward_transforms = traits.List(
592592
File(exists=True), desc="List of output transforms for forward registration"
593593
)
594+
reverse_forward_transforms = traits.List(
595+
File(exists=True), desc="List of output transforms for forward registration reversed for antsApplyTransform"
596+
)
594597
reverse_transforms = traits.List(
595598
File(exists=True), desc="List of output transforms for reverse registration"
596599
)
597600
forward_invert_flags = traits.List(
598601
traits.Bool(), desc="List of flags corresponding to the forward transforms"
599602
)
603+
reverse_forward_invert_flags = traits.List(
604+
traits.Bool(), desc="List of flags corresponding to the forward transforms reversed for antsApplyTransform"
605+
)
600606
reverse_invert_flags = traits.List(
601607
traits.Bool(), desc="List of flags corresponding to the reverse transforms"
602608
)
@@ -797,6 +803,8 @@ class Registration(ANTSCommand):
797803
'inverse_composite_transform': '...data/output_InverseComposite.h5',
798804
'inverse_warped_image': <undefined>,
799805
'metric_value': <undefined>,
806+
'reverse_forward_invert_flags': [],
807+
'reverse_forward_transforms': [],
800808
'reverse_invert_flags': [],
801809
'reverse_transforms': [],
802810
'save_state': '...data/trans.mat',
@@ -826,6 +834,9 @@ class Registration(ANTSCommand):
826834
'inverse_composite_transform': <undefined>,
827835
'inverse_warped_image': <undefined>,
828836
'metric_value': <undefined>,
837+
'reverse_forward_invert_flags': [False, False],
838+
'reverse_forward_transforms': ['...data/output_1Warp.nii.gz',
839+
'...data/output_0GenericAffine.mat'],
829840
'reverse_invert_flags': [True, False],
830841
'reverse_transforms': ['...data/output_0GenericAffine.mat', \
831842
'...data/output_1InverseWarp.nii.gz'],
@@ -1472,6 +1483,10 @@ def _list_outputs(self):
14721483
outputs["metric_value"] = self._metric_value
14731484
if self._elapsed_time:
14741485
outputs["elapsed_time"] = self._elapsed_time
1486+
1487+
outputs["reverse_forward_transforms"] = outputs["forward_transforms"][::-1]
1488+
outputs["reverse_forward_invert_flags"] = outputs["forward_invert_flags"][::-1]
1489+
14751490
return outputs
14761491

14771492

nipype/interfaces/ants/tests/test_auto_Registration.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -97,6 +97,8 @@ def test_Registration_outputs():
9797
inverse_composite_transform=dict(extensions=None,),
9898
inverse_warped_image=dict(extensions=None,),
9999
metric_value=dict(),
100+
reverse_forward_invert_flags=dict(),
101+
reverse_forward_transforms=dict(),
100102
reverse_invert_flags=dict(),
101103
reverse_transforms=dict(),
102104
save_state=dict(extensions=None,),

0 commit comments

Comments
 (0)