mirror of
https://github.com/Brandon-Rozek/matmod.git
synced 2025-01-06 20:45:02 +00:00
45 lines
1.4 KiB
Python
Executable file
45 lines
1.4 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
from os import cpu_count
|
|
import argparse
|
|
import multiprocessing
|
|
|
|
from parse_magic import (
|
|
SourceFile,
|
|
parse_matrices
|
|
)
|
|
from vsp import has_vsp, VSP_Result
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(description="VSP Checker")
|
|
parser.add_argument("--verbose", action='store_true', help="Print out all parsed matrices")
|
|
parser.add_argument("-i", type=str, help="Path to MaGIC ugly data file")
|
|
args = vars(parser.parse_args())
|
|
|
|
data_file_path = args.get("i")
|
|
if data_file_path is None:
|
|
data_file_path = input("Path to MaGIC Ugly Data File: ")
|
|
|
|
solutions = []
|
|
with open(data_file_path, "r") as data_file:
|
|
solutions = parse_matrices(SourceFile(data_file))
|
|
print(f"Parsed {len(solutions)} matrices")
|
|
|
|
num_has_vsp = 0
|
|
with multiprocessing.Pool(processes=max(cpu_count() - 2, 1)) as pool:
|
|
results = [
|
|
pool.apply_async(has_vsp, (model, interpretation,))
|
|
for model, interpretation in solutions
|
|
]
|
|
|
|
for i, result in enumerate(results):
|
|
vsp_result: VSP_Result = result.get()
|
|
print(vsp_result)
|
|
|
|
if args['verbose'] or vsp_result.has_vsp:
|
|
model = solutions[i][0]
|
|
print(model)
|
|
|
|
if vsp_result.has_vsp:
|
|
num_has_vsp += 1
|
|
|
|
print(f"Tested {len(solutions)} models, {num_has_vsp} of which satisfy VSP")
|