3
0
Fork 0
mirror of https://github.com/Z3Prover/z3 synced 2026-03-10 15:20:30 +00:00
z3/src/ast/simplifiers/dependent_expr_state.cpp
Nikolaj Bjorner b084821a0c wip - dependent expr simpliifer
- simplify iterator over current indices
- add more simplifiers used by asserted_formulas
- improve diagnostics printing
2022-11-30 13:41:40 +07:00

119 lines
3.2 KiB
C++

/*++
Copyright (c) 2022 Microsoft Corporation
Module Name:
dependent_expr_state.cpp
Author:
Nikolaj Bjorner (nbjorner) 2022-11-2.
--*/
#include "ast/simplifiers/dependent_expr_state.h"
#include "ast/recfun_decl_plugin.h"
#include "ast/for_each_ast.h"
unsigned dependent_expr_state::num_exprs() {
expr_fast_mark1 visited;
unsigned r = 0;
for (unsigned i = 0; i < qtail(); i++)
r += get_num_exprs((*this)[i].fml(), visited);
return r;
}
void dependent_expr_state::freeze(func_decl* f) {
if (m_frozen.is_marked(f) || !is_uninterp(f))
return;
m_frozen_trail.push_back(f);
m_frozen.mark(f, true);
}
void dependent_expr_state::freeze(expr* term) {
if (is_app(term))
freeze(to_app(term)->get_decl());
}
/**
* Freeze functions appearing as sub-expressions of 'e'.
* The only_as_array flag indicates whether to only freeze occurrences of as-array
* from elimination.
*/
void dependent_expr_state::freeze_terms(expr* e, bool only_as_array, ast_mark& visited) {
auto& m = m_frozen_trail.get_manager();
struct proc {
bool only_as_array;
array_util a;
dependent_expr_state& st;
proc(ast_manager& m, bool o, dependent_expr_state& d) :
only_as_array(o), a(m), st(d) {}
void operator()(func_decl* f) {
if (!only_as_array)
st.freeze(f);
if (a.is_as_array(f, f) && is_uninterp(f))
st.freeze(f);
}
void operator()(ast* s) {}
};
proc proc(m, only_as_array, *this);
for_each_ast(proc, visited, e);
}
/**
* Freeze all functions used in recursive definitions
*/
void dependent_expr_state::freeze_recfun() {
if (m_recfun_frozen)
return;
m_recfun_frozen = true;
auto& m = m_frozen_trail.get_manager();
recfun::util rec(m);
ast_mark visited;
for (func_decl* f : rec.get_rec_funs())
freeze_terms(rec.get_def(f).get_rhs(), false, visited);
}
/**
* The current qhead is to be updated to qtail.
* Before this update, freeze all functions appearing in formulas.
*/
void dependent_expr_state::freeze_prefix() {
ast_mark visited;
for (unsigned i = qhead(); i < qtail(); ++i)
freeze_terms((*this)[i].fml(), false, visited);
}
/**
* Freeze functions in the unprocessed suffix that appear in dependencies and in as-array.
*/
void dependent_expr_state::freeze_suffix() {
if (m_suffix_frozen)
return;
m_suffix_frozen = true;
auto& m = m_frozen_trail.get_manager();
freeze_recfun();
ast_mark visited;
ptr_vector<expr> es;
for (unsigned i = qhead(); i < qtail(); ++i) {
auto d = (*this)[i];
if (d.dep()) {
es.reset();
m.linearize(d.dep(), es);
for (expr* e : es)
freeze_terms(e, false, visited);
}
freeze_terms(d.fml(), true, visited);
}
}
bool dependent_expr_state::has_quantifiers() {
if (m_has_quantifiers != l_undef)
return m_has_quantifiers == l_true;
bool found = false;
for (unsigned i = qhead(); i < qtail(); ++i)
found |= ::has_quantifiers((*this)[i].fml());
m_has_quantifiers = found ? l_true : l_false;
return m_has_quantifiers == l_true;
}