update fmap wf in base

nipreps · Dec 11, 2020 · ab515da · ab515da
1 parent 329c1d4
commit ab515da
Showing 1 changed file with 33 additions and 20 deletions.
diff --git a/dmriprep/workflows/base.py b/dmriprep/workflows/base.py
@@ -300,29 +300,34 @@ def init_single_subject_wf(subject_id):
 and a *b=0* average for reference to the subsequent steps of preprocessing was calculated.
 """
     )
-
+    dwi_preproc_list = []
     for dwi_file in subject_data["dwi"]:
         dwi_preproc_wf = init_dwi_preproc_wf(dwi_file)
+        dwi_preproc_list.append(dwi_preproc_wf)
+
+    dwi_preproc_list_wf = pe.Node(niu.IdentityInterface(fields=["dwi_workflows"]),
+                                  name="dwi_preproc_list_wf")
+    dwi_preproc_list_wf.iterables = [("dwi_workflows", dwi_preproc_list)]
 
-        # fmt: off
-        workflow.connect([
-            (anat_preproc_wf, dwi_preproc_wf,
-             [("outputnode.t1w_preproc", "inputnode.t1w_preproc"),
-              ("outputnode.t1w_mask", "inputnode.t1w_mask"),
-              ("outputnode.t1w_dseg", "inputnode.t1w_dseg"),
-              ("outputnode.t1w_aseg", "inputnode.t1w_aseg"),
-              ("outputnode.t1w_aparc", "inputnode.t1w_aparc"),
-              ("outputnode.t1w_tpms", "inputnode.t1w_tpms"),
-              ("outputnode.template", "inputnode.template"),
-              ("outputnode.anat2std_xfm", "inputnode.anat2std_xfm"),
-              ("outputnode.std2anat_xfm", "inputnode.std2anat_xfm"),
-              # Undefined if --fs-no-reconall, but this is safe
-              ("outputnode.subjects_dir", "inputnode.subjects_dir"),
-              ("outputnode.t1w2fsnative_xfm", "inputnode.t1w2fsnative_xfm"),
-              ("outputnode.fsnative2t1w_xfm", "inputnode.fsnative2t1w_xfm")]),
-            (bids_info, dwi_preproc_wf, [("subject", "inputnode.subject_id")]),
-        ])
-        # fmt: on
+    # fmt: off
+    workflow.connect([
+        (anat_preproc_wf, dwi_preproc_list_wf,
+            [("outputnode.t1w_preproc", "inputnode.t1w_preproc"),
+            ("outputnode.t1w_mask", "inputnode.t1w_mask"),
+            ("outputnode.t1w_dseg", "inputnode.t1w_dseg"),
+            ("outputnode.t1w_aseg", "inputnode.t1w_aseg"),
+            ("outputnode.t1w_aparc", "inputnode.t1w_aparc"),
+            ("outputnode.t1w_tpms", "inputnode.t1w_tpms"),
+            ("outputnode.template", "inputnode.template"),
+            ("outputnode.anat2std_xfm", "inputnode.anat2std_xfm"),
+            ("outputnode.std2anat_xfm", "inputnode.std2anat_xfm"),
+            # Undefined if --fs-no-reconall, but this is safe
+            ("outputnode.subjects_dir", "inputnode.subjects_dir"),
+            ("outputnode.t1w2fsnative_xfm", "inputnode.t1w2fsnative_xfm"),
+            ("outputnode.fsnative2t1w_xfm", "inputnode.fsnative2t1w_xfm")]),
+        (bids_info, dwi_preproc_list_wf, [("subject", "inputnode.subject_id")]),
+    ])
+    # fmt: on
 
     if "fieldmap" in config.workflow.ignore:
         return workflow
@@ -348,6 +353,14 @@ def init_single_subject_wf(subject_id):
         subject=subject_id,
     )
 
+    workflow.connect([
+        (fmap_wf, dwi_preproc_list_wf,
+            [("outputnode.fmap", "inputnode.fmap"),
+            ("outputnode.fmap_ref", "inputnode.fmap_ref"),
+            ("outputnode.fmap_coeff", "inputnode.fmap_coeff"),
+            ("outputnode.fmap_mask", "inputnode.fmap_mask")]),
+    ])
+
     # Overwrite ``out_path_base`` of sdcflows's DataSinks
     for node in fmap_wf.list_node_names():
         if node.split(".")[-1].startswith("ds_"):