3
0
Fork 0
mirror of https://github.com/Z3Prover/z3 synced 2025-04-10 03:07:07 +00:00
z3/test/var_subst.cpp
Leonardo de Moura 68269c43a6 other components
Signed-off-by: Leonardo de Moura <leonardo@microsoft.com>
2012-10-02 11:48:48 -07:00

125 lines
3.5 KiB
C++

/*++
Copyright (c) 2006 Microsoft Corporation
Module Name:
var_subst.cpp
Abstract:
<abstract>
Author:
Leonardo de Moura (leonardo) 2008-06-12.
Revision History:
--*/
#include"var_subst.h"
#include"smtparser.h"
#include"ast_pp.h"
#include"arith_decl_plugin.h"
#include"bv_decl_plugin.h"
#include"array_decl_plugin.h"
#include"for_each_expr.h"
namespace find_q {
struct proc {
quantifier * m_q;
proc():m_q(0) {}
void operator()(var * n) {}
void operator()(app * n) {}
void operator()(quantifier * n) { m_q = n; }
};
};
quantifier * find_quantifier(expr * n) {
find_q::proc p;
for_each_expr(p, n);
return p.m_q;
}
void tst_instantiate(ast_manager & m, expr * f) {
if (is_quantifier(f)) {
tst_instantiate(m, to_quantifier(f)->get_expr());
return;
}
quantifier * q = find_quantifier(f);
if (q) {
expr_ref_vector cnsts(m);
for (unsigned i = 0; i < q->get_num_decls(); i++)
cnsts.push_back(m.mk_fresh_const("a", q->get_decl_sort(i)));
expr_ref r(m);
instantiate(m, q, cnsts.c_ptr(), r);
TRACE("var_subst", tout << "quantifier:\n" << mk_pp(q, m) << "\nresult:\n" << mk_pp(r, m) << "\n";);
}
}
void tst_subst(ast_manager& m) {
func_decl_ref p(m);
sort_ref s(m);
obj_ref<var, ast_manager> x(m), y(m), z(m), u(m), v(m);
expr_ref e1(m), e2(m), e3(m);
expr_ref t1(m), t2(m), t3(m);
s = m.mk_sort(symbol("S"));
sort* ss[2] = { s.get(), s.get() };
symbol names[2] = { symbol("y"), symbol("x") };
p = m.mk_func_decl(symbol("p"), 2, ss, m.mk_bool_sort());
x = m.mk_var(0, s);
y = m.mk_var(1, s);
z = m.mk_var(2, s);
u = m.mk_var(3, s);
v = m.mk_var(4, s);
e1 = m.mk_and(m.mk_app(p, x.get(), y.get()), m.mk_app(p, z.get(), u.get()));
e2 = m.mk_forall(1, ss, names, e1);
t1 = m.mk_forall(1, ss, names,
m.mk_and(m.mk_app(p, x.get(), z.get()), m.mk_app(p, y.get(), u.get())));
t2 = m.mk_forall(2, ss, names,
m.mk_and(m.mk_app(p, x.get(), y.get()), m.mk_app(p, u.get(), z.get())));
var_subst subst(m);
expr_ref_vector sub1(m);
sub1.push_back(x);
sub1.push_back(y);
// replace #1 -> #2, #2 -> #1
subst(e2, 2, sub1.c_ptr(), e3);
std::cout << mk_pp(e2, m) << "\n";
std::cout << mk_pp(e3, m) << "\n";
std::cout << mk_pp(t1, m) << "\n";
SASSERT(e3.get() == t1.get());
// replace #2 -> #3, #3 -> #2
e2 = m.mk_forall(2, ss, names, e1);
subst(e2, 2, sub1.c_ptr(), e3);
std::cout << mk_pp(e2, m) << "\n";
std::cout << mk_pp(e3, m) << "\n";
std::cout << mk_pp(t2, m) << "\n";
SASSERT(e3.get() == t2.get());
}
void tst_var_subst() {
ast_manager m;
m.register_decl_plugins();
tst_subst(m);
scoped_ptr<smtlib::parser> parser = smtlib::parser::create(m);
parser->initialize_smtlib();
parser->parse_string(
"(benchmark samples :logic AUFLIA\n"
" :extrafuns ((f Int Int) (g Int Int Int) (a Int) (b Int))\n"
" :formula (forall (x Int) (or (= (f x) x) (forall (y Int) (z Int) (= (g x y) (f z)))))\n"
" :formula (forall (x Int) (w Int) (or (= (f x) x) (forall (y Int) (z Int) (or (= (g x y) (g w z)) (forall (x1 Int) (= (f x1) (g x y)))))))\n"
")"
);
smtlib::benchmark* b = parser->get_benchmark();
smtlib::theory::expr_iterator it = b->begin_formulas();
smtlib::theory::expr_iterator end = b->end_formulas();
for (; it != end; ++it)
tst_instantiate(m, *it);
}