enh: output cortex mask

Author: mgxd

nibabies/data/io_spec_anat.json

@@ -232,6 +232,14 @@
           "desc": "msmsulc",
           "suffix": "sphere",
           "extension": ".surf.gii"
+        },
+        "cortex_mask": {
+          "datatype": "anat",
+          "hemi": ["L", "R"],
+          "space": null,
+          "desc": "cortex",
+          "suffix": "mask",
+          "extension": ".label.gii"
         }
       },
       "masks": {

nibabies/workflows/anatomical/apply.py

@@ -75,7 +75,7 @@ def init_infant_anat_apply_wf(
     )
 
     outputnode = pe.Node(
-        niu.IdentityInterface(fields=['anat_aseg', 'anat_aparc', 'midthickness_fsLR', 'roi']),
+        niu.IdentityInterface(fields=['anat_aseg', 'anat_aparc', 'midthickness_fsLR']),
         name='outputnode',
     )
 
@@ -242,10 +242,6 @@ def init_infant_anat_apply_wf(
                     ('outputnode.curv', 'inputnode.curv'),
                     ('outputnode.sulc', 'inputnode.sulc'),
                     ('outputnode.thickness', 'inputnode.thickness'),
-                    ('outputnode.roi', 'inputnode.roi'),
-                ]),
-                (hcp_morphometrics_wf, outputnode, [
-                    ('outputnode.roi', 'roi'),
                 ]),
                 (resample_surfaces_wf, morph_grayords_wf, [
                     ('outputnode.midthickness_fsLR', 'inputnode.midthickness_fsLR'),

nibabies/workflows/anatomical/fit.py

@@ -13,21 +13,23 @@
 from niworkflows.utils.connections import pop_file
 from smriprep.workflows.anatomical import (
     _is_skull_stripped,
-    init_anat_ribbon_wf,
     init_anat_template_wf,
 )
 from smriprep.workflows.fit.registration import init_register_template_wf
 from smriprep.workflows.outputs import (
     init_ds_dseg_wf,
     init_ds_fs_registration_wf,
     init_ds_mask_wf,
+    init_ds_surface_masks_wf,
     init_ds_surface_metrics_wf,
     init_ds_surfaces_wf,
     init_ds_template_registration_wf,
     init_ds_template_wf,
     init_ds_tpms_wf,
 )
 from smriprep.workflows.surfaces import (
+    init_anat_ribbon_wf,
+    init_cortex_masks_wf,
     init_fsLR_reg_wf,
     init_gifti_morphometrics_wf,
     init_gifti_surfaces_wf,
@@ -147,6 +149,7 @@ def init_infant_anat_fit_wf(
                 'sphere_reg',
                 'sphere_reg_fsLR',
                 'sphere_reg_msm',
+                'cortex_mask',
                 'anat_ribbon',
                 # Reverse transform; not computable from forward transform
                 'std2anat_xfm',
@@ -1413,6 +1416,32 @@ def init_infant_anat_fit_wf(
         else:
             LOGGER.info('ANAT Stage 9: Found pre-computed fsLR registration sphere')
             fsLR_buffer.inputs.sphere_reg_fsLR = sorted(precomputed['sphere_reg_fsLR'])
+
+    # Stage 10: Cortical surface mask
+    if len(precomputed.get('cortex_mask', [])) < 2:
+        LOGGER.info('ANAT Stage 11: Creating cortical surface mask')
+
+        cortex_masks_wf = init_cortex_masks_wf()
+        ds_cortex_masks_wf = init_ds_surface_masks_wf(
+            output_dir=output_dir,
+            mask_type='cortex',
+            name='ds_cortex_masks_wf',
+        )
+
+        workflow.connect([
+            (surfaces_buffer, cortex_masks_wf, [
+                ('midthickness', 'inputnode.midthickness'),
+                ('thickness', 'inputnode.thickness'),
+            ]),
+            (cortex_masks_wf, ds_cortex_masks_wf, [
+                ('outputnode.cortex_masks', 'inputnode.mask_files'),
+                ('outputnode.source_files', 'inputnode.source_files'),
+            ]),
+            (ds_cortex_masks_wf, outputnode, [('outputnode.mask_files', 'cortex_mask')]),
+        ])  # fmt:skip
+    else:
+        LOGGER.info('ANAT Stage 11: Found pre-computed cortical surface mask')
+        outputnode.inputs.cortex_mask = sorted(precomputed['cortex_mask'])
     return workflow
 
 
@@ -1476,6 +1505,7 @@ def init_infant_single_anat_fit_wf(
                 'sphere_reg',
                 'sphere_reg_fsLR',
                 'sphere_reg_msm',
+                'cortex_mask',
                 'anat_ribbon',
                 # Reverse transform; not computable from forward transform
                 'std2anat_xfm',
@@ -2342,4 +2372,30 @@ def init_infant_single_anat_fit_wf(
         else:
             LOGGER.info('ANAT Stage 9: Found pre-computed fsLR registration sphere')
             fsLR_buffer.inputs.sphere_reg_fsLR = sorted(precomputed['sphere_reg_fsLR'])
+
+    # Stage 10: Cortical surface mask
+    if len(precomputed.get('cortex_mask', [])) < 2:
+        LOGGER.info('ANAT Stage 11: Creating cortical surface mask')
+
+        cortex_masks_wf = init_cortex_masks_wf()
+        ds_cortex_masks_wf = init_ds_surface_masks_wf(
+            output_dir=output_dir,
+            mask_type='cortex',
+            name='ds_cortex_masks_wf',
+        )
+
+        workflow.connect([
+            (surfaces_buffer, cortex_masks_wf, [
+                ('midthickness', 'inputnode.midthickness'),
+                ('thickness', 'inputnode.thickness'),
+            ]),
+            (cortex_masks_wf, ds_cortex_masks_wf, [
+                ('outputnode.cortex_masks', 'inputnode.mask_files'),
+                ('outputnode.source_files', 'inputnode.source_files'),
+            ]),
+            (ds_cortex_masks_wf, outputnode, [('outputnode.mask_files', 'cortex_mask')]),
+        ])  # fmt:skip
+    else:
+        LOGGER.info('ANAT Stage 11: Found pre-computed cortical surface mask')
+        outputnode.inputs.cortex_mask = sorted(precomputed['cortex_mask'])
     return workflow

nibabies/workflows/base.py

@@ -809,8 +809,10 @@ def init_single_subject_wf(
 
             if config.workflow.cifti_output:
                 workflow.connect([
+                    (anat_fit_wf, bold_wf, [
+                        ('outputnode.cortex_mask', 'inputnode.cortex_mask'),
+                    ]),
                     (anat_apply_wf, bold_wf, [
-                        ('outputnode.roi', 'inputnode.cortex_mask'),
                         ('outputnode.midthickness_fsLR', 'inputnode.midthickness_fsLR'),
                         ('outputnode.anat_aseg', 'inputnode.anat_aseg'),
                     ]),

pyproject.toml

@@ -25,8 +25,7 @@ dependencies = [
     "nireports >= 23.2.0",
     "nitime",
     "nitransforms >= 24.1.1",
-    #"niworkflows >= 1.13.1",
-    "niworkflows @ git+https://github.com/nipreps/niworkflows.git@master",
+    "niworkflows >= 1.14.1",
     "numpy >= 1.21.0",
     "packaging",
     "pandas < 3",
@@ -35,7 +34,7 @@ dependencies = [
     "pybids >= 0.15.0",
     "requests",
     "sdcflows >= 2.13.0",
-    "smriprep >= 0.17.0",
+    "smriprep >= 0.19.1",
     "tedana >= 23.0.2",
     "templateflow >= 24.2.0",
     "toml",