mirror of
https://github.com/Z3Prover/z3
synced 2026-07-16 12:05:43 +00:00
94 lines
3.1 KiB
C++
94 lines
3.1 KiB
C++
#include "model/model.h"
|
|
#include "model/model_evaluator.h"
|
|
#include "model/func_interp.h"
|
|
#include "model/model_pp.h"
|
|
#include "ast/arith_decl_plugin.h"
|
|
#include "ast/reg_decl_plugins.h"
|
|
#include "ast/ast_pp.h"
|
|
#include <iostream>
|
|
|
|
void tst_model_evaluator() {
|
|
ast_manager m;
|
|
reg_decl_plugins(m);
|
|
arith_util a(m);
|
|
|
|
model mdl(m);
|
|
|
|
sort* sI = a.mk_int();
|
|
|
|
sort* dom[2] = { sI, m.mk_bool_sort() };
|
|
func_decl_ref f(m.mk_func_decl(symbol("f"), 2, dom, sI), m);
|
|
func_decl_ref g(m.mk_func_decl(symbol("g"), 2, dom, sI), m);
|
|
func_decl_ref h(m.mk_func_decl(symbol("h"), 2, dom, sI), m);
|
|
func_decl_ref F(m.mk_func_decl(symbol("F"), 2, dom, sI), m);
|
|
func_decl_ref G(m.mk_func_decl(symbol("G"), 2, dom, sI), m);
|
|
expr_ref vI0(m.mk_var(0, sI), m);
|
|
expr_ref vI1(m.mk_var(1, sI), m);
|
|
expr_ref vB0(m.mk_var(0, m.mk_bool_sort()), m);
|
|
expr_ref vB1(m.mk_var(1, m.mk_bool_sort()), m);
|
|
expr_ref vB2(m.mk_var(2, m.mk_bool_sort()), m);
|
|
expr* vI0p = vI0.get();
|
|
expr* vI1p = vI1.get();
|
|
expr* vB1p = vB1.get();
|
|
expr* vB2p = vB2.get();
|
|
|
|
expr_ref f01(m.mk_app(f, vI0p, vB1p), m);
|
|
expr_ref g01(m.mk_app(g, vI0p, vB1p), m);
|
|
expr_ref h01(m.mk_app(h, vI0p, vB1p), m);
|
|
func_interp* fi = alloc(func_interp, m, 2);
|
|
func_interp* gi = alloc(func_interp, m, 2);
|
|
func_interp* hi = alloc(func_interp, m, 2);
|
|
hi->set_else(m.mk_ite(vB1p, m.mk_app(f, vI0p, vB1p), m.mk_app(g, vI0p, vB1p)));
|
|
mdl.register_decl(h, hi);
|
|
|
|
|
|
model_evaluator eval(mdl);
|
|
|
|
expr_ref e(m), v(m);
|
|
|
|
{
|
|
symbol nI("N");
|
|
fi->set_else(m.mk_ite(m.mk_exists(1, &sI, &nI, a.mk_le(vI0p, m.mk_app(F, vI1p, vB2p))), vI0p, a.mk_int(1)));
|
|
gi->set_else(m.mk_ite(m.mk_exists(1, &sI, &nI, a.mk_le(vI0p, m.mk_app(G, vI1p, vB2p))), a.mk_int(2), vI0p));
|
|
mdl.register_decl(g, gi);
|
|
mdl.register_decl(f, fi);
|
|
model_pp(std::cout, mdl);
|
|
e = m.mk_app(h, vI0p, vB1p);
|
|
eval(e, v);
|
|
std::cout << e << " " << v << "\n";
|
|
}
|
|
|
|
{
|
|
fi->set_else(m.mk_app(F, vI0p, vB1p));
|
|
gi->set_else(m.mk_app(G, vI0p, vB1p));
|
|
model_pp(std::cout, mdl);
|
|
e = m.mk_app(h, vI0p, vB1p);
|
|
eval(e, v);
|
|
std::cout << e << " " << v << "\n";
|
|
}
|
|
|
|
{
|
|
func_interp fi2(m, 1);
|
|
expr_ref zero(a.mk_int(0), m);
|
|
expr_ref one(a.mk_int(1), m);
|
|
fi2.set_else(zero);
|
|
for (unsigned i = 0; i < 600; ++i) {
|
|
expr_ref arg(a.mk_int(rational(i)), m);
|
|
expr* args[1] = { arg.get() };
|
|
fi2.insert_entry(args, i == 599 ? one.get() : zero.get());
|
|
}
|
|
fi2.compress();
|
|
SASSERT(fi2.num_entries() == 1);
|
|
|
|
expr_ref removed_arg(a.mk_int(0), m);
|
|
[[maybe_unused]] expr* removed_args[1] = { removed_arg.get() };
|
|
SASSERT(fi2.get_entry(removed_args) == nullptr);
|
|
|
|
expr_ref kept_arg(a.mk_int(599), m);
|
|
expr* kept_args[1] = { kept_arg.get() };
|
|
[[maybe_unused]] func_entry* kept = fi2.get_entry(kept_args);
|
|
SASSERT(kept != nullptr);
|
|
SASSERT(kept->get_result() == one.get());
|
|
}
|
|
|
|
}
|