added instance name and ext_nr to result csv

This commit is contained in:
Tom Krüger
2021-01-21 02:46:39 +01:00
parent 2aaf89d01f
commit a9fe4f1ae0

18
main.py
View File

@@ -58,18 +58,32 @@ def parse_result(res_str):
return res return res
def parse_instance_path(ipath):
name = pl.Path(ipath).name.strip()
ext_nr = int(name[:-len(".cnf.bz2")][-4:])
instance = name[:-len("_0000.cnf.bz2")]
return instance, ext_nr
if __name__ == '__main__': if __name__ == '__main__':
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument("-i", "--instance", help="The instance.") parser.add_argument("-i", "--instance", help="The instance.")
args = parser.parse_args() args = parser.parse_args()
res_str = run_alg(args.instance) res_str = run_alg(args.instance)
iname, ext_nr = parse_instance_path(args.instance)
res = parse_result(res_str) res = parse_result(res_str)
res_vals = [iname, str(ext_nr)] + list(map(str, res.values()))
print(','.join(map(str, res.values()))) print(','.join(res_vals))