matmod/parse_magic.py

253 lines
7.1 KiB
Python

"""
Parses the Magic Ugly Data File Format
Assumes the base logic is R with no extra connectives
"""
import sys
from typing import TextIO, List, Optional, Tuple, Set, Dict
from model import Model, ModelValue, ModelFunction
from logic import (
Implication,
Conjunction,
Negation,
Disjunction
)
from vsp import has_vsp
def parse_matrices(infile: TextIO) -> List[Tuple[Model, Dict]]:
next(infile) # Skip header line
solutions: List[Tuple[Model, Dict]] = []
while True:
size = parse_size(infile)
if size is None:
break
carrier_set = carrier_set_from_size(size)
while True:
mnegation = parse_negation(infile, size)
if mnegation is None:
break
while True:
result = parse_order(infile, size)
if result is None:
break
mconjunction, mdisjunction = result
while True:
designated_values = parse_designated(infile, size)
if designated_values is None:
break
results = parse_implication(infile, size)
if result is None:
break
for mimplication in results:
logical_operations = {
mnegation, mconjunction, mdisjunction,
mimplication
}
model = Model(carrier_set, logical_operations, designated_values)
interpretation = {
Negation: mnegation,
Conjunction: mconjunction,
Disjunction: mdisjunction,
Implication: mimplication
}
solutions.append((model, interpretation))
print(f"Parsed Matrix {len(solutions)}")
return solutions
def carrier_set_from_size(size: int):
return {
mvalue_from_index(i) for i in range(size + 1)
}
def parse_size(infile: TextIO) -> Optional[int]:
# Elements are represented in hexidecimal
size = int(next(infile), 16)
if size == -1:
return None
assert size > 0, "Unexpected size"
return size
def parse_negation(infile: TextIO, size: int) -> Optional[ModelFunction]:
line = next(infile).strip()
if line == '-1':
return None
row = line.split(" ")
assert len(row) == size + 1, "Negation table doesn't match size"
mapping = {}
for i, j in zip(range(size + 1), row):
x = mvalue_from_index(i)
y = parse_mvalue(j)
mapping[(x, )] = y
return ModelFunction(1, mapping, "Negation")
def mvalue_from_index(i: int):
return ModelValue(f"a{hex(i)[-1]}")
def parse_mvalue(x: str) -> ModelValue:
return mvalue_from_index(int(x, 16))
def determine_cresult(size: int, ordering: Dict[ModelValue, ModelValue], a: ModelValue, b: ModelValue) -> ModelValue:
for i in range(size + 1):
c = mvalue_from_index(i)
if not ordering[(c, a)]:
continue
if not ordering[(c, b)]:
continue
invalid = False
for j in range(size + 1):
d = mvalue_from_index(j)
if c == d:
continue
if ordering[(c, d)]:
if ordering[(d, a)] and ordering [(d, b)]:
invalid = True
if not invalid:
return c
print(a, "&", b, "is not defined")
def determine_dresult(size: int, ordering: Dict[ModelValue, ModelValue], a: ModelValue, b: ModelValue) -> ModelValue:
for i in range(size + 1):
c = mvalue_from_index(i)
if not ordering[(a, c)]:
continue
if not ordering[(b, c)]:
continue
invalid = False
for j in range(size + 1):
d = mvalue_from_index(j)
if d == c:
continue
if ordering[(d, c)]:
if ordering[(a, d)] and ordering[(b, d)]:
invalid = True
if not invalid:
return c
print(a, "|", b, "is not defined")
def parse_order(infile: TextIO, size: int) -> Optional[Tuple[ModelFunction, ModelFunction]]:
line = next(infile).strip()
if line == '-1':
return None
table = line.split(" ")
assert len(table) == (size + 1)**2
omapping = {}
table_i = 0
for i in range(size + 1):
x = mvalue_from_index(i)
for j in range(size + 1):
y = mvalue_from_index(j)
omapping[(x, y)] = table[table_i] == '1'
table_i += 1
# NOTE: Print omapping for debugging
for (x, y) in omapping.keys():
print(x, y, "maps to", omapping[(x, y)])
cmapping = {}
dmapping = {}
for i in range(size + 1):
x = mvalue_from_index(i)
for j in range(size + 1):
y = mvalue_from_index(j)
cmapping[(x, y)] = determine_cresult(size, omapping, x, y)
dmapping[(x, y)] = determine_dresult(size, omapping, x, y)
mconjunction = ModelFunction(2, cmapping, "Conjunction")
mdisjunction = ModelFunction(2, dmapping, "Disjunction")
return mconjunction, mdisjunction
def parse_designated(infile: TextIO, size: int) -> Optional[Set[ModelValue]]:
line = next(infile).strip()
if line == '-1':
return None
row = line.split(" ")
assert len(row) == size + 1, "Designated table doesn't match size"
designated_values = set()
for i, j in zip(range(size + 1), row):
if j == '1':
x = mvalue_from_index(i)
designated_values.add(x)
return designated_values
def parse_implication(infile: TextIO, size: int) -> Optional[List[ModelFunction]]:
line = next(infile).strip()
if line == '-1':
return None
# Split and remove the last '-1' character
table = line.split(" ")[:-1]
assert len(table) % (size + 1)**2 == 0
table_i = 0
mimplications: List[ModelFunction] = []
for _ in range(len(table) // (size + 1)**2):
mapping = {}
for i in range(size + 1):
x = mvalue_from_index(i)
for j in range(size + 1):
y = mvalue_from_index(j)
r = parse_mvalue(table[table_i])
table_i += 1
mapping[(x, y)] = r
mimplication = ModelFunction(2, mapping, "Implication")
mimplications.append(mimplication)
return mimplications
if __name__ == "__main__":
solutions: List[Model] = parse_matrices(sys.stdin)
print(f"Parsed {len(solutions)} matrices")
for i, (model, interpretation) in enumerate(solutions):
# TODO: Check if conjunction and disjunction are well defined while parsing
model.logical_operations -= {interpretation[Conjunction], interpretation[Disjunction]}
del interpretation[Conjunction]
del interpretation[Disjunction]
# print(model)
if has_vsp(model, interpretation):
print(model)
print("Has VSP")
else:
print("Model", i, "does not have VSP")