jfz-2023-s473555/TaskC04/run.py

174 lines
5.1 KiB
Python

import sys
import csv
from tokenize import Comment
from numpy import append
DEBUG = False
alfabet = "abc"
EPS = "<eps>"
# sys.setrecursionlimit(4100)
def print_debug(*args, **kwargs):
if DEBUG:
print(*args, **kwargs)
if len(sys.argv) == 1:
print("Default arguments parsed\n")
sys.argv.append("simple2.arg")
sys.argv.append("simple2.in")
sys.argv.append("simple2.exp")
with open(sys.argv[1], mode="r", newline="", encoding="utf8") as csvfile:
filereader = csv.reader(csvfile, delimiter="\t", quotechar="|")
fsa_description = list(filereader)
# skip first line with comment
if fsa_description[0][0][0] == "#":
fsa_description = fsa_description[1:]
# get accepting states
accepting_state = []
for item in reversed(fsa_description):
if len(item) == 1:
accepting_state.append(item[0])
else:
break
fsa_description = fsa_description[: -len(accepting_state)]
fsa_description_map: dict[tuple[str, str], list[str]] = {}
is_error = False
for num, item in enumerate(fsa_description):
if EPS in item[2]:
item[2] = item[2].replace(EPS, "")
tuple = (item[0], EPS)
if (item[0], EPS) in fsa_description_map:
print_debug(
f"WARNING - duplicate letter: key = ({item[0]} | {EPS}) | line: {num + 1}"
)
# append to description map or create it if it doesn't exist
fsa_description_map[tuple] = fsa_description_map.get(tuple, []) + [item[1]]
for letter in item[2]:
if letter not in alfabet:
print_debug(
"WARNING - letter not in alfabet: ", letter, "| line:", num + 1
)
tuple = (item[0], letter)
if (item[0], letter) in fsa_description_map:
print_debug(
f"WARNING - duplicate letter: key = ({item[0]} | {letter}) | line: {num + 1}"
)
# append to description map or create it if it doesn't exist
fsa_description_map[tuple] = fsa_description_map.get(tuple, []) + [item[1]]
if is_error:
exit(-1)
states = {}
for item in fsa_description_map:
states[item[0]] = states[item[0]] + item[1] if item[0] in states else item[1]
from collections import Counter
# check if all letters are used once
def is_permutation(str1, str2):
return Counter(str1) == Counter(str2)
def find_missing_letters(str1, str2) -> str:
missing_letters = ""
for char in str2:
if char not in str1:
missing_letters += char
return missing_letters
for state in states:
if not is_permutation(states[state], alfabet):
print_debug(
f"WARNING - state {state} doesn't match full alphabet: {states[state]} | {alfabet} | diff - {find_missing_letters(states[state], alfabet)}"
)
# exit(-1)
with open(sys.argv[2], mode="r", newline="", encoding="utf8") as file:
content = file.read()
test_in = content.splitlines()
with open(sys.argv[3], mode="r", newline="", encoding="utf8") as file:
content = file.read()
test_out = content.splitlines()
for i, _ in enumerate(test_out):
test_out[i] = test_out[i].split(" ")[0]
def is_correct(current_state: str) -> bool:
if current_state in accepting_state:
return True
return False
def find_final_state(current_state: str, word: str, letter: str) -> str:
next_states = fsa_description_map[(current_state, letter)]
for next_state in next_states:
if letter != EPS:
word = word[1:]
current_state = traverse_states(next_state, word)
if is_correct(current_state):
return current_state # found a correct state
return current_state
def traverse_states(current_state: str, word: str) -> str:
if word == "" and (current_state, EPS) not in fsa_description_map:
return current_state
if (current_state, EPS) in fsa_description_map:
final_state = find_final_state(current_state, word, EPS)
if is_correct(final_state):
return final_state # found a correct state
letter = word[0]
if (current_state, letter) not in fsa_description_map:
return "-1" # unspecified next state
final_state = find_final_state(current_state, word, letter)
if is_correct(final_state):
return final_state # found a correct state
return "-2" # no correct state found
is_difference = []
for i, word in enumerate(test_in):
current_state = "0"
if len(word) != 0:
current_state = traverse_states(current_state, word)
print(str(i + 1) + "\t", end="")
if current_state == "-3":
print("FATAL ERROR")
exit(-1)
if current_state in accepting_state:
print("YES\t", end="")
if "YES" != test_out[i] and "TRUE" != test_out[i]:
print("ERROR", word, end="")
is_difference.append(i + 1)
else:
print("NO\t", end="")
if "NO" != test_out[i] and "FALSE" != test_out[i]:
print("ERROR", word, end="")
is_difference.append(i + 1)
print()
if len(is_difference) != 0:
print(is_difference)