27
27
import nibabel as nb
28
28
from nipype .interfaces import utility as niu
29
29
from nipype .pipeline import engine as pe
30
- from niworkflows .func .util import init_enhance_and_skullstrip_bold_wf
30
+ from niworkflows .func .util import init_enhance_and_skullstrip_bold_wf , init_skullstrip_bold_wf
31
31
from niworkflows .interfaces .header import ValidateImage
32
32
from niworkflows .interfaces .nitransforms import ConcatenateXFMs
33
33
from niworkflows .interfaces .utility import KeySelect
@@ -553,7 +553,8 @@ def init_bold_fit_wf(
553
553
)
554
554
unwarp_wf .inputs .inputnode .metadata = layout .get_metadata (bold_file )
555
555
556
- # fmt:off
556
+ skullstrip_bold_wf = init_skullstrip_bold_wf ()
557
+
557
558
workflow .connect ([
558
559
(inputnode , fmap_select , [
559
560
('fmap_ref' , 'fmap_ref' ),
@@ -578,28 +579,28 @@ def init_bold_fit_wf(
578
579
(unwarp_wf , ds_coreg_boldref_wf , [
579
580
('outputnode.corrected' , 'inputnode.boldref' ),
580
581
]),
581
- (unwarp_wf , ds_boldmask_wf , [
582
- ('outputnode.corrected_mask' , 'inputnode.boldmask' ),
582
+ (unwarp_wf , skullstrip_bold_wf , [
583
+ ('outputnode.corrected' , 'inputnode.in_file' ),
584
+ ]),
585
+ (skullstrip_bold_wf , ds_boldmask_wf , [
586
+ ('outputnode.mask_file' , 'inputnode.boldmask' ),
583
587
]),
584
588
(fmap_select , func_fit_reports_wf , [('fmap_ref' , 'inputnode.fmap_ref' )]),
585
589
(fmap_select , summary , [('sdc_method' , 'distortion_correction' )]),
586
590
(fmapreg_buffer , func_fit_reports_wf , [
587
591
('boldref2fmap_xfm' , 'inputnode.boldref2fmap_xfm' ),
588
592
]),
589
593
(unwarp_wf , func_fit_reports_wf , [('outputnode.fieldmap' , 'inputnode.fieldmap' )]),
590
- ])
591
- # fmt:on
594
+ ]) # fmt:skip
592
595
else :
593
- # fmt:off
594
596
workflow .connect ([
595
597
(enhance_boldref_wf , ds_coreg_boldref_wf , [
596
598
('outputnode.bias_corrected_file' , 'inputnode.boldref' ),
597
599
]),
598
600
(enhance_boldref_wf , ds_boldmask_wf , [
599
601
('outputnode.mask_file' , 'inputnode.boldmask' ),
600
602
]),
601
- ])
602
- # fmt:on
603
+ ]) # fmt:skip
603
604
else :
604
605
config .loggers .workflow .info ('Found coregistration reference - skipping Stage 3' )
605
606
regref_buffer .inputs .boldref = precomputed ['coreg_boldref' ]
0 commit comments