'''Training script generator for subject-specific 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()

command_template = 'python train_within.py $datapath $outpath -start {0} -end {1} -gpu {2} > $logpath/stdout.f{2}.out &'

commands = []

for i in range(8):

    start = i * 7 + 1

    end = min(i * 7 + 8, 55)

    commands.append(command_template.format(start, end, i))

commands.append('wait')

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

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

    f.write(content)

content = pbs_template.format(0)

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

    f.write(content)