Split bold_reg_wf

Author: emdupre

fmriprep/workflows/bold/__init__.py

@@ -24,7 +24,10 @@
 from .hmc import init_bold_hmc_wf
 from .stc import init_bold_stc_wf
 from .t2s import init_bold_t2s_wf
-from .registration import init_bold_reg_wf
+from .registration import (
+    init_bold_apply_reg_wf,
+    init_bold_calc_reg_wf,
+)
 from .resampling import (
     init_bold_mni_trans_wf,
     init_bold_surf_wf,

fmriprep/workflows/bold/base.py

@@ -34,7 +34,7 @@
 from .hmc import init_bold_hmc_wf
 from .stc import init_bold_stc_wf
 from .t2s import init_bold_t2s_wf
-from .registration import init_bold_reg_wf
+from .registration import init_bold_apply_reg_wf, init_bold_calc_reg_wf
 from .resampling import (
     init_bold_surf_wf,
     init_bold_mni_trans_wf,
@@ -211,7 +211,8 @@ def init_func_preproc_wf(bold_file, ignore, freesurfer,
         * :py:func:`~fmriprep.workflows.bold.stc.init_bold_stc_wf`
         * :py:func:`~fmriprep.workflows.bold.hmc.init_bold_hmc_wf`
         * :py:func:`~fmriprep.workflows.bold.t2s.init_bold_t2s_wf`
-        * :py:func:`~fmriprep.workflows.bold.registration.init_bold_reg_wf`
+        * :py:func:`~fmriprep.workflows.bold.registration.init_bold_apply_reg_wf`
+        * :py:func:`~fmriprep.workflows.bold.registration.init_bold_calc_reg_wf`
         * :py:func:`~fmriprep.workflows.bold.confounds.init_bold_confounds_wf`
         * :py:func:`~fmriprep.workflows.bold.confounds.init_ica_aroma_wf`
         * :py:func:`~fmriprep.workflows.bold.resampling.init_bold_mni_trans_wf`
@@ -380,14 +381,20 @@ def init_func_preproc_wf(bold_file, ignore, freesurfer,
                                    omp_nthreads=omp_nthreads)
 
     # mean BOLD registration to T1w
-    bold_reg_wf = init_bold_reg_wf(name='bold_reg_wf',
-                                   freesurfer=freesurfer,
-                                   use_bbr=use_bbr,
-                                   bold2t1w_dof=bold2t1w_dof,
-                                   mem_gb=mem_gb['resampled'],
-                                   omp_nthreads=omp_nthreads,
-                                   use_compression=False,
-                                   use_fieldwarp=(fmaps is not None or use_syn))
+    bold_calc_reg_wf = init_bold_calc_reg_wf(name='bold_calc_reg_wf',
+                                             freesurfer=freesurfer,
+                                             use_bbr=use_bbr,
+                                             bold2t1w_dof=bold2t1w_dof,
+                                             mem_gb=mem_gb['resampled'],
+                                             omp_nthreads=omp_nthreads,
+                                             use_compression=False)
+
+    # mean BOLD registration to T1w
+    bold_apply_reg_wf = init_bold_apply_reg_wf(name='bold_apply_reg_wf',
+                                               mem_gb=mem_gb['resampled'],
+                                               omp_nthreads=omp_nthreads,
+                                               use_compression=False,
+                                               use_fieldwarp=(fmaps is not None or use_syn))
 
     # get confounds
     bold_confounds_wf = init_bold_confs_wf(
@@ -449,7 +456,7 @@ def init_func_preproc_wf(bold_file, ignore, freesurfer,
             ('outputnode.raw_ref_image', 'inputnode.raw_ref_image'),
             ('outputnode.bold_file', 'inputnode.bold_file')]),
         # EPI-T1 registration workflow
-        (inputnode, bold_reg_wf, [
+        (inputnode, bold_calc_reg_wf, [
             ('bold_file', 'inputnode.name_source'),
             ('t1_preproc', 'inputnode.t1_preproc'),
             ('t1_brain', 'inputnode.t1_brain'),
@@ -461,12 +468,15 @@ def init_func_preproc_wf(bold_file, ignore, freesurfer,
             ('subjects_dir', 'inputnode.subjects_dir'),
             ('subject_id', 'inputnode.subject_id'),
             ('t1_2_fsnative_reverse_transform', 'inputnode.t1_2_fsnative_reverse_transform')]),
-        (bold_split, bold_reg_wf, [('out_files', 'inputnode.bold_split')]),
-        (bold_hmc_wf, bold_reg_wf, [('outputnode.xforms', 'inputnode.hmc_xforms')]),
-        (bold_reg_wf, outputnode, [('outputnode.bold_t1', 'bold_t1'),
-                                   ('outputnode.bold_aseg_t1', 'bold_aseg_t1'),
-                                   ('outputnode.bold_aparc_t1', 'bold_aparc_t1')]),
-        (bold_reg_wf, summary, [('outputnode.fallback', 'fallback')]),
+        (bold_split, bold_calc_reg_wf, [('out_files', 'inputnode.bold_split')]),
+        (bold_hmc_wf, bold_apply_reg_wf, [('outputnode.xforms', 'inputnode.hmc_xforms')]),
+        (bold_calc_reg_wf, outputnode, [('outputnode.bold_aseg_t1', 'bold_aseg_t1'),
+                                        ('outputnode.bold_aparc_t1', 'bold_aparc_t1')]),
+        (bold_calc_reg_wf, bold_apply_reg_wf, [
+            ('outputnode.reference_image', 'inputnode.reference_grid'),
+            ('outputnode.itk_bold_to_t1', 'inputnode.itk_bold_to_t1')]),
+        (bold_apply_reg_wf, outputnode, [('outputnode.bold_t1', 'bold_t1')]),
+        (bold_calc_reg_wf, summary, [('outputnode.fallback', 'fallback')]),
         # SDC (or pass-through workflow)
         (inputnode, bold_sdc_wf, [
             ('t1_brain', 'inputnode.t1_brain'),
@@ -475,9 +485,10 @@ def init_func_preproc_wf(bold_file, ignore, freesurfer,
             ('outputnode.ref_image', 'inputnode.bold_ref'),
             ('outputnode.ref_image_brain', 'inputnode.bold_ref_brain'),
             ('outputnode.bold_mask', 'inputnode.bold_mask')]),
-        (bold_sdc_wf, bold_reg_wf, [
+        (bold_sdc_wf, bold_calc_reg_wf, [
             ('outputnode.bold_ref_brain', 'inputnode.ref_bold_brain'),
-            ('outputnode.bold_mask', 'inputnode.ref_bold_mask'),
+            ('outputnode.bold_mask', 'inputnode.ref_bold_mask')]),
+        (bold_sdc_wf, bold_apply_reg_wf, [
             ('outputnode.out_warp', 'inputnode.fieldwarp')]),
         (bold_sdc_wf, bold_bold_trans_wf, [
             ('outputnode.out_warp', 'inputnode.fieldwarp'),
@@ -488,7 +499,7 @@ def init_func_preproc_wf(bold_file, ignore, freesurfer,
                                         ('t1_mask', 'inputnode.t1_mask')]),
         (bold_hmc_wf, bold_confounds_wf, [
             ('outputnode.movpar_file', 'inputnode.movpar_file')]),
-        (bold_reg_wf, bold_confounds_wf, [
+        (bold_calc_reg_wf, bold_confounds_wf, [
             ('outputnode.itk_t1_to_bold', 'inputnode.t1_bold_xform')]),
         (bold_confounds_wf, outputnode, [
             ('outputnode.confounds_file', 'confounds'),
@@ -519,7 +530,7 @@ def init_func_preproc_wf(bold_file, ignore, freesurfer,
                 ('t1_seg', 'inputnode.in_seg')]),
             (bold_reference_wf, fmap_unwarp_report_wf, [
                 ('outputnode.ref_image', 'inputnode.in_pre')]),
-            (bold_reg_wf, fmap_unwarp_report_wf, [
+            (bold_calc_reg_wf, fmap_unwarp_report_wf, [
                 ('outputnode.itk_t1_to_bold', 'inputnode.in_xfm')]),
             (bold_sdc_wf, fmap_unwarp_report_wf, [
                 ('outputnode.bold_ref', 'inputnode.in_post')]),
@@ -533,7 +544,7 @@ def init_func_preproc_wf(bold_file, ignore, freesurfer,
                     ('t1_seg', 'inputnode.in_seg')]),
                 (bold_reference_wf, syn_unwarp_report_wf, [
                     ('outputnode.ref_image', 'inputnode.in_pre')]),
-                (bold_reg_wf, syn_unwarp_report_wf, [
+                (bold_calc_reg_wf, syn_unwarp_report_wf, [
                     ('outputnode.itk_t1_to_bold', 'inputnode.in_xfm')]),
                 (bold_sdc_wf, syn_unwarp_report_wf, [
                     ('outputnode.syn_bold_ref', 'inputnode.in_post')]),
@@ -571,14 +582,14 @@ def init_func_preproc_wf(bold_file, ignore, freesurfer,
 
             # Replace EPI-to-T1w registration inputs
             workflow.disconnect([
-                (bold_sdc_wf, bold_reg_wf, [
+                (bold_sdc_wf, bold_calc_reg_wf, [
                     ('outputnode.bold_ref_brain', 'inputnode.ref_bold_brain'),
                     ('outputnode.bold_mask', 'inputnode.ref_bold_mask')]),
             ])
             workflow.connect([
                 (bold_hmc_wf, bold_t2s_wf, [
                     ('outputnode.xforms', 'inputnode.hmc_xforms')]),
-                (bold_t2s_wf, bold_reg_wf, [
+                (bold_t2s_wf, bold_calc_reg_wf, [
                     ('outputnode.t2s_map', 'inputnode.ref_bold_brain'),
                     ('outputnode.oc_mask', 'inputnode.ref_bold_mask')]),
             ])
@@ -596,7 +607,7 @@ def init_func_preproc_wf(bold_file, ignore, freesurfer,
         workflow.connect([
             (bold_bold_trans_wf, boldmask_to_t1w, [
                 ('outputnode.bold_mask', 'input_image')]),
-            (bold_reg_wf, boldmask_to_t1w, [
+            (bold_calc_reg_wf, boldmask_to_t1w, [
                 ('outputnode.bold_mask_t1', 'reference_image'),
                 ('outputnode.itk_bold_to_t1', 'transforms')]),
             (boldmask_to_t1w, outputnode, [
@@ -628,7 +639,7 @@ def init_func_preproc_wf(bold_file, ignore, freesurfer,
                 ('out_files', 'inputnode.bold_split')]),
             (bold_hmc_wf, bold_mni_trans_wf, [
                 ('outputnode.xforms', 'inputnode.hmc_xforms')]),
-            (bold_reg_wf, bold_mni_trans_wf, [
+            (bold_calc_reg_wf, bold_mni_trans_wf, [
                 ('outputnode.itk_bold_to_t1', 'inputnode.itk_bold_to_t1')]),
             (bold_bold_trans_wf, bold_mni_trans_wf, [
                 ('outputnode.bold_mask', 'inputnode.bold_mask')]),
@@ -641,7 +652,7 @@ def init_func_preproc_wf(bold_file, ignore, freesurfer,
                 ('outputnode.bold_mask', 'inputnode.bold_mask')]),
             (inputnode, carpetplot_wf, [
                 ('t1_2_mni_reverse_transform', 'inputnode.t1_2_mni_reverse_transform')]),
-            (bold_reg_wf, carpetplot_wf, [
+            (bold_calc_reg_wf, carpetplot_wf, [
                 ('outputnode.itk_t1_to_bold', 'inputnode.t1_bold_xform')]),
             (bold_confounds_wf, carpetplot_wf, [
                 ('outputnode.confounds_file', 'inputnode.confounds_file')]),
@@ -679,7 +690,7 @@ def init_func_preproc_wf(bold_file, ignore, freesurfer,
                 (bold_hmc_wf, ica_aroma_wf, [
                     ('outputnode.movpar_file', 'inputnode.movpar_file'),
                     ('outputnode.xforms', 'inputnode.hmc_xforms')]),
-                (bold_reg_wf, ica_aroma_wf, [
+                (bold_calc_reg_wf, ica_aroma_wf, [
                     ('outputnode.itk_bold_to_t1', 'inputnode.itk_bold_to_t1')]),
                 (bold_bold_trans_wf, ica_aroma_wf, [
                     ('outputnode.bold_mask', 'inputnode.bold_mask')]),
@@ -709,7 +720,7 @@ def init_func_preproc_wf(bold_file, ignore, freesurfer,
                 ('subjects_dir', 'inputnode.subjects_dir'),
                 ('subject_id', 'inputnode.subject_id'),
                 ('t1_2_fsnative_forward_transform', 'inputnode.t1_2_fsnative_forward_transform')]),
-            (bold_reg_wf, bold_surf_wf, [('outputnode.bold_t1', 'inputnode.source_file')]),
+            (bold_apply_reg_wf, bold_surf_wf, [('outputnode.bold_t1', 'inputnode.source_file')]),
             (bold_surf_wf, outputnode, [('outputnode.surfaces', 'surfaces')]),
         ])