-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathinit_connectome_tract.py
31 lines (25 loc) · 1.26 KB
/
init_connectome_tract.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
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"]
rep_str = copier.arrToInterpolateString(repititions)
algs = [("tens", "dtidet", "lmax2"), ("detr", "csddet", "lmax8"), ("prob", "csdprob", "lmax8")]
mapping = {}
mapping["stn"] = {
"input": "/N/dc2/projects/lifebid/HCP/Brent/cogs610/reps_data/stn_{}_{}_{}_rep{}.mat",
"output": root + "O3D_STN/derivatives/connectome_tract/sub-{}/dwi/sub-{}_dwi_var-{}life_run-{}"
}
mapping["hcp3t"] = {
"input": "/N/dc2/projects/lifebid/HCP/Brent/cogs610/reps_data/hcp_{}_{}_{}_rep{}.mat",
"output": root + "O3D_HCP3T/derivatives/connectome_tract/sub-{}/dwi/sub-{}_dwi_var-{}life_run-{}"
}
mapping["hcp7t"] = {
"input": "/N/dc2/projects/lifebid/HCP/Brent/cogs610/7t_reps_data/7T_{}_{}_{}_rep{}.mat",
"output": root + "O3D_HCP7T/derivatives/connectome_tract/sub-{}/dwi/sub-{}_dwi_var-{}life_run-{}"
}
for a in algs:
inalg, outalg, lmax = a
in_str = mapping[dataset]["input"].format(name, inalg, lmax, rep_str)
out_str = mapping[dataset]["output"].format(subject, subject, outalg, rep_str)
copier.copy(in_str, out_str, action = "mat2fiber", dummy = False)