mirrors/z3
3
0
Fork 0
mirror of https://github.com/Z3Prover/z3 synced 2025-04-24 09:35:32 +00:00
Code Activity

univariate::find_max

Browse source
This commit is contained in:
Jakob Rath 2022-12-15 15:59:15 +01:00
parent 31e0add966
commit 8a5f1af3d1
3 changed files with 117 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

69
src/math/polysat/univariate/univariate_solver.cpp
View file

@ -35,6 +35,7 @@ namespace polysat {
unsigned bit_width;
func_decl_ref x_decl;
expr_ref x;
vector<rational> model_cache;
public:
univariate_bitblast_solver(solver_factory& mk_solver, unsigned bit_width) :
@ -48,15 +49,30 @@ namespace polysat {
s = mk_solver(m, p, false, true, true, symbol::null);
x_decl = m.mk_const_decl("x", bv->mk_sort(bit_width));
x = m.mk_const(x_decl);
model_cache.push_back(rational(-1));
}
~univariate_bitblast_solver() override = default;
void reset_cache() {
model_cache.back() = -1;
}
void push_cache() {
model_cache.push_back(model_cache.back());
}
void pop_cache() {
model_cache.pop_back();
}
void push() override {
push_cache();
s->push();
}
void pop(unsigned n) override {
pop_cache();
s->pop(n);
}
@ -114,6 +130,7 @@ namespace polysat {
#endif
void add(expr* e, bool sign, dep_t dep) {
reset_cache();
if (sign)
e = m.mk_not(e);
expr* a = m.mk_const(m.mk_const_decl(symbol(dep), m.mk_bool_sort()));
@ -194,15 +211,19 @@ namespace polysat {
}
rational model() override {
model_ref model;
s->get_model(model);
SASSERT(model);
app* val = to_app(model->get_const_interp(x_decl));
SASSERT(val->get_decl_kind() == OP_BV_NUM);
SASSERT(val->get_num_parameters() == 2);
auto const& p = val->get_parameter(0);
SASSERT(p.is_rational());
return p.get_rational();
rational& cached_model = model_cache.back();
if (cached_model.is_neg()) {
model_ref model;
s->get_model(model);
SASSERT(model);
app* val = to_app(model->get_const_interp(x_decl));
SASSERT(val->get_decl_kind() == OP_BV_NUM);
SASSERT(val->get_num_parameters() == 2);
auto const& p = val->get_parameter(0);
SASSERT(p.is_rational());
cached_model = p.get_rational();
}
return cached_model;
}
bool find_min(rational& val) override {
@ -214,6 +235,7 @@ namespace polysat {
add_bit0(k, 0);
continue;
}
// try decreasing k-th bit
push();
add_bit0(k, 0);
lbool result = check();
@ -233,6 +255,35 @@ namespace polysat {
return true;
}
bool find_max(rational& val) override {
val = model();
push();
// try increasing val by setting bits to 1, starting at the msb.
for (unsigned k = bit_width; k-- > 0; ) {
if (val.get_bit(k)) {
add_bit1(k, 0);
continue;
}
// try increasing k-th bit
push();
add_bit1(k, 0);
lbool result = check();
if (result == l_true) {
SASSERT(model() > val);
val = model();
}
pop(1);
if (result == l_true)
add_bit1(k, 0);
else if (result == l_false)
add_bit0(k, 0);
else
return false;
}
pop(1);
return true;
}
std::ostream& display(std::ostream& out) const override {
return out << *s;
}

24
src/math/polysat/univariate/univariate_solver.h
View file

@ -42,16 +42,32 @@ namespace polysat {
virtual lbool check() = 0;
// Precondition: check() returned l_false
/**
* Precondition: check() returned l_false
*/
virtual dep_vector unsat_core() = 0;
// Precondition: check() returned l_true
/**
* Precondition: check() returned l_true
*/
virtual rational model() = 0;
// Precondition: check() returned l_true
// Returns false on resource out.
/**
* Find minimal model.
*
* Precondition: check() returned l_true
* Returns: true on success, false on resource out.
*/
virtual bool find_min(rational& out_min) = 0;
/**
* Find maximal model.
*
* Precondition: check() returned l_true
* Returns: true on success, false on resource out.
*/
virtual bool find_max(rational& out_max) = 0;
virtual void add_ule(univariate const& lhs, univariate const& rhs, bool sign, dep_t dep) = 0;
virtual void add_umul_ovfl(univariate const& lhs, univariate const& rhs, bool sign, dep_t dep) = 0;
virtual void add_smul_ovfl(univariate const& lhs, univariate const& rhs, bool sign, dep_t dep) = 0;