import json, re, sys, os, glob
from os import path

def eprint(*args, **kwargs):
	print(*args, file=sys.stderr, flush=True, **kwargs)

def process_run(tree, runPath):

	with open(".garlic/total_time_start", "r") as f:
		total_time_start = float(f.readline().strip())

	with open(".garlic/total_time_end", "r") as f:
		total_time_end = float(f.readline().strip())

	with open("stdout.log", "r") as f:
		lines = [line.strip() for line in f.readlines()]

	time_line = None
	for line in lines:

		if re.match(r'^ ?time .*', line):
			time_line = line
			break

	if time_line is None:
		eprint("missing time line, aborting")
		eprint("stdout file = {}/stdout.log".format(runPath))
		exit(1)

	time_str = time_line.split()[1]

	tree['time'] = float(time_str)
	tree['total_time'] = total_time_end - total_time_start

	print(json.dumps(tree))

def process_result_tree(resultTree):

	eprint("processing resultTree: " + resultTree)

	os.chdir(resultTree)

	experiments = glob.glob(resultTree + "/*-experiment")

	for exp in glob.glob("*-experiment"):
		eprint("found experiment: " + exp)
		expPath = path.join(resultTree, exp)
		os.chdir(expPath)

		for unit in glob.glob("*-unit"):
			eprint("found unit: " + unit)
			unitPath = path.join(resultTree, exp, unit)
			os.chdir(unitPath)

			with open('garlic_config.json') as json_file:
				garlic_conf = json.load(json_file)

			tree = {"exp":exp, "unit":unit, "config":garlic_conf}

			for i in range(garlic_conf['loops']):
				run = str(i + 1)
				runPath = path.join(resultTree, exp, unit, run)
				if path.isdir(runPath) == False:
					eprint("missing run {}, aborting".format(run))
					exit(1)

				tree["run"] = run
				os.chdir(runPath)

				process_run(tree, runPath)


if len(sys.argv) != 2:
	eprint("usage: python {} <resultTree>".format(argv[0]))
	exit(1)

process_result_tree(sys.argv[1])