import os

sim_pack = 'qiskit'
task = 'hdyn'
com_cap = 'st'
prec = 'dp'

gen_path = os.path.abspath(os.getcwd()) + '/'
# print(gen_path)
par_dir_path = os.path.abspath(os.path.join(gen_path, os.pardir))
# print(par_dir_path)
sim_path = par_dir_path + '/{}/'.format(sim_pack)

# print(sim_path)

print(' ############################ ')

print('### Create job scripts folder for package: {}, task: {},  compute cap: {}, precision: {}'.format(sim_pack, task, com_cap, prec))
job_scripts_path = sim_path+'job_scripts_{}_{}'.format(com_cap, prec)
print(job_scripts_path)
os.mkdir(job_scripts_path)

print('### Create job scripts')
os.chdir(job_scripts_path)
cjs = gen_path+'create_job_script.py {} {} {} {}'.format(sim_pack, task, com_cap, prec)
os.system('python3 {}'.format(cjs))

print(' ############################ ')

print('### Create bash scripts folder for package: {}, task: {},  compute cap: {}, precision: {}'.format(sim_pack, task, com_cap, prec))
bash_scripts_path = sim_path+'bash_scripts_{}_{}'.format(com_cap, prec)
os.mkdir(bash_scripts_path)

print('### Create bash scripts')
os.chdir(bash_scripts_path)
cbs = gen_path+'create_bash_script.py {} {} {} {}'.format(sim_pack, task, com_cap, prec)
os.system('python3 {}'.format(cbs))

print(' ############################ ')

print('### Create output folder')
os.mkdir(sim_path+'output_{}_{}'.format(com_cap, prec))

print('### Create error folder')
os.mkdir(sim_path+'error_{}_{}'.format(com_cap, prec))

print('### Create data folder')
os.mkdir(sim_path+'data_{}_{}'.format(com_cap, prec))

print(' ############################ ')

print('### Create run files folder package: {}, task: {},  compute cap: {}, precision: {}'.format(sim_pack, task, com_cap, prec))
run_files_path = sim_path+'run_files_{}_{}'.format(com_cap, prec)
os.mkdir(run_files_path)

# sim_parser = input("Enter the python file including the path for the parser: ")
sim_parser = par_dir_path + '/qasm_parser/qasm_qiskit_parser_2pt0_new.py'
os.system('python3 {}'.format(sim_parser))

print(' ############################ ')

print('### Create bash scripts to run jobs')
os.chdir(sim_path)

with open('run_jobs_{}_{}.sh'.format(com_cap, prec), 'w') as f:

    f.write('path={}job_scripts_{}_{}\n'.format(sim_path, com_cap, prec))
    f.write('cd $path\n')
    f.write('for job_script in *.sh; do\n')
    f.write('echo $job_script;\n')
    f.write('sbatch $job_script;\n')
    f.write('done\n')