'''Training script generator for subject-independent classification.

'''

with open('script.sh.template', 'r') as f:

    sh_template = f.read()

with open('train.pbs.template', 'r') as f:

    pbs_template = f.read()

def list_to_str(l):

    return ' '.join(map(str, l))

all_folds = list(range(54))

command_template = "python train_base.py $datapath $outpath -fold {0} -gpu {1} > $logpath/stdout.f{0}.out &"

for i, j in enumerate(range(0, len(all_folds), 8)):

    folds = all_folds[j:j+8]

    commands = []

    for ind, fold in enumerate(folds):

        commands.append(command_template.format(fold, ind))

    commands.append('wait')

    content = sh_template + '\n'.join(commands)

    with open('script.' + str(i) + '.sh', 'w') as f:

        f.write(content)

    content = pbs_template.format(i)

    with open('train.' + str(i) + '.pbs', 'w') as f:

        f.write(content)