mirror of https://github.com/Lyxal/Keta.git
206 lines
4.7 KiB
Python
206 lines
4.7 KiB
Python
import Parse
|
|
import TokenLib
|
|
import string
|
|
|
|
dyads: str = string.punctuation
|
|
monads: str = string.ascii_uppercase
|
|
nilads: str = "1234567890"
|
|
|
|
source: str = str(reversed(input()))
|
|
|
|
def balance(source: str) -> str:
|
|
|
|
|
|
'''
|
|
|
|
a[bc --> a[bc]
|
|
ab} --> {ab}
|
|
m{ab]c -> [m{ab}]c
|
|
|
|
'''
|
|
|
|
final: str = ""
|
|
brackets: [str] = []
|
|
temp: str = ""
|
|
escaped: str = False
|
|
|
|
for char in source:
|
|
|
|
if escaped:
|
|
final += "\\" + char
|
|
escaped = False
|
|
|
|
elif char == "\\":
|
|
escaped = True
|
|
|
|
elif char in "[{":
|
|
brackets.append(char)
|
|
final += char
|
|
|
|
elif char == "}":
|
|
if brackets:
|
|
if brackets[-1] == "{":
|
|
final += char
|
|
|
|
else:
|
|
final = "{" + final + "]}"
|
|
|
|
brackets.pop()
|
|
|
|
else:
|
|
final = "[" + final + "]"
|
|
|
|
elif char == "]":
|
|
if brackets:
|
|
if brackets[-1] == "[":
|
|
final += char
|
|
|
|
else:
|
|
final = "[" + final + "}]"
|
|
|
|
brackets.pop()
|
|
|
|
else:
|
|
final = "[" + final + "]"
|
|
|
|
else:
|
|
final += char
|
|
|
|
if brackets:
|
|
for char in brackets:
|
|
if char == "[":
|
|
final += "]"
|
|
else:
|
|
final += "}"
|
|
return final.replace("{", "[").replace("}", "]")
|
|
|
|
|
|
def arities(source: [TokenLib.Token]) -> [(int, TokenLib.Token)]:
|
|
|
|
arity_list: [(int, TokenLib.Token)] = []
|
|
for token in source:
|
|
if token.get_name() == TokenLib.BLOCK:
|
|
arity_list.append([0,
|
|
arities(token.get_data())])
|
|
|
|
|
|
'''
|
|
arities = []
|
|
for char in source:
|
|
if char in dyads:
|
|
arities.append((2, char))
|
|
|
|
elif char in monads:
|
|
arities.append((1, char))
|
|
|
|
else:
|
|
arities.append((0, char))
|
|
|
|
exprs = []
|
|
expr = []
|
|
patterns = ["0", "1", "2",
|
|
"020", "021", "022", "02", "10", "11", "12", "20", "21", "22",
|
|
"102", "110", "111", "112", "120", "121", "122",
|
|
"202", "210", "211", "212", "220", "221", "222"]
|
|
pattern = ""
|
|
while len(arities):
|
|
if pattern in patterns and pattern + str(arities[-1][0]) not in patterns:
|
|
exprs.append([pattern, expr])
|
|
expr = []
|
|
pattern = ""
|
|
|
|
pattern += str(arities[-1][0])
|
|
expr.append(arities[-1][1])
|
|
arities.pop()
|
|
|
|
if expr and pattern in patterns:
|
|
exprs.append([pattern, expr])
|
|
expr = []
|
|
pattern = ""
|
|
|
|
print(exprs)
|
|
|
|
|
|
for exp in exprs:
|
|
pattern, fns = exp
|
|
|
|
if pattern == "0":
|
|
print(fns[0])
|
|
|
|
elif pattern == "1":
|
|
print(fns[0] + "(R)")
|
|
|
|
elif pattern == "2":
|
|
print(fns[0] + "(L, R)")
|
|
|
|
elif pattern == "020":
|
|
print(fns[1] + "(" + fns[0] + ", " + fns[2] + ")")
|
|
|
|
elif pattern == "021":
|
|
print(fns[1] + "(" + fns[0] + ", " + fns[2] + "(R))")
|
|
|
|
elif pattern == "022":
|
|
print(fns[1] + "(" + fns[0] + ", " + fns[2] + "(L, R))")
|
|
|
|
elif pattern == "02":
|
|
print(fns[1] + "(" + fns[0] + ", R)")
|
|
|
|
elif pattern == "10":
|
|
print(fns[0] + "(" + fns[1] + ")")
|
|
|
|
elif pattern == "11":
|
|
print(fns[0] + "(" + fns[1] + "(R))")
|
|
|
|
elif pattern == "12":
|
|
print(fns[1] + "(" + fns[0] + ", " + "R)")
|
|
|
|
elif pattern == "20":
|
|
print(fns[0] + "(L, " + fns[1] + ")")
|
|
|
|
elif pattern == "21":
|
|
print(fns[0] + "(L, " + fns[1] + ")")
|
|
|
|
elif pattern == "22":
|
|
print(fns[1] + "(" + fns[0] + "(L, R), R*)")
|
|
|
|
elif pattern == "102":
|
|
print(fns[2] + "(" + fns[0] + "(" + fns[1] + "), R)")
|
|
|
|
elif pattern == "110":
|
|
print(fns[0] + "(" + fns[1] + "(" + fns[2] + "))")
|
|
|
|
elif pattern == "111":
|
|
print(fns[0] + "(" + fns[1] + "(" + fns[2] + "(R)))")
|
|
|
|
elif pattern == "120":
|
|
print(fns[1] + "(" + fns[0] + "(R), " + fns[2] + ")")
|
|
|
|
elif pattern == "121":
|
|
print(fns[1] + "(" + fns[0] + "(R), " + fns[2] + "(R*))")
|
|
|
|
elif pattern == "122":
|
|
print(fns[1] + "(" + fns[0] + "(R), " + fns[2] + "(L*, R*))")
|
|
|
|
elif pattern == "202":
|
|
print(fns[2] + "(" + fns[0] + "(L, " + fns[1] + "), R*)")
|
|
|
|
elif pattern == "210":
|
|
print(fns[0] + "(L, " + fns[1] + "(" + fns[2] + "))")
|
|
|
|
elif pattern == "211":
|
|
print(fns[0] + "(L, " + fns[1] + "(" + fns[2] + "(R*)))")
|
|
|
|
elif pattern == "212":
|
|
print(fns[2] + "(" + fns[0] + "(L, " + fns[1] + "(R*)), R&)")
|
|
|
|
elif pattern == "220":
|
|
print(fns[1] + "(" + fns[0] + "(L, R), " + fns[2] + ")")
|
|
|
|
elif pattern == "221":
|
|
print(fns[1] + "(" + fns[0] + "(L, R), " + fns[2] + "(R*))")
|
|
|
|
elif pattern == "222":
|
|
print(fns[2] + "(" + fns[1] + "(" + fns[0] + "(L, R), R*), R&)")
|
|
'''
|
|
|