mirror of
https://github.com/Z3Prover/z3
synced 2026-01-20 01:03:20 +00:00
2141 lines
81 KiB
C++
2141 lines
81 KiB
C++
/*++
|
|
Copyright (c) 2025 Microsoft Corporation
|
|
|
|
|
|
vanishing(v, p) = (q, r) with q = 0, r >= 0 if p := q*v + r, M(q) = 0
|
|
|
|
TODOs:
|
|
|
|
Assumptions are added to impose sign constraints on non-linear multipliers
|
|
or to identify vanishing polynomials.
|
|
The sign of the assumption is based on the current candidate interpretation.
|
|
It could be that the negation of this assumption is asserted (and is
|
|
false under the current candiate interpretation).
|
|
The procedure could loop around refining a false assumption.
|
|
To handle this we could also track bounds of polynomials.
|
|
Then we can check if a bound on a polynomial is already asserted,
|
|
either as a case split or as an input assertion.
|
|
To retract consequences of temporary assumed bounds we would associate
|
|
levels with constraints and also undo and replay constraints during backjumping/backtracking.
|
|
A bound assumption corresponds to a decision. A conflict that depends on a bound assumption
|
|
reverts the assumptiom, and justifies it by the conflict dependencies.
|
|
In this way we would eliminate the double nested loop: saturate looping over the search loop.
|
|
|
|
--*/
|
|
|
|
#include <algorithm>
|
|
#include <ostream>
|
|
#include <unordered_map>
|
|
#include <utility>
|
|
#include <variant>
|
|
#include "util/debug.h"
|
|
#include "util/dependency.h"
|
|
#include "util/lbool.h"
|
|
#include "util/memory_manager.h"
|
|
#include "util/params.h"
|
|
#include "util/rational.h"
|
|
#include "util/trace.h"
|
|
#include "util/trail.h"
|
|
#include "util/uint_set.h"
|
|
#include "util/util.h"
|
|
#include "util/vector.h"
|
|
#include "params/smt_params_helper.hpp"
|
|
#include "math/dd/pdd_eval.h"
|
|
#include "math/lp/nla_core.h"
|
|
#include "math/lp/nla_coi.h"
|
|
#include "math/lp/nla_stellensatz2.h"
|
|
#include "math/dd/dd_pdd.h"
|
|
#include "math/lp/explanation.h"
|
|
#include "math/lp/int_solver.h"
|
|
#include "math/lp/lar_solver.h"
|
|
#include "math/lp/lia_move.h"
|
|
#include "math/lp/lp_settings.h"
|
|
#include "math/lp/lp_types.h"
|
|
#include "math/lp/nla_common.h"
|
|
#include "math/lp/nla_defs.h"
|
|
#include "math/lp/nla_types.h"
|
|
|
|
|
|
namespace nla {
|
|
|
|
lpvar stellensatz2::monomial_factory::mk_monomial(lp::lar_solver &lra, svector<lpvar> const &vars) {
|
|
lpvar v = lp::null_lpvar;
|
|
if (vars.empty())
|
|
return v;
|
|
if (vars.size() == 1)
|
|
return vars[0];
|
|
svector<lpvar> _vars(vars);
|
|
std::sort(_vars.begin(), _vars.end());
|
|
if (m_vars2mon.find(_vars, v))
|
|
return v;
|
|
auto is_int = all_of(vars, [&](lpvar v) { return lra.var_is_int(v); });
|
|
auto nv = lra.number_of_vars();
|
|
v = lra.add_var(nv, is_int);
|
|
m_vars2mon.insert(_vars, v);
|
|
m_mon2vars.insert(v, _vars);
|
|
return v;
|
|
}
|
|
|
|
lpvar stellensatz2::monomial_factory::get_monomial(svector<lpvar> const &vars) {
|
|
lpvar v = lp::null_lpvar;
|
|
if (vars.empty())
|
|
return v;
|
|
if (vars.size() == 1)
|
|
return vars[0];
|
|
svector<lpvar> _vars(vars);
|
|
std::sort(_vars.begin(), _vars.end());
|
|
if (m_vars2mon.find(_vars, v))
|
|
return v;
|
|
return lp::null_lpvar;
|
|
}
|
|
|
|
void stellensatz2::monomial_factory::init(lpvar v, svector<lpvar> const &_vars) {
|
|
svector<lpvar> vars(_vars);
|
|
std::sort(vars.begin(), vars.end());
|
|
m_vars2mon.insert(vars, v);
|
|
m_mon2vars.insert(v, vars);
|
|
}
|
|
|
|
stellensatz2::stellensatz2(core* core) :
|
|
common(core),
|
|
m_solver(*this),
|
|
m_coi(*core),
|
|
pddm(core->lra_solver().number_of_vars()),
|
|
m_di(m_dm, core->reslim()) {
|
|
}
|
|
|
|
lbool stellensatz2::saturate() {
|
|
unsigned num_conflicts = 0;
|
|
init_solver();
|
|
TRACE(arith, display(tout << "stellensatz before saturation\n"));
|
|
start_saturate:
|
|
lbool r = search();
|
|
|
|
TRACE(arith, tout << "search " << r << ": " << m_core << "\n");
|
|
|
|
if (r == l_undef)
|
|
r = m_solver.solve(m_core);
|
|
TRACE(arith, display(tout << "stellensatz after saturation " << r << "\n"));
|
|
if (r == l_false) {
|
|
++num_conflicts;
|
|
IF_VERBOSE(2, verbose_stream() << "(nla.stellensatz :conflicts " << num_conflicts << ":constraints "
|
|
<< m_constraints.size() << ")\n");
|
|
if (num_conflicts >= m_config.max_conflicts)
|
|
return l_undef;
|
|
while (backtrack(m_core)) {
|
|
++c().lp_settings().stats().m_stellensatz.m_num_backtracks;
|
|
lbool rb = m_solver.solve(m_core);
|
|
TRACE(arith, tout << "backtrack search " << rb << ": " << m_core << "\n");
|
|
// verbose_stream() << rb << " " << num_conflicts << " " << m_config.max_conflicts << "\n";
|
|
if (rb == l_false)
|
|
continue;
|
|
if (rb == l_undef) {
|
|
//verbose_stream() << "undetermined solve\n";
|
|
// return rb;
|
|
}
|
|
m_solver.update_values(m_values);
|
|
goto start_saturate;
|
|
}
|
|
++c().lp_settings().stats().m_stellensatz.m_num_conflicts;
|
|
conflict(m_core);
|
|
}
|
|
if (r == l_true && !set_model())
|
|
r = l_undef;
|
|
if (r == l_undef) {
|
|
IF_VERBOSE(2, display(verbose_stream()));
|
|
}
|
|
// verbose_stream() << "result " << r << "\n";
|
|
return r;
|
|
}
|
|
|
|
void stellensatz2::init_solver() {
|
|
m_num_scopes = 0;
|
|
m_monomial_factory.reset();
|
|
m_coi.init();
|
|
m_values.reset();
|
|
init_vars();
|
|
simplify();
|
|
init_occurs();
|
|
}
|
|
|
|
void stellensatz2::init_vars() {
|
|
auto const& src = c().lra_solver();
|
|
auto sz = src.number_of_vars();
|
|
m_idx2bound.reset();
|
|
m_split_count.reset();
|
|
m_split_count.reserve(sz, 0);
|
|
for (unsigned v = 0; v < sz; ++v) {
|
|
m_values.push_back(c().val(v));
|
|
if (!m_coi.vars().contains(v))
|
|
continue;
|
|
if (c().is_monic_var(v))
|
|
init_monomial(v);
|
|
// assert bounds on v in the new solver.
|
|
if (src.column_has_lower_bound(v)) {
|
|
auto lo_bound = src.get_lower_bound(v);
|
|
SASSERT(lo_bound.y >= 0);
|
|
auto k = lo_bound.y > 0 ? lp::lconstraint_kind::GT : lp::lconstraint_kind::GE;
|
|
auto rhs = lo_bound.x;
|
|
auto dep = src.get_column_lower_bound_witness(v);
|
|
add_var_bound(v, k, rhs, external_justification(dep));
|
|
}
|
|
if (src.column_has_upper_bound(v)) {
|
|
auto hi_bound = src.get_upper_bound(v);
|
|
SASSERT(hi_bound.y <= 0);
|
|
auto k = hi_bound.y < 0 ? lp::lconstraint_kind::LT : lp::lconstraint_kind::LE;
|
|
auto rhs = hi_bound.x;
|
|
auto dep = src.get_column_upper_bound_witness(v);
|
|
add_var_bound(v, k, rhs, external_justification(dep));
|
|
}
|
|
}
|
|
}
|
|
|
|
// set the model based on m_values computed by the solver
|
|
bool stellensatz2::set_model() {
|
|
if (any_of(m_constraints, [&](auto const &c) { return !constraint_is_true(c); }))
|
|
return false;
|
|
auto &src = c().lra_solver();
|
|
c().m_use_nra_model = true;
|
|
m_values.reserve(src.number_of_vars());
|
|
for (unsigned v = 0; v < src.number_of_vars(); ++v) {
|
|
if (c().is_monic_var(v)) {
|
|
auto& mon = c().emons()[v];
|
|
rational val(1);
|
|
for (auto w : mon.vars())
|
|
val *= m_values[w];
|
|
m_values[v] = val;
|
|
}
|
|
else if (src.column_has_term(v)) {
|
|
auto const& t = src.get_term(v);
|
|
rational val(0);
|
|
for (auto cv : t)
|
|
val += m_values[cv.j()] * cv.coeff();
|
|
m_values[v] = val;
|
|
}
|
|
else if (m_coi.vars().contains(v)) {
|
|
// variable is in coi, m_values[v] is set.
|
|
}
|
|
else {
|
|
m_values[v] = c().val(v);
|
|
}
|
|
c().m_nra.set_value(v, m_values[v]);
|
|
}
|
|
return true;
|
|
}
|
|
|
|
dd::pdd stellensatz2::to_pdd(lpvar v) {
|
|
if (m_coi.mons().contains(v)) {
|
|
auto& mon = c().emons()[v];
|
|
dd::pdd prod(pddm);
|
|
prod = 1;
|
|
for (auto w : mon.vars())
|
|
prod *= to_pdd(w);
|
|
return prod;
|
|
}
|
|
if (m_coi.terms().contains(v)) {
|
|
auto const& t = c().lra_solver().get_term(v);
|
|
dd::pdd sum(pddm);
|
|
sum = 0;
|
|
for (auto cv : t)
|
|
sum += to_pdd(cv.j())*cv.coeff();
|
|
return sum;
|
|
}
|
|
return pddm.mk_var(v);
|
|
}
|
|
|
|
stellensatz2::term_offset stellensatz2::to_term_offset(dd::pdd const &p) {
|
|
term_offset to;
|
|
for (auto const &[coeff, vars] : p) {
|
|
if (vars.empty())
|
|
to.second += coeff;
|
|
else
|
|
to.first.add_monomial(coeff, m_monomial_factory.get_monomial(vars));
|
|
}
|
|
return to;
|
|
}
|
|
|
|
bool stellensatz2::has_term_offset(dd::pdd const &p) {
|
|
for (auto const &[coeff, vars] : p)
|
|
if (!vars.empty() && lp::null_lpvar == m_monomial_factory.get_monomial(vars))
|
|
return false;
|
|
return true;
|
|
}
|
|
|
|
void stellensatz2::init_monomial(unsigned mon_var) {
|
|
auto &mon = c().emons()[mon_var];
|
|
m_monomial_factory.init(mon_var, mon.vars());
|
|
}
|
|
|
|
lp::constraint_index stellensatz2::add_var_bound(lp::lpvar v, lp::lconstraint_kind k, rational const& rhs, justification j) {
|
|
auto p = to_pdd(v) - rhs;
|
|
rational d(1);
|
|
for (auto const& [c, is_constant] : p.coefficients())
|
|
d = lcm(d, denominator(c));
|
|
if (d != 1)
|
|
p *= d;
|
|
return gcd_normalize(add_constraint(p, k, j));
|
|
}
|
|
|
|
lp::constraint_index stellensatz2::add_constraint(dd::pdd p, lp::lconstraint_kind k, justification j) {
|
|
if (k == lp::lconstraint_kind::LE) {
|
|
k = lp::lconstraint_kind::GE;
|
|
p = -p;
|
|
}
|
|
if (k == lp::lconstraint_kind::LT) {
|
|
k = lp::lconstraint_kind::GT;
|
|
p = -p;
|
|
}
|
|
if (k == lp::lconstraint_kind::GT && is_int(p)) {
|
|
k = lp::lconstraint_kind::GE;
|
|
p -= rational(1);
|
|
}
|
|
lp::constraint_index ci = lp::null_ci;
|
|
|
|
if (m_constraint_index.find({p.index(), k}, ci))
|
|
return ci;
|
|
ci = m_constraints.size();
|
|
|
|
unsigned level = get_level(j);
|
|
if (is_decision(j)) {
|
|
level = m_num_scopes;
|
|
++m_num_scopes;
|
|
}
|
|
|
|
m_constraints.push_back({p, k});
|
|
m_levels.push_back(level);
|
|
m_justifications.push_back(j);
|
|
m_constraint_index.insert({p.index(), k}, ci);
|
|
push_bound(ci);
|
|
++c().lp_settings().stats().m_stellensatz.m_num_constraints;
|
|
m_has_occurs.reserve(ci + 1);
|
|
return ci;
|
|
}
|
|
|
|
void stellensatz2::push_bound(lp::constraint_index ci) {
|
|
SASSERT(ci == m_bounds.size());
|
|
auto [p, k] = m_constraints[ci];
|
|
auto bound = -(p.offset());
|
|
auto q = p + bound;
|
|
auto mq = -q;
|
|
m_dm.push_scope();
|
|
auto q_index = q.index();
|
|
auto last_index = q_index < m_idx2bound.size() ? m_idx2bound[q_index] : UINT_MAX;
|
|
m_idx2bound.reserve(std::max(mq.index(), q_index) + 1, UINT_MAX);
|
|
m_idx2bound[q_index] = ci;
|
|
m_bounds.push_back(bound_info{k, bound, last_index, m_dm.mk_leaf(ci), q, mq});
|
|
}
|
|
|
|
void stellensatz2::pop_bound() {
|
|
auto const &[k, bound, last_bound, d, q, mq] = m_bounds.back();
|
|
m_idx2bound[q.index()] = last_bound;
|
|
m_idx2bound.pop_back();
|
|
m_dm.pop_scope(1);
|
|
}
|
|
|
|
lp::constraint_index stellensatz2::resolve(lp::constraint_index conflict, lp::constraint_index ci) {
|
|
SASSERT(ci != lp::null_ci);
|
|
if (conflict == lp::null_ci)
|
|
return lp::null_ci;
|
|
auto const &[p, k] = m_constraints[ci];
|
|
if (p.is_var())
|
|
return resolve_variable(p.var(), ci, conflict);
|
|
if (p.is_minus())
|
|
return resolve_variable((-p).var(), ci, conflict);
|
|
return lp::null_ci;
|
|
}
|
|
|
|
lbool stellensatz2::sign(dd::pdd const &p) {
|
|
if (p.is_val()) {
|
|
if (p.val() > 0)
|
|
return l_false;
|
|
if (p.val() < 0)
|
|
return l_true;
|
|
return l_undef;
|
|
}
|
|
auto val = value(p);
|
|
if (p.is_var()) {
|
|
if (val > 0)
|
|
return l_false;
|
|
if (val < 0)
|
|
return l_true;
|
|
return l_undef;
|
|
}
|
|
|
|
auto offset = p.offset();
|
|
auto q = p - offset;
|
|
if (has_lo(q)) {
|
|
auto lo_bound = lo_val(q);
|
|
// q >= lo_bound
|
|
// q + offset >= lo_bound + offset
|
|
if (lo_bound + offset > 0)
|
|
return l_false;
|
|
if (lo_bound + offset == 0)
|
|
return lo_is_strict(q) ? l_false : l_undef;
|
|
}
|
|
if (has_hi(q)) {
|
|
auto hi_bound = hi_val(q);
|
|
// q <= hi_bound
|
|
// q + offset <= hi_bound + offset
|
|
if (hi_bound + offset < 0)
|
|
return l_true;
|
|
if (hi_bound + offset == 0)
|
|
return hi_is_strict(q) ? l_true : l_undef;
|
|
}
|
|
if (val > 0)
|
|
return l_false;
|
|
if (val < 0)
|
|
return l_true;
|
|
return l_undef;
|
|
}
|
|
|
|
lp::constraint_index stellensatz2::resolve_variable(lpvar x, lp::constraint_index ci, lp::constraint_index other_ci) {
|
|
TRACE(arith, tout << "resolve j" << x << " between ci " << (int)ci << " and other_ci " << (int)other_ci << "\n");
|
|
if (ci == lp::null_ci || other_ci == lp::null_ci)
|
|
return lp::null_ci;
|
|
auto f = factor(x, ci);
|
|
auto g = factor(x, other_ci);
|
|
if (f.degree != 1)
|
|
return lp::null_ci; // future - could handle this
|
|
if (g.degree != 1)
|
|
return lp::null_ci; // not handling degree > 1
|
|
|
|
//
|
|
// check that p_value1 and p_value2 have different signs
|
|
// check that other_p is disjoint from tabu
|
|
// compute overlaps extending x
|
|
//
|
|
auto f_sign = sign(f.p);
|
|
if (f_sign == l_undef)
|
|
return lp::null_ci;
|
|
auto g_sign = sign(g.p);
|
|
if (g_sign == l_undef)
|
|
return lp::null_ci;
|
|
if (f_sign == g_sign)
|
|
return lp::null_ci;
|
|
auto p_value1 = value(f.p);
|
|
auto p_value2 = value(g.p);
|
|
|
|
for (unsigned i = 0; i < f.degree; ++i)
|
|
f.p *= to_pdd(x);
|
|
for (unsigned i = 0; i < g.degree; ++i)
|
|
g.p *= to_pdd(x);
|
|
|
|
auto [m1, f_p] = f.p.var_factors();
|
|
auto [m2, g_p] = g.p.var_factors();
|
|
unsigned_vector m1m2;
|
|
dd::pdd::merge(m1, m2, m1m2);
|
|
SASSERT(m1m2.contains(x));
|
|
f_p = f_p.mul(m1);
|
|
g_p = g_p.mul(m2);
|
|
|
|
// TRACE(arith, tout << "m1 " << m1 << " m2 " << m2 << " m1m2: " << m1m2 << "\n");
|
|
|
|
// m1m2 * f_p + f.q >= 0
|
|
// m1m2 * g_p + g.q >= 0
|
|
//
|
|
lp::constraint_index ci_a, ci_b;
|
|
auto const &[other_p, other_k] = m_constraints[other_ci];
|
|
auto const &[p, k] = m_constraints[ci];
|
|
|
|
bool g_strict = other_k == lp::lconstraint_kind::GT;
|
|
bool f_strict = k == lp::lconstraint_kind::GT;
|
|
if (g_p.is_val() && f_p.is_val() && g_p.val() == -f_p.val())
|
|
ci_a = ci;
|
|
else if (g_p.is_val())
|
|
ci_a = multiply(ci, pddm.mk_val(g_p.val()));
|
|
else if (f_sign == l_true)
|
|
ci_a = multiply(ci, assume(g_p, f_strict ? lp::lconstraint_kind::LT : lp::lconstraint_kind::LE));
|
|
else
|
|
ci_a = multiply(ci, assume(g_p, f_strict ? lp::lconstraint_kind::GT : lp::lconstraint_kind::GE));
|
|
|
|
if (g_p.is_val() && f_p.is_val() && g_p.val() == -f_p.val())
|
|
ci_b = other_ci;
|
|
else if (f_p.is_val())
|
|
ci_b = multiply(other_ci, pddm.mk_val(f_p.val()));
|
|
else if (f_sign == l_false)
|
|
ci_b = multiply(other_ci, assume(f_p, g_strict ? lp::lconstraint_kind::LT : lp::lconstraint_kind::LE));
|
|
else
|
|
ci_b = multiply(other_ci, assume(f_p, g_strict ? lp::lconstraint_kind::GT : lp::lconstraint_kind::GE));
|
|
|
|
auto new_ci = add(ci_a, ci_b);
|
|
|
|
++c().lp_settings().stats().m_stellensatz.m_num_resolutions;
|
|
|
|
TRACE(arith, tout << "eliminate j" << x << ":\n"; display_constraint(tout, ci) << "\n";
|
|
display_constraint(tout, other_ci) << "\n"; display_constraint(tout, ci_a) << "\n";
|
|
display_constraint(tout, ci_b) << "\n"; display_constraint(tout, new_ci) << "\n");
|
|
|
|
new_ci = factor(new_ci);
|
|
|
|
init_occurs(new_ci);
|
|
|
|
// display_constraint(verbose_stream(), new_ci) << "\n";
|
|
|
|
return new_ci;
|
|
}
|
|
|
|
void stellensatz2::conflict(svector<lp::constraint_index> const& core) {
|
|
lp::explanation ex;
|
|
lemma_builder new_lemma(c(), "stellensatz");
|
|
m_constraints_in_conflict.reset();
|
|
for (auto ci : core)
|
|
explain_constraint(new_lemma, ci, ex);
|
|
new_lemma &= ex;
|
|
IF_VERBOSE(2, verbose_stream() << "stellensatz unsat \n" << new_lemma << "\n");
|
|
TRACE(arith, tout << "unsat\n" << new_lemma << "\n");
|
|
c().lra_solver().settings().stats().m_nla_stellensatz++;
|
|
}
|
|
|
|
//
|
|
// for px + q >= 0
|
|
// If M(p) = 0, then
|
|
// derive the constraint q >= 0 using the inference
|
|
//
|
|
// px + q >= 0 p = 0
|
|
// --------------------
|
|
// q >= 0
|
|
//
|
|
// The equality p = 0 is tracked as an assumption.
|
|
//
|
|
lp::constraint_index stellensatz2::vanishing(lpvar x, factorization const &f, lp::constraint_index ci) {
|
|
if (f.p.is_zero())
|
|
return lp::null_ci;
|
|
auto p_value = value(f.p);
|
|
if (p_value != 0)
|
|
return lp::null_ci;
|
|
|
|
++c().lp_settings().stats().m_stellensatz.m_num_vanishings;
|
|
// add p = 0 as assumption and reduce to q.
|
|
auto p_is_0 = assume(f.p, lp::lconstraint_kind::EQ);
|
|
// ci & -p_is_0*x^f.degree => new_ci
|
|
dd::pdd r = pddm.mk_val(rational(-1));
|
|
for (unsigned i = 0; i < f.degree; ++i)
|
|
r = r * pddm.mk_var(x);
|
|
p_is_0 = multiply(p_is_0, r);
|
|
auto new_ci = add(ci, p_is_0);
|
|
TRACE(arith, display_constraint(tout << "j" << x << " ", ci) << "\n";
|
|
display_constraint(tout << "reduced ", new_ci) << "\n";
|
|
display_constraint(tout, p_is_0) << "\n");
|
|
init_occurs(new_ci);
|
|
return new_ci;
|
|
}
|
|
|
|
//
|
|
// reduce x * y * p >= 0
|
|
// to
|
|
// p >= 0 based on assumptions x > 0, y > 0 or x < 0, y < 0
|
|
// -p >= 0 based on assumptions x > 0, y < 0 or x < 0, y > 0
|
|
//
|
|
// reduce x * y * p > 0
|
|
// to
|
|
// p > 0 based on assumptions x >= 0, y >= 0 or x <= 0, y <= 0
|
|
// -p > 0 based on assumptions x >= 0, y <= 0 or x <= 0, y >= 0
|
|
//
|
|
lp::constraint_index stellensatz2::factor(lp::constraint_index ci) {
|
|
auto const &[p, k] = m_constraints[ci];
|
|
auto [vars, q] = p.var_factors(); // p = vars * q
|
|
|
|
bool is_gt = k == lp::lconstraint_kind::GT;
|
|
unsigned i = 0;
|
|
for (auto v : vars)
|
|
if (is_gt || value(v) != 0)
|
|
vars[i++] = v;
|
|
else
|
|
q *= pddm.mk_var(v);
|
|
vars.shrink(i);
|
|
if (vars.empty())
|
|
return ci;
|
|
if (vars.size() == 1 && q.is_val())
|
|
return ci;
|
|
if (q.is_val()) {
|
|
q *= pddm.mk_var(vars.back());
|
|
vars.pop_back();
|
|
}
|
|
|
|
vector<dd::pdd> muls;
|
|
muls.push_back(q);
|
|
for (auto v : vars)
|
|
muls.push_back(muls.back() * pddm.mk_var(v));
|
|
auto new_ci = ci;
|
|
SASSERT(muls.back() == p);
|
|
int sign = 1;
|
|
for (unsigned i = vars.size(); i-- > 0;) {
|
|
auto pv = pddm.mk_var(vars[i]);
|
|
auto val = value(vars[i]);
|
|
auto k1 = val == 0 ? lp::lconstraint_kind::GE : (val > 0 ? lp::lconstraint_kind::GT : lp::lconstraint_kind::LT);
|
|
if (val < 0)
|
|
sign = -sign;
|
|
new_ci = divide(new_ci, assume(pv, k1), sign * muls[i]);
|
|
TRACE(arith_verbose, display_constraint(tout, new_ci) << "\n"; display_constraint(tout, assume(pv, k1)) << "\n";);
|
|
}
|
|
TRACE(arith, display_constraint(tout << "factor ", new_ci) << "\n");
|
|
return new_ci;
|
|
}
|
|
|
|
unsigned stellensatz2::degree_of_var_in_constraint(lpvar var, lp::constraint_index ci) const {
|
|
return m_constraints[ci].p.degree(var);
|
|
}
|
|
|
|
stellensatz2::factorization stellensatz2::factor(lpvar v, lp::constraint_index ci) {
|
|
auto const& [p, k] = m_constraints[ci];
|
|
auto degree = degree_of_var_in_constraint(v, ci);
|
|
dd::pdd lc(pddm), rest(pddm);
|
|
p.factor(v, degree, lc, rest);
|
|
return {degree, lc, rest};
|
|
}
|
|
|
|
//
|
|
// check if core depends on an assumption
|
|
// identify the maximal assumption
|
|
// undo m_constraints down to max_ci.
|
|
// negate max_ci
|
|
// propagate it using remaining external and assumptions.
|
|
// find a new satisfying assignment (if any) before continuing.
|
|
//
|
|
bool stellensatz2::backtrack(svector<lp::constraint_index> const &core) {
|
|
// reset_bounds();
|
|
m_constraints_in_conflict.reset();
|
|
svector<lp::constraint_index> external, assumptions;
|
|
for (auto ci : core)
|
|
explain_constraint(ci, external, assumptions);
|
|
TRACE(arith, display(tout << "backtrack core " << core << "external " << external << " assumptions " << assumptions << "\n"));
|
|
if (assumptions.empty())
|
|
return false;
|
|
lp::constraint_index max_ci = 0;
|
|
for (auto ci : assumptions)
|
|
max_ci = std::max(ci, max_ci);
|
|
assumptions.erase(max_ci);
|
|
external.append(assumptions);
|
|
backtrack(max_ci, external);
|
|
return true;
|
|
}
|
|
|
|
bool stellensatz2::core_is_linear(svector<lp::constraint_index> const &core) {
|
|
m_constraints_in_conflict.reset();
|
|
svector<lp::constraint_index> external, assumptions;
|
|
for (auto ci : core)
|
|
explain_constraint(ci, external, assumptions);
|
|
return all_of(assumptions, [&](auto ci) { return has_term_offset(m_constraints[ci].p); });
|
|
}
|
|
|
|
void stellensatz2::explain_constraint(lp::constraint_index ci, svector<lp::constraint_index> &external,
|
|
svector<lp::constraint_index> &assumptions) {
|
|
if (m_constraints_in_conflict.contains(ci))
|
|
return;
|
|
m_constraints_in_conflict.insert(ci);
|
|
|
|
auto &j = m_justifications[ci];
|
|
if (std::holds_alternative<external_justification>(j)) {
|
|
external.push_back(ci);
|
|
}
|
|
else if (is_decision(j)) {
|
|
assumptions.push_back(ci);
|
|
}
|
|
else {
|
|
for (auto cij : justification_range(*this, j))
|
|
explain_constraint(cij, external, assumptions);
|
|
}
|
|
}
|
|
|
|
//
|
|
// a constraint can be explained by a set of bounds used as justifications
|
|
// and by an original constraint.
|
|
//
|
|
void stellensatz2::explain_constraint(lemma_builder &new_lemma, lp::constraint_index ci, lp::explanation &ex) {
|
|
svector<lp::constraint_index> external, assumptions;
|
|
explain_constraint(ci, external, assumptions);
|
|
for (auto ci : external) {
|
|
auto &j = m_justifications[ci];
|
|
auto dep = std::get<external_justification>(j);
|
|
c().lra_solver().push_explanation(dep.dep, ex);
|
|
}
|
|
for (auto ci : assumptions) {
|
|
auto &[p, k] = m_constraints[ci];
|
|
auto [t, coeff] = to_term_offset(p);
|
|
new_lemma |= ineq(t, negate(k), -coeff);
|
|
}
|
|
}
|
|
|
|
rational stellensatz2::value(dd::pdd const& p) const {
|
|
dd::pdd_eval eval;
|
|
eval.var2val() = [&](unsigned v) -> rational { return value(v); };
|
|
return eval(p);
|
|
}
|
|
|
|
//
|
|
// normalize constraint by dividing by gcd of coefficients
|
|
//
|
|
lp::constraint_index stellensatz2::gcd_normalize(lp::constraint_index ci) {
|
|
auto [p, k] = m_constraints[ci];
|
|
rational g(0);
|
|
bool _is_int = is_int(p);
|
|
for (auto const& [c, is_constant] : p.coefficients())
|
|
if (!is_constant || !_is_int)
|
|
g = gcd(g, c);
|
|
if (g == 0 || g == 1)
|
|
return ci;
|
|
// g := gcd of non-constant coefficients, or all coefficients if not integer
|
|
switch (k) {
|
|
case lp::lconstraint_kind::GE: {
|
|
// p + k >= 0
|
|
// -> (p + k) / g >= 0
|
|
// -> p / g + k/g + floor(k/g) - k/g >= 0
|
|
// -> p / g + floor(k/g) >= 0
|
|
auto q = p * (1/ g);
|
|
q += (floor(q.offset()) - q.offset());
|
|
return add_constraint(q, k, gcd_justification(ci));
|
|
}
|
|
case lp::lconstraint_kind::GT: {
|
|
// p + k > 0
|
|
// p / g + floor(k/g) >= 0 if k/g is not integer
|
|
// p / q + k/g - 1 >= 0 if k/g is integer
|
|
auto q = p * (1 / g);
|
|
auto offset = q.offset();
|
|
q += (floor(offset) - offset);
|
|
if (_is_int) {
|
|
k = lp::lconstraint_kind::GE;
|
|
if (offset.is_int())
|
|
q -= rational(1);
|
|
}
|
|
return add_constraint(q, k, gcd_justification(ci));
|
|
}
|
|
case lp::lconstraint_kind::LT:
|
|
case lp::lconstraint_kind::LE:
|
|
UNREACHABLE();
|
|
case lp::lconstraint_kind::EQ:
|
|
case lp::lconstraint_kind::NE:
|
|
if (!divides(g, p.offset()))
|
|
return ci;
|
|
return add_constraint(p * (1/g), k, gcd_justification(ci));
|
|
default:
|
|
UNREACHABLE();
|
|
return ci;
|
|
}
|
|
}
|
|
|
|
lp::constraint_index stellensatz2::assume(dd::pdd const& p, lp::lconstraint_kind k) {
|
|
if (k == lp::lconstraint_kind::EQ) {
|
|
auto left = assume(p, lp::lconstraint_kind::GE);
|
|
auto right = assume(-p, lp::lconstraint_kind::GE);
|
|
return add_constraint(p, k, eq_justification{left, right});
|
|
}
|
|
if (k == lp::lconstraint_kind::LE)
|
|
return assume(-p, lp::lconstraint_kind::GE);
|
|
if (k == lp::lconstraint_kind::LT)
|
|
return assume(-p, lp::lconstraint_kind::GT);
|
|
|
|
u_dependency *d = nullptr;
|
|
auto has_bound = [&](rational const& a, lpvar x, rational const& b) {
|
|
rational bound;
|
|
bool is_strict = false;
|
|
if (a == 1 && k == lp::lconstraint_kind::GE && c().lra_solver().has_lower_bound(x, d, bound, is_strict) &&
|
|
bound >= -b) {
|
|
return true;
|
|
}
|
|
if (a == 1 && k == lp::lconstraint_kind::GT && c().lra_solver().has_lower_bound(x, d, bound, is_strict) &&
|
|
(bound > -b || (is_strict && bound >= -b))) {
|
|
return true;
|
|
}
|
|
if (a == -1 && k == lp::lconstraint_kind::GE && c().lra_solver().has_upper_bound(x, d, bound, is_strict) &&
|
|
bound <= -b) {
|
|
return true;
|
|
}
|
|
if (a == -1 && k == lp::lconstraint_kind::GT && c().lra_solver().has_upper_bound(x, d, bound, is_strict) &&
|
|
(bound < -b || (is_strict && bound <= -b))) {
|
|
return true;
|
|
}
|
|
return false;
|
|
};
|
|
|
|
if (p.is_unilinear() && has_bound(p.hi().val(), p.var(), p.lo().val()))
|
|
// ax + b ~k~ 0
|
|
return add_constraint(p, k, external_justification(d));
|
|
return add_constraint(p, k, assumption_justification());
|
|
}
|
|
|
|
// p1 >= lo1, p2 >= lo2 => p1 + p2 >= lo1 + lo2
|
|
lp::constraint_index stellensatz2::add(lp::constraint_index left, lp::constraint_index right) {
|
|
auto const &[p, k1] = m_constraints[left];
|
|
auto const &[q, k2] = m_constraints[right];
|
|
lp::lconstraint_kind k = join(k1, k2);
|
|
return gcd_normalize(add_constraint(p + q, k, addition_justification{left, right}));
|
|
}
|
|
|
|
// p >= 0 => a * p >= 0 if a > 0,
|
|
// p = 0 => p * q = 0 no matter what q
|
|
lp::constraint_index stellensatz2::multiply(lp::constraint_index ci, dd::pdd q) {
|
|
auto const& [p, k] = m_constraints[ci];
|
|
auto k1 = k;
|
|
if (q.is_val() && q.val() < 0)
|
|
k1 = swap_side(k1);
|
|
SASSERT(q.is_val() || k1 == lp::lconstraint_kind::EQ);
|
|
return add_constraint(p * q, k1, multiplication_poly_justification{ci, q});
|
|
}
|
|
|
|
lp::constraint_index stellensatz2::multiply(lp::constraint_index left, lp::constraint_index right) {
|
|
auto const &[p, k1] = m_constraints[left];
|
|
auto const &[q, k2] = m_constraints[right];
|
|
lp::lconstraint_kind k = meet(k1, k2);
|
|
return add_constraint(p*q, k, multiplication_justification{left, right});
|
|
}
|
|
|
|
// p k 0, d > 0 -> p/d k 0, where q := d / d
|
|
// q is the quotient obtained by dividing the divisor constraint, which is of the form d - 1 >= 0 or d > 0
|
|
lp::constraint_index stellensatz2::divide(lp::constraint_index ci, lp::constraint_index divisor, dd::pdd q) {
|
|
auto const &[p, k] = m_constraints[ci];
|
|
return add_constraint(q, k, division_justification{ci, divisor});
|
|
}
|
|
|
|
lp::constraint_index stellensatz2::substitute(lp::constraint_index ci, lp::constraint_index ci_eq, lpvar v,
|
|
dd::pdd p) {
|
|
auto const &[p1, k1] = m_constraints[ci];
|
|
auto const &[p2, k2] = m_constraints[ci_eq];
|
|
SASSERT(k2 == lp::lconstraint_kind::EQ);
|
|
auto q = p1.subst_pdd(v, p);
|
|
return add_constraint(q, k1, substitute_justification{ci, ci_eq, v, p});
|
|
}
|
|
|
|
void stellensatz2::init_occurs() {
|
|
m_occurs.reset();
|
|
m_occurs.reserve(num_vars());
|
|
for (lp::constraint_index ci = 0; ci < m_constraints.size(); ++ci)
|
|
init_occurs(ci);
|
|
}
|
|
|
|
void stellensatz2::init_occurs(lp::constraint_index ci) {
|
|
if (ci == lp::null_ci)
|
|
return;
|
|
if (m_has_occurs[ci])
|
|
return;
|
|
m_has_occurs[ci] = true;
|
|
auto const &con = m_constraints[ci];
|
|
for (auto v : con.p.free_vars())
|
|
m_occurs[v].push_back(ci);
|
|
}
|
|
|
|
void stellensatz2::remove_occurs(lp::constraint_index ci) {
|
|
if (!m_has_occurs[ci])
|
|
return;
|
|
m_has_occurs[ci] = false;
|
|
auto const &con = m_constraints[ci];
|
|
for (auto v : con.p.free_vars())
|
|
m_occurs[v].pop_back();
|
|
}
|
|
|
|
bool stellensatz2::is_int(svector<lp::lpvar> const& vars) const {
|
|
return all_of(vars, [&](lpvar v) { return var_is_int(v); });
|
|
}
|
|
|
|
bool stellensatz2::is_int(dd::pdd const &p) const {
|
|
return is_int(p.free_vars());
|
|
}
|
|
|
|
bool stellensatz2::var_is_int(lp::lpvar v) const {
|
|
return c().lra_solver().var_is_int(v);
|
|
}
|
|
|
|
bool stellensatz2::constraint_is_true(lp::constraint_index ci) const {
|
|
return ci != lp::null_ci && constraint_is_true(m_constraints[ci]);
|
|
}
|
|
|
|
bool stellensatz2::constraint_is_false(lp::constraint_index ci) const {
|
|
return ci != lp::null_ci && constraint_is_false(m_constraints[ci]);
|
|
}
|
|
|
|
bool stellensatz2::constraint_is_true(constraint const &c) const {
|
|
auto const& [p, k] = c;
|
|
auto lhs = value(p);
|
|
switch (k) {
|
|
case lp::lconstraint_kind::GT: return lhs > 0;
|
|
case lp::lconstraint_kind::GE: return lhs >= 0;
|
|
case lp::lconstraint_kind::LE: return lhs <= 0;
|
|
case lp::lconstraint_kind::LT: return lhs < 0;
|
|
case lp::lconstraint_kind::EQ: return lhs == 0;
|
|
case lp::lconstraint_kind::NE: return lhs != 0;
|
|
default: UNREACHABLE(); return false;
|
|
}
|
|
}
|
|
|
|
bool stellensatz2::constraint_is_trivial(lp::constraint_index ci) const {
|
|
return ci != lp::null_ci && constraint_is_trivial(m_constraints[ci]);
|
|
}
|
|
|
|
bool stellensatz2::constraint_is_trivial(constraint const &c) const {
|
|
auto const &[p, k] = c;
|
|
if (!p.is_val())
|
|
return false;
|
|
if (p.val() > 0)
|
|
return k == lp::lconstraint_kind::GE || k == lp::lconstraint_kind::GT || k == lp::lconstraint_kind::NE;
|
|
else if (p.val() == 0)
|
|
return k == lp::lconstraint_kind::EQ || k == lp::lconstraint_kind::GE || k == lp::lconstraint_kind::LE;
|
|
else // p.val() < 0
|
|
return k == lp::lconstraint_kind::LE || k == lp::lconstraint_kind::LT || k == lp::lconstraint_kind::NE;
|
|
}
|
|
|
|
bool stellensatz2::constraint_is_conflict(lp::constraint_index ci) const {
|
|
return ci != lp::null_ci && constraint_is_conflict(m_constraints[ci]);
|
|
}
|
|
|
|
bool stellensatz2::constraint_is_conflict(constraint const& c) const {
|
|
auto const &[p, k] = c;
|
|
return p.is_val() &&
|
|
((p.val() < 0 && k == lp::lconstraint_kind::GE)
|
|
|| (p.val() <= 0 && k == lp::lconstraint_kind::GT));
|
|
}
|
|
|
|
bool stellensatz2::constraint_is_bound_conflict(lp::constraint_index ci) {
|
|
auto const &c = m_constraints[ci];
|
|
auto const& [p, k] = c;
|
|
scoped_dep_interval iv(m_di);
|
|
interval(p, iv);
|
|
if (iv->m_upper_inf)
|
|
return false;
|
|
if (iv->m_upper > 0)
|
|
return false;
|
|
bool is_negative = iv->m_upper < 0 || iv->m_upper_open;
|
|
SASSERT(is_negative || iv->m_upper == 0);
|
|
bool is_conflict = k == lp::lconstraint_kind::GT || (k == lp::lconstraint_kind::GE && is_negative);
|
|
if (!is_conflict)
|
|
return false;
|
|
m_conflict_dep.reset();
|
|
m_dm.linearize(iv->m_lower_dep, m_conflict_dep);
|
|
TRACE(arith, tout << "constraint is bound conflict: "; display_constraint(tout, ci) << "\n";);
|
|
return true;
|
|
}
|
|
|
|
bool stellensatz2::var_is_bound_conflict(lpvar v) {
|
|
return is_bound_conflict(pddm.mk_var(v));
|
|
}
|
|
|
|
bool stellensatz2::is_bound_conflict(dd::pdd const &p) {
|
|
if (!has_lo(p) || !has_hi(p))
|
|
return false;
|
|
if (lo_val(p) < hi_val(p))
|
|
return false;
|
|
if (lo_val(p) == hi_val(p) && !lo_is_strict(p) && !hi_is_strict(p))
|
|
return false;
|
|
m_conflict_dep.reset();
|
|
m_conflict_dep.push_back(lo_constraint(p));
|
|
m_conflict_dep.push_back(hi_constraint(p));
|
|
return true;
|
|
}
|
|
|
|
//
|
|
// Based on current bounds for variables, compute bounds of polynomial
|
|
//
|
|
void stellensatz2::interval(dd::pdd p, scoped_dep_interval& iv) {
|
|
auto &m = iv.m();
|
|
if (p.is_val()) {
|
|
m.set_value(iv, p.val());
|
|
return;
|
|
}
|
|
scoped_dep_interval x(m), lo(m), hi(m);
|
|
auto v = p.var();
|
|
auto lb = get_lower(v);
|
|
if (lb == UINT_MAX) {
|
|
m.set_lower_is_inf(x, true);
|
|
}
|
|
else {
|
|
auto const& lo = m_bounds[lb];
|
|
m.set_lower_is_inf(x, false);
|
|
m.set_lower_is_open(x, lo.m_kind == lp::lconstraint_kind::GT);
|
|
m.set_lower(x, lo.m_value);
|
|
m.set_lower_dep(x, lo.d);
|
|
}
|
|
auto ub = get_upper(v);
|
|
if (ub == UINT_MAX) {
|
|
m.set_upper_is_inf(x, true);
|
|
}
|
|
else {
|
|
auto const& hi = m_bounds[ub];
|
|
m.set_upper_is_inf(x, false);
|
|
m.set_upper_is_open(x, hi.m_kind == lp::lconstraint_kind::LT);
|
|
m.set_upper(x, hi.m_value);
|
|
m.set_upper_dep(x, hi.d);
|
|
}
|
|
interval(p.hi(), hi);
|
|
interval(p.lo(), lo);
|
|
m.mul<dep_intervals::with_deps>(x, hi, hi);
|
|
m.add<dep_intervals::with_deps>(lo, hi, iv);
|
|
}
|
|
|
|
//
|
|
// Main search loop.
|
|
// Resolve conflicts, propagate and case split on bounds.
|
|
//
|
|
lbool stellensatz2::search() {
|
|
init_search();
|
|
lbool is_sat = l_undef;
|
|
while (is_sat == l_undef && c().reslim().inc()) {
|
|
if (is_conflict())
|
|
is_sat = resolve_conflict();
|
|
else if (should_propagate())
|
|
propagate();
|
|
else if (!decide())
|
|
is_sat = l_true;
|
|
}
|
|
if (is_sat == l_true)
|
|
return all_of(m_constraints, [&](auto const &c) { return constraint_is_true(c); }) ? l_true : l_undef;
|
|
return is_sat;
|
|
}
|
|
|
|
void stellensatz2::init_search() {
|
|
m_prop_qhead = 0;
|
|
m_level2var.reset();
|
|
m_var2level.reset();
|
|
for (unsigned v = 0; v < m_values.size(); ++v)
|
|
m_level2var.push_back(v);
|
|
init_levels();
|
|
}
|
|
|
|
void stellensatz2::init_levels() {
|
|
random_gen rand(c().random());
|
|
shuffle(m_level2var.size(), m_level2var.begin(), rand);
|
|
|
|
m_var2level.resize(m_values.size(), m_level2var.size());
|
|
for (unsigned i = 0; i < m_level2var.size(); ++i)
|
|
m_var2level[m_level2var[i]] = i;
|
|
for (auto const &c : m_constraints)
|
|
if (constraint_is_false(c))
|
|
for (auto v : c.p.free_vars())
|
|
move_up(v);
|
|
|
|
for (auto &v : m_max_occurs)
|
|
v.reset();
|
|
m_max_occurs.reserve(num_vars());
|
|
for (unsigned ci = 0; ci < m_constraints.size(); ++ci) {
|
|
auto const &c = m_constraints[ci];
|
|
if (c.p.degree() > m_config.max_degree)
|
|
continue;
|
|
unsigned level = 0;
|
|
lpvar max_var = lp::null_lpvar;
|
|
for (auto v : c.p.free_vars()) {
|
|
if (m_var2level[v] > level) {
|
|
level = m_var2level[v];
|
|
max_var = v;
|
|
}
|
|
}
|
|
if (max_var != lp::null_lpvar)
|
|
m_max_occurs[max_var].push_back(ci);
|
|
}
|
|
}
|
|
|
|
//
|
|
// Conflict resolution is similar to CDCL
|
|
// walk m_bounds based on the propagated bounds
|
|
// flip the last decision and backjump to the UIP.
|
|
//
|
|
lbool stellensatz2::resolve_conflict() {
|
|
TRACE(arith, tout << "resolving conflict: "; display_constraint(tout, m_conflict) << "\n"; display(tout););
|
|
SASSERT(is_conflict());
|
|
|
|
m_conflict_marked_ci.reset();
|
|
unsigned conflict_level = 0;
|
|
for (auto ci : m_conflict_dep) {
|
|
mark_dependencies(ci);
|
|
conflict_level = std::max(conflict_level, m_levels[ci]);
|
|
}
|
|
|
|
bool found_decision = false;
|
|
TRACE(arith, tout << "conflict level " << conflict_level << " constraints: " << m_conflict_marked_ci << "\n");
|
|
unsigned ci = m_constraints.size();
|
|
unsigned sz = ci;
|
|
while (!found_decision && ci > 0 && !m_conflict_marked_ci.empty()) {
|
|
--ci;
|
|
if (!m_conflict_marked_ci.contains(ci))
|
|
continue;
|
|
|
|
mark_dependencies(ci);
|
|
m_conflict_marked_ci.remove(ci);
|
|
m_conflict = resolve(m_conflict, ci);
|
|
if (conflict_level == 0)
|
|
m_core.push_back(ci);
|
|
|
|
found_decision = is_decision(ci);
|
|
|
|
TRACE(arith, tout << "num constraints: " << m_constraints.size() << "\n";
|
|
tout << "is_decision: " << found_decision << "\n"; display_constraint(tout, ci) << "\n";
|
|
tout << "new conflict: "; display_constraint(tout, m_conflict) << "\n";);
|
|
}
|
|
SASSERT(found_decision == (conflict_level != 0));
|
|
if (conflict_level == 0) {
|
|
m_core.reset();
|
|
for (auto ci : m_conflict_marked_ci)
|
|
m_core.push_back(ci);
|
|
reset_conflict();
|
|
return l_false;
|
|
}
|
|
if (constraint_is_conflict(m_conflict)) {
|
|
m_core.reset();
|
|
m_core.push_back(m_conflict);
|
|
reset_conflict();
|
|
return l_false;
|
|
}
|
|
|
|
SASSERT(is_decision(ci));
|
|
|
|
svector<lp::constraint_index> deps;
|
|
for (auto ci : m_conflict_marked_ci)
|
|
deps.push_back(ci);
|
|
|
|
backtrack(ci, deps);
|
|
return l_undef;
|
|
}
|
|
|
|
stellensatz2::constraint stellensatz2::negate_constraint(constraint const &c) {
|
|
auto [p, k] = c;
|
|
switch (k) {
|
|
case lp::lconstraint_kind::GE:
|
|
if (is_int(p)) {
|
|
k = lp::lconstraint_kind::LE;
|
|
p = p - rational(1);
|
|
}
|
|
else
|
|
k = lp::lconstraint_kind::LT;
|
|
break;
|
|
case lp::lconstraint_kind::GT: k = lp::lconstraint_kind::LE; break;
|
|
case lp::lconstraint_kind::LT: k = lp::lconstraint_kind::GE; break;
|
|
case lp::lconstraint_kind::LE:
|
|
if (is_int(p)) {
|
|
k = lp::lconstraint_kind::GE;
|
|
p = p + rational(1);
|
|
}
|
|
else
|
|
k = lp::lconstraint_kind::GT;
|
|
break;
|
|
}
|
|
return {p, k};
|
|
}
|
|
|
|
void stellensatz2::backtrack(unsigned ci, svector<lp::constraint_index> const &deps) {
|
|
auto &[p, k] = m_constraints[ci];
|
|
unsigned propagation_level = 0;
|
|
for (auto ci : deps)
|
|
propagation_level = std::max(propagation_level, m_levels[ci]);
|
|
m_constraint_index.erase({p.index(), k});
|
|
|
|
// propagate negated constraint
|
|
m_constraints[ci] = negate_constraint(m_constraints[ci]);
|
|
m_justifications[ci] = propagation_justification(deps);
|
|
m_levels[ci] = propagation_level;
|
|
m_num_scopes = propagation_level;
|
|
lp::constraint_index head = ci + 1, tail = ci + 1;
|
|
// replay other constraints
|
|
unsigned sz = m_constraints.size();
|
|
svector<lp::constraint_index> tail2head;
|
|
tail2head.resize(sz - ci);
|
|
auto translate_ci = [&](lp::constraint_index old_ci) -> lp::constraint_index {
|
|
return old_ci < ci ? old_ci : tail2head[sz - old_ci];
|
|
};
|
|
for (; tail < m_constraints.size(); ++tail) {
|
|
auto [p, k] = m_constraints[tail];
|
|
auto level = get_level(m_justifications[tail]);
|
|
auto has_occurs = m_has_occurs[tail];
|
|
remove_occurs(tail);
|
|
m_constraint_index.erase({p.index(), k});
|
|
if (level > m_num_scopes)
|
|
continue;
|
|
m_constraints[head] = m_constraints[tail];
|
|
m_justifications[head] = translate_j(translate_ci, m_justifications[tail]);
|
|
m_levels[head] = is_decision(head) ? ++m_num_scopes : get_level(m_justifications[tail]);
|
|
tail2head[sz - tail] = head;
|
|
if (has_occurs)
|
|
init_occurs(head);
|
|
m_constraint_index.insert({p.index(), k}, head);
|
|
++head;
|
|
}
|
|
m_constraints.shrink(head);
|
|
m_justifications.shrink(head);
|
|
m_levels.shrink(head);
|
|
// re-insert bounds
|
|
for (; m_bounds.size() >= ci;)
|
|
pop_bound();
|
|
for (; m_bounds.size() < head;)
|
|
push_bound(m_bounds.size());
|
|
|
|
SASSERT(well_formed_last_bound());
|
|
reset_conflict();
|
|
m_prop_qhead = ci;
|
|
}
|
|
|
|
stellensatz2::justification stellensatz2::translate_j(std::function<lp::constraint_index(lp::constraint_index)> const& f,
|
|
justification const& j) const {
|
|
return std::visit([&](auto const& arg) -> justification {
|
|
using T = std::decay_t<decltype(arg)>;
|
|
if constexpr (std::is_same_v<T, bound_propagation_justification>) {
|
|
svector<lp::constraint_index> cs;
|
|
for (auto ci : arg.cs)
|
|
cs.push_back(f(ci));
|
|
return bound_propagation_justification{f(arg.ci), cs};
|
|
}
|
|
else if constexpr (std::is_same_v<T, assumption_justification>) {
|
|
return assumption_justification{};
|
|
}
|
|
else if constexpr (std::is_same_v<T, external_justification>) {
|
|
return external_justification{arg.dep};
|
|
}
|
|
else if constexpr (std::is_same_v<T, gcd_justification>) {
|
|
return gcd_justification{f(arg.ci)};
|
|
}
|
|
else if constexpr (std::is_same_v<T, substitute_justification>) {
|
|
return substitute_justification{f(arg.ci), f(arg.ci_eq), arg.v, arg.p};
|
|
}
|
|
else if constexpr (std::is_same_v<T, addition_justification>) {
|
|
return addition_justification{f(arg.left), f(arg.right)};
|
|
}
|
|
else if constexpr (std::is_same_v<T, division_justification>) {
|
|
return division_justification{f(arg.ci), f(arg.divisor)};
|
|
}
|
|
else if constexpr (std::is_same_v<T, eq_justification>) {
|
|
return eq_justification{f(arg.left), f(arg.right)};
|
|
}
|
|
else if constexpr (std::is_same_v<T, multiplication_justification>) {
|
|
return multiplication_justification{f(arg.left), f(arg.right)};
|
|
}
|
|
else if constexpr (std::is_same_v<T, multiplication_poly_justification>) {
|
|
return multiplication_poly_justification{f(arg.ci), arg.p};
|
|
}
|
|
else {
|
|
UNREACHABLE();
|
|
return arg;
|
|
}
|
|
}, j);
|
|
}
|
|
|
|
void stellensatz2::mark_dependencies(u_dependency* d) {
|
|
auto cdeps = antecedents(d);
|
|
for (auto a : cdeps)
|
|
m_conflict_marked_ci.insert(a);
|
|
}
|
|
|
|
void stellensatz2::mark_dependencies(lp::constraint_index ci) {
|
|
for (auto cij : justification_range(*this, m_justifications[ci]))
|
|
m_conflict_marked_ci.insert(cij);
|
|
}
|
|
|
|
lp::constraint_index stellensatz2::get_constraint_index(justification const& j, unsigned index) {
|
|
if (std::holds_alternative<bound_propagation_justification>(j)) {
|
|
auto const &bpj = std::get<bound_propagation_justification>(j);
|
|
return index == 0 ? bpj.ci : bpj.cs[index - 1];
|
|
}
|
|
if (std::holds_alternative<assumption_justification>(j)) {
|
|
UNREACHABLE();
|
|
}
|
|
if (std::holds_alternative<external_justification>(j)) {
|
|
UNREACHABLE();
|
|
}
|
|
if (std::holds_alternative<gcd_justification>(j)) {
|
|
SASSERT(index == 0);
|
|
return std::get<gcd_justification>(j).ci;
|
|
}
|
|
if (std::holds_alternative<substitute_justification>(j)) {
|
|
SASSERT(index < 2);
|
|
auto const &sj = std::get<substitute_justification>(j);
|
|
return index == 0 ? sj.ci : sj.ci_eq;
|
|
}
|
|
if (std::holds_alternative<addition_justification>(j)) {
|
|
SASSERT(index < 2);
|
|
auto const &aj = std::get<addition_justification>(j);
|
|
return index == 0 ? aj.left : aj.right;
|
|
}
|
|
if (std::holds_alternative<division_justification>(j)) {
|
|
SASSERT(index < 2);
|
|
auto const &dj = std::get<division_justification>(j);
|
|
return index == 0 ? dj.ci : dj.divisor;
|
|
}
|
|
if (std::holds_alternative<eq_justification>(j)) {
|
|
SASSERT(index < 2);
|
|
auto const &ej = std::get<eq_justification>(j);
|
|
return index == 0 ? ej.left : ej.right;
|
|
}
|
|
if (std::holds_alternative<multiplication_justification>(j)) {
|
|
SASSERT(index < 2);
|
|
auto const &mj = std::get<multiplication_justification>(j);
|
|
return index == 0 ? mj.left : mj.right;
|
|
}
|
|
if (std::holds_alternative<multiplication_poly_justification>(j)) {
|
|
SASSERT(index == 0);
|
|
return std::get<multiplication_poly_justification>(j).ci;
|
|
}
|
|
UNREACHABLE();
|
|
return lp::null_ci;
|
|
}
|
|
|
|
unsigned stellensatz2::num_constraints(justification const &j) {
|
|
if (std::holds_alternative<bound_propagation_justification>(j))
|
|
return 1 + std::get<bound_propagation_justification>(j).cs.size();
|
|
if (std::holds_alternative<assumption_justification>(j))
|
|
return 0;
|
|
if (std::holds_alternative<external_justification>(j))
|
|
return 0;
|
|
if (std::holds_alternative<gcd_justification>(j))
|
|
return 1;
|
|
if (std::holds_alternative<substitute_justification>(j))
|
|
return 2;
|
|
if (std::holds_alternative<addition_justification>(j))
|
|
return 2;
|
|
if (std::holds_alternative<division_justification>(j))
|
|
return 2;
|
|
if (std::holds_alternative<eq_justification>(j))
|
|
return 2;
|
|
if (std::holds_alternative<multiplication_justification>(j))
|
|
return 2;
|
|
if (std::holds_alternative<multiplication_poly_justification>(j))
|
|
return 1;
|
|
UNREACHABLE();
|
|
return 0;
|
|
}
|
|
|
|
void stellensatz2::propagate() {
|
|
if (m_prop_qhead == m_constraints.size())
|
|
return;
|
|
|
|
for (; m_prop_qhead < m_constraints.size(); ++m_prop_qhead) {
|
|
if (!c().reslim().inc())
|
|
return;
|
|
auto q = m_bounds[m_prop_qhead].q;
|
|
if (is_bound_conflict(q))
|
|
return;
|
|
|
|
if (!q.is_var())
|
|
continue;
|
|
auto v = q.var();
|
|
|
|
for (unsigned i = 0; i < m_occurs[v].size(); ++i) {
|
|
lp::constraint_index ci = m_occurs[v][i];
|
|
|
|
TRACE(arith, tout << "propagate v" << v << " (" << ci << ")\n");
|
|
|
|
// detect conflict or propagate dep_intervals
|
|
|
|
if (constraint_is_bound_conflict(ci))
|
|
return;
|
|
|
|
lpvar w;
|
|
rational value;
|
|
lp::lconstraint_kind k;
|
|
svector<lp::constraint_index> cs;
|
|
if (!constraint_is_propagating(ci, cs, w, k, value))
|
|
continue;
|
|
|
|
TRACE(arith, display_constraint(tout << "constraint is propagating ", ci) << "\n";
|
|
tout << "v" << w << " " << k << " " << value << "\n";);
|
|
|
|
// block repeated bounds propagation
|
|
if (propagation_cycle(w, value, k, ci, cs))
|
|
continue;
|
|
|
|
ci = add_constraint(pddm.mk_var(w) - value, k, bound_propagation_justification{ci, cs});
|
|
|
|
if (constraint_is_bound_conflict(ci))
|
|
return;
|
|
|
|
set_in_bounds(w);
|
|
|
|
CTRACE(arith, !well_formed_last_bound(), display(tout));
|
|
SASSERT(well_formed_last_bound());
|
|
SASSERT(well_formed_var(w));
|
|
}
|
|
}
|
|
}
|
|
|
|
bool stellensatz2::propagation_cycle(lpvar v, rational const &value, lp::lconstraint_kind k, lp::constraint_index ci, svector<lp::constraint_index>& cs) const {
|
|
if (!in_bounds(v, value))
|
|
return false;
|
|
auto level = m_levels[ci];
|
|
for (auto a : cs)
|
|
level = std::max(level, m_levels[a]);
|
|
SASSERT(level <= m_num_scopes);
|
|
bool is_upper = k == lp::lconstraint_kind::LE || k == lp::lconstraint_kind::LT;
|
|
auto last_bound = is_upper ? get_upper(v) : get_lower(v);
|
|
while (last_bound != UINT_MAX && m_levels[last_bound] == level) {
|
|
auto const &j = m_justifications[last_bound];
|
|
if (std::holds_alternative<bound_propagation_justification>(j) &&
|
|
ci == std::get<bound_propagation_justification>(j).ci)
|
|
return true;
|
|
last_bound = m_bounds[last_bound].m_last_bound;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
//
|
|
// Attempt to repair variables in some order
|
|
// If hitting a variable that cannot be repaired, create a decision based on the value conflict.
|
|
// Attempts to repair a variable can result in a new conflict obtained from vanishing polynomials
|
|
// or conflicting bounds. The conflicts are assumed to contain variables of lower levels and
|
|
// repair of those constraints are re-attempted.
|
|
// If a variable is in a false constraint that cannot be repaired, pick a non-fixed
|
|
// variable from the constraint and tighten its bound.
|
|
//
|
|
bool stellensatz2::decide() {
|
|
rational value;
|
|
lp::lconstraint_kind k;
|
|
lpvar w;
|
|
|
|
init_levels();
|
|
unsigned num_fixed = 0;
|
|
unsigned num_levels = m_level2var.size();
|
|
unsigned num_rounds = 0;
|
|
unsigned max_rounds = num_levels * 5;
|
|
m_tabu.reset();
|
|
for (unsigned level = 0; level < num_levels && c().reslim().inc() && num_rounds < max_rounds; ++level) {
|
|
w = m_level2var[level];
|
|
++num_rounds;
|
|
lp::constraint_index conflict = repair_variable(w);
|
|
TRACE(arith, display_constraint(tout << "repair lvl:" << level << " v" << w << ": ", conflict)
|
|
<< " in bounds " << in_bounds(w) << " is tabu " << m_tabu.contains(conflict) << "\n");
|
|
if (conflict == lp::null_ci)
|
|
continue;
|
|
SASSERT(constraint_is_false(conflict));
|
|
if (constraint_is_conflict(conflict)) {
|
|
SASSERT(m_conflict_dep.empty());
|
|
m_conflict_dep.push_back(conflict);
|
|
return true;
|
|
}
|
|
|
|
if (m_tabu.contains(conflict)) // already attempted to repair this constraint without success
|
|
continue;
|
|
|
|
m_tabu.insert(conflict);
|
|
|
|
auto p = m_constraints[conflict].p;
|
|
SASSERT(!p.free_vars().empty());
|
|
if (!p.free_vars().contains(w))
|
|
// backtrack decision to max variable in ci
|
|
level = std::min(max_level(m_constraints[conflict]) - 1, level);
|
|
}
|
|
|
|
if (m_num_scopes < 4 && find_split(w, value, k)) {
|
|
SASSERT(k == lp::lconstraint_kind::LE || k == lp::lconstraint_kind::GE);
|
|
bool is_upper = k == lp::lconstraint_kind::LE;
|
|
SASSERT(!is_upper || !has_lo(w) || lo_val(w) <= value);
|
|
SASSERT( is_upper || !has_hi(w) || hi_val(w) >= value);
|
|
add_constraint(pddm.mk_var(w) - value, k, assumption_justification());
|
|
m_values[w] = value;
|
|
TRACE(arith, tout << "decide v" << w << " " << k << " " << value << "\n");
|
|
IF_VERBOSE(3, verbose_stream() << "decide v" << w << " " << k << " " << value << "\n");
|
|
SASSERT(in_bounds(w));
|
|
SASSERT(well_formed_var(w));
|
|
SASSERT(well_formed_last_bound());
|
|
++c().lp_settings().stats().m_stellensatz.m_num_decisions;
|
|
// verbose_stream() << "split " << m_num_scopes << "\n";
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
unsigned stellensatz2::max_level(constraint const &c) const {
|
|
unsigned level = 0;
|
|
auto const &vars = c.p.free_vars();
|
|
for (auto v : vars)
|
|
level = std::max(level, m_var2level[v]);
|
|
return level;
|
|
}
|
|
|
|
unsigned stellensatz2::get_level(justification const& j) const {
|
|
unsigned level = 0;
|
|
for (auto cij : justification_range(*this, j))
|
|
level = std::max(level, m_levels[cij]);
|
|
return level;
|
|
}
|
|
|
|
//
|
|
// Compute intervals for polynomials p, q, in constraint px + q >= 0
|
|
// Determine bounds on x implied by intervals on p, q.
|
|
// If a tighter bound is computed for x, produce the bound propagation.
|
|
//
|
|
bool stellensatz2::constraint_is_propagating(lp::constraint_index ci, svector<lp::constraint_index> &cs, lpvar &v,
|
|
lp::lconstraint_kind &k, rational &value) {
|
|
auto [p, ck] = m_constraints[ci];
|
|
cs.reset();
|
|
for (auto x : p.free_vars()) {
|
|
auto f = factor(x, ci);
|
|
if (f.degree > 1)
|
|
continue;
|
|
scoped_dep_interval ivp(m_di), ivq(m_di);
|
|
interval(f.p, ivp);
|
|
interval(-f.q, ivq);
|
|
TRACE(arith_verbose, tout << "variable v" << x << " in " << p << "\n";
|
|
m_di.display(tout << "interval: " << f.p << ": ", ivp) << "\n";
|
|
m_di.display(tout << "interval: " << -f.q << ": ", ivq) << "\n";
|
|
display_constraint(tout, ci) << "\n");
|
|
|
|
// hi_p > 0
|
|
// 0 <= lo_q <= -q
|
|
// => x >= lo_q / hi_p
|
|
//
|
|
if (!ivp->m_upper_inf && ivp->m_upper > 0 && !ivq->m_lower_inf && ivq->m_lower >= 0) {
|
|
// v >= -q / p
|
|
auto quot = ivq->m_lower / ivp->m_upper;
|
|
if (var_is_int(x))
|
|
quot = ceil(quot);
|
|
bool is_strict = !var_is_int(x) && (ck == lp::lconstraint_kind::GT || ivq->m_lower_open || ivp->m_lower_open);
|
|
if (!has_lo(x) || quot > lo_val(x) || (!lo_is_strict(x) && quot == lo_val(x) && is_strict)) {
|
|
TRACE(arith, tout << "new lower bound v" << x << " " << quot << "\n");
|
|
auto d = m_dm.mk_join(ivq->m_lower_dep, ivp->m_upper_dep);
|
|
m_dm.linearize(d, cs);
|
|
k = is_strict ? lp::lconstraint_kind::GT : lp::lconstraint_kind::GE;
|
|
v = x;
|
|
value = quot;
|
|
return true;
|
|
}
|
|
}
|
|
|
|
// hi_p >= p >= lo_p > 0
|
|
// 0 > hi_q >= -q >= lo_q
|
|
// => x >= lo_q / lo_p
|
|
if (!ivp->m_lower_inf && ivp->m_lower > 0 && !ivq->m_lower_inf && ivq->m_lower <= 0) {
|
|
auto quot = ivq->m_lower / ivp->m_lower;
|
|
if (var_is_int(x))
|
|
quot = ceil(quot);
|
|
bool is_strict =
|
|
!var_is_int(x) && (ck == lp::lconstraint_kind::GT || ivq->m_lower_open || ivp->m_lower_open);
|
|
if (!has_lo(x) || quot > lo_val(x) || (!lo_is_strict(x) && quot == lo_val(x) && is_strict)) {
|
|
TRACE(arith, tout << "new lower bound v" << x << " " << quot << "\n");
|
|
auto d = m_dm.mk_join(ivp->m_lower_dep, ivq->m_lower_dep);
|
|
m_dm.linearize(d, cs);
|
|
k = is_strict ? lp::lconstraint_kind::GT : lp::lconstraint_kind::GE;
|
|
v = x;
|
|
value = quot;
|
|
return true;
|
|
}
|
|
}
|
|
|
|
// p <= hi_p < 0
|
|
// lo_q <= q <= hi_q < 0
|
|
// => x <= lo_q / hi_p
|
|
if (!ivp->m_upper_inf && ivp->m_upper < 0 && !ivq->m_upper_inf && !ivq->m_lower_inf && ivq->m_upper <= 0) {
|
|
// v <= -q / p
|
|
auto quot = ivq->m_lower / ivp->m_upper;
|
|
if (var_is_int(x))
|
|
quot = floor(quot);
|
|
bool is_strict =
|
|
!var_is_int(x) && (ck == lp::lconstraint_kind::GT || ivq->m_lower_open || ivp->m_upper_open);
|
|
if (!has_hi(x) || quot < hi_val(x) || (!hi_is_strict(x) && quot == hi_val(x) && is_strict)) {
|
|
TRACE(arith, tout << "new upper bound v" << x << " " << quot << "\n");
|
|
auto d = m_dm.mk_join(ivq->m_upper_dep, m_dm.mk_join(ivq->m_lower_dep, ivp->m_upper_dep));
|
|
m_dm.linearize(d, cs);
|
|
k = is_strict ? lp::lconstraint_kind::LT : lp::lconstraint_kind::LE;
|
|
v = x;
|
|
value = quot;
|
|
return true;
|
|
}
|
|
}
|
|
// lo_p <= p < 0
|
|
// 0 <= lo_q <= -q
|
|
// => x <= lo_q / lo_p
|
|
//
|
|
if (!ivp->m_upper_inf && ivp->m_upper < 0 && !ivp->m_lower_inf &&
|
|
!ivq->m_lower_inf && ivq->m_lower >= 0) {
|
|
auto quot = ivq->m_lower / ivp->m_lower;
|
|
if (var_is_int(x))
|
|
quot = floor(quot);
|
|
bool is_strict =
|
|
!var_is_int(x) && (ck == lp::lconstraint_kind::GT || ivq->m_lower_open || ivp->m_lower_open);
|
|
if (!has_hi(x) || quot < hi_val(x) || (!hi_is_strict(x) && quot == hi_val(x) && is_strict)) {
|
|
TRACE(arith, tout << "new upper bound v" << x << " " << quot << "\n");
|
|
auto d = m_dm.mk_join(ivp->m_upper_dep, m_dm.mk_join(ivp->m_lower_dep, ivq->m_lower_dep));
|
|
m_dm.linearize(d, cs);
|
|
k = is_strict ? lp::lconstraint_kind::LT : lp::lconstraint_kind::LE;
|
|
v = x;
|
|
value = quot;
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool stellensatz2::well_formed() const {
|
|
for (unsigned v = 0; v < num_vars(); ++v)
|
|
if (!well_formed_var(v))
|
|
return false;
|
|
for (unsigned i = 0; i < m_constraints.size(); ++i)
|
|
if (!well_formed_bound(i))
|
|
return false;
|
|
return true;
|
|
}
|
|
|
|
//
|
|
// integer variables have only non-strict bounds
|
|
// bounds are assigned at monotonically increasing levels
|
|
// previous bounds are the same sign (lower or upper bounds)
|
|
// previous bounds are weaker
|
|
// previous bounds are for the same variable
|
|
//
|
|
bool stellensatz2::well_formed_bound(unsigned bound_index) const {
|
|
return true;
|
|
}
|
|
|
|
//
|
|
// values of variables are within bounds unless the bounds are conflicting
|
|
// m_lower[v], m_upper[v] are annotated by the same variable v.
|
|
//
|
|
bool stellensatz2::well_formed_var(lpvar v) const {
|
|
// if (var_is_bound_conflict(v))
|
|
// return true;
|
|
auto const &value = m_values[v];
|
|
|
|
#if 0
|
|
if (has_lo(v) && value < lo_val(v))
|
|
return false;
|
|
if (has_lo(v) && lo_is_strict(v) && value == lo_val(v))
|
|
return false;
|
|
if (has_lo(v) && lo_kind(v) != lp::lconstraint_kind::GE && lo_kind(v) != lp::lconstraint_kind::GT)
|
|
return false;
|
|
if (has_hi(v) && value > hi_val(v))
|
|
return false;
|
|
if (has_hi(v) && hi_is_strict(v) && value == hi_val(v))
|
|
return false;
|
|
if (has_hi(v) && hi_kind(v) != lp::lconstraint_kind::LE && hi_kind(v) != lp::lconstraint_kind::LT)
|
|
return false;
|
|
#endif
|
|
return true;
|
|
}
|
|
|
|
unsigned_vector stellensatz2::antecedents(u_dependency *d) const {
|
|
unsigned_vector cs;
|
|
m_dm.linearize(d, cs);
|
|
return cs;
|
|
}
|
|
|
|
|
|
void stellensatz2::updt_params(params_ref const& p) {
|
|
smt_params_helper sp(p);
|
|
m_config.max_degree = sp.arith_nl_stellensatz_max_degree();
|
|
m_config.max_conflicts = sp.arith_nl_stellensatz_max_conflicts();
|
|
m_config.max_constraints = sp.arith_nl_stellensatz_max_constraints();
|
|
m_config.strategy = sp.arith_nl_stellensatz_strategy();
|
|
}
|
|
|
|
|
|
// Solver component
|
|
// check LRA feasibilty
|
|
// (partial) check LIA feasibility
|
|
// try patch LIA model
|
|
// option: iterate by continuing saturation based on partial results
|
|
// option: add incremental linearization axioms
|
|
// option: detect squares and add axioms for violated squares
|
|
// option: add NIA filters (non-linear divisbility axioms)
|
|
|
|
|
|
void stellensatz2::solver::init() {
|
|
lra_solver = alloc(lp::lar_solver);
|
|
int_solver = alloc(lp::int_solver, *lra_solver);
|
|
m_ex.clear();
|
|
m_internal2external_constraints.reset();
|
|
m_monomial_factory.reset();
|
|
auto &src = s.c().lra_solver();
|
|
auto &dst = *lra_solver;
|
|
for (unsigned v = 0; v < src.number_of_vars(); ++v)
|
|
dst.add_var(v, src.var_is_int(v));
|
|
|
|
for (lp::constraint_index ci = 0; ci < s.m_constraints.size(); ++ci) {
|
|
auto const &[p, k] = s.m_constraints[ci];
|
|
auto [t, offset] = to_term_offset(p);
|
|
auto coeffs = t.coeffs_as_vector();
|
|
if (coeffs.empty())
|
|
continue;
|
|
SASSERT(!coeffs.empty());
|
|
auto ti = dst.add_term(coeffs, dst.number_of_vars());
|
|
auto new_ci = dst.add_var_bound(ti, k, -offset);
|
|
m_internal2external_constraints.setx(new_ci, ci, ci);
|
|
}
|
|
}
|
|
|
|
stellensatz2::term_offset stellensatz2::solver::to_term_offset(dd::pdd const &p) {
|
|
term_offset to;
|
|
for (auto const &[coeff, vars] : p) {
|
|
if (vars.empty())
|
|
to.second += coeff;
|
|
else
|
|
to.first.add_monomial(coeff, m_monomial_factory.mk_monomial(*lra_solver, vars));
|
|
}
|
|
return to;
|
|
}
|
|
|
|
lbool stellensatz2::solver::solve(svector<lp::constraint_index> &core) {
|
|
init();
|
|
lbool r = solve_lra();
|
|
if (r == l_true)
|
|
r = solve_lia();
|
|
if (r == l_false) {
|
|
core.reset();
|
|
for (auto p : m_ex)
|
|
core.push_back(m_internal2external_constraints[p.ci()]);
|
|
}
|
|
return r;
|
|
}
|
|
|
|
lbool stellensatz2::solver::solve_lra() {
|
|
auto status = lra_solver->find_feasible_solution();;
|
|
if (lra_solver->is_feasible())
|
|
return l_true;
|
|
if (status == lp::lp_status::INFEASIBLE) {
|
|
lra_solver->get_infeasibility_explanation(m_ex);
|
|
return l_false;
|
|
}
|
|
return l_undef;
|
|
}
|
|
|
|
lbool stellensatz2::solver::solve_lia() {
|
|
switch (int_solver->check(&m_ex)) {
|
|
case lp::lia_move::sat:
|
|
return l_true;
|
|
case lp::lia_move::conflict:
|
|
return l_false;
|
|
default: // TODO: an option is to perform (bounded) search here to get an LIA verdict.
|
|
return l_undef;
|
|
}
|
|
return l_undef;
|
|
}
|
|
|
|
// update values using the model
|
|
void stellensatz2::solver::update_values(vector<rational>& values) {
|
|
std::unordered_map<lpvar, rational> _values;
|
|
lra_solver->get_model(_values);
|
|
unsigned sz = values.size();
|
|
for (unsigned i = 0; i < sz; ++i) {
|
|
values[i] = _values[i];
|
|
if (s.var_is_int(i) && !values[i].is_int())
|
|
values[i] = ceil(values[i]);
|
|
}
|
|
TRACE(arith, for (unsigned i = 0; i < sz; ++i) tout << "j" << i << " := " << values[i] << "\n";);
|
|
}
|
|
|
|
lp::constraint_index stellensatz2::repair_variable(lp::lpvar v) {
|
|
auto [inf, sup, conflict, lo, hi] = find_bounds(v);
|
|
|
|
CTRACE(arith, inf != lp::null_ci || sup != lp::null_ci || conflict != lp::null_ci,
|
|
tout << "bounds for v" << v << " @ " << m_var2level[v] << "\n";
|
|
display_constraint(tout << "lo: ", inf) << "\n";
|
|
display_constraint(tout << "hi: ", sup) << "\n";
|
|
display_constraint(tout << "conflict: ", conflict) << "\n");
|
|
|
|
if (conflict != lp::null_ci)
|
|
return conflict;
|
|
|
|
if (inf == lp::null_ci && sup == lp::null_ci)
|
|
return inf;
|
|
|
|
if (!constraint_is_false(inf) && !constraint_is_false(sup))
|
|
return lp::null_ci;
|
|
|
|
TRACE(arith, tout << "v" << v << " @ " << m_var2level[v] << "\n");
|
|
|
|
bool strict_lo = inf != lp::null_ci && m_constraints[inf].k == lp::lconstraint_kind::GT;
|
|
bool strict_hi = sup != lp::null_ci && m_constraints[sup].k == lp::lconstraint_kind::GT;
|
|
bool strict = strict_lo || strict_hi;
|
|
|
|
if (inf == lp::null_ci) {
|
|
SASSERT(sup != lp::null_ci);
|
|
if (strict_hi) {
|
|
if (has_lo(v))
|
|
hi = (hi + lo_val(v)) / 2;
|
|
else
|
|
hi = hi - 1;
|
|
}
|
|
CTRACE(arith, !in_bounds(v, hi), display_var_range(tout << "repair v" << v << " to hi " << hi << "\n", v) << "\n");
|
|
SASSERT(in_bounds(v, hi));
|
|
m_values[v] = hi;
|
|
SASSERT(in_bounds(v));
|
|
}
|
|
else if (sup == lp::null_ci) {
|
|
SASSERT(inf != lp::null_ci);
|
|
if (strict_lo) {
|
|
if (has_hi(v))
|
|
lo = (lo + hi_val(v)) / 2;
|
|
else
|
|
lo = lo + 1;
|
|
}
|
|
CTRACE(arith, !in_bounds(v, lo), display_var_range(tout << "repair v" << v << " to lo " << lo << "\n", v) << "\n");
|
|
SASSERT(in_bounds(v, lo));
|
|
m_values[v] = lo;
|
|
SASSERT(in_bounds(v));
|
|
}
|
|
else if ((!strict && lo <= hi) || lo < hi) {
|
|
// repair v by setting it between lo and hi assuming it is integral when v is integer
|
|
|
|
auto mid = (lo + hi) / 2;
|
|
if (var_is_int(v) && ceil(lo) <= hi)
|
|
mid = ceil(lo);
|
|
|
|
SASSERT(in_bounds(v, mid));
|
|
TRACE(arith, tout << "repair v" << v << " := " << mid << " / " << m_values[v] << " lo: " << lo
|
|
<< " hi: " << hi << "\n");
|
|
m_values[v] = mid;
|
|
SASSERT(in_bounds(v));
|
|
}
|
|
else {
|
|
TRACE(arith, tout << "cannot repair v" << v << " between " << lo << " and " << hi << " " << (lo > hi)
|
|
<< " is int " << var_is_int(v) << "\n";
|
|
display_constraint(tout, inf) << "\n"; display_constraint(tout, sup) << "\n";);
|
|
auto res = resolve_variable(v, inf, sup);
|
|
TRACE(arith, display_constraint(tout << "resolve ", res) << " " << constraint_is_false(res) << "\n");
|
|
if (constraint_is_false(res))
|
|
return res;
|
|
}
|
|
return lp::null_ci;
|
|
}
|
|
|
|
bool stellensatz2::find_split(lpvar &v, rational &r, lp::lconstraint_kind &k) {
|
|
uint_set tried;
|
|
bool found = false;
|
|
unsigned n = 0;
|
|
for (lpvar w = 0; w < m_level2var.size(); ++w) {
|
|
if (var_is_int(w) && !m_values[w].is_int()) {
|
|
if (is_fixed(w))
|
|
continue;
|
|
if (m_split_count[w] > m_config.max_splits_per_var)
|
|
continue;
|
|
if (c().random(++n) != 0)
|
|
continue;
|
|
r = floor(m_values[w]);
|
|
k = lp::lconstraint_kind::LE;
|
|
v = w;
|
|
found = true;
|
|
}
|
|
}
|
|
if (found)
|
|
++m_split_count[v];
|
|
return found;
|
|
}
|
|
|
|
void stellensatz2::set_in_bounds(lpvar v) {
|
|
if (in_bounds(v))
|
|
return;
|
|
if (!has_lo(v))
|
|
m_values[v] = hi_is_strict(v) ? hi_val(v) - 1 : hi_val(v);
|
|
else if (!has_hi(v))
|
|
m_values[v] = lo_is_strict(v) ? lo_val(v) + 1 : lo_val(v);
|
|
else if (lo_is_strict(v) || hi_is_strict(v))
|
|
m_values[v] = (lo_val(v) + hi_val(v)) / 2;
|
|
else
|
|
m_values[v] = lo_val(v);
|
|
SASSERT(in_bounds(v));
|
|
}
|
|
|
|
bool stellensatz2::in_bounds(lpvar v, rational const& value) const {
|
|
if (has_lo(v)) {
|
|
if (value < lo_val(v))
|
|
return false;
|
|
if (lo_is_strict(v) && value <= lo_val(v))
|
|
return false;
|
|
}
|
|
if (has_hi(v)) {
|
|
if (value > hi_val(v))
|
|
return false;
|
|
if (hi_is_strict(v) && value >= hi_val(v))
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
//
|
|
// Enumerate polynomial inequalities p*v + q >= 0
|
|
// where variables in p, q are at levels below v.
|
|
// If M(p) = 0 and M(q) < 0, return q >= 0 and assume p = 0
|
|
// Otherwise, return greatest lower and least upper bounds for v based on polynomial inequalities.
|
|
//
|
|
stellensatz2::repair_var_info stellensatz2::find_bounds(lpvar v) {
|
|
repair_var_info result;
|
|
auto &[inf, sup, van, lo, hi] = result;
|
|
for (auto ci : m_max_occurs[v]) {
|
|
//
|
|
// consider only constraints where v is maximal
|
|
// and where the degree of constraints is bounded
|
|
// for lower and upper bounds only constraints where v
|
|
// occurs pseudo-linearly are considered
|
|
//
|
|
auto const &p = m_constraints[ci].p;
|
|
auto const &vars = p.free_vars();
|
|
TRACE(arith_verbose, display_constraint(tout, ci) << "\n"; for (auto j : vars) tout
|
|
<< "j" << j << " deg: " << p.degree(j)
|
|
<< " lvl: " << m_var2level[j]
|
|
<< "\n";);
|
|
|
|
|
|
auto f = factor(v, ci);
|
|
auto q_ge_0 = vanishing(v, f, ci);
|
|
if (q_ge_0 != lp::null_ci) {
|
|
if (!constraint_is_true(q_ge_0)) {
|
|
van = q_ge_0;
|
|
return result;
|
|
}
|
|
TRACE(arith_verbose, display_constraint(tout << "vanished j" << v << " in ", ci) << "\n";
|
|
display_constraint(tout << " to ", q_ge_0) << "\n";);
|
|
continue;
|
|
}
|
|
|
|
if (f.degree > 1)
|
|
continue;
|
|
auto p_value = value(f.p);
|
|
auto q_value = value(f.q);
|
|
SASSERT(f.degree == 1);
|
|
SASSERT(p_value != 0);
|
|
auto quot = -q_value / p_value;
|
|
if (p_value < 0) {
|
|
if (var_is_int(v))
|
|
quot = floor(quot);
|
|
// p*x + q >= 0 => x <= -q / p if p < 0
|
|
if (sup == lp::null_ci || hi > quot) {
|
|
hi = quot;
|
|
sup = ci;
|
|
}
|
|
else if (hi == quot && m_constraints[ci].k == lp::lconstraint_kind::GT) {
|
|
sup = ci;
|
|
}
|
|
}
|
|
else {
|
|
if (var_is_int(v))
|
|
quot = ceil(quot);
|
|
// p*x + q >= 0 => x >= -q / p if p > 0
|
|
if (inf == lp::null_ci || lo < quot) {
|
|
lo = quot;
|
|
inf = ci;
|
|
}
|
|
else if (lo == quot && m_constraints[ci].k == lp::lconstraint_kind::GT) {
|
|
inf = ci;
|
|
}
|
|
}
|
|
}
|
|
return result;
|
|
}
|
|
|
|
void stellensatz2::move_up(lpvar v) {
|
|
auto l = m_var2level[v];
|
|
if (l == 0)
|
|
return;
|
|
auto l0 = c().random(l);
|
|
auto w = m_level2var[l0];
|
|
m_var2level[v] = l0;
|
|
m_var2level[w] = l;
|
|
m_level2var[l0] = v;
|
|
m_level2var[l] = w;
|
|
}
|
|
|
|
|
|
std::ostream &stellensatz2::display(std::ostream &out) const {
|
|
for (unsigned v = 0; v < num_vars(); ++v)
|
|
display_var_range(out, v) << "\n";
|
|
|
|
for (unsigned ci = 0; ci < m_constraints.size(); ++ci) {
|
|
display_constraint(out, ci) << "\n";
|
|
display(out << "\t<- ", m_justifications[ci]) << "\n";
|
|
}
|
|
return out;
|
|
}
|
|
|
|
std::ostream &stellensatz2::display_var_range(std::ostream &out, lpvar v) const {
|
|
out << "v" << v << " " << m_values[v] << " ";
|
|
if (has_lo(v)) {
|
|
if (lo_is_strict(v))
|
|
out << "(";
|
|
else
|
|
out << "[";
|
|
out << lo_val(v);
|
|
}
|
|
else
|
|
out << "(-oo";
|
|
out << ", ";
|
|
if (has_hi(v)) {
|
|
out << hi_val(v);
|
|
if (hi_is_strict(v))
|
|
out << ")";
|
|
else
|
|
out << "]";
|
|
}
|
|
else
|
|
out << "+oo)";
|
|
if (has_lo(v))
|
|
out << " " << get_lower(v);
|
|
if (has_hi(v))
|
|
out << " " << get_upper(v);
|
|
return out;
|
|
}
|
|
|
|
std::ostream &stellensatz2::display_product(std::ostream &out, svector<lpvar> const &vars) const {
|
|
bool first = true;
|
|
for (auto v : vars) {
|
|
if (first)
|
|
first = false;
|
|
else
|
|
out << " * ";
|
|
out << "j" << v;
|
|
}
|
|
return out;
|
|
}
|
|
|
|
std::ostream &stellensatz2::display(std::ostream &out, term_offset const &t) const {
|
|
bool first = true;
|
|
for (auto [v, coeff] : t.first) {
|
|
c().display_coeff(out, first, coeff);
|
|
first = false;
|
|
out << pp_j(*this, v);
|
|
}
|
|
if (t.second != 0)
|
|
out << " + " << t.second;
|
|
return out;
|
|
}
|
|
|
|
std::ostream &stellensatz2::display_var(std::ostream &out, lpvar j) const {
|
|
if (c().is_monic_var(j))
|
|
return display_product(out, c().emon(j).vars());
|
|
else
|
|
return out << "j" << j;
|
|
}
|
|
|
|
std::ostream &stellensatz2::display_constraint(std::ostream &out, lp::constraint_index ci) const {
|
|
if (ci == lp::null_ci)
|
|
return out << "(null) ";
|
|
bool is_true = constraint_is_true(ci);
|
|
auto const &[p, k] = m_constraints[ci];
|
|
auto level = m_levels[ci];
|
|
return display_constraint(out << "(" << ci << ") @ " << level << " ", m_constraints[ci])
|
|
<< (is_true ? " [true] " : " [false] ") << "(" << value(p) << " " << k << " 0)\n";
|
|
auto const &j = m_justifications[ci];
|
|
display(out, j) << "\n";
|
|
}
|
|
|
|
std::ostream &stellensatz2::display_constraint(std::ostream &out, constraint const &c) const {
|
|
auto const &[p, k] = c;
|
|
p.display(out);
|
|
return out << " " << k << " 0";
|
|
}
|
|
|
|
std::ostream &stellensatz2::display(std::ostream &out, justification const &j) const {
|
|
if (std::holds_alternative<external_justification>(j)) {
|
|
auto dep = std::get<external_justification>(j).dep;
|
|
unsigned_vector cs;
|
|
c().lra_solver().dep_manager().linearize(dep, cs);
|
|
for (auto c : cs)
|
|
out << "[o " << c << "] "; // constraint index from c().lra_solver.
|
|
}
|
|
else if (std::holds_alternative<addition_justification>(j)) {
|
|
auto m = std::get<addition_justification>(j);
|
|
out << "(" << m.left << ") + (" << m.right << ")";
|
|
}
|
|
else if (std::holds_alternative<eq_justification>(j)) {
|
|
auto &m = std::get<eq_justification>(j);
|
|
out << "(" << m.left << ") & (" << m.right << ")";
|
|
}
|
|
else if (std::holds_alternative<substitute_justification>(j)) {
|
|
auto m = std::get<substitute_justification>(j);
|
|
out << "(" << m.ci << ") (" << m.ci_eq << ") by j" << m.v << " := " << m.p;
|
|
}
|
|
else if (std::holds_alternative<propagation_justification>(j)) {
|
|
auto &m = std::get<propagation_justification>(j);
|
|
out << "propagate ";
|
|
for (auto c : m.cs)
|
|
out << "(" << c << ") ";
|
|
}
|
|
else if (std::holds_alternative<multiplication_justification>(j)) {
|
|
auto m = std::get<multiplication_justification>(j);
|
|
out << "(" << m.left << ") * (" << m.right << ")";
|
|
}
|
|
else if (std::holds_alternative<multiplication_poly_justification>(j)) {
|
|
auto m = std::get<multiplication_poly_justification>(j);
|
|
out << m.p << " * (" << m.ci << ")";
|
|
}
|
|
else if (std::holds_alternative<division_justification>(j)) {
|
|
auto m = std::get<division_justification>(j);
|
|
out << "(" << m.ci << ") / (" << m.divisor << ")";
|
|
}
|
|
else if (std::holds_alternative<assumption_justification>(j)) {
|
|
out << "assumption";
|
|
}
|
|
else if (std::holds_alternative<gcd_justification>(j)) {
|
|
auto m = std::get<gcd_justification>(j);
|
|
out << " gcd (" << m.ci << ")";
|
|
}
|
|
else
|
|
UNREACHABLE();
|
|
return out;
|
|
}
|
|
|
|
std::ostream &stellensatz2::display_lemma(std::ostream &out, lp::explanation const &ex) {
|
|
m_constraints_in_conflict.reset();
|
|
svector<lp::constraint_index> todo;
|
|
for (auto c : ex)
|
|
todo.push_back(c.ci());
|
|
for (unsigned i = 0; i < todo.size(); ++i) {
|
|
auto ci = todo[i];
|
|
if (m_constraints_in_conflict.contains(ci))
|
|
continue;
|
|
m_constraints_in_conflict.insert(ci);
|
|
out << "(" << ci << ") ";
|
|
display_constraint(out, ci);
|
|
auto const& j = m_justifications[ci];
|
|
for (auto cij : justification_range(*this, j))
|
|
todo.push_back(cij);
|
|
}
|
|
return out;
|
|
}
|
|
|
|
//
|
|
// collect variables that are defined x + p >= 0: d1, -x - p >= 0 : d2.
|
|
// substitute x by p elsewhere.
|
|
// Accumulate dependencies from d1, d2 into substituted (can be a propagation justification)
|
|
//
|
|
void stellensatz2::simplify() {
|
|
using var_ci_vector = svector<std::pair<unsigned, lp::constraint_index>>;
|
|
struct eq_info {
|
|
lpvar v;
|
|
dd::pdd eq;
|
|
lp::constraint_index ci1, ci2;
|
|
};
|
|
u_map<var_ci_vector> bounds;
|
|
vector<dd::pdd> pdds;
|
|
vector<eq_info> eqs;
|
|
for (unsigned ci = 0; ci < m_constraints.size(); ++ci) {
|
|
auto const& [p, k] = m_constraints[ci];
|
|
bool is_le = true;
|
|
if (k != lp::lconstraint_kind::GE)
|
|
continue;
|
|
for (auto const &[coeff, vars] : p) {
|
|
if (vars.size() != 1)
|
|
continue;
|
|
if (coeff != 1 && coeff != -1)
|
|
continue;
|
|
auto v = vars[0];
|
|
auto &qs = bounds.insert_if_not_there(v, var_ci_vector());
|
|
auto mp = -p;
|
|
for (auto [q, ci2] : qs) {
|
|
auto Q = pdds[q];
|
|
if (mp != Q)
|
|
continue;
|
|
eqs.push_back({v, coeff == 1 ? -p : p, ci, ci2});
|
|
}
|
|
qs.push_back({pdds.size(), ci});
|
|
pdds.push_back(p);
|
|
}
|
|
}
|
|
uint_set tracked;
|
|
for (auto [v, p, ci1, ci2] : eqs) {
|
|
if (tracked.contains(ci1) || tracked.contains(ci2))
|
|
continue;
|
|
auto q = p + pddm.mk_var(v);
|
|
if (q.free_vars().contains(v))
|
|
continue;
|
|
tracked.insert(ci1);
|
|
tracked.insert(ci2);
|
|
unsigned sz = m_constraints.size();
|
|
for (unsigned ci = 0; ci < sz; ++ci) {
|
|
auto const &[p, k] = m_constraints[ci];
|
|
if (!p.free_vars().contains(v))
|
|
continue;
|
|
auto r = p.subst_pdd(v, q);
|
|
if (r.is_val())
|
|
continue;
|
|
|
|
svector<lp::constraint_index> assumptions;
|
|
assumptions.push_back(ci1);
|
|
assumptions.push_back(ci2);
|
|
assumptions.push_back(ci);
|
|
propagation_justification prop{assumptions};
|
|
add_constraint(r, k, prop);
|
|
}
|
|
}
|
|
}
|
|
}
|