mirror of
https://github.com/Z3Prover/z3
synced 2025-04-05 17:14:07 +00:00
88 lines
2.7 KiB
C++
88 lines
2.7 KiB
C++
/*++
|
|
Copyright (c) 2019 Microsoft Corporation
|
|
|
|
Module Name:
|
|
|
|
hoist_rewriter.h
|
|
|
|
Abstract:
|
|
|
|
Hoist predicates over disjunctions
|
|
|
|
Author:
|
|
|
|
Nikolaj Bjorner (nbjorner) 2019-2-4
|
|
|
|
Notes:
|
|
|
|
--*/
|
|
#pragma once
|
|
|
|
#include "ast/ast.h"
|
|
#include "ast/rewriter/rewriter.h"
|
|
#include "ast/rewriter/expr_safe_replace.h"
|
|
#include "util/params.h"
|
|
#include "util/union_find.h"
|
|
#include "util/obj_hashtable.h"
|
|
|
|
class bool_rewriter;
|
|
|
|
class hoist_rewriter {
|
|
ast_manager & m;
|
|
expr_ref_vector m_args1, m_args2, m_refs;
|
|
obj_hashtable<expr> m_preds1, m_preds2;
|
|
basic_union_find m_uf1, m_uf2, m_uf0;
|
|
ptr_vector<expr> m_es;
|
|
svector<std::pair<expr*,expr*>> m_eqs;
|
|
u_map<expr*> m_roots;
|
|
expr_safe_replace m_subst;
|
|
obj_map<expr, unsigned> m_expr2var;
|
|
ptr_vector<expr> m_var2expr;
|
|
expr_mark m_mark;
|
|
bool m_elim_and = false;
|
|
|
|
bool is_and(expr* e, expr_ref_vector* args);
|
|
expr_ref mk_and(expr_ref_vector const& args);
|
|
expr_ref mk_or(expr_ref_vector const& args);
|
|
|
|
bool is_var(expr* e) { return m_expr2var.contains(e); }
|
|
expr* mk_expr(unsigned v) { return m_var2expr[v]; }
|
|
unsigned mk_var(expr* e);
|
|
|
|
void reset(basic_union_find& uf);
|
|
|
|
expr_ref hoist_predicates(obj_hashtable<expr> const& p, unsigned num_args, expr* const* args);
|
|
|
|
|
|
public:
|
|
hoist_rewriter(ast_manager & m, params_ref const & p = params_ref());
|
|
family_id get_fid() const { return m.get_basic_family_id(); }
|
|
bool is_eq(expr * t) const { return m.is_eq(t); }
|
|
void updt_params(params_ref const & p) {}
|
|
static void get_param_descrs(param_descrs & r) {}
|
|
br_status mk_app_core(func_decl * f, unsigned num_args, expr * const * args, expr_ref & result);
|
|
br_status mk_or(unsigned num_args, expr * const * args, expr_ref & result);
|
|
void set_elim_and(bool b) { m_elim_and = b; }
|
|
};
|
|
|
|
struct hoist_rewriter_cfg : public default_rewriter_cfg {
|
|
hoist_rewriter m_r;
|
|
bool rewrite_patterns() const { return false; }
|
|
br_status reduce_app(func_decl * f, unsigned num, expr * const * args, expr_ref & result, proof_ref & result_pr) {
|
|
result_pr = nullptr;
|
|
if (f->get_family_id() != m_r.get_fid())
|
|
return BR_FAILED;
|
|
return m_r.mk_app_core(f, num, args, result);
|
|
}
|
|
hoist_rewriter_cfg(ast_manager & m, params_ref const & p):m_r(m, p) {}
|
|
};
|
|
|
|
class hoist_rewriter_star : public rewriter_tpl<hoist_rewriter_cfg> {
|
|
hoist_rewriter_cfg m_cfg;
|
|
public:
|
|
hoist_rewriter_star(ast_manager & m, params_ref const & p = params_ref()):
|
|
rewriter_tpl<hoist_rewriter_cfg>(m, false, m_cfg),
|
|
m_cfg(m, p) {}
|
|
};
|
|
|