#include "ast/for_each_expr.h"
#include "ast/ast_ll_pp.h"
#include "ast/rewriter/macro_replacer.h"
#include "ast/simplifiers/model_reconstruction_trail.h"
#include "ast/simplifiers/dependent_expr_state.h"
#include "ast/converters/generic_model_converter.h"
#include "ast/ast_translation.h"
void model_reconstruction_trail::add_vars(expr* e, ast_mark& free_vars) {
for (expr* t : subterms::all(expr_ref(e, m))) {
if (is_app(t) && is_uninterp(t)) {
func_decl* f = to_app(t)->get_decl();
TRACE(simplifier, tout << "add var " << f->get_name() << "\n");
free_vars.mark(f, true);
if (m_model_vars.is_marked(f))
m_intersects_with_model = true;
}
}
}
void model_reconstruction_trail::replay(unsigned qhead, expr_ref_vector& assumptions, dependent_expr_state& st) {
if (m_trail.empty())
return;
if (qhead == st.qtail() && assumptions.empty())
return;
ast_mark free_vars;
m_intersects_with_model = false;
scoped_ptr<expr_replacer> rp = mk_default_expr_replacer(m, st.proofs_enabled());
for (unsigned i = qhead; i < st.qtail(); ++i)
add_vars(st[i], free_vars);
for (expr* a : assumptions)
add_vars(a, free_vars);
TRACE(simplifier,
tout << "intersects " << m_intersects_with_model << "\n";
for (unsigned i = qhead; i < st.qtail(); ++i)
tout << mk_bounded_pp(st[i].fml(), m) << "\n";
);
if (!m_intersects_with_model)
return;
for (auto& t : m_trail) {
TRACE(simplifier, tout << " active " << t->m_active << " hide " << t->is_hide() << " intersects " << t->intersects(free_vars) << " loose " << t->is_loose() << "\n");
if (!t->m_active)
continue;
if (t->is_hide())
continue;
if (!t->intersects(free_vars))
continue;
if (t->is_loose_subst() || (st.proofs_enabled() && !assumptions.empty() && t->is_subst())) {
for (auto const& [k, v] : t->m_subst->sub()) {
add_vars(v, free_vars);
proof* pr = nullptr;
expr_dependency* dep = nullptr;
if (st.proofs_enabled()) {
expr* value = nullptr;
if (!t->m_subst->find(k, value, pr, dep) || !pr)
throw default_exception("cannot replay a substitution without its proof");
}
st.add(dependent_expr(m, m.mk_eq(k, v), pr, dep));
}
m_trail_stack.push(value_trail(t->m_active));
t->m_active = false;
continue;
}
if (t->is_loose_constraint()) {
for (auto r : t->m_removed) {
add_vars(r, free_vars);
TRACE(simplifier, tout << "replay removed " << r << "\n");
st.add(r);
}
m_trail_stack.push(value_trail(t->m_active));
t->m_active = false;
continue;
}
bool all_const = true;
for (auto const& [d, def, dep] : t->m_defs)
all_const &= d->get_arity() == 0;
if (t->is_loose() && (!t->is_def() || !all_const || t->is_subst())) {
for (auto r : t->m_removed) {
add_vars(r, free_vars);
TRACE(simplifier, tout << "replay removed " << r << "\n");
st.add(r);
}
m_trail_stack.push(value_trail(t->m_active));
t->m_active = false;
continue;
}
if (t->is_def()) {
if (st.proofs_enabled())
throw default_exception("cannot replay a macro definition without its proof");
macro_replacer mrp(m);
for (auto const& [d, def, dep] : t->m_defs) {
app_ref head(m);
ptr_buffer<expr> args;
for (unsigned i = 0; i < d->get_arity(); ++i)
args.push_back(m.mk_var(i, d->get_domain(i)));
head = m.mk_app(d, args);
mrp.insert(head, def, dep);
TRACE(simplifier, tout << mk_pp(d, m) << " " << mk_pp(def,m) << " " << "\n");
dependent_expr de(m, def, nullptr, dep);
add_vars(de, free_vars);
}
for (unsigned i = qhead; i < st.qtail(); ++i) {
auto [f, p, dep1] = st[i]();
expr_ref g(m);
expr_dependency_ref dep2(m);
mrp(f, dep1, g, dep2);
CTRACE(simplifier, f != g, tout << "updated " << mk_pp(g, m) << "\n");
if (f != g)
st.update(i, dependent_expr(m, g, nullptr, dep2));
}
for (unsigned i = 0; i < assumptions.size(); ++i) {
expr* a = assumptions.get(i);
expr_ref g(m);
expr_dependency_ref dep(m);
mrp(a, nullptr, g, dep);
if (a != g)
assumptions[i] = g;
}
if (t->is_loose()) {
SASSERT(all_const);
SASSERT(!t->is_subst());
for (auto const& [d, def, dep] : t->m_defs)
st.add(dependent_expr(m, m.mk_eq(m.mk_const(d), def), nullptr, nullptr));
}
continue;
}
rp->set_substitution(t->m_subst.get());
ptr_vector<expr> dep_exprs;
expr_ref_vector trail(m);
for (unsigned i = qhead; i < st.qtail(); ++i) {
auto [f, p, dep1] = st[i]();
expr_ref g(m);
proof_ref rewrite_pr(m);
expr_dependency_ref dep2(m);
(*rp)(f, g, rewrite_pr, dep2);
if (dep1) {
dep_exprs.reset();
trail.reset();
m.linearize(dep1, dep_exprs);
for (auto*& d : dep_exprs) {
auto [h, dep3] = rp->replace_with_dep(d);
if (h != d) {
trail.push_back(h);
d = h;
dep2 = m.mk_join(dep2, dep3);
}
}
if (!trail.empty())
dep1 = m.mk_join(dep_exprs.size(), dep_exprs.data());
}
proof_ref pr(p, m);
if (st.proofs_enabled() && f != g) {
if (!pr || !rewrite_pr)
throw default_exception("cannot replay an assertion substitution without its proof");
pr = m.mk_modus_ponens(pr, rewrite_pr);
}
dependent_expr d(m, g, pr, m.mk_join(dep1, dep2));
CTRACE(simplifier, f != g, tout << "updated " << mk_pp(g, m) << "\n");
add_vars(d, free_vars);
st.update(i, d);
}
for (unsigned i = 0; i < assumptions.size(); ++i) {
expr* a = assumptions.get(i);
auto [g, dep2] = rp->replace_with_dep(a);
if (a != g)
assumptions[i] = g;
}
}
TRACE(simplifier, st.display(tout));
}
model_converter_ref model_reconstruction_trail::get_model_converter() {
generic_model_converter_ref mc = alloc(generic_model_converter, m, "dependent-expr-model");
append(*mc);
return model_converter_ref(mc.get());
}
void model_reconstruction_trail::append(generic_model_converter& mc) {
for (auto* t : m_trail) {
if (!t->m_active)
continue;
else if (t->is_hide())
mc.hide(t->m_decl);
else if (t->is_def())
for (auto const& [f, def, dep] : t->m_defs)
mc.add(f, def);
else {
for (auto const& [v, def] : t->m_subst->sub())
mc.add(v, def);
}
}
TRACE(simplifier, display(tout); mc.display(tout));
}
std::ostream& model_reconstruction_trail::display(std::ostream& out) const {
for (auto* t : m_trail) {
if (!t->m_active)
continue;
else if (t->is_hide())
out << "hide " << t->m_decl->get_name() << "\n";
else if (t->is_def()) {
for (auto const& [f, def, dep] : t->m_defs)
out << "def: " << f->get_name() << " <- " << mk_pp(def, m) << "\n";
}
else {
for (auto const& [v, def] : t->m_subst->sub())
out << "sub: " << mk_pp(v, m) << " -> " << mk_pp(def, m) << "\n";
}
for (auto const& d : t->m_removed)
out << "rm: " << d << "\n";
}
return out;
}
void model_reconstruction_trail::translate(model_reconstruction_trail const& src, ast_translation& tr) {
SASSERT(m_trail.empty());
expr_dependency_translation dep_tr(tr);
for (entry* e : src.m_trail) {
vector<dependent_expr> removed;
for (dependent_expr const& d : e->m_removed)
removed.push_back(dependent_expr(tr, d));
if (e->is_hide()) {
hide(tr(e->m_decl.get()));
}
else if (e->is_def()) {
vector<std::tuple<func_decl_ref, expr_ref, expr_dependency_ref>> defs;
for (auto const& [f, def, dep] : e->m_defs)
defs.push_back({
func_decl_ref(tr(f.get()), m),
expr_ref(tr(def.get()), m),
expr_dependency_ref(dep_tr(dep.get()), m)
});
push(defs, removed);
}
else {
auto* subst = alloc(expr_substitution, m, e->m_subst->unsat_core_enabled(), e->m_subst->proofs_enabled());
for (auto const& [k, v] : e->m_subst->sub()) {
proof* pr = nullptr;
expr_dependency* dep = nullptr;
expr* value = nullptr;
VERIFY(e->m_subst->find(k, value, pr, dep));
subst->insert(tr(k), tr(value), tr(pr), dep_tr(dep));
}
push(subst, removed, e->is_loose_constraint());
}
m_trail.back()->m_active = e->m_active;
}
}