mirror of
https://github.com/Brandon-Rozek/matmod.git
synced 2025-07-31 21:01:59 +00:00
30 lines
887 B
Python
30 lines
887 B
Python
"""
|
|
Print a model given it's name
|
|
and ugly data file
|
|
"""
|
|
import argparse
|
|
import os
|
|
import sys
|
|
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|
|
|
from parse_magic import SourceFile, parse_matrices
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(description="Show hassee diagram for model")
|
|
parser.add_argument("uglyfile", type=str, help="Path to ugly data file")
|
|
parser.add_argument("modelname", type=str, help="Name of model within file")
|
|
args = vars(parser.parse_args())
|
|
|
|
data_file = open(args['uglyfile'], "r")
|
|
solutions = parse_matrices(SourceFile(data_file))
|
|
|
|
model_found = False
|
|
|
|
for model, _ in solutions:
|
|
if model.name == args['modelname']:
|
|
model_found = True
|
|
print(model)
|
|
break
|
|
|
|
if not model_found:
|
|
print("Model", args['modelname'], "not found.")
|