enh: use *NiWorkflows*' EPI-reference workflow

Replaces our workflows under ``dmriprep.workflows.dwi.util``.

Author: oesteban

dmriprep/workflows/dwi/base.py

@@ -64,16 +64,16 @@ def init_dwi_preproc_wf(dwi_file, has_fieldmap=False):
 
     See Also
     --------
-    * :py:func:`~dmriprep.workflows.dwi.util.init_dwi_reference_wf`
     * :py:func:`~dmriprep.workflows.dwi.outputs.init_dwi_derivatives_wf`
     * :py:func:`~dmriprep.workflows.dwi.outputs.init_reportlets_wf`
 
     """
     from niworkflows.interfaces.reportlets.registration import (
         SimpleBeforeAfterRPT as SimpleBeforeAfter,
     )
+    from niworkflows.workflows.epi.refmap import init_epi_reference_wf
+
     from ...interfaces.vectors import CheckGradientTable
-    from .util import init_dwi_reference_wf
     from .outputs import init_dwi_derivatives_wf, init_reportlets_wf
     from .eddy import init_eddy_wf
 
@@ -143,21 +143,22 @@ def init_dwi_preproc_wf(dwi_file, has_fieldmap=False):
 
     gradient_table = pe.Node(CheckGradientTable(), name="gradient_table")
 
-    dwi_reference_wf = init_dwi_reference_wf(
-        mem_gb=config.DEFAULT_MEMORY_MIN_GB, omp_nthreads=config.nipype.omp_nthreads
+    dwi_reference_wf = init_epi_reference_wf(
+        omp_nthreads=config.nipype.omp_nthreads,
+        name="dwi_reference_wf",
     )
 
     dwi_derivatives_wf = init_dwi_derivatives_wf(output_dir=str(config.execution.output_dir))
 
     # MAIN WORKFLOW STRUCTURE
     # fmt: off
     workflow.connect([
+        (inputnode, dwi_derivatives_wf, [("dwi_file", "inputnode.source_file")]),
         (inputnode, gradient_table, [("dwi_file", "dwi_file"),
                                      ("in_bvec", "in_bvec"),
                                      ("in_bval", "in_bval")]),
-        (inputnode, dwi_reference_wf, [("dwi_file", "inputnode.dwi_file")]),
-        (inputnode, dwi_derivatives_wf, [("dwi_file", "inputnode.source_file")]),
-        (gradient_table, dwi_reference_wf, [("b0_ixs", "inputnode.b0_ixs")]),
+        (inputnode, dwi_reference_wf, [(("dwi_file", _aslist), "inputnode.in_files")]),
+        (gradient_table, dwi_reference_wf, [(("b0_mask", _aslist), "inputnode.t_masks")]),
         (gradient_table, outputnode, [("out_rasb", "gradients_rasb")]),
         (outputnode, dwi_derivatives_wf, [
             ("dwi_reference", "inputnode.dwi_ref"),
@@ -205,7 +206,7 @@ def _bold_reg_suffix(fallback):
             (inputnode, ds_report_reg, [("dwi_file", "source_file")]),
             # BBRegister
             (dwi_reference_wf, bbr_wf, [
-                ("outputnode.ref_image", "inputnode.in_file")
+                ("outputnode.epi_ref_file", "inputnode.in_file")
             ]),
             (bbr_wf, ds_report_reg, [
                 ('outputnode.out_report', 'in_file'),
@@ -247,7 +248,7 @@ def _bold_reg_suffix(fallback):
                 ("in_bvec", "inputnode.in_bvec"),
                 ("in_bval", "inputnode.in_bval")
             ]),
-            (dwi_reference_wf, eddy_report, [("outputnode.ref_image", "before")]),
+            (dwi_reference_wf, eddy_report, [("outputnode.epi_ref_file", "before")]),
             (eddy_wf, eddy_report, [('outputnode.eddy_ref_image', 'after')]),
             (dwi_reference_wf, ds_report_eddy, [("outputnode.dwi_file", "source_file")]),
             (eddy_report, ds_report_eddy, [("out_report", "in_file")]),
@@ -275,8 +276,7 @@ def _bold_reg_suffix(fallback):
     if not has_fieldmap:
         # fmt: off
         workflow.connect([
-            (dwi_reference_wf, outputnode, [("outputnode.ref_image", "dwi_reference"),
-                                            ("outputnode.dwi_mask", "dwi_mask")]),
+            (dwi_reference_wf, outputnode, [("outputnode.epi_ref_file", "dwi_reference")]),
         ])
         # fmt: on
         return workflow
@@ -328,12 +328,11 @@ def _bold_reg_suffix(fallback):
             ("fmap_coeff", "inputnode.fmap_coeff"),
             ("fmap_mask", "inputnode.fmap_mask")]),
         (dwi_reference_wf, coeff2epi_wf, [
-            ("outputnode.ref_image", "inputnode.target_ref"),
-            ("outputnode.dwi_mask", "inputnode.target_mask")]),
-        (dwi_reference_wf, unwarp_wf, [("outputnode.ref_image", "inputnode.distorted")]),
+            ("outputnode.epi_ref_file", "inputnode.target_ref")]),
+        (dwi_reference_wf, unwarp_wf, [("outputnode.epi_ref_file", "inputnode.distorted")]),
         (coeff2epi_wf, unwarp_wf, [
             ("outputnode.fmap_coeff", "inputnode.fmap_coeff")]),
-        (dwi_reference_wf, sdc_report, [("outputnode.ref_image", "before")]),
+        (dwi_reference_wf, sdc_report, [("outputnode.epi_ref_file", "before")]),
         (unwarp_wf, sdc_report, [("outputnode.corrected", "after"),
                                  ("outputnode.corrected_mask", "wm_seg")]),
         (sdc_report, reportlets_wf, [("out_report", "inputnode.sdc_report")]),
@@ -363,3 +362,7 @@ def _get_wf_name(filename):
     fname = Path(filename).name.rpartition(".nii")[0].replace("_dwi", "_wf")
     fname_nosub = "_".join(fname.split("_")[1:])
     return f"dwi_preproc_{fname_nosub.replace('.', '_').replace(' ', '').replace('-', '_')}"
+
+
+def _aslist(value):
+    return [value]