fix for images where flip -> unflip does not return to original space

- this was happening with some hippb500 images, but fixed it for any modality, in case it happens again.. - the fix is to copy the transform from the original (unflipped) header after unflipping. we just use `get_nnunet_input` for this..
khanlab · Mar 27, 2023 · 358e6ff · 358e6ff
1 parent 2a80f37
commit 358e6ff
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 4 deletions.
diff --git a/hippunfold/workflow/rules/autotop.smk b/hippunfold/workflow/rules/autotop.smk
@@ -231,6 +231,7 @@ rule unflip_coords:
             hemi="{hemi}flip",
             **config["subj_wildcards"]
         ),
+        unflip_ref=get_nnunet_input,
     output:
         nii=bids(
             root=work,
@@ -248,7 +249,8 @@ rule unflip_coords:
     group:
         "subj"
     shell:
-        "c3d {input} -flip x {output}"
+        "c3d {input.nii} -flip x -popas FLIPPED "
+        " {input.unflip_ref} -push FLIPPED -copy-transform -o {output.nii} "
 
 
 rule unflip_coords_equivol:
@@ -264,6 +266,7 @@ rule unflip_coords_equivol:
             hemi="{hemi}flip",
             **config["subj_wildcards"]
         ),
+        unflip_ref=get_nnunet_input,
     output:
         nii=bids(
             root=work,
@@ -281,4 +284,5 @@ rule unflip_coords_equivol:
     group:
         "subj"
     shell:
-        "c3d {input} -flip x {output}"
+        "c3d {input.nii} -flip x -popas FLIPPED "
+        " {input.unflip_ref} -push FLIPPED -copy-transform -o {output.nii} "
diff --git a/hippunfold/workflow/rules/nnunet.smk b/hippunfold/workflow/rules/nnunet.smk
@@ -158,6 +158,7 @@ rule unflip_nnunet_nii:
             space="corobl",
             hemi="{hemi}flip"
         ),
+        unflip_ref=get_nnunet_input,
     output:
         nnunet_seg=bids(
             root=work,
@@ -173,7 +174,8 @@ rule unflip_nnunet_nii:
     group:
         "subj"
     shell:
-        "c3d {input} -flip x {output}"
+        "c3d {input.nnunet_seg} -flip x -popas FLIPPED "
+        " {input.unflip_ref} -push FLIPPED -copy-transform -o {output.nnunet_seg} "
 
 
 def get_f3d_ref(wildcards):

diff --git a/hippunfold/workflow/rules/shape_inject.smk b/hippunfold/workflow/rules/shape_inject.smk
@@ -386,6 +386,7 @@ rule unflip_postproc:
             hemi="{hemi}flip",
             **config["subj_wildcards"]
         ),
+        unflip_ref=get_nnunet_input,
     output:
         nii=bids(
             root=work,
@@ -401,4 +402,5 @@ rule unflip_postproc:
     group:
         "subj"
     shell:
-        "c3d {input} -flip x {output}"
+        "c3d {input.nii} -flip x -popas FLIPPED "
+        " {input.unflip_ref} -push FLIPPED -copy-transform -o {output.nii} "