Я пытаюсь собрать все переменные в формулу (количественная формула в Z3py). Небольшой пример
w, x, y, z = Bools('w x y z')
fml = And( ForAll(x, ForAll(y, And(x, y))), ForAll(z, ForAll(w, And(z, w))) )
varSet = traverse( fml )
Код, который я использую для прохождения,
def traverse(e):
r = set()
def collect(e):
if is_quantifier(e):
# Lets assume there is only one type of quantifier
if e.is_forall():
collect(e.body())
else:
if ( is_and(e) ):
n = e.num_args()
for i in range(n):
collect( e.arg(i) )
if ( is_or(e) ):
n = e.num_args()
for i in range(n):
collect( e.arg(i) )
if ( is_not(e) ):
collect( e.arg(0) )
if ( is_var(e) ):
r.add( e )
collect(e)
return r
И я получаю: set([Var(0), Var(1)]). Насколько я понимаю, это связано с тем, что Z3 использует индекс Де Брюйна. Можно ли этого избежать и получить нужный набор: set( [Var(0), Var(1), Var(2), Var(3)] ).