fix: re-enable priors respecting sd_priors argument

Author: oesteban

sdcflows/workflows/fit/syn.py

@@ -44,13 +44,11 @@
 
 def init_syn_sdc_wf(
     *,
-    atlas_threshold=3,
     sloppy=False,
     debug=False,
     name="syn_sdc_wf",
     omp_nthreads=1,
     laplacian_weight=None,
-    sd_prior=True,
     **kwargs,
 ):
     """
@@ -59,10 +57,6 @@ def init_syn_sdc_wf(
     SyN deformation is restricted to the phase-encoding (PE) direction.
     If no PE direction is specified, anterior-posterior PE is assumed.
 
-    SyN deformation is also restricted to regions that are expected to have a
-    >3mm (approximately 1 voxel) warp, based on the fieldmap atlas.
-
-
     Workflow Graph
         .. workflow ::
             :graph2use: orig
@@ -73,9 +67,6 @@ def init_syn_sdc_wf(
 
     Parameters
     ----------
-    atlas_threshold : :obj:`float`
-        Exclude from the registration metric computation areas with average distortions
-        below this threshold (in mm).
     sloppy : :obj:`bool`
         Whether a fast (less accurate) configuration of the workflow should be applied.
     debug : :obj:`bool`
@@ -150,16 +141,15 @@ def init_syn_sdc_wf(
     workflow = Workflow(name=name)
     workflow.__desc__ = f"""\
 A deformation field to correct for susceptibility distortions was estimated
-based on *fMRIPrep*'s *fieldmap-less* approach.
+based on *SDCFlows*' *fieldmap-less* approach.
 The deformation field is that resulting from co-registering the EPI reference
-to the same-subject T1w-reference with its intensity inverted [@fieldmapless1;
-@fieldmapless2].
+to the same-subject's T1w-reference [@fieldmapless1; @fieldmapless2].
 Registration is performed with `antsRegistration`
 (ANTs {ants_version or "-- version unknown"}), and
 the process regularized by constraining deformation to be nonzero only
-along the phase-encoding direction, and modulated with an average fieldmap
-template [@fieldmapless3].
+along the phase-encoding direction.
 """
+
     inputnode = pe.Node(niu.IdentityInterface(INPUT_FIELDS), name="inputnode")
     outputnode = pe.Node(
         niu.IdentityInterface(
@@ -211,10 +201,11 @@ def init_syn_sdc_wf(
 
     epi_umask = pe.Node(Union(), name="epi_umask")
     moving_masks = pe.Node(
-        niu.Merge(3),
+        niu.Merge(2),
         name="moving_masks",
         run_without_submitting=True,
     )
+    moving_masks.inputs.in1 = "NULL"
 
     fixed_masks = pe.Node(
         niu.Merge(2),
@@ -287,7 +278,7 @@ def init_syn_sdc_wf(
         (inputnode, amask2epi, [("epi_mask", "reference_image")]),
         (inputnode, zooms_bmask, [("anat_mask", "input_image")]),
         (inputnode, fixed_masks, [("anat_mask", "in1"),
-                                  ("anat_mask", "in2")]),
+                                  ("sd_prior", "in2")]),
         (inputnode, anat_dilmsk, [("anat_mask", "in_file")]),
         (inputnode, warp_dir, [("anat_ref", "fixed_image")]),
         (inputnode, anat_merge, [("anat_ref", "in1")]),
@@ -298,9 +289,7 @@ def init_syn_sdc_wf(
         (inputnode, epi_umask, [("epi_mask", "in1")]),
         (lap_anat, lap_anat_norm, [("output_image", "in_file")]),
         (lap_anat_norm, anat_merge, [("out", "in2")]),
-        (epi_umask, moving_masks, [("out_file", "in1"),
-                                   ("out_file", "in2"),
-                                   ("out_file", "in3")]),
+        (epi_umask, moving_masks, [("out_file", "in2")]),
         (clip_epi, epi_merge, [("out_file", "in1")]),
         (clip_epi, lap_epi, [("out_file", "op1")]),
         (clip_epi, zooms_epi, [("out_file", "in_file")]),
@@ -339,6 +328,7 @@ def init_syn_sdc_wf(
 
 def init_syn_preprocessing_wf(
     *,
+    atlas_threshold=3,
     debug=False,
     name="syn_preprocessing_wf",
     omp_nthreads=1,
@@ -359,6 +349,9 @@ def init_syn_preprocessing_wf(
 
     Parameters
     ----------
+    atlas_threshold : :obj:`float`
+        Mask excluding areas with average distortions below this threshold (in mm)
+        on the prior.
     debug : :obj:`bool`
         Whether a fast (less accurate) configuration of the workflow should be applied.
     name : :obj:`str`
@@ -499,6 +492,8 @@ def _remove_first_mask(in_file):
     sampling_ref = pe.Node(GenerateSamplingReference(), name="sampling_ref")
 
     if sd_prior:
+        from niworkflows.interfaces.nibabel import Binarize
+
         # Mapping & preparing prior knowledge
         # Concatenate transform files:
         # 1) MNI -> anat; 2) ATLAS -> MNI
@@ -521,19 +516,20 @@ def _remove_first_mask(in_file):
             mem_gb=0.3,
         )
 
+        prior_msk = pe.Node(Binarize(thresh_low=atlas_threshold), name="prior_msk")
+
         workflow.connect([
             (inputnode, transform_list, [("std2anat_xfm", "in2")]),
             (epi2anat, transform_list, [("forward_transforms", "in1")]),
             (transform_list, prior2epi, [("out", "transforms")]),
             (sampling_ref, prior2epi, [("out_file", "reference_image")]),
-            (prior2epi, outputnode, [("output_image", "sd_prior")]),
+            (prior2epi, prior_msk, [("output_image", "in_file")]),
+            (prior_msk, outputnode, [("out_mask", "sd_prior")]),
         ])  # fmt:skip
 
     else:
-        # no prior to be used
-        # MG: Future goal is to allow using alternative mappings
-        # i.e. in the case of infants, where priors change depending on development
-        outputnode.inputs.sd_prior = Undefined
+        # no prior to be used -> set anatomical mask as prior
+        workflow.connect(mask_dtype, "out", outputnode, "sd_prior")
 
     workflow.connect([
         (inputnode, epi_reference_wf, [("in_epis", "inputnode.in_files")]),