Something went wrong. Try again.
This repository has no description
Something went wrong. Try again.
3.5 kB · 104 lines
Python
at develop
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105#!/usr/bin/env python3import minizincimport sysimport csvimport osimport timeimport subprocessimport refrom tqdm import tqdmfrom pathlib import Pathfrom datetime import timedelta
MZNCC = "../build/mzncc"MZNASM = "../build/mznasm"REPEATS = 10
# Instances Selection (File location)if len(sys.argv) < 3: print("Usage: new_flatstats.py <instances> <library>") exit(1)
instances_location = sys.argv[1]library = sys.argv[2]
def parse_statistics(stdout: bytes, stderr: bytes): statistics: Dict[str, StatisticsType] = {} matches = re.findall(rb"%%%mzn-stat:? (\w*)=([^\r\n]*)", stdout) for m in matches: minizinc.result.set_stat(statistics, m[0].decode(), m[1].decode())
matches = re.findall(rb"([A-Z].*): (.*)\n", stderr) for m in matches: statistics[m[0].decode()] = m[1].decode() return statistics
writer = Nonenum_lines = sum(1 for line in open(instances_location)) - 1with open(instances_location) as instances_file: with tqdm(total=num_lines*REPEATS, file=sys.stderr) as pbar: reader = csv.reader(instances_file, dialect="unix") next(reader, None) # Skip the header line mem = "" for row in reader: assert len(row) == 3 pbar.set_description( "Flattening %s - %s" % (row[1].split(os.sep)[-1], row[2].split(os.sep)[-1]) )
# Compile model if mem != row[1]: clock = time.perf_counter() os.system(f"{MZNCC} -G{library} {row[1]} > _temp.uzn") clock = timedelta( microseconds=int((time.perf_counter() - clock) * 1000000) ) mem = row[1]
# Flatten instance cmd = [ '/usr/bin/time', '-v', MZNASM, "--solver", "org.minizinc.mzn-fzn", "-c", "-s", "_temp.uzn", ] if row[2]: cmd.append(row[2]) for i in range(REPEATS): output = subprocess.run(cmd, capture_output=True,)
if output.returncode != 0: print( f"mznasm ERROR {output.returncode}: {row[1].split(os.sep)[-1]} - {row[2].split(os.sep)[-1]}:\n{output.stderr}", file=sys.stderr, ) break else: stats = parse_statistics(output.stdout, output.stderr) stats["problem"] = row[0] stats["model"] = row[1] stats["data_file"] = row[2] stats["compileTime"] = clock stats["iteration"] = i + 1 if writer is None: keys = list( set( ["problem", "model", "data_file", "status"] + list(minizinc.result.StdStatisticTypes.keys()) + list(stats.keys()) ) ) writer = csv.DictWriter( sys.stdout, keys, dialect="unix", extrasaction="ignore" ) writer.writeheader() writer.writerow(stats) pbar.update(1) sys.stdout.flush()