Z3py:如何从公式中获取变量列表?

时间:2012-12-29 09:34:17

标签: z3 z3py

在Z3Py中,我有一个公式。如何在公式中检索变量列表?

感谢。

1 个答案:

答案 0 :(得分:6)

Vu Nguyen为Z3Py贡献了几个有用的程序。 他实现了一个过程get_vars来收集已使用变量的列表。 可以找到此程序和许多其他程序here。 他的贡献将在下一个正式版本中提供。 以下是get_vars版本,可在rise4fun在线执行。

# Wrapper for allowing Z3 ASTs to be stored into Python Hashtables. 
class AstRefKey:
    def __init__(self, n):
        self.n = n
    def __hash__(self):
        return self.n.hash()
    def __eq__(self, other):
        return self.n.eq(other.n)
    def __repr__(self):
        return str(self.n)

def askey(n):
    assert isinstance(n, AstRef)
    return AstRefKey(n)

def get_vars(f):
    r = set()
    def collect(f):
      if is_const(f): 
          if f.decl().kind() == Z3_OP_UNINTERPRETED and not askey(f) in r:
              r.add(askey(f))
      else:
          for c in f.children():
              collect(c)
    collect(f)
    return r

x,y = Ints('x y') 
a,b = Bools('a b') 
print get_vars(Implies(And(x + y == 0, x*2 == 10),Or(a, Implies(a, b == False))))