def flirt_t1_epi(subject_id, data_dir, run, sink_dir):
def get_niftis(subject_id, data_dir, run, sink_dir):
from os.path import join, exists
t1 = join(
data_dir,
subject_id,
"session-1",
"anatomical",
"anatomical-0",
"anatomical.nii.gz",
)
# t1_brain_mask = join(data_dir, subject, 'session-1', 'anatomical', 'anatomical-0', 'fsl', 'anatomical-bet.nii.gz')
ex_func = join(
sink_dir,
subject_id,
"{0}-{1}_retr-example_func.nii.gz".format(subject_id, run),
)
assert exists(t1), "t1 does not exist"
assert exists(ex_func), "example func does not exist"
standard = "/home/applications/fsl/5.0.8/data/standard/MNI152_T1_2mm.nii.gz"
return t1, ex_func, standard
coreg = join(sink_dir, "qa", "{0}-{1}_t1_flirt.png".format(subject, run))
data = Function(
function=get_niftis,
input_names=["subject_id", "data_dir", "run", "sink_dir"],
output_names=["t1", "ex_func", "standard"],
)
data.inputs.data_dir = data_dir
data.inputs.sink_dir = sink_dir
data.inputs.subject_id = subject
data.inputs.run = run
grabber = data.run()
re_flit = FLIRT(
cost_func="normmi",
dof=12,
searchr_x=[-90, 90],
searchr_y=[-90, 90],
searchr_z=[-90, 90],
interp="trilinear",
bins=256,
)
re_flit.inputs.reference = grabber.outputs.ex_func
re_flit.inputs.in_file = grabber.outputs.t1
re_flit.inputs.out_file = join(
sink_dir, subject, "{0}-{1}_t1-flirt-retr.nii.gz".format(subject, run))
re_flit.inputs.out_matrix_file = join(
sink_dir, subject, "{0}-{1}_t1-flirt-retr.mat".format(subject, run))
reg_func = re_flit.run()
display = plotting.plot_anat(grabber.outputs.ex_func, dim=0)
display.add_edges(reg_func.outputs.out_file)
display.savefig(coreg, dpi=300)
return
def func_to_mni(subject, reg_dir, data_dir, qc_dir, sink_dir, standard, qc):
from nipype.interfaces.fsl import ApplyXFM
from nipype.interfaces.utility import Function
from os.path import join
def get_niftis(subject_id, data_dir, reg_dir, standard):
from os.path import join, exists
map = join(data_dir, '{0}_sbc_hb.nii.gz'.format(subject_id))
xfm = join(reg_dir, 'example_func2standard.mat')
assert exists(map), "map does not exist"
assert exists(xfm), "xfm does not exist"
if standard == 'mni':
template = '/home/applications/fsl/5.0.8/data/standard/MNI152_T1_2mm_brain.nii.gz'
else:
template = standard
return map, xfm, template
data = Function(
function=get_niftis,
input_names=['subject_id', 'data_dir', 'reg_dir', 'standard'],
output_names=['map', 'xfm', 'template'])
data.inputs.data_dir = data_dir
data.inputs.subject_id = subject
data.inputs.reg_dir = reg_dir
data.inputs.standard = standard
grabber = data.run()
applyxfm = ApplyXFM()
applyxfm.inputs.in_file = grabber.outputs.map
applyxfm.inputs.in_matrix_file = grabber.outputs.xfm
applyxfm.inputs.out_file = join(sink_dir, '{0}_mni.nii.gz'.format(subject))
applyxfm.inputs.reference = grabber.outputs.template
applyxfm.inputs.apply_xfm = True
result = applyxfm.run()
if qc == True:
from nilearn.plotting import plot_stat_map
plot_stat_map(result.outputs.out_file,
grabber.outputs.template,
output_file=join(qc_dir, '{0}_mni.png'))
else:
return result.outputs.out_file
Created on Mon Aug 29 14:30:53 2016
@author: fbeyer
"""
import os
from nipype.pipeline.engine import Node, Workflow
from nipype.interfaces.utility import Function
import nipype.interfaces.utility as util
import nipype.interfaces.io as nio
import nipype.interfaces.fsl as fsl
from strip_rois import strip_rois_func
from moco import create_moco_pipeline
#from ICA_AROMA2 import create_ica_aroma
from smoothing import create_smoothing_pipeline
import ICA_AROMA_functions as aromafunc
runICA = Function(input_names=["fslDir", "outDir", "inFile", "melDirIn", "mask", "dim", "TR"],
output_names=["mdir"],
function=aromafunc.runICA)
runICA.inputs.fslDir= '/usr/share/fsl/5.0/bin/'
#os.path.join(os.environ["FSLDIR"],'bin','')
runICA.inputs.dim=0
runICA.inputs.TR=2
runICA.inputs.melDirIn=""
runICA.inputs.outDir="/scr/lessing2/data_fbeyer/FTO_YFAS/WDR/"
runICA.inputs.inFile="/scr/lessing2/data_fbeyer/FTO_YFAS/Subjects/LI00037838/aroma_inputs/func_preproc_smoothed.nii"
runICA.inputs.mask="/scr/lessing2/data_fbeyer/FTO_YFAS/Subjects/LI00037838/aroma_inputs/func_brain_mask.nii"
runICA.run()
def preproc(data_dir, sink_dir, subject, task, session, run, masks,
motion_thresh, moco):
from nipype.interfaces.fsl import MCFLIRT, FLIRT, FNIRT, ExtractROI, ApplyWarp, MotionOutliers, InvWarp, FAST
#from nipype.interfaces.afni import AlignEpiAnatPy
from nipype.interfaces.utility import Function
from nilearn.plotting import plot_anat
from nilearn import input_data
#WRITE A DARA GRABBER
def get_niftis(subject_id, data_dir, task, run, session):
from os.path import join, exists
t1 = join(data_dir, subject_id, 'session-{0}'.format(session),
'anatomical', 'anatomical-0', 'anatomical.nii.gz')
#t1_brain_mask = join(data_dir, subject_id, 'session-1', 'anatomical', 'anatomical-0', 'fsl', 'anatomical-bet.nii.gz')
epi = join(data_dir, subject_id, 'session-{0}'.format(session), task,
'{0}-{1}'.format(task, run), '{0}.nii.gz'.format(task))
assert exists(t1), "t1 does not exist at {0}".format(t1)
assert exists(epi), "epi does not exist at {0}".format(epi)
standard = '/home/applications/fsl/5.0.8/data/standard/MNI152_T1_2mm.nii.gz'
return t1, epi, standard
data = Function(
function=get_niftis,
input_names=["subject_id", "data_dir", "task", "run", "session"],
output_names=["t1", "epi", "standard"])
data.inputs.data_dir = data_dir
data.inputs.subject_id = subject
data.inputs.run = run
data.inputs.session = session
data.inputs.task = task
grabber = data.run()
if session == 0:
sesh = 'pre'
if session == 1:
sesh = 'post'
#reg_dir = '/home/data/nbc/physics-learning/data/first-level/{0}/session-1/retr/retr-{1}/retr-5mm.feat/reg'.format(subject, run)
#set output paths for quality assurance pngs
qa1 = join(
sink_dir, 'qa',
'{0}-session-{1}_{2}-{3}_t1_flirt.png'.format(subject, session, task,
run))
qa2 = join(
sink_dir, 'qa',
'{0}-session-{1}_{2}-{3}_mni_flirt.png'.format(subject, session, task,
run))
qa3 = join(
sink_dir, 'qa',
'{0}-session-{1}_{2}-{3}_mni_fnirt.png'.format(subject, session, task,
run))
confound_file = join(
sink_dir, sesh, subject,
'{0}-session-{1}_{2}-{3}_confounds.txt'.format(subject, session, task,
run))
#run motion correction if indicated
if moco == True:
mcflirt = MCFLIRT(ref_vol=144, save_plots=True, output_type='NIFTI_GZ')
mcflirt.inputs.in_file = grabber.outputs.epi
#mcflirt.inputs.in_file = join(data_dir, subject, 'session-1', 'retr', 'retr-{0}'.format(run), 'retr.nii.gz')
mcflirt.inputs.out_file = join(
sink_dir, sesh, subject,
'{0}-session-{1}_{2}-{3}_mcf.nii.gz'.format(
subject, session, task, run))
flirty = mcflirt.run()
motion = np.genfromtxt(flirty.outputs.par_file)
else:
print "no moco needed"
motion = 0
#calculate motion outliers
try:
mout = MotionOutliers(metric='fd', threshold=motion_thresh)
mout.inputs.in_file = grabber.outputs.epi
mout.inputs.out_file = join(
sink_dir, sesh, subject,
'{0}-session-{1}_{2}-{3}_fd-gt-{3}mm'.format(
subject, session, task, run, motion_thresh))
mout.inputs.out_metric_plot = join(
sink_dir, sesh, subject,
'{0}-session-{1}_{2}-{3}_metrics.png'.format(
subject, session, task, run))
mout.inputs.out_metric_values = join(
sink_dir, sesh, subject,
'{0}-session-{1}_{2}-{3}_fd.txt'.format(subject, session, task,
run))
moutliers = mout.run()
outliers = np.genfromtxt(moutliers.outputs.out_file)
e = 'no errors in motion outliers, yay'
except Exception as e:
print(e)
outliers = np.genfromtxt(mout.inputs.out_metric_values)
#set everything above the threshold to 1 and everything below to 0
outliers[outliers > motion_thresh] = 1
outliers[outliers < motion_thresh] = 0
#concatenate motion parameters and motion outliers to form confounds file
#outliers = outliers.reshape((outliers.shape[0],1))
conf = outliers
np.savetxt(confound_file, conf, delimiter=',')
#extract an example volume for normalization
ex_fun = ExtractROI(t_min=144, t_size=1)
ex_fun.inputs.in_file = flirty.outputs.out_file
ex_fun.inputs.roi_file = join(
sink_dir, sesh, subject,
'{0}-session-{1}_{2}-{3}-example_func.nii.gz'.format(
subject, session, task, run))
fun = ex_fun.run()
warp = ApplyWarp(interp="nn", abswarp=True)
if not exists(
'/home/data/nbc/physics-learning/data/first-level/{0}/session-{1}/{2}/{2}-{3}/{2}-5mm.feat/reg/example_func2standard_warp.nii.gz'
.format(subject, session, task, run)):
#two-step normalization using flirt and fnirt, outputting qa pix
flit = FLIRT(cost_func="corratio", dof=12)
reg_func = flit.run(
reference=fun.outputs.roi_file,
in_file=grabber.outputs.t1,
searchr_x=[-180, 180],
searchr_y=[-180, 180],
out_file=join(
sink_dir, sesh, subject,
'{0}-session-{1}_{2}-{3}_t1-flirt.nii.gz'.format(
subject, session, task, run)),
out_matrix_file=join(
sink_dir, sesh, subject,
'{0}-session-{1}_{2}-{3}_t1-flirt.mat'.format(
subject, session, task, run)))
reg_mni = flit.run(
reference=grabber.outputs.t1,
in_file=grabber.outputs.standard,
searchr_y=[-180, 180],
searchr_z=[-180, 180],
out_file=join(
sink_dir, sesh, subject,
'{0}-session-{1}_{2}-{3}_mni-flirt-t1.nii.gz'.format(
subject, session, task, run)),
out_matrix_file=join(
sink_dir, sesh, subject,
'{0}-session-{1}_{2}-{3}_mni-flirt-t1.mat'.format(
subject, session, task, run)))
#plot_stat_map(aligner.outputs.out_file, bg_img=fun.outputs.roi_file, colorbar=True, draw_cross=False, threshold=1000, output_file=qa1a, dim=-2)
display = plot_anat(fun.outputs.roi_file, dim=-1)
display.add_edges(reg_func.outputs.out_file)
display.savefig(qa1, dpi=300)
display.close()
display = plot_anat(grabber.outputs.t1, dim=-1)
display.add_edges(reg_mni.outputs.out_file)
display.savefig(qa2, dpi=300)
display.close()
perf = FNIRT(output_type='NIFTI_GZ')
perf.inputs.warped_file = join(
sink_dir, sesh, subject,
'{0}-session-{1}_{2}-{3}_mni-fnirt-t1.nii.gz'.format(
subject, session, task, run))
perf.inputs.affine_file = reg_mni.outputs.out_matrix_file
perf.inputs.in_file = grabber.outputs.standard
perf.inputs.subsampling_scheme = [8, 4, 2, 2]
perf.inputs.fieldcoeff_file = join(
sink_dir, sesh, subject,
'{0}-session-{1}_{2}-{3}_mni-fnirt-t1-warpcoeff.nii.gz'.format(
subject, session, task, run))
perf.inputs.field_file = join(
sink_dir, sesh, subject,
'{0}-session-{1}_{2}-{3}_mni-fnirt-t1-warp.nii.gz'.format(
subject, session, task, run))
perf.inputs.ref_file = grabber.outputs.t1
reg2 = perf.run()
warp.inputs.field_file = reg2.outputs.field_file
#plot fnirted MNI overlaid on example func
display = plot_anat(grabber.outputs.t1, dim=-1)
display.add_edges(reg2.outputs.warped_file)
display.savefig(qa3, dpi=300)
display.close()
else:
warpspeed = InvWarp(output_type='NIFTI_GZ')
warpspeed.inputs.warp = '/home/data/nbc/physics-learning/data/first-level/{0}/session-{1}/{2}/{2}-{3}/{2}-5mm.feat/reg/example_func2standard_warp.nii.gz'.format(
subject, session, task, run)
warpspeed.inputs.reference = fun.outputs.roi_file
warpspeed.inputs.inverse_warp = join(
sink_dir, sesh, subject,
'{0}-session-{1}_{2}-{3}_mni-fnirt-t1-warp.nii.gz'.format(
subject, session, task, run))
mni2epiwarp = warpspeed.run()
warp.inputs.field_file = mni2epiwarp.outputs.inverse_warp
for key in masks.keys():
#warp takes us from mni to epi
warp.inputs.in_file = masks[key]
warp.inputs.ref_file = fun.outputs.roi_file
warp.inputs.out_file = join(
sink_dir, sesh, subject,
'{0}-session-{1}_{2}-{3}_{4}.nii.gz'.format(
subject, session, task, run, key))
net_warp = warp.run()
qa_file = join(
sink_dir, 'qa', '{0}-session-{1}_{2}-{3}_qa_{4}.png'.format(
subject, session, task, run, key))
display = plotting.plot_roi(net_warp.outputs.out_file,
bg_img=fun.outputs.roi_file,
colorbar=True,
vmin=0,
vmax=18,
draw_cross=False)
display.savefig(qa_file, dpi=300)
display.close()
return flirty.outputs.out_file, confound_file, e
def fnirt_again(data_dir, sink_dir, subject, run, masks, mask_names):
import numpy as np
def get_niftis(subject, data_dir, sink_dir, run, masks):
from os.path import join, exists
t1 = join(
data_dir,
subject,
"session-1",
"anatomical",
"anatomical-0",
"anatomical.nii.gz",
)
# t1_brain_mask = join(data_dir, subject, 'session-1', 'anatomical', 'anatomical-0', 'fsl', 'anatomical-bet.nii.gz')
example_func = join(
sink_dir, subject,
"{0}-{1}_retr-example_func.nii.gz".format(subject, run))
assert exists(t1), "t1 does not exist"
assert exists(example_func), "example_func does not exist"
standard = "/home/applications/fsl/5.0.8/data/standard/MNI152_T1_2mm.nii.gz"
mni2t1 = join(sink_dir, subject,
"{0}-{1}_mni-flirt-t1.mat".format(subject, run))
t12epi = join(sink_dir, subject,
"{0}-{1}_t1-flirt-retr.mat".format(subject, run))
masks = masks
return t1, example_func, standard, mni2t1, t12epi, masks
data = Function(
function=get_niftis,
input_names=["subject", "data_dir", "sink_dir", "run", "masks"],
output_names=[
"t1", "example_func", "standard", "mni2t1", "t12epi", "masks"
],
)
data.inputs.data_dir = data_dir
data.inputs.sink_dir = sink_dir
data.inputs.subject = subject
data.inputs.masks = masks
data.inputs.run = run
grabber = data.run()
perf = FNIRT(output_type="NIFTI_GZ")
perf.inputs.warped_file = join(
sink_dir, subject, "{0}-{1}_mni-fnirt-t1.nii.gz".format(subject, run))
perf.inputs.affine_file = grabber.outputs.mni2t1
perf.inputs.in_file = grabber.outputs.standard
perf.inputs.subsampling_scheme = [8, 4, 2, 2]
perf.inputs.fieldcoeff_file = join(
sink_dir, subject,
"{0}-{1}_mni-fnirt-t1-warpcoef.nii.gz".format(subject, run))
perf.inputs.field_file = join(
sink_dir, subject,
"{0}-{1}_mni-fnirt-t1-warp.nii.gz".format(subject, run))
perf.inputs.ref_file = grabber.outputs.t1
reg2 = perf.run()
for i in np.arange(0, len(masks)):
# warp takes us from mni to t1, postmat
warp = ApplyWarp(interp="nn", abswarp=True)
warp.inputs.in_file = grabber.outputs.masks[i]
warp.inputs.ref_file = grabber.outputs.example_func
warp.inputs.field_file = reg2.outputs.field_file
warp.inputs.postmat = grabber.outputs.t12epi
warp.inputs.out_file = join(
sink_dir,
subject,
"{0}-{1}_{2}_retr.nii.gz".format(subject, run, mask_names[i]),
)
net_warp = warp.run()
return "yay"
