mirror of
https://github.com/Z3Prover/z3
synced 2025-04-22 16:45:31 +00:00
adding drat forward checking
Signed-off-by: Nikolaj Bjorner <nbjorner@microsoft.com>
This commit is contained in:
parent
61341b8879
commit
5f70e4823d
3 changed files with 304 additions and 39 deletions
|
@ -26,7 +26,8 @@ Notes:
|
|||
namespace sat {
|
||||
drat::drat(solver& s):
|
||||
s(s),
|
||||
m_out(0)
|
||||
m_out(0),
|
||||
m_inconsistent(false)
|
||||
{
|
||||
if (s.m_config.m_drat && s.m_config.m_drat_file != symbol()) {
|
||||
m_out = alloc(std::ofstream, s.m_config.m_drat_file.str().c_str());
|
||||
|
@ -35,6 +36,15 @@ namespace sat {
|
|||
|
||||
drat::~drat() {
|
||||
dealloc(m_out);
|
||||
for (unsigned i = 0; i < m_proof.size(); ++i) {
|
||||
clause* c = m_proof[i];
|
||||
if (m_status[i] == status::deleted || m_status[i] == status::external) {
|
||||
s.m_cls_allocator.del_clause(c);
|
||||
}
|
||||
else if (c && c->size() == 2) {
|
||||
s.m_cls_allocator.del_clause(c);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
std::ostream& operator<<(std::ostream& out, drat::status st) {
|
||||
|
@ -46,15 +56,15 @@ namespace sat {
|
|||
}
|
||||
}
|
||||
|
||||
void drat::dump(unsigned sz, literal const* c, status st) {
|
||||
if (is_cleaned(sz, c)) return;
|
||||
void drat::dump(unsigned n, literal const* c, status st) {
|
||||
if (is_cleaned(n, c)) return;
|
||||
switch (st) {
|
||||
case status::asserted: return;
|
||||
case status::learned: break;
|
||||
case status::deleted: (*m_out) << "d "; break;
|
||||
}
|
||||
literal last = null_literal;
|
||||
for (unsigned i = 0; i < sz; ++i) {
|
||||
for (unsigned i = 0; i < n; ++i) {
|
||||
if (c[i] != last) {
|
||||
(*m_out) << c[i] << " ";
|
||||
last = c[i];
|
||||
|
@ -72,22 +82,251 @@ namespace sat {
|
|||
return false;
|
||||
}
|
||||
|
||||
void drat::append(unsigned n, literal const* c, status st) {
|
||||
if (is_cleaned(n, c)) return;
|
||||
m_status.push_back(st);
|
||||
m_proof.push_back(0); // TBD
|
||||
|
||||
|
||||
std::cout << st << " ";
|
||||
void drat::trace(std::ostream& out, unsigned n, literal const* c, status st) {
|
||||
out << st << " ";
|
||||
literal last = null_literal;
|
||||
for (unsigned i = 0; i < n; ++i) {
|
||||
declare(c[i]);
|
||||
if (c[i] != last) {
|
||||
std::cout << c[i] << " ";
|
||||
out << c[i] << " ";
|
||||
last = c[i];
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
std::cout << "\n";
|
||||
out << "\n";
|
||||
}
|
||||
|
||||
void drat::append(literal l, status st) {
|
||||
trace(std::cout, 1, &l, st);
|
||||
if (st == status::learned) {
|
||||
verify(1, &l);
|
||||
}
|
||||
if (st == status::deleted) {
|
||||
return;
|
||||
}
|
||||
assign_propagate(l);
|
||||
}
|
||||
|
||||
void drat::append(literal l1, literal l2, status st) {
|
||||
literal lits[2] = { l1, l2 };
|
||||
trace(std::cout, 2, lits, st);
|
||||
if (st == status::deleted) {
|
||||
// noop
|
||||
// don't record binary as deleted.
|
||||
}
|
||||
else {
|
||||
if (st == status::learned) {
|
||||
verify(2, lits);
|
||||
}
|
||||
clause* c = s.m_cls_allocator.mk_clause(2, lits, st == status::learned);
|
||||
m_proof.push_back(c);
|
||||
m_status.push_back(st);
|
||||
m_watches[(~l1).index()].push_back(c);
|
||||
m_watches[(~l2).index()].push_back(c);
|
||||
|
||||
if (value(l1) == l_false && value(l2) == l_false) {
|
||||
m_inconsistent = true;
|
||||
}
|
||||
else if (value(l1) == l_false) {
|
||||
assign_propagate(l2);
|
||||
}
|
||||
else if (value(l2) == l_false) {
|
||||
assign_propagate(l1);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void drat::append(clause& c, status st) {
|
||||
unsigned n = c.size();
|
||||
if (is_cleaned(n, c.begin())) return;
|
||||
trace(std::cout, n, c.begin(), st);
|
||||
|
||||
if (st == status::learned) {
|
||||
verify(n, c.begin());
|
||||
}
|
||||
|
||||
m_status.push_back(st);
|
||||
m_proof.push_back(&c);
|
||||
if (st == status::deleted) {
|
||||
del_watch(c, c[0]);
|
||||
del_watch(c, c[1]);
|
||||
return;
|
||||
}
|
||||
literal l1 = null_literal, l2 = null_literal;
|
||||
for (unsigned i = 0; i < n; ++i) {
|
||||
if (value(c[i]) != l_false) {
|
||||
if (l1 == null_literal) {
|
||||
l1 = c[i];
|
||||
}
|
||||
else {
|
||||
l2 = c[i];
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
if (l2 == null_literal && l1 != null_literal) {
|
||||
assign_propagate(l1);
|
||||
}
|
||||
else if (l1 == null_literal) {
|
||||
m_inconsistent = true;
|
||||
}
|
||||
else {
|
||||
m_watches[(~l1).index()].push_back(&c);
|
||||
m_watches[(~l2).index()].push_back(&c);
|
||||
}
|
||||
}
|
||||
|
||||
void drat::del_watch(clause& c, literal l) {
|
||||
watch& w = m_watches[(~l).index()];
|
||||
for (unsigned i = 0; i < w.size(); ++i) {
|
||||
if (w[i] == &c) {
|
||||
w[i] = w.back();
|
||||
w.pop_back();
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
void drat::declare(literal l) {
|
||||
unsigned n = static_cast<unsigned>(l.var());
|
||||
while (m_assignment.size() <= n) {
|
||||
m_assignment.push_back(l_undef);
|
||||
m_watches.push_back(watch());
|
||||
m_watches.push_back(watch());
|
||||
}
|
||||
}
|
||||
|
||||
void drat::verify(unsigned n, literal const* c) {
|
||||
if (m_inconsistent) {
|
||||
std::cout << "inconsistent\n";
|
||||
return;
|
||||
}
|
||||
unsigned num_units = m_units.size();
|
||||
for (unsigned i = 0; !m_inconsistent && i < n; ++i) {
|
||||
assign_propagate(~c[i]);
|
||||
}
|
||||
for (unsigned i = num_units; i < m_units.size(); ++i) {
|
||||
m_assignment[m_units[i].var()] = l_undef;
|
||||
}
|
||||
m_units.resize(num_units);
|
||||
bool ok = m_inconsistent;
|
||||
m_inconsistent = false;
|
||||
if (ok) {
|
||||
std::cout << "Verified\n";
|
||||
}
|
||||
else {
|
||||
std::cout << "Verification failed\n";
|
||||
display(std::cout);
|
||||
}
|
||||
}
|
||||
|
||||
void drat::display(std::ostream& out) const {
|
||||
out << "units: " << m_units << "\n";
|
||||
#if 0
|
||||
for (unsigned i = 0; i < m_assignment.size(); ++i) {
|
||||
lbool v = value(literal(i, false));
|
||||
if (v != l_undef) std::cout << i << ": " << v << "\n";
|
||||
}
|
||||
#endif
|
||||
for (unsigned i = 0; i < m_proof.size(); ++i) {
|
||||
clause* c = m_proof[i];
|
||||
if (m_status[i] != status::deleted && c) {
|
||||
out << i << ": " << *c << "\n";
|
||||
}
|
||||
}
|
||||
#if 0
|
||||
for (unsigned i = 0; i < m_assignment.size(); ++i) {
|
||||
watch const& w1 = m_watches[2*i];
|
||||
watch const& w2 = m_watches[2*i + 1];
|
||||
if (!w1.empty()) {
|
||||
out << i << " |-> ";
|
||||
for (unsigned i = 0; i < w1.size(); ++i) out << w1[i] << " ";
|
||||
out << "\n";
|
||||
}
|
||||
if (!w2.empty()) {
|
||||
out << "-" << i << " |-> ";
|
||||
for (unsigned i = 0; i < w2.size(); ++i) out << w2[i] << " ";
|
||||
out << "\n";
|
||||
}
|
||||
}
|
||||
#endif
|
||||
}
|
||||
|
||||
lbool drat::value(literal l) const {
|
||||
lbool v = m_assignment[l.var()];
|
||||
return v == l_undef || !l.sign() ? v : ~v;
|
||||
}
|
||||
|
||||
void drat::assign(literal l) {
|
||||
lbool new_value = l.sign() ? l_false : l_true;
|
||||
lbool old_value = value(l);
|
||||
if (new_value != old_value) {
|
||||
if (old_value == l_undef) {
|
||||
m_assignment[l.var()] = new_value;
|
||||
m_units.push_back(l);
|
||||
}
|
||||
else {
|
||||
m_inconsistent = true;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void drat::assign_propagate(literal l) {
|
||||
unsigned num_units = m_units.size();
|
||||
assign(l);
|
||||
for (unsigned i = num_units; !m_inconsistent && i < m_units.size(); ++i) {
|
||||
propagate(m_units[i]);
|
||||
}
|
||||
}
|
||||
|
||||
void drat::propagate(literal l) {
|
||||
watch& clauses = m_watches[l.index()];
|
||||
watch::iterator it = clauses.begin();
|
||||
watch::iterator it2 = it;
|
||||
watch::iterator end = clauses.end();
|
||||
for (; it != end; ++it) {
|
||||
clause& c = *(*it);
|
||||
if (c[0] == ~l) {
|
||||
std::swap(c[0], c[1]);
|
||||
}
|
||||
if (c[1] != ~l) {
|
||||
*it2 = *it;
|
||||
it2++;
|
||||
continue;
|
||||
}
|
||||
SASSERT(c[1] == ~l);
|
||||
if (value(c[0]) == l_true) {
|
||||
it2++;
|
||||
}
|
||||
else {
|
||||
literal * l_it = c.begin() + 2;
|
||||
literal * l_end = c.end();
|
||||
bool done = false;
|
||||
for (; l_it != l_end && !done; ++l_it) {
|
||||
if (value(*l_it) != l_false) {
|
||||
c[1] = *l_it;
|
||||
*l_it = ~l;
|
||||
m_watches[(~c[1]).index()].push_back(&c);
|
||||
done = true;
|
||||
}
|
||||
}
|
||||
if (done)
|
||||
continue;
|
||||
else if (value(c[0]) == l_false) {
|
||||
m_inconsistent = true;
|
||||
goto end_process_watch;
|
||||
}
|
||||
else {
|
||||
*it2 = *it;
|
||||
it2++;
|
||||
assign(c[0]);
|
||||
}
|
||||
}
|
||||
}
|
||||
end_process_watch:
|
||||
for (; it != end; ++it, ++it2)
|
||||
*it2 = *it;
|
||||
clauses.set_end(it2);
|
||||
}
|
||||
|
||||
drat::status drat::get_status(bool learned) const {
|
||||
|
@ -96,41 +335,44 @@ namespace sat {
|
|||
|
||||
void drat::add() {
|
||||
if (m_out) (*m_out) << "0\n";
|
||||
if (s.m_config.m_drat_check) append(0, 0, status::learned);
|
||||
if (s.m_config.m_drat_check) {
|
||||
if (m_inconsistent) std::cout << "Verified\n";
|
||||
else std::cout << "Failed to verify\n";
|
||||
}
|
||||
}
|
||||
void drat::add(literal l, bool learned) {
|
||||
status st = get_status(learned);
|
||||
if (m_out) dump(1, &l, st);
|
||||
if (s.m_config.m_drat_check) append(1, &l, st);
|
||||
if (s.m_config.m_drat_check) append(l, st);
|
||||
}
|
||||
void drat::add(literal l1, literal l2, bool learned) {
|
||||
literal ls[2] = {l1, l2};
|
||||
status st = get_status(learned);
|
||||
if (m_out) dump(2, ls, st);
|
||||
if (s.m_config.m_drat_check) append(2, ls, st);
|
||||
}
|
||||
void drat::add(literal l1, literal l2, literal l3, bool learned) {
|
||||
literal ls[3] = {l1, l2, l3};
|
||||
status st = get_status(learned);
|
||||
if (m_out) dump(3, ls, st);
|
||||
if (s.m_config.m_drat_check) append(3, ls, get_status(learned));
|
||||
if (s.m_config.m_drat_check) append(l1, l2, st);
|
||||
}
|
||||
void drat::add(clause& c, bool learned) {
|
||||
status st = get_status(learned);
|
||||
if (m_out) dump(c.size(), c.begin(), st);
|
||||
if (s.m_config.m_drat_check) append(c.size(), c.begin(), get_status(learned));
|
||||
if (s.m_config.m_drat_check) append(c, get_status(learned));
|
||||
}
|
||||
void drat::add(unsigned n, literal const* lits, unsigned m, premise * const* premises) {
|
||||
if (s.m_config.m_drat_check) {
|
||||
clause* c = s.m_cls_allocator.mk_clause(n, lits, true);
|
||||
append(*c, status::external);
|
||||
}
|
||||
}
|
||||
void drat::del(literal l) {
|
||||
if (m_out) dump(1, &l, status::deleted);
|
||||
if (s.m_config.m_drat_check) append(1, &l, status::deleted);
|
||||
if (s.m_config.m_drat_check) append(l, status::deleted);
|
||||
}
|
||||
void drat::del(literal l1, literal l2) {
|
||||
literal ls[2] = {l1, l2};
|
||||
if (m_out) dump(2, ls, status::deleted);
|
||||
if (s.m_config.m_drat_check) append(2, ls, status::deleted);
|
||||
if (s.m_config.m_drat_check) append(l1, l2, status::deleted);
|
||||
}
|
||||
void drat::del(clause& c) {
|
||||
if (m_out) dump(c.size(), c.begin(), status::deleted);
|
||||
if (s.m_config.m_drat_check) append(c.size(), c.begin(), status::deleted);
|
||||
if (s.m_config.m_drat_check) append(c, status::deleted);
|
||||
}
|
||||
}
|
||||
|
|
|
@ -21,30 +21,51 @@ Notes:
|
|||
|
||||
namespace sat {
|
||||
class drat {
|
||||
enum status { asserted, learned, deleted };
|
||||
|
||||
enum status { asserted, learned, deleted, external };
|
||||
typedef ptr_vector<clause> watch;
|
||||
struct premise {
|
||||
enum { t_clause, t_unit, t_ext } m_type;
|
||||
union evidence {
|
||||
clause* m_clause;
|
||||
literal m_literal;
|
||||
} m_evidence;
|
||||
};
|
||||
solver& s;
|
||||
std::ostream* m_out;
|
||||
ptr_vector<clause> m_proof;
|
||||
svector<status> m_status;
|
||||
literal_vector m_units;
|
||||
vector<watch_list> m_watches;
|
||||
char_vector m_assignment;
|
||||
vector<watch> m_watches;
|
||||
svector<lbool> m_assignment;
|
||||
bool m_inconsistent;
|
||||
|
||||
void dump(unsigned n, literal const* lits, status st);
|
||||
void append(unsigned n, literal const* lits, status st);
|
||||
void dump(unsigned n, literal const* c, status st);
|
||||
void append(literal l, status st);
|
||||
void append(literal l1, literal l2, status st);
|
||||
void append(clause& c, status st);
|
||||
friend std::ostream& operator<<(std::ostream & out, status st);
|
||||
status get_status(bool learned) const;
|
||||
bool is_cleaned(unsigned n, literal const* lits) const;
|
||||
|
||||
void declare(literal l);
|
||||
void assign(literal l);
|
||||
void propagate(literal l);
|
||||
void assign_propagate(literal l);
|
||||
void del_watch(clause& c, literal l);
|
||||
void verify(unsigned n, literal const* c);
|
||||
lbool value(literal l) const;
|
||||
void trace(std::ostream& out, unsigned n, literal const* c, status st);
|
||||
void display(std::ostream& out) const;
|
||||
|
||||
public:
|
||||
drat(solver& s);
|
||||
~drat();
|
||||
void add();
|
||||
void add(literal l, bool learned);
|
||||
void add(literal l1, literal l2, bool learned);
|
||||
void add(literal l1, literal l2, literal l3, bool learned);
|
||||
void add(clause& c, bool learned);
|
||||
void add(unsigned n, literal const* c, unsigned m, premise* const* premises);
|
||||
|
||||
void del(literal l);
|
||||
void del(literal l1, literal l2);
|
||||
void del(clause& c);
|
||||
|
|
|
@ -207,7 +207,9 @@ namespace sat {
|
|||
if (m_config.m_drat) {
|
||||
m_drat.del(c);
|
||||
}
|
||||
m_cls_allocator.del_clause(&c);
|
||||
else if (!m_config.m_drat || !m_config.m_drat_check) {
|
||||
m_cls_allocator.del_clause(&c);
|
||||
}
|
||||
m_stats.m_del_clause++;
|
||||
}
|
||||
|
||||
|
@ -276,13 +278,12 @@ namespace sat {
|
|||
|
||||
|
||||
clause * solver::mk_ter_clause(literal * lits, bool learned) {
|
||||
if (m_config.m_drat)
|
||||
m_drat.add(lits[0], lits[1], lits[2], learned);
|
||||
m_stats.m_mk_ter_clause++;
|
||||
clause * r = m_cls_allocator.mk_clause(3, lits, learned);
|
||||
bool reinit = attach_ter_clause(*r);
|
||||
if (reinit && !learned) push_reinit_stack(*r);
|
||||
|
||||
if (m_config.m_drat) m_drat.add(*r, learned);
|
||||
|
||||
if (learned)
|
||||
m_learned.push_back(r);
|
||||
else
|
||||
|
@ -494,6 +495,7 @@ namespace sat {
|
|||
void solver::dettach_bin_clause(literal l1, literal l2, bool learned) {
|
||||
get_wlist(~l1).erase(watched(l2, learned));
|
||||
get_wlist(~l2).erase(watched(l1, learned));
|
||||
if (m_config.m_drat) m_drat.del(l1, l2);
|
||||
}
|
||||
|
||||
void solver::dettach_clause(clause & c) {
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue