def process_runs(rundir, subid, run3meanfunc, logger):
_, run = os.path.split(rundir)
rawdir = os.path.join(rundir, 'raw')
exists, mvdir, xfmdir = check_make_dirs(rundir, logger)
if exists:
return
_, run3 = os.path.split(run3meanfunc)
cmd = 'ln -s %s %s'%(run3meanfunc, xfmdir)
os.system(cmd)
run3meanfunc = os.path.join(xfmdir, run3)
func = os.path.join(rawdir, '{0}_4d_func.nii.gz'.format(subid))
funcs = split_file(func, xfmdir, subid, run, logger)
if funcs is None:
return
movement, affines, moved = make_affines(funcs, logger)
logger.info('Write movement')
save_movement(movement, mvdir)
meanfunc = reg.make_mean(moved)
for f in moved[1:]:
os.remove(f)
threshold_mean(meanfunc, logger, thresh=275)
b4meanfunc = reg.n4_biascorrect(meanfunc)
if b4meanfunc is None:
logger.error('{0}: Bias correction failed'.format(meanfunc))
return
movedfuncs = align_means(funcs, b4meanfunc, run3meanfunc, affines, logger)
exists, refunc = join_files(movedfuncs, rundir, subid, run)
if not exists:
return
def process_run(rundir, subid, logger):
"""
1. Find raw
2. Split func into xfm dir
3. Align func, create mean and 4d
4. Bias correct anat
5. Invxfm anat2func
6. Generate movement files
7. Clean up
"""
_, run = os.path.split(rundir)
rawdir = os.path.join(rundir, 'raw')
exists, mvdir, xfmdir = check_make_dirs(rundir, logger)
if exists:
return
anat, aparc, func = find_raw_files(rawdir, subid, logger)
if anat is None:
return
funcs = split_file(func, xfmdir, subid, run, logger)
if funcs is None:
return
movement, movedfuncs = align_funcs(funcs, logger)
exists, refunc = join_files(movedfuncs, rundir, subid, run)
if not exists:
return
#Write movement and displacement
logger.info('Write movement')
save_movement(movement, mvdir)
#Make mean from aligned
meanfunc = reg.make_mean(movedfuncs)
#Bias correct anat
b4anat = reg.n4_biascorrect(anat)
if b4anat is None:
logger.error('{0}: Bias correction failed'.format(anat))
return
#Bias correct mean
b4meanfunc = reg.n4_biascorrect(meanfunc)
if b4meanfunc is None:
logger.error('{0}: Bias correction failed'.format(meanfunc))
return
#Reslice mean to anat space
rb4meanfunc = reslice_mean(b4anat, b4meanfunc)
#Make mask by growing func (5x5x5 kernel) and mask with anat
mask = reg.grow_mask(rb4meanfunc, size= 3)
#Find mapping from func to anat
func2anat_xfm = reg.affine_register_mi(b4anat, b4meanfunc, mask= mask)
#Apply inverse mapping to anat
anat2func = reg.apply_transform(b4anat, '-i %s'%(func2anat_xfm),
target=meanfunc, use_nn = False)
#Apply transform to aparc nearest neighbor
aparc2func = reg.apply_transform(aparc, '-i %s'%(func2anat_xfm),
target=meanfunc, use_nn = True)
#Move
for item in (anat2func, aparc2func):
cmd = 'mv %s %s'%(item, rundir)
os.system(cmd)
def process_run3(rundir, subid, datadir, logger):
"""
1. Find raw
2. Split func into xfm dir
3. Center func
4. Align func, create mean and 4d
5. Bias correct anat
6. Invxfm anat2func
7. Generate movement files
8. Clean up
"""
_, run = os.path.split(rundir)
rawdir = os.path.join(rundir, 'raw')
exists, mvdir, xfmdir = check_make_dirs(rundir, logger)
if exists:
return
anat, aparc, func = find_raw_files(rawdir, subid, logger)
if anat is None:
return
funcs = split_file(func, xfmdir, subid, run, logger)
if funcs is None:
return
cfuncs = suzanne_center(funcs)
movement, movedfuncs = align_funcs(cfuncs, logger)
exists, refunc = join_files(movedfuncs, rundir, subid, run)
if not exists:
return
logger.info('Write movement')
save_movement(movement, mvdir)
meanfunc = reg.make_mean(movedfuncs)
threshold_mean(meanfunc, logger, thresh=275)
b4anat = bias_correct(anat, logger, xfmdir)
b4meanfunc = bias_correct(meanfunc, logger, xfmdir)
func2anat_xfm = reg.affine_register_mi(b4anat, b4meanfunc, mask= None)
anat2func = reg.apply_transform(b4anat, '-i %s'%(func2anat_xfm),
target=b4meanfunc, use_nn = False)
aparc2func = reg.apply_transform(aparc, '-i %s'%(func2anat_xfm),
target=b4meanfunc, use_nn = True)
pth, aparcnme = os.path.split(aparc2func)
aparc_mask = os.path.join(datadir, 'mask_{0}'.format(aparcnme))
mask_from_aparc(aparc2func, aparc_mask)
for item in (anat2func, aparc2func):
cmd = 'mv %s %s'%(item, datadir)
os.system(cmd)
return b4meanfunc
maskimg = ni.Nifti1Image(mask, img.get_affine())
maskimg.to_filename(outfile)
path = '/home/jagust/fmri-pstask/subjects/'
subjs = sorted(glob(os.path.join(path, 'B*')))
for subj in subjs:
jnk, subid = os.path.split(subj)
xfmdir = glob(os.path.join(subj, '*run3/xfm'))[0]
rawdir = glob(os.path.join(subj, '*run3/raw'))[0]
movedfuncs = glob(os.path.join(xfmdir, 'B*0000*nii.gz'))
xfmfuncs = glob(os.path.join(xfmdir, 'xfm*'))
for xfm in xfmfuncs:
movedfuncs.append(xfm)
meanfunc = reg.make_mean(movedfuncs)
threshold_mean(meanfunc, thresh=250) #****Change****
b4meanfunc = bias_correct(meanfunc, xfmdir)
# anat = os.path.join(rawdir, '%s_brainmask.nii.gz'%(subid))
# b4anat = bias_correct(anat, xfmdir)
b4anat = os.path.join(xfmdir, 'n4_%s_brainmask.nii.gz'%(subid))
aparc = os.path.join(rawdir, '%s_aparcaseg.nii.gz'%(subid))
func2anat_xfm = reg.affine_register_mi(b4anat, b4meanfunc, mask= None)
anat2func = reg.apply_transform(b4anat, '-i %s'%(func2anat_xfm),
target=b4meanfunc, use_nn = False)
aparc2func = reg.apply_transform(aparc, '-i %s'%(func2anat_xfm),
target=b4meanfunc, use_nn = True)
pth, aparcnme = os.path.split(aparc2func)
aparc_mask = os.path.join(subj, 'mask_{0}'.format(aparcnme))
def process_run3(rundir, subid, logger):
"""
1. Find raw
2. Split func into xfm dir
3. Align func, create mean and 4d
4. Bias correct anat
5. Invxfm anat2func
6. Generate movement files
7. Clean up
"""
_, run = os.path.split(rundir)
rawdir = os.path.join(rundir, 'raw')
exists, mvdir, xfmdir = check_make_dirs(rundir, logger)
if exists:
return
anat, aparc, func = find_raw_files(rawdir, subid, logger)
if anat is None:
return
funcs = split_file(func, xfmdir, subid, run, logger)
if funcs is None:
return
movement, movedfuncs = align_funcs(funcs, logger)
exists, refunc = join_files(movedfuncs, rundir, subid, run)
if not exists:
return
#Write movement and displacement
logger.info('Write movement')
save_movement(movement, mvdir)
#Make mean from aligned
meanfunc = reg.make_mean(movedfuncs)
#Bias correct anat
b4anat = reg.n4_biascorrect(anat)
if b4anat is None:
logger.error('{0}: Bias correction failed'.format(anat))
return
_, b4anatname = os.path.split(b4anat)
cmd = 'mv %s %s'%(b4anat, xfmdir)
os.system(cmd)
b4anat = os.path.join(xfmdir, b4anatname)
# threshold mean to remove "cruft" current value == 300
threshold_mean(meanfunc, logger, thresh=300)
#Bias correct mean
#b4meanfunc = reg.n4_biascorrect(meanfunc)
#if b4meanfunc is None:
# logger.error('{0}: Bias correction failed'.format(meanfunc))
# return
#Reslice mean to anat space
b4meanfunc = meanfunc
rb4meanfunc = reslice_mean(b4anat, b4meanfunc)
#Make mask by growing func (5x5x5 kernel) and mask with anat
# mask = reg.grow_mask(rb4meanfunc, size= 3)
#Find mapping from func to anat
func2anat_xfm = reg.affine_register_mi(b4anat, b4meanfunc, mask= None)
#Apply inverse mapping to anat
anat2func = reg.apply_transform(b4anat, '-i %s'%(func2anat_xfm),
target=meanfunc, use_nn = False)
#Apply transform to aparc nearest neighbor
aparc2func = reg.apply_transform(aparc, '-i %s'%(func2anat_xfm),
target=meanfunc, use_nn = True)
#Skull strip shit
pth, aparcnme = os.path.split(aparc2func)
aparc_mask = os.path.join(datadir, 'mask_{0}'.format(aparcnme))
mask_from_aparc(aparc2func, aparc_mask)
#masked_b4meanfunc = b4meanfunc.replace('n4_', 'masked_n4_')
#mask_img(b4meanfunc, aparc_mask, masked_b4meanfunc)
#Move
for item in (anat2func, aparc2func):
cmd = 'mv %s %s'%(item, rundir)
os.system(cmd)
return b4meanfunc
def process_runs(rundir, subid, run3meanfunc, logger):
_, run = os.path.split(rundir)
rawdir = os.path.join(rundir, 'raw')
exists, mvdir, xfmdir = check_make_dirs(rundir, logger)
if exists:
return
_, run3 = os.path.split(run3meanfunc)
cmd = 'cp %s %s'%(run3meanfunc, xfmdir)
os.system(cmd)
run3meanfunc = os.path.join(xfmdir, run3)
anat, aparc, func = find_raw_files(rawdir, subid, logger)
if anat is None:
return
funcs = split_file(func, xfmdir, subid, run, logger)
if funcs is None:
return
movement, affines, moved = make_affines(funcs, logger)
logger.info('Write movement')
save_movement(movement, mvdir)
meanfunc = reg.make_mean(moved)
#Remove old movedfuncs
newdir = os.path.join(xfmdir, 'first_realign')
os.mkdir(newdir)
for f in moved[1:]:
cmd = 'mv %s %s'%(f, newdir)
os.system(cmd)
#Bias correct anat
b4anat = reg.n4_biascorrect(anat)
if b4anat is None:
logger.error('{0}: Bias correction failed'.format(anat))
return
_, b4anatname = os.path.split(b4anat)
cmd = 'mv %s %s'%(b4anat, xfmdir)
os.system(cmd)
b4anat = os.path.join(xfmdir, b4anatname)
# threshold mean to remove "cruft" current value == 300
threshold_mean(meanfunc, logger, thresh=300)
#Bias correct mean
#b4meanfunc = reg.n4_biascorrect(meanfunc)
#if b4meanfunc is None:
# logger.error('{0}: Bias correction failed'.format(meanfunc))
# return
b4meanfunc = meanfunc
#Reslice mean to anat space
rb4meanfunc = reslice_mean(b4anat, b4meanfunc)
#Make mask by growing func (5x5x5 kernel) and mask with anat
# mask = reg.grow_mask(rb4meanfunc, size= 3)
#Find mapping from func to anat
func2anat_xfm = reg.affine_register_mi(b4anat, b4meanfunc, mask= None)
#Apply inverse mapping to anat
anat2func = reg.apply_transform(b4anat, '-i %s'%(func2anat_xfm),
target=meanfunc, use_nn = False)
#Apply transform to aparc nearest neighbor
aparc2func = reg.apply_transform(aparc, '-i %s'%(func2anat_xfm),
target=meanfunc, use_nn = True)
#Skull strip shit
pth, aparcnme = os.path.split(aparc2func)
aparc_mask = os.path.join(rundir, 'mask_{0}'.format(aparcnme))
mask_from_aparc(aparc2func, aparc_mask)
#masked_b4meanfunc = b4meanfunc.replace('n4_', 'masked_n4_')
#mask_img(b4meanfunc, aparc_mask, masked_b4meanfunc)
#Make new moved funcs based on 2 affines
movedfuncs = align_means(funcs, b4meanfunc, run3meanfunc, affines, logger)
exists, refunc = join_files(movedfuncs, rundir, subid, run)
if not exists:
return
#Move
for item in (anat2func, aparc2func):
cmd = 'mv %s %s'%(item, rundir)
os.system(cmd)
