import argparse
from time import sleep
import subprocess as sub

# Construct an argument parser
all_args = argparse.ArgumentParser()

all_args.add_argument("-b", "--Value1", action="store", dest="b",
   type=int, choices=range(1,6), help="Bayesian model of size x")
all_args.add_argument("-f", "--Value2", action="store", dest="f",
   type=int, choices=range(1,6), help="Frequentist model of size x")
args = vars(all_args.parse_args())


check = list(args.values())
if all(v is None for v in check):
    raise Exception("One argument required")
elif None in check:
    if args['f'] is not None:
        cmd = ["python", "main_frequentist.py"]
    elif args['b'] is not None:
        cmd = ["python", "main_bayesian.py"]
else:
    raise Exception("Only one argument allowed")


wide = args["f"] or args["b"]

with open("tmp", "w") as file:
    file.write(str(wide))

sleep(3)


if cmd[1] == "main_frequentist.py":
    cmd2 = ["./cpu_watt.sh", "freq_{}_cpu_watts".format(wide)]
    cmd3 = ["./mem_free.sh", "freq_{}_ram_use".format(wide)]
    with open("frq", "w") as file:
        file.write(str(1))
    with open("bay", "w") as file:
        file.write(str(0))
elif cmd[1] == "main_bayesian.py":
    cmd2 = ["./cpu_watt.sh", "bayes_{}_cpu_watts".format(wide)]
    cmd3 = ["./mem_free.sh", "bayes_{}_ram_use".format(wide)]
    with open("bay", "w") as file:
        file.write(str(1))
    with open("frw", "w") as file:
        file.write(str(0))


path = sub.check_output(['pwd'])
path = path.decode()
path = path.replace('\n', '')

#startWattCounter = 'python ' + path + '/gpu_sample_draw.py'

#test = startNODE.split()
#test.append(pythonEnd)
#test.append(pythonEnd2)

#startNODE = test

##print(startNODE)
##print(startWattCounter)

p1 = sub.Popen(cmd)
#p2 = sub.Popen(startWattCounter.split())
p3 = sub.Popen(cmd2)
p4 = sub.Popen(cmd3)

retcode = p1.wait()
print("Return code: {}".format(retcode))

p1.kill()
#p2.kill()
p3.kill()
p4.kill()