init_dissection_afq_trk.py

import sys
import utils.copy_template as copier

dataset, subject, root = copier.parseCommandLine(sys.argv)

name = copier.subjectNameFromNumber(subject)
repititions = ['0' + `i` for i in range(1,10)] + ["10"]
lmax = ['lmax0' + `i` if i < 10 else 'lmax' + `i` for i in range(2, 14, 2)]
lmax_str = copier.arrToInterpolateString(lmax)
algs = ['SD_PROB', 'SD_STREAM']
algs_str = copier.arrToInterpolateString(algs)
out_algs = copier.arrToInterpolateString(['csdproblife', 'csddetlife'])
rep_str = copier.arrToInterpolateString(repititions)
anatomy = copier.getAnatomy(root, dataset, subject)

lifebid_root = "/N/dc2/projects/lifebid/"

mapping = {}
mapping["stn"] = {
    "mat_in": lifebid_root + "code/ccaiafa/Caiafa_Pestilli_paper2015/Results/ETC_Dec2015/Single_TC/fe_structure_{}_96dirs_b2000_1p5iso_STC_run01_500000_{}_connNUM{}_TRACTS.mat",
    "mat_out": root + "O3D_STN/derivatives/dissection_afq_trk/sub-{}/dwi/sub-{}_dwi_DTI_var-{}_run-{}_tract_var-afq"
}

mapping["hcp3t"] = {
    "mat_in": lifebid_root + "code/ccaiafa/Caiafa_Pestilli_paper2015/Results/ETC_Dec2015/Single_TC/fe_structure_{}_STC_run01_500000_{}_connNUM{}_TRACTS.mat",
    "mat_out": root + "O3D_HCP3T/derivatives/dissection_afq_trk/sub-{}/dwi/sub-{}_dwi_DTI_var-{}_run-{}_tract_var-afq"
}

# mapping["hcp7t"] = {
#     "mat_in": lifebid_root + "code/ccaiafa/Caiafa_Pestilli_paper2015/paper_datasets/HCP7T/sub-{}/fe_structures/fe_structure_{}_STC_run01_SD_STREAM_lmax8_connNUM{}.mat",
#     "anatomy": lifebid_root + "code/ccaiafa/Caiafa_Pestilli_paper2015/paper_datasets/HCP7T/sub-{}/dwi/data_b2000.nii.gz",
#     "mat_out": root + "O3D_HCP7T/derivatives/tracking_csddet_trk/sub-{}/dwi/sub-{}_dwi_var-csddetlife_run-{}_tract.trk"
# }

if dataset == "hcp7t":
    print "Cannot run for hcp7t yet, these files don't exist"
    sys.exit()


mat_in = mapping[dataset]["mat_in"].format(name, algs_str, 'lmax08', rep_str)
mat_out = mapping[dataset]["mat_out"].format(subject, subject, out_algs, rep_str)
copier.copy(mat_in, mat_out, anatomy = anatomy, action = "afq2trk", dummy = False)

mat_in = mapping[dataset]["mat_in"].format(name, 'tensor', '', rep_str)
mat_out = mapping[dataset]["mat_out"].format(subject, subject, 'dtidetlife', rep_str)
copier.copy(mat_in, mat_out, anatomy = anatomy, action = "afq2trk", dummy = False)