def test_ComputeMeshWarp_outputs():
output_map = dict(distance=dict(),
out_file=dict(),
out_warp=dict(),
)
outputs = ComputeMeshWarp.output_spec()
for key, metadata in output_map.items():
for metakey, value in metadata.items():
yield assert_equal, getattr(outputs.traits()[key], metakey), value
def test_ComputeMeshWarp_outputs():
output_map = dict(
distance=dict(),
out_file=dict(),
out_warp=dict(),
)
outputs = ComputeMeshWarp.output_spec()
for key, metadata in output_map.items():
for metakey, value in metadata.items():
yield assert_equal, getattr(outputs.traits()[key], metakey), value
def test_ComputeMeshWarp_inputs():
input_map = dict(
ignore_exception=dict(
nohash=True,
usedefault=True,
),
metric=dict(usedefault=True, ),
out_file=dict(usedefault=True, ),
out_warp=dict(usedefault=True, ),
surface1=dict(mandatory=True, ),
surface2=dict(mandatory=True, ),
weighting=dict(usedefault=True, ),
)
inputs = ComputeMeshWarp.input_spec()
for key, metadata in input_map.items():
for metakey, value in metadata.items():
yield assert_equal, getattr(inputs.traits()[key], metakey), value
def test_ComputeMeshWarp_inputs():
input_map = dict(ignore_exception=dict(nohash=True,
usedefault=True,
),
metric=dict(usedefault=True,
),
out_file=dict(usedefault=True,
),
out_warp=dict(usedefault=True,
),
surface1=dict(mandatory=True,
),
surface2=dict(mandatory=True,
),
weighting=dict(usedefault=True,
),
)
inputs = ComputeMeshWarp.input_spec()
for key, metadata in input_map.items():
for metakey, value in metadata.items():
yield assert_equal, getattr(inputs.traits()[key], metakey), value
def hcp_workflow(name='Evaluation_HCP',
settings={},
map_metric=False,
compute_fmb=False):
"""
The regseg evaluation workflow for the human connectome project (HCP)
"""
from nipype.pipeline import engine as pe
from nipype.interfaces import utility as niu
from nipype.interfaces import io as nio
from nipype.algorithms.mesh import ComputeMeshWarp, WarpPoints
from nipype.algorithms.misc import AddCSVRow
from nipype.workflows.dmri.fsl.artifacts import sdc_fmb
from .. import data
from ..interfaces.utility import (ExportSlices, TileSlicesGrid,
SlicesGridplot)
from .registration import regseg_wf, sdc_t2b
from .preprocess import preprocess
from .fieldmap import process_vsm
from .dti import mrtrix_dti
import evaluation as ev
wf = pe.Workflow(name=name)
inputnode = pe.Node(
niu.IdentityInterface(fields=['subject_id', 'data_dir']),
name='inputnode')
inputnode.inputs.data_dir = settings['data_dir']
inputnode.iterables = [('subject_id', settings['subject_id'])]
# Generate the distorted set, including surfaces
pre = preprocess()
rdti = mrtrix_dti('ReferenceDTI')
wdti = mrtrix_dti('WarpedDTI')
mdti = pe.Node(niu.Merge(2), name='MergeDTI')
wf.connect([
(inputnode, pre, [('subject_id', 'inputnode.subject_id'),
('data_dir', 'inputnode.data_dir')]),
(pre, rdti, [('outputnode.dwi', 'inputnode.in_dwi'),
('outputnode.dwi_mask', 'inputnode.in_mask'),
('outputnode.bvec', 'inputnode.in_bvec'),
('outputnode.bval', 'inputnode.in_bval')]),
(pre, wdti, [('outputnode.warped_dwi', 'inputnode.in_dwi'),
('outputnode.warped_msk', 'inputnode.in_mask'),
('outputnode.bvec', 'inputnode.in_bvec'),
('outputnode.bval', 'inputnode.in_bval')]),
(wdti, mdti, [('outputnode.fa', 'in1'), ('outputnode.md', 'in2')]),
])
regseg = regseg_wf(usemask=True)
regseg.inputs.inputnode.options = data.get('regseg_hcp')
exprs = pe.Node(ExportSlices(slices=[38, 48, 57, 67, 76, 86],
axis=['axial', 'sagittal']),
name='ExportREGSEG')
gridrs = pe.Node(SlicesGridplot(label=['regseg', 'regseg'],
slices=[38, 48, 57, 67, 76, 86],
view=['axial', 'sagittal']),
name='GridPlotREGSEG')
meshrs = pe.MapNode(ComputeMeshWarp(),
iterfield=['surface1', 'surface2'],
name='REGSEGSurfDistance')
csvrs = pe.Node(AddCSVRow(in_file=settings['out_csv']),
name="REGSEGAddRow")
csvrs.inputs.method = 'REGSEG'
wf.connect([(mdti, regseg, [('out', 'inputnode.in_fixed')]),
(pre, regseg, [('outputnode.surf', 'inputnode.in_surf'),
('outputnode.warped_msk', 'inputnode.in_mask')
]),
(pre, exprs, [('outputnode.warped_surf', 'sgreen')]),
(regseg, exprs, [('outputnode.out_surf', 'syellow')]),
(wdti, exprs, [('outputnode.fa', 'reference')]),
(exprs, gridrs, [('out_files', 'in_files')]),
(pre, meshrs, [('outputnode.warped_surf', 'surface1')]),
(regseg, meshrs, [('outputnode.out_surf', 'surface2')]),
(inputnode, csvrs, [('subject_id', 'subject_id')]),
(meshrs, csvrs, [('distance', 'surf_dist')])])
if compute_fmb:
cmethod0 = sdc_fmb()
selbmap = pe.Node(niu.Split(splits=[1, 1], squeeze=True),
name='SelectBmap')
dfm = process_vsm()
dfm.inputs.inputnode.scaling = 1.0
dfm.inputs.inputnode.enc_dir = 'y-'
wrpsurf = pe.MapNode(WarpPoints(),
iterfield=['points'],
name='UnwarpSurfs')
export0 = pe.Node(ExportSlices(slices=[38, 48, 57, 67, 76, 86],
axis=['axial', 'sagittal']),
name='ExportFMB')
mesh0 = pe.MapNode(ComputeMeshWarp(),
iterfield=['surface1', 'surface2'],
name='FMBSurfDistance')
grid0 = pe.Node(SlicesGridplot(label=['FMB'] * 2,
slices=[38, 48, 57, 67, 76, 86],
view=['axial', 'sagittal']),
name='GridPlotFMB')
csv0 = pe.Node(AddCSVRow(in_file=settings['out_csv']),
name="FMBAddRow")
csv0.inputs.method = 'FMB'
wf.connect([
(pre, cmethod0, [('outputnode.warped_dwi', 'inputnode.in_file'),
('outputnode.warped_msk', 'inputnode.in_mask'),
('outputnode.bval', 'inputnode.in_bval'),
('outputnode.mr_param', 'inputnode.settings')]),
(pre, selbmap, [('outputnode.bmap_wrapped', 'inlist')]),
(selbmap, cmethod0, [('out1', 'inputnode.bmap_mag'),
('out2', 'inputnode.bmap_pha')]),
(cmethod0, dfm, [('outputnode.out_vsm', 'inputnode.vsm')]),
(pre, dfm, [('outputnode.warped_msk', 'inputnode.reference')]),
(dfm, wrpsurf, [('outputnode.dfm', 'warp')]),
(pre, wrpsurf, [('outputnode.surf', 'points')
])(wrpsurf, export0, [('out_points', 'syellow')]),
(pre, export0, [('outputnode.warped_surf', 'sgreen')]),
(wdti, export0, [('outputnode.fa', 'reference')]),
(export0, grid0, [('out_files', 'in_files')]),
(pre, mesh0, [('outputnode.warped_surf', 'surface1')]),
(wrpsurf, mesh0, [('out_points', 'surface2')]),
(inputnode, csv0, [('subject_id', 'subject_id')]),
(mesh0, csv0, [('distance', 'surf_dist')])
])
cmethod1 = sdc_t2b(num_threads=settings['nthreads'])
export1 = pe.Node(ExportSlices(slices=[38, 48, 57, 67, 76, 86],
axis=['axial', 'sagittal']),
name='ExportT2B')
grid1 = pe.Node(SlicesGridplot(label=['T2B'] * 2,
slices=[38, 48, 57, 67, 76, 86],
view=['axial', 'sagittal']),
name='GridPlotT2B')
mesh1 = pe.MapNode(ComputeMeshWarp(),
iterfield=['surface1', 'surface2'],
name='T2BSurfDistance')
csv1 = pe.Node(AddCSVRow(in_file=settings['out_csv']), name="T2BAddRow")
csv1.inputs.method = 'T2B'
wf.connect([(pre, cmethod1,
[('outputnode.warped_dwi', 'inputnode.in_dwi'),
('outputnode.warped_msk', 'inputnode.dwi_mask'),
('outputnode.t2w_brain', 'inputnode.in_t2w'),
('outputnode.t1w_mask', 'inputnode.t2w_mask'),
('outputnode.surf', 'inputnode.in_surf'),
('outputnode.bval', 'inputnode.in_bval'),
('outputnode.mr_param', 'inputnode.in_param')]),
(cmethod1, export1, [('outputnode.out_surf', 'syellow')]),
(pre, export1, [('outputnode.warped_surf', 'sgreen')]),
(wdti, export1, [('outputnode.fa', 'reference')]),
(export1, grid1, [('out_files', 'in_files')]),
(pre, mesh1, [('outputnode.warped_surf', 'surface1')]),
(cmethod1, mesh1, [('outputnode.out_surf', 'surface2')]),
(inputnode, csv1, [('subject_id', 'subject_id')]),
(mesh1, csv1, [('distance', 'surf_dist')])])
tile = pe.Node(TileSlicesGrid(), name='TileGridplots')
csvtile = pe.Node(AddCSVRow(
in_file=op.join(op.dirname(settings['out_csv']), 'tiles.csv')),
name="TileAddRow")
wf.connect([(inputnode, tile, [('subject_id', 'out_file')]),
(gridrs, tile, [('out_file', 'in_reference')]),
(grid1, tile, [('out_file', 'in_competing')]),
(tile, csvtile, [('out_file', 'names')])])
if map_metric:
out_csv = op.abspath(op.join(name, 'energiesmapping.csv'))
mapen = ev.map_energy(out_csv=out_csv)
wf.connect([
(inputnode, mapen, [('subject_id', 'inputnode.subject_id')]),
(regseg, mapen, [('outputnode.out_enh', 'inputnode.reference'),
('outputnode.reg_msk', 'inputnode.in_mask')]),
(pre, mapen, [('outputnode.warped_surf', 'inputnode.surfaces0'),
('outputnode.surf', 'inputnode.surfaces1')])
])
return wf
