mirror of
https://github.com/Z3Prover/z3
synced 2025-11-04 05:19:11 +00:00
b52fd8d954
plugin setting allows adding equality saturation within the E-graph propagation without involving externalizing theory solver dispatch. It makes equality saturation independent of SAT integration. Add a special relation operator to support ad-hoc AC symbols.
131 lines
5.6 KiB
C++
131 lines
5.6 KiB
C++
/*++
|
|
Copyright (c) 2015 Microsoft Corporation
|
|
|
|
Module Name:
|
|
|
|
special_relations_decl_plugin.h
|
|
|
|
Abstract:
|
|
|
|
<abstract>
|
|
|
|
Author:
|
|
|
|
Nikolaj Bjorner (nbjorner) 2015-15-9.
|
|
Ashutosh Gupta 2016
|
|
|
|
Revision History:
|
|
|
|
2023-11-27: Added ac-op for E-graph plugin
|
|
|
|
--*/
|
|
#pragma once
|
|
|
|
#include "ast/ast.h"
|
|
|
|
|
|
enum special_relations_op_kind {
|
|
OP_SPECIAL_RELATION_LO,
|
|
OP_SPECIAL_RELATION_PO,
|
|
OP_SPECIAL_RELATION_PLO,
|
|
OP_SPECIAL_RELATION_TO,
|
|
OP_SPECIAL_RELATION_TC,
|
|
OP_SPECIAL_RELATION_AC,
|
|
LAST_SPECIAL_RELATIONS_OP
|
|
};
|
|
|
|
class special_relations_decl_plugin : public decl_plugin {
|
|
symbol m_lo;
|
|
symbol m_po;
|
|
symbol m_plo;
|
|
symbol m_to;
|
|
symbol m_tc;
|
|
symbol m_ac;
|
|
bool m_has_special_relation = false;
|
|
public:
|
|
special_relations_decl_plugin();
|
|
|
|
decl_plugin * mk_fresh() override {
|
|
return alloc(special_relations_decl_plugin);
|
|
}
|
|
|
|
func_decl * mk_func_decl(decl_kind k, unsigned num_parameters, parameter const * parameters,
|
|
unsigned arity, sort * const * domain, sort * range) override;
|
|
|
|
void get_op_names(svector<builtin_name> & op_names, symbol const & logic) override;
|
|
|
|
sort * mk_sort(decl_kind k, unsigned num_parameters, parameter const * parameters) override { return nullptr; }
|
|
|
|
bool has_special_relation() const { return m_has_special_relation; }
|
|
};
|
|
|
|
enum sr_property {
|
|
sr_none = 0x00,
|
|
sr_transitive = 0x01, // Rxy & Ryz -> Rxz
|
|
sr_reflexive = 0x02, // Rxx
|
|
sr_antisymmetric = 0x04, // Rxy & Ryx -> x = y
|
|
sr_lefttree = 0x08, // Ryx & Rzx -> Ryz | Rzy
|
|
sr_righttree = 0x10, // Rxy & Rxz -> Ryx | Rzy
|
|
sr_total = 0x20, // Rxy | Ryx
|
|
sr_po = 0x01 | 0x02 | 0x04, // partial order
|
|
sr_to = 0x01 | 0x02 | 0x04 | 0x10, // right-tree
|
|
sr_plo = 0x01 | 0x02 | 0x04 | 0x08 | 0x10, // piecewise linear order
|
|
sr_lo = 0x01 | 0x02 | 0x04 | 0x20, // linear order
|
|
sr_tc = 0x40, // transitive closure of relation
|
|
};
|
|
|
|
class special_relations_util {
|
|
ast_manager& m;
|
|
mutable family_id m_fid;
|
|
func_decl* mk_rel_decl(func_decl* f, decl_kind k) {
|
|
SASSERT(f);
|
|
parameter p(f); SASSERT(f->get_arity() == 2);
|
|
return m.mk_func_decl(fid(), k, 1, &p, 2, f->get_domain(), f->get_range());
|
|
}
|
|
family_id fid() const {
|
|
if (null_family_id == m_fid)
|
|
m_fid = m.get_family_id("specrels");
|
|
return m_fid;
|
|
}
|
|
public:
|
|
special_relations_util(ast_manager& m) : m(m), m_fid(null_family_id) { }
|
|
|
|
family_id get_family_id() const { return fid(); }
|
|
|
|
bool has_special_relation() const { return static_cast<special_relations_decl_plugin*>(m.get_plugin(m.mk_family_id("specrels")))->has_special_relation(); }
|
|
|
|
bool is_special_relation(func_decl* f) const { return f->get_family_id() == fid(); }
|
|
bool is_special_relation(expr* e) const { return is_app(e) && is_special_relation(to_app(e)->get_decl()); }
|
|
sr_property get_property(func_decl* f) const;
|
|
sr_property get_property(app* e) const { return get_property(e->get_decl()); }
|
|
func_decl* get_relation(func_decl* f) const { SASSERT(is_special_relation(f)); return to_func_decl(f->get_parameter(0).get_ast()); }
|
|
func_decl* get_relation(expr* e) const { SASSERT(is_special_relation(e)); return to_func_decl(to_app(e)->get_parameter(0).get_ast()); }
|
|
|
|
func_decl* mk_to_decl(func_decl* f) { return mk_rel_decl(f, OP_SPECIAL_RELATION_TO); }
|
|
func_decl* mk_po_decl(func_decl* f) { return mk_rel_decl(f, OP_SPECIAL_RELATION_PO); }
|
|
func_decl* mk_plo_decl(func_decl* f) { return mk_rel_decl(f, OP_SPECIAL_RELATION_PLO); }
|
|
func_decl* mk_lo_decl(func_decl* f) { return mk_rel_decl(f, OP_SPECIAL_RELATION_LO); }
|
|
func_decl* mk_tc_decl(func_decl* f) { return mk_rel_decl(f, OP_SPECIAL_RELATION_TC); }
|
|
|
|
bool is_lo(expr const * e) const { return is_app_of(e, fid(), OP_SPECIAL_RELATION_LO); }
|
|
bool is_po(expr const * e) const { return is_app_of(e, fid(), OP_SPECIAL_RELATION_PO); }
|
|
bool is_plo(expr const * e) const { return is_app_of(e, fid(), OP_SPECIAL_RELATION_PLO); }
|
|
bool is_to(expr const * e) const { return is_app_of(e, fid(), OP_SPECIAL_RELATION_TO); }
|
|
bool is_tc(expr const * e) const { return is_app_of(e, fid(), OP_SPECIAL_RELATION_TC); }
|
|
bool is_ac(expr const* e) const { return is_app_of(e, fid(), OP_SPECIAL_RELATION_AC); }
|
|
|
|
bool is_lo(func_decl const * e) const { return is_decl_of(e, fid(), OP_SPECIAL_RELATION_LO); }
|
|
bool is_po(func_decl const * e) const { return is_decl_of(e, fid(), OP_SPECIAL_RELATION_PO); }
|
|
bool is_plo(func_decl const * e) const { return is_decl_of(e, fid(), OP_SPECIAL_RELATION_PLO); }
|
|
bool is_to(func_decl const * e) const { return is_decl_of(e, fid(), OP_SPECIAL_RELATION_TO); }
|
|
bool is_tc(func_decl const * e) const { return is_decl_of(e, fid(), OP_SPECIAL_RELATION_TC); }
|
|
bool is_ac(func_decl const* e) const { return is_decl_of(e, fid(), OP_SPECIAL_RELATION_AC); }
|
|
|
|
app * mk_lo (expr * arg1, expr * arg2) { return m.mk_app( fid(), OP_SPECIAL_RELATION_LO, arg1, arg2); }
|
|
app * mk_po (expr * arg1, expr * arg2) { return m.mk_app( fid(), OP_SPECIAL_RELATION_PO, arg1, arg2); }
|
|
app * mk_plo(expr * arg1, expr * arg2) { return m.mk_app( fid(), OP_SPECIAL_RELATION_PLO, arg1, arg2); }
|
|
app * mk_to (expr * arg1, expr * arg2) { return m.mk_app( fid(), OP_SPECIAL_RELATION_TO, arg1, arg2); }
|
|
|
|
};
|
|
|
|
|