1
0
Fork 0
mirror of https://github.com/Lyxal/Keta.git synced 2024-08-14 23:53:13 +00:00
Keta/ekg/chainify.py
2019-12-28 11:14:53 +11:00

205 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&)")
'''