3
0
Fork 0
mirror of https://github.com/Z3Prover/z3 synced 2025-04-13 20:38:43 +00:00

working on tab Horn solver

Signed-off-by: Nikolaj Bjorner <nbjorner@microsoft.com>
This commit is contained in:
Nikolaj Bjorner 2013-01-17 18:03:34 -08:00
parent 79dafcea81
commit 8daf100c65
5 changed files with 214 additions and 26 deletions

View file

@ -49,6 +49,10 @@ bool matcher::operator()(expr * e1, expr * e2, substitution & s) {
if (is_var(p.second))
return false;
if (!is_app(p.first))
return false;
if (!is_app(p.second))
return false;
app * n1 = to_app(p.first);
app * n2 = to_app(p.second);

View file

@ -32,7 +32,7 @@ class matcher {
ast_manager & m_manager;
substitution * m_subst;
cache m_cache;
// cache m_cache;
svector<expr_pair> m_todo;
void reset();

View file

@ -518,7 +518,7 @@ namespace datalog {
if (m.is_false(itail.get())) {
//the tail member is never true, so we may delete the rule
TRACE("dl", r->display(m_context, tout << "rule in infeasible\n"););
TRACE("dl", r->display(m_context, tout << "rule is infeasible\n"););
return false;
}
if (!m.is_true(itail.get())) {

View file

@ -22,6 +22,8 @@ Revision History:
#include "dl_rule_set.h"
#include "dl_context.h"
#include "dl_mk_rule_inliner.h"
#include "smt_kernel.h"
#include "matcher.h"
namespace datalog {
@ -42,6 +44,129 @@ namespace datalog {
}
};
// subsumption index structure.
class tab_index {
ast_manager& m;
rule_manager& rm;
context& m_ctx;
app_ref_vector m_preds;
expr_ref m_precond;
rule_ref_vector m_rules;
svector<unsigned> m_num_vars;
unsigned m_index;
matcher m_matcher;
substitution m_subst;
public:
tab_index(ast_manager& m, rule_manager& rm, context& ctx):
m(m),
rm(rm),
m_ctx(ctx),
m_preds(m),
m_precond(m),
m_rules(rm),
m_matcher(m),
m_subst(m) {}
void insert(rule* r) {
m_rules.push_back(r);
m_num_vars.push_back(1+rm.get_var_counter().get_max_var(*r));
}
void setup(rule const& r) {
m_preds.reset();
m_index = 0;
expr_ref_vector fmls(m);
expr_ref_vector vars(m);
expr_ref fml(m);
ptr_vector<sort> sorts;
r.get_vars(sorts);
unsigned utsz = r.get_uninterpreted_tail_size();
unsigned tsz = r.get_tail_size();
var_subst vs(m, false);
for (unsigned i = 0; i < sorts.size(); ++i) {
if (!sorts[i]) {
sorts[i] = m.mk_bool_sort();
}
vars.push_back(m.mk_const(symbol(i), sorts[i]));
}
for (unsigned i = 0; i < utsz; ++i) {
fml = r.get_tail(i);
vs(fml, vars.size(), vars.c_ptr(), fml);
m_preds.push_back(to_app(fml));
}
for (unsigned i = utsz; i < tsz; ++i) {
fml = r.get_tail(i);
vs(fml, vars.size(), vars.c_ptr(), fml);
fmls.push_back(fml);
}
m_precond = m.mk_and(fmls.size(), fmls.c_ptr());
IF_VERBOSE(1, r.display(m_ctx, verbose_stream() << "setup-match\n"););
}
// extract pre_cond => post_cond validation obligation from match.
bool next_match(expr_ref& pre_cond, expr_ref& post_cond) {
for (; m_index < m_rules.size(); ++m_index) {
if (try_match(pre_cond, post_cond)) {
return true;
}
}
return false;
}
private:
//
// check that each predicate in r is matched by some predicate in premise.
// for now: skip multiple matches within the same rule (incomplete).
//
bool try_match(expr_ref& pre_cond, expr_ref& post_cond) {
rule const& r = *m_rules[m_index];
unsigned num_vars = m_num_vars[m_index];
m_subst.reset();
m_subst.reserve(2, num_vars);
unsigned deltas[2] = {0, 0};
expr_ref_vector fmls(m);
expr_ref q(m);
unsigned utsz = r.get_uninterpreted_tail_size();
unsigned tsz = r.get_tail_size();
IF_VERBOSE(1, r.display(m_ctx, verbose_stream() << "try-match\n"););
for (unsigned i = 0; i < utsz; ++i) {
m_subst.push_scope();
if (!try_match(r.get_tail(i))) {
return false;
}
}
for (unsigned i = utsz; i < tsz; ++i) {
app* p = r.get_tail(i);
m_subst.apply(2, deltas, expr_offset(p, 0), q);
fmls.push_back(q);
}
pre_cond = m_precond;
post_cond = m.mk_and(fmls.size(), fmls.c_ptr());
IF_VERBOSE(1, verbose_stream() << "match: " << mk_pp(post_cond, m) << "\n";);
// TBD:
// - existential closure of post_cond
// - eliminate variables from post_cond
return true;
}
bool try_match(expr* q) {
for (unsigned i = 0; i < m_preds.size(); ++i) {
if (m_matcher(q, m_preds[i].get(), m_subst)) {
return true;
}
else {
// undo effect of failed match attempt.
m_subst.pop_scope();
m_subst.push_scope();
}
}
return false;
}
};
enum tab_instruction {
SELECT_RULE,
SELECT_PREDICATE,
@ -54,13 +179,13 @@ namespace datalog {
std::ostream& operator<<(std::ostream& out, tab_instruction i) {
switch(i) {
case SELECT_RULE: return out << "select-rule";
case SELECT_RULE: return out << "select-rule";
case SELECT_PREDICATE: return out << "select-predicate";
case BACKTRACK: return out << "backtrack";
case NEXT_RULE: return out << "next-rule";
case SATISFIABLE: return out << "sat";
case UNSATISFIABLE: return out << "unsat";
case CANCEL: return out << "cancel";
case BACKTRACK: return out << "backtrack";
case NEXT_RULE: return out << "next-rule";
case SATISFIABLE: return out << "sat";
case UNSATISFIABLE: return out << "unsat";
case CANCEL: return out << "cancel";
}
return out << "unmatched instruction";
}
@ -77,6 +202,9 @@ namespace datalog {
context& m_ctx;
ast_manager& m;
rule_manager& rm;
tab_index m_subsumption_index;
smt_params m_fparams;
smt::kernel m_solver;
rule_unifier m_unifier;
rule_set m_rules;
trail_stack<imp> m_trail;
@ -92,6 +220,8 @@ namespace datalog {
m_ctx(ctx),
m(ctx.get_manager()),
rm(ctx.get_rule_manager()),
m_subsumption_index(m, rm, ctx),
m_solver(m, m_fparams),
m_unifier(ctx),
m_rules(ctx),
m_trail(*this),
@ -101,7 +231,11 @@ namespace datalog {
m_predicate_index(0),
m_rule_index(0),
m_cancel(false)
{}
{
// m_fparams.m_relevancy_lvl = 0;
m_fparams.m_mbqi = false;
m_fparams.m_soft_timeout = 1000;
}
~imp() {}
@ -148,13 +282,15 @@ namespace datalog {
}
else {
m_instruction = SELECT_RULE;
m_predicate_index = 0; // TBD replace by better selection function.
m_predicate_index = 0; // TBD replace by better selection function.
m_rule_index = 0;
IF_VERBOSE(1, verbose_stream() << mk_pp(m_query->get_tail(m_predicate_index), m) << "\n";);
}
}
void apply_rule(rule const& r) {
m_rule_index++;
IF_VERBOSE(1, r.display(m_ctx, verbose_stream()););
bool can_unify = m_unifier.unify_rules(*m_query, m_predicate_index, r);
if (can_unify) {
m_stats.m_num_unfold++;
@ -162,18 +298,24 @@ namespace datalog {
m_trail.push(value_trail<imp,unsigned>(m_rule_index));
m_trail.push(value_trail<imp,unsigned>(m_predicate_index));
rule_ref new_query(rm);
m_unifier.apply(*m_query, m_predicate_index, r, new_query);
m_trail.push(restore_rule<imp>(m_query_trail, m_query));
m_query = new_query;
TRACE("dl", m_query->display(m_ctx, tout););
if (l_false == query_is_sat()) {
m_instruction = BACKTRACK;
}
else if (l_true == query_is_subsumed()) {
NOT_IMPLEMENTED_YET();
bool is_feasible = m_unifier.apply(*m_query, m_predicate_index, r, new_query);
if (is_feasible) {
TRACE("dl", m_query->display(m_ctx, tout););
if (l_false == query_is_sat(*new_query.get())) {
m_instruction = BACKTRACK;
}
else if (l_true == query_is_subsumed(*new_query.get())) {
m_instruction = BACKTRACK;
}
else {
m_subsumption_index.insert(m_query.get());
m_trail.push(restore_rule<imp>(m_query_trail, m_query));
m_query = new_query;
m_instruction = SELECT_PREDICATE;
}
}
else {
m_instruction = SELECT_PREDICATE;
m_instruction = BACKTRACK;
}
}
else {
@ -212,7 +354,7 @@ namespace datalog {
lbool run() {
m_instruction = SELECT_PREDICATE;
while (true) {
IF_VERBOSE(1, verbose_stream() << "run " << m_instruction << "\n";);
IF_VERBOSE(1, verbose_stream() << "run " << m_trail.get_num_scopes() << " " << m_instruction << "\n";);
if (m_cancel) {
cleanup();
return l_undef;
@ -241,16 +383,55 @@ namespace datalog {
}
}
lbool query_is_sat() {
lbool query_is_sat(rule const& query) {
expr_ref_vector fmls(m);
expr_ref fml(m);
unsigned utsz = query.get_uninterpreted_tail_size();
unsigned tsz = query.get_tail_size();
for (unsigned i = utsz; i < tsz; ++i) {
fmls.push_back(query.get_tail(i));
}
ptr_vector<sort> sorts;
svector<symbol> names;
query.get_vars(sorts);
sorts.reverse();
for (unsigned i = 0; i < sorts.size(); ++i) {
if (!sorts[i]) {
sorts[i] = m.mk_bool_sort();
}
names.push_back(symbol(i));
}
fml = m.mk_and(fmls.size(), fmls.c_ptr());
fml = m.mk_exists(sorts.size(), sorts.c_ptr(), names.c_ptr(), fml);
m_solver.push();
m_solver.assert_expr(fml);
lbool is_sat = m_solver.check();
m_solver.pop(1);
return l_undef;
TRACE("dl", tout << is_sat << ":\n" << mk_pp(fml, m) << "\n";);
return is_sat;
}
lbool query_is_subsumed() {
return l_undef;
lbool query_is_subsumed(rule const& query) {
lbool is_subsumed = l_false;
m_subsumption_index.setup(query);
expr_ref precond(m), postcond(m);
while (m_subsumption_index.next_match(precond, postcond)) {
if (is_ground(postcond)) {
postcond = m.mk_not(postcond);
m_solver.push();
m_solver.assert_expr(precond);
m_solver.assert_expr(postcond);
lbool is_sat = m_solver.check();
m_solver.pop(1);
if (is_sat == l_false) {
return l_true;
}
}
}
return is_subsumed;
}
};
tab::tab(context& ctx):

View file

@ -19,6 +19,8 @@ Revision History:
#ifdef _WINDOWS
#include"matcher.h"
#include"ast_pp.h"
#include "reg_decl_plugins.h"
void tst_match(ast_manager & m, app * t, app * i) {
substitution s(m);
@ -72,6 +74,7 @@ void tst_match(ast_manager & m, app * t, app * i) {
void tst1() {
ast_manager m;
reg_decl_plugins(m);
sort_ref s( m.mk_sort(symbol("S")), m);
func_decl_ref g( m.mk_func_decl(symbol("g"), s, s), m);
func_decl_ref h( m.mk_func_decl(symbol("h"), s, s), m);