def test_annotations(self):
x = Symbol('x')
x_next = Symbol('x.next')
f = Iff(x, Not(x_next))
ann = Annotations()
ann.add(x, 'next', x_next.symbol_name())
ann.add(f, 'trans', 'true')
ann.add(x, 'init', 'true')
tree_buf = StringIO()
dag_buf = StringIO()
tree_printer = SmtPrinter(tree_buf, annotations=ann)
dag_printer = SmtDagPrinter(dag_buf, annotations=ann)
dag_printer.printer(f)
tree_printer.printer(f)
self.assertEqual(
tree_buf.getvalue(),
"(! (= (! x :next x.next :init true) (not x.next)) :trans true)")
self.assertEqual(
dag_buf.getvalue(),
"(let ((.def_0 (not x.next))) (let ((.def_1 (= (! x :next x.next :init true) .def_0))) (! .def_1 :trans true)))"
)
def test_daggify(self):
x = Symbol("x")
f = And(x, x)
for _ in range(10):
f = And(f, f)
tree_buf = StringIO()
dag_buf = StringIO()
tree_printer = SmtPrinter(tree_buf)
dag_printer = SmtDagPrinter(dag_buf)
dag_printer.printer(f)
tree_printer.printer(f)
short_f_str = dag_buf.getvalue()
long_f_str = tree_buf.getvalue()
self.assertTrue(len(short_f_str) < len(long_f_str))
def test_daggify(self):
x = Symbol("x")
f = And(x,x)
for _ in xrange(10):
f = And(f,f)
tree_buf = cStringIO()
dag_buf = cStringIO()
tree_printer = SmtPrinter(tree_buf)
dag_printer = SmtDagPrinter(dag_buf)
dag_printer.printer(f)
tree_printer.printer(f)
short_f_str = dag_buf.getvalue()
long_f_str = tree_buf.getvalue()
self.assertTrue(len(short_f_str) < len(long_f_str))
def serialize(self, outstream, daggify=False):
"""Serializes the SmtLibScript expanding commands"""
if daggify:
printer = SmtDagPrinter(outstream)
else:
printer = SmtPrinter(outstream)
for cmd in self.commands:
cmd.serialize(printer=printer)
outstream.write("\n")
def serialize(self, outstream=None, printer=None, daggify=False):
"""Serializes the SmtLibCommand into outstream using the given printer.
Exactly one of outstream or printer must be specified. When
specifying the printer, the associated outstream will be used.
If printer is not specified, daggify controls the printer to
be created. If true a daggified formula is produced, otherwise
a tree printing is done.
"""
if (outstream is None) and (printer is not None):
outstream = printer.stream
elif (outstream is not None) and (printer is None):
if daggify:
printer = SmtDagPrinter(outstream)
else:
printer = SmtPrinter(outstream)
else:
assert (outstream is not None and printer is not None) or \
(outstream is None and printer is None), \
"Exactly one of outstream and printer must be set."
if self.name in [smtcmd.SET_OPTION, smtcmd.SET_INFO]:
outstream.write("(%s %s %s)" %
(self.name, self.args[0], self.args[1]))
elif self.name == smtcmd.ASSERT:
outstream.write("(%s " % self.name)
printer.printer(self.args[0])
outstream.write(")")
elif self.name == smtcmd.GET_VALUE:
outstream.write("(%s (" % self.name)
for a in self.args:
printer.printer(a)
outstream.write(" ")
outstream.write("))")
elif self.name in [
smtcmd.CHECK_SAT, smtcmd.EXIT, smtcmd.RESET_ASSERTIONS
]:
outstream.write("(%s)" % self.name)
elif self.name == smtcmd.SET_LOGIC:
outstream.write("(%s %s)" % (self.name, self.args[0]))
elif self.name in [smtcmd.DECLARE_FUN, smtcmd.DECLARE_CONST]:
symbol = self.args[0]
type_str = symbol.symbol_type().as_smtlib()
outstream.write("(%s %s %s)" %
(self.name, symbol.symbol_name(), type_str))
elif self.name == smtcmd.DEFINE_FUN:
raise NotImplementedError
elif self.name in [smtcmd.PUSH, smtcmd.POP]:
outstream.write("(%s %d)" % (self.name, self.args[0]))
def serialize(self, outstream=None, printer=None, daggify=False):
"""Serializes the SmtLibCommand into outstream using the given printer.
Exactly one of outstream or printer must be specified. When
specifying the printer, the associated outstream will be used.
If printer is not specified, daggify controls the printer to
be created. If true a daggified formula is produced, otherwise
a tree printing is done.
"""
if (outstream is None) and (printer is not None):
outstream = printer.stream
elif (outstream is not None) and (printer is None):
if daggify:
printer = SmtDagPrinter(outstream)
else:
printer = SmtPrinter(outstream)
else:
assert (outstream is not None and printer is not None) or \
(outstream is None and printer is None), \
"Exactly one of outstream and printer must be set."
if self.name in [smtcmd.SET_OPTION,smtcmd.SET_INFO]:
outstream.write("(%s %s %s)" % (self.name,self.args[0],self.args[1]))
elif self.name == smtcmd.ASSERT:
outstream.write("(%s " % self.name)
printer.printer(self.args[0])
outstream.write(")")
elif self.name == smtcmd.GET_VALUE:
outstream.write("(%s (" % self.name)
for a in self.args:
printer.printer(a)
outstream.write(" ")
outstream.write("))")
elif self.name in [smtcmd.CHECK_SAT, smtcmd.EXIT,
smtcmd.RESET_ASSERTIONS]:
outstream.write("(%s)" % self.name)
elif self.name == smtcmd.SET_LOGIC:
outstream.write("(%s %s)" % (self.name, self.args[0]))
elif self.name in [smtcmd.DECLARE_FUN, smtcmd.DECLARE_CONST]:
symbol = self.args[0]
type_str = symbol.symbol_type().as_smtlib()
outstream.write("(%s %s %s)" % (self.name,
quote(symbol.symbol_name()),
type_str))
elif self.name == smtcmd.DEFINE_FUN:
raise NotImplementedError
elif self.name in [smtcmd.PUSH, smtcmd.POP]:
outstream.write("(%s %d)" % (self.name, self.args[0]))
def serialize(self, outstream=None, printer=None, daggify=True):
"""Serializes the SmtLibCommand into outstream using the given printer.
Exactly one of outstream or printer must be specified. When
specifying the printer, the associated outstream will be used.
If printer is not specified, daggify controls the printer to
be created. If true a daggified formula is produced, otherwise
a tree printing is done.
"""
if (outstream is None) and (printer is not None):
outstream = printer.stream
elif (outstream is not None) and (printer is None):
if daggify:
printer = SmtDagPrinter(outstream)
else:
printer = SmtPrinter(outstream)
else:
assert (outstream is not None and printer is not None) or \
(outstream is None and printer is None), \
"Exactly one of outstream and printer must be set."
if self.name == smtcmd.SET_OPTION:
outstream.write("(%s %s %s)" %
(self.name, self.args[0], self.args[1]))
elif self.name == smtcmd.SET_INFO:
outstream.write("(%s %s %s)" %
(self.name, self.args[0], quote(self.args[1])))
elif self.name == smtcmd.ASSERT:
outstream.write("(%s " % self.name)
printer.printer(self.args[0])
outstream.write(")")
elif self.name == smtcmd.GET_VALUE:
outstream.write("(%s (" % self.name)
for a in self.args:
printer.printer(a)
outstream.write(" ")
outstream.write("))")
elif self.name in [
smtcmd.CHECK_SAT, smtcmd.EXIT, smtcmd.RESET_ASSERTIONS
]:
outstream.write("(%s)" % self.name)
elif self.name == smtcmd.SET_LOGIC:
outstream.write("(%s %s)" % (self.name, self.args[0]))
elif self.name in [smtcmd.DECLARE_FUN, smtcmd.DECLARE_CONST]:
symbol = self.args[0]
type_str = symbol.symbol_type().as_smtlib()
outstream.write("(%s %s %s)" %
(self.name, quote(symbol.symbol_name()), type_str))
elif self.name == smtcmd.DEFINE_FUN:
name = self.args[0]
params_list = self.args[1]
params = " ".join(
["(%s %s)" % (v, v.symbol_type()) for v in params_list])
rtype = self.args[2]
expr = self.args[3]
outstream.write("(%s %s (%s) %s " %
(self.name, name, params, rtype))
printer.printer(expr)
outstream.write(")")
elif self.name in [smtcmd.PUSH, smtcmd.POP]:
outstream.write("(%s %d)" % (self.name, self.args[0]))
elif self.name == smtcmd.DEFINE_SORT:
name = self.args[0]
params_list = self.args[1]
params = " ".join(params_list)
rtype = self.args[2]
outstream.write("(%s %s (%s) %s)" %
(self.name, name, params, rtype))
elif self.name in smtcmd.ALL_COMMANDS:
raise NotImplementedError("'%s' is a valid SMT-LIB command "\
"but it is currently not supported. "\
"Please open a bug-report." % self.name)
else:
raise UnknownSmtLibCommandError(self.name)
def serialize(self, outstream=None, printer=None, daggify=True):
"""Serializes the SmtLibCommand into outstream using the given printer.
Exactly one of outstream or printer must be specified. When
specifying the printer, the associated outstream will be used.
If printer is not specified, daggify controls the printer to
be created. If true a daggified formula is produced, otherwise
a tree printing is done.
"""
if (outstream is None) and (printer is not None):
outstream = printer.stream
elif (outstream is not None) and (printer is None):
if daggify:
printer = SmtDagPrinter(outstream)
else:
printer = SmtPrinter(outstream)
else:
assert (outstream is not None and printer is not None) or \
(outstream is None and printer is None), \
"Exactly one of outstream and printer must be set."
if self.name == smtcmd.SET_OPTION:
outstream.write("(%s %s %s)" % (self.name,self.args[0],self.args[1]))
elif self.name == smtcmd.SET_INFO:
outstream.write("(%s %s %s)" % (self.name,self.args[0],
quote(self.args[1])))
elif self.name == smtcmd.ASSERT:
outstream.write("(%s " % self.name)
printer.printer(self.args[0])
outstream.write(")")
elif self.name == smtcmd.GET_VALUE:
outstream.write("(%s (" % self.name)
for a in self.args:
printer.printer(a)
outstream.write(" ")
outstream.write("))")
elif self.name in [smtcmd.CHECK_SAT, smtcmd.EXIT,
smtcmd.RESET_ASSERTIONS, smtcmd.GET_UNSAT_CORE,
smtcmd.GET_ASSIGNMENT, smtcmd.GET_MODEL]:
outstream.write("(%s)" % self.name)
elif self.name == smtcmd.SET_LOGIC:
outstream.write("(%s %s)" % (self.name, self.args[0]))
elif self.name in [smtcmd.DECLARE_FUN, smtcmd.DECLARE_CONST]:
symbol = self.args[0]
type_str = symbol.symbol_type().as_smtlib()
outstream.write("(%s %s %s)" % (self.name,
quote(symbol.symbol_name()),
type_str))
elif self.name == smtcmd.DEFINE_FUN:
name = self.args[0]
params_list = self.args[1]
params = " ".join(["(%s %s)" % (v, v.symbol_type()) for v in params_list])
rtype = self.args[2]
expr = self.args[3]
outstream.write("(%s %s (%s) %s " % (self.name,
name,
params,
rtype))
printer.printer(expr)
outstream.write(")")
elif self.name in [smtcmd.PUSH, smtcmd.POP]:
outstream.write("(%s %d)" % (self.name, self.args[0]))
elif self.name == smtcmd.DEFINE_SORT:
name = self.args[0]
params_list = self.args[1]
params = " ".join(params_list)
rtype = self.args[2]
outstream.write("(%s %s (%s) %s)" % (self.name,
name,
params,
rtype))
elif self.name == smtcmd.DECLARE_SORT:
type_decl = self.args[0]
outstream.write("(%s %s %d)" % (self.name,
type_decl.name,
type_decl.arity))
elif self.name in smtcmd.ALL_COMMANDS:
raise NotImplementedError("'%s' is a valid SMT-LIB command "\
"but it is currently not supported. "\
"Please open a bug-report." % self.name)
else:
raise UnknownSmtLibCommandError(self.name)
def _formula_to_smt2(self, formula):
buf = cStringIO()
printer = SmtDagPrinter(buf)
printer.printer(formula)
return buf.getvalue()
