2024-01-04 22:51:33 +08:00
|
|
|
|
|
|
|
|
|
|
|
|
|
from sympy.parsing.sympy_parser import parse_expr
|
|
|
|
|
from sympy import symbols, Not, Or, And, to_dnf
|
|
|
|
|
|
|
|
|
|
|
2024-04-10 19:59:13 +08:00
|
|
|
# goal = "(On_Coffee_Bar | On_Yogur_Bar) & At_Robot_Bar"
|
2024-01-04 22:51:33 +08:00
|
|
|
# goal = "On(Coffee,WaterTable),On(Coffee,BrightTable6)"
|
2024-04-10 19:59:13 +08:00
|
|
|
goal = "Not (On_Coffee_Bar | On_Yogur_Bar) & At_Robot_Bar"
|
2024-01-04 22:51:33 +08:00
|
|
|
|
|
|
|
|
goal_dnf = str(to_dnf(goal, simplify=True))
|
|
|
|
|
# print(goal_dnf)
|
|
|
|
|
goal_set=[]
|
|
|
|
|
if '|' in goal or '&' in goal or 'Not' in goal:
|
|
|
|
|
goal_ls = goal_dnf.split("|")
|
|
|
|
|
for g in goal_ls:
|
|
|
|
|
g_set = set()
|
|
|
|
|
g = g.replace(" ", "").replace("(", "").replace(")", "")
|
|
|
|
|
g = g.split("&")
|
|
|
|
|
for literal in g:
|
|
|
|
|
if '_' in literal:
|
|
|
|
|
first_part, rest = literal.split('_', 1)
|
|
|
|
|
literal = first_part + '(' + rest
|
|
|
|
|
# 添加 ')' 到末尾
|
|
|
|
|
literal += ')'
|
|
|
|
|
# 替换剩余的 '_' 为 ','
|
|
|
|
|
literal = literal.replace('_', ',')
|
|
|
|
|
g_set.add(literal)
|
|
|
|
|
goal_set.append(g_set)
|
|
|
|
|
else:
|
|
|
|
|
g_set = set()
|
|
|
|
|
w = goal.split(")")
|
|
|
|
|
g_set.add(w[0] + ")")
|
|
|
|
|
if len(w) > 1:
|
|
|
|
|
for x in w[1:]:
|
|
|
|
|
if x != "":
|
|
|
|
|
g_set.add(x[1:] + ")")
|
|
|
|
|
goal_set.append(g_set)
|
|
|
|
|
|
|
|
|
|
# goal_set = [set(["On(Coffee,Bar)", "At(Robot,Bar)"]), set(["On(Yogurt,Bar)", "At(Robot,Bar)"])]
|
|
|
|
|
print(goal_set)
|
