Skip to content

Commit 20572de

Browse files
committed
Change reg workflow to produce consistent outputs
1 parent 013aaab commit 20572de

File tree

1 file changed

+23
-49
lines changed

1 file changed

+23
-49
lines changed

CPAC/registration/registration.py

Lines changed: 23 additions & 49 deletions
Original file line numberDiff line numberDiff line change
@@ -1421,15 +1421,9 @@ def FSL_registration_connector(wf_name, cfg, orig="T1w", opt=None,
14211421

14221422

14231423
if opt == 'FSL':
1424-
if cfg.registration_workflows['anatomical_registration']['registration']['FSL-FNIRT']['ref_resolution'] == \
1425-
cfg.registration_workflows['anatomical_registration']['resolution_for_anat']:
1426-
fnirt_reg_anat_mni = create_fsl_fnirt_nonlinear_reg(
1427-
f'anat_mni_fnirt_register{symm}'
1428-
)
1429-
else:
1430-
fnirt_reg_anat_mni = create_fsl_fnirt_nonlinear_reg_nhp(
1431-
f'anat_mni_fnirt_register{symm}'
1432-
)
1424+
fnirt_reg_anat_mni = create_fsl_fnirt_nonlinear_reg_nhp(
1425+
f'anat_mni_fnirt_register{symm}'
1426+
)
14331427

14341428
wf.connect(inputNode, 'input_brain',
14351429
fnirt_reg_anat_mni, 'inputspec.input_brain')
@@ -1454,33 +1448,22 @@ def FSL_registration_connector(wf_name, cfg, orig="T1w", opt=None,
14541448
wf.connect(inputNode, 'fnirt_config',
14551449
fnirt_reg_anat_mni, 'inputspec.fnirt_config')
14561450

1457-
if cfg.registration_workflows['anatomical_registration']['registration']['FSL-FNIRT']['ref_resolution'] == \
1458-
cfg.registration_workflows['anatomical_registration']['resolution_for_anat']:
1459-
# NOTE: this is an UPDATE because of the opt block above
1460-
added_outputs = {
1461-
f'space-{sym}template_desc-preproc_{orig}': (
1462-
fnirt_reg_anat_mni, 'outputspec.output_brain'),
1463-
f'from-{orig}_to-{sym}{tmpl}template_mode-image_xfm': (
1464-
fnirt_reg_anat_mni, 'outputspec.nonlinear_xfm')
1465-
}
1466-
outputs.update(added_outputs)
1467-
else:
1468-
# NOTE: this is an UPDATE because of the opt block above
1469-
added_outputs = {
1470-
f'space-{sym}template_desc-preproc_{orig}': (
1471-
fnirt_reg_anat_mni, 'outputspec.output_brain'),
1472-
f'space-{sym}template_desc-head_{orig}': (
1473-
fnirt_reg_anat_mni, 'outputspec.output_head'),
1474-
f'space-{sym}template_desc-{orig}_mask': (
1475-
fnirt_reg_anat_mni, 'outputspec.output_mask'),
1476-
f'space-{sym}template_desc-T1wT2w_biasfield': (
1477-
fnirt_reg_anat_mni, 'outputspec.output_biasfield'),
1478-
f'from-{orig}_to-{sym}{tmpl}template_mode-image_xfm': (
1479-
fnirt_reg_anat_mni, 'outputspec.nonlinear_xfm'),
1480-
f'from-{orig}_to-{sym}{tmpl}template_mode-image_warp': (
1481-
fnirt_reg_anat_mni, 'outputspec.nonlinear_warp')
1482-
}
1483-
outputs.update(added_outputs)
1451+
# NOTE: this is an UPDATE because of the opt block above
1452+
added_outputs = {
1453+
f'space-{sym}template_desc-preproc_{orig}': (
1454+
fnirt_reg_anat_mni, 'outputspec.output_brain'),
1455+
f'space-{sym}template_desc-head_{orig}': (
1456+
fnirt_reg_anat_mni, 'outputspec.output_head'),
1457+
f'space-{sym}template_desc-{orig}_mask': (
1458+
fnirt_reg_anat_mni, 'outputspec.output_mask'),
1459+
f'space-{sym}template_desc-T1wT2w_biasfield': (
1460+
fnirt_reg_anat_mni, 'outputspec.output_biasfield'),
1461+
f'from-{orig}_to-{sym}{tmpl}template_mode-image_xfm': (
1462+
fnirt_reg_anat_mni, 'outputspec.nonlinear_xfm'),
1463+
f'from-{orig}_to-{sym}{tmpl}template_mode-image_warp': (
1464+
fnirt_reg_anat_mni, 'outputspec.nonlinear_warp')
1465+
}
1466+
outputs.update(added_outputs)
14841467

14851468
return (wf, outputs)
14861469

@@ -2030,20 +2013,11 @@ def register_FSL_anat_to_template(wf, cfg, strat_pool, pipe_num, opt=None):
20302013
node, out = connect
20312014
wf.connect(node, out, fsl, 'inputspec.input_brain')
20322015

2033-
if cfg.registration_workflows['anatomical_registration']['registration']['FSL-FNIRT']['ref_resolution'] == \
2034-
cfg.registration_workflows['anatomical_registration']['resolution_for_anat']:
2035-
2036-
node, out = strat_pool.get_data('T1w-brain-template')
2037-
wf.connect(node, out, fsl, 'inputspec.reference_brain')
2038-
2039-
node, out = strat_pool.get_data('T1w-template')
2040-
wf.connect(node, out, fsl, 'inputspec.reference_head')
2041-
else:
2042-
node, out = strat_pool.get_data('FNIRT-T1w-brain-template')
2043-
wf.connect(node, out, fsl, 'inputspec.reference_brain')
2016+
node, out = strat_pool.get_data('T1w-brain-template')
2017+
wf.connect(node, out, fsl, 'inputspec.reference_brain')
20442018

2045-
node, out = strat_pool.get_data('FNIRT-T1w-template')
2046-
wf.connect(node, out, fsl, 'inputspec.reference_head')
2019+
node, out = strat_pool.get_data('T1w-template')
2020+
wf.connect(node, out, fsl, 'inputspec.reference_head')
20472021

20482022
node, out = strat_pool.get_data(["desc-preproc_T1w",
20492023
"space-longitudinal_desc-reorient_T1w"])

0 commit comments

Comments
 (0)