Added utility scripts

utils/hasse.py

@@ -0,0 +1,54 @@
+"""
+Given a model, create a Hasse diagram.
+
+Note: This has a dependency on the hasse-diagram library
+https://pypi.org/project/hasse-diagram/
+"""
+import argparse
+import os
+import sys
+sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+
+from model import Model
+from parse_magic import SourceFile, parse_matrices
+
+import numpy as np
+import hassediagram
+
+__all__ = ['plot_model_hassee']
+
+def plot_model_hassee(model: Model):
+    assert model.ordering is not None
+    carrier_list = list(model.carrier_set)
+    hasse_ordering = []
+    for elem1 in carrier_list:
+        elem_ordering = []
+        for elem2 in carrier_list:
+            elem_ordering.append(
+                1 if model.ordering.is_lt(elem1, elem2) else 0
+            )
+        hasse_ordering.append(elem_ordering)
+    hassediagram.plot_hasse(np.array(hasse_ordering), carrier_list)
+
+
+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))
+
+    requested_model = None
+
+    for model, _ in solutions:
+        if model.name == args['modelname']:
+            requested_model = model
+            break
+
+    if requested_model is None:
+        print("Model name", args['modelname'], "not found.")
+        sys.exit(0)
+
+    plot_model_hassee(requested_model)