mirrors/z3
3
0
Fork 0
mirror of https://github.com/Z3Prover/z3 synced 2025-06-28 08:58:44 +00:00
Code Activity

simplify and order nla_expr

Browse source 
Signed-off-by: Lev Nachmanson <levnach@hotmail.com>
This commit is contained in:
Lev Nachmanson 2019-07-15 20:27:30 -07:00
parent 8eaa2bfb02
commit 9cba261a9c
2 changed files with 115 additions and 61 deletions
Download patch file Download diff file Expand all files Collapse all files

69
src/math/lp/nla_expr.h
View file

@ -125,6 +125,43 @@ public:
m_children.sort(); m_children.sort();
} }
} }
void simplify() {
if (is_sum()) {
bool has_sum = false;
for (unsigned j = 0; j < m_children.es().size(); j++) {
auto& e = m_children.es()[j];
e.simplify();
if (e.is_sum())
has_sum = true;
}
if (has_sum) {
nla_expr n(expr_type::SUM);
for (auto &e : m_children.es()) {
n += e;
}
*this = n;
}
} else if (is_mul()) {
bool has_mul = false;
for (unsigned j = 0; j < m_children.es().size(); j++) {
auto& e = m_children.es()[j];
e.simplify();
if (e.is_mul())
has_mul = true;
}
if (has_mul) {
nla_expr n(expr_type::MUL);
for (auto &e : m_children.es()) {
n *= e;
}
*this = n;
}
TRACE("nla_cn", tout << "simplified " << *this << "\n";);
}
}
std::ostream & print_mul(std::ostream& out) const { std::ostream & print_mul(std::ostream& out) const {
bool first = true; bool first = true;
for (unsigned j : m_children.order()) { for (unsigned j : m_children.order()) {
@ -278,10 +315,36 @@ public:
SASSERT(false); SASSERT(false);
return false; return false;
} }
} }
nla_expr& operator*=(const nla_expr& b) {
if (is_mul()) {
if (b.is_mul()) {
for (auto& e: b.children())
add_child(e);
} else {
add_child(b);
}
return *this;
}
SASSERT(false); // not impl
return *this;
}
nla_expr& operator+=(const nla_expr& b) {
if (is_sum()) {
if (b.is_sum()) {
for (auto& e: b.children())
add_child(e);
} else {
add_child(b);
}
return *this;
}
SASSERT(false); // not impl
return *this;
}
}; };
template <typename T> template <typename T>
nla_expr<T> operator+(const nla_expr<T>& a, const nla_expr<T>& b) { nla_expr<T> operator+(const nla_expr<T>& a, const nla_expr<T>& b) {

43
src/test/lp/lp.cpp
View file

@ -67,37 +67,28 @@ void test_basic_lemma_for_mon_zero_from_factors_to_monomial();
void test_basic_lemma_for_mon_neutral_from_monomial_to_factors(); void test_basic_lemma_for_mon_neutral_from_monomial_to_factors();
void test_basic_lemma_for_mon_neutral_from_factors_to_monomial(); void test_basic_lemma_for_mon_neutral_from_factors_to_monomial();
void test_cn_on_expr(horner::nex t) {
TRACE("nla_cn", tout << "t=" << t << '\n';);
cross_nested cn(t, [](const horner::nex& n) {
TRACE("nla_cn", tout << n << "\n";);
auto nn = n;
nn.simplify();
nn.sort();
TRACE("nla_cn", tout << "ordered version\n" << nn << "\n______________________\n";);
} );
cn.run();
}
void test_cn() { void test_cn() {
typedef horner::nex nex; typedef horner::nex nex;
enable_trace("nla_cn"); enable_trace("nla_cn");
// (a(a+(b+c)c+d)d + e(a(e+c)+d) enable_trace("nla_cn_details");
nex a = nex::var(0), b = nex::var(1), c = nex::var(2), d = nex::var(3), e = nex::var(4); nex a = nex::var(0), b = nex::var(1), c = nex::var(2), d = nex::var(3), e = nex::var(4);
{
nex t = a*a*d + a*b*c*d + a*c*c*d + a*d*d + e*a*e + e*a*c + e*d;
std::cout << "t = " << t << "\n";
TRACE("nla_cn", tout << "t=" << t << '\n';);
cross_nested cn(t, [](const nex& n) {
std::cout << n << "\n";
auto nn = n;
nn.sort();
std::cout << "ordered version\n" << nn << "\n______________________\n";
} ); nex t = a*a*d + a*b*c*d + a*a*c*c*d + a*d*d + e*a*e + e*a*c + e*d;
cn.run(); test_cn_on_expr(t);
} test_cn_on_expr(a*b*d + a*b*c);
{
nex t = a*b*d + a*b*c;
std::cout << "t = " << t << "\n";
TRACE("nla_cn", tout << "t=" << t << '\n';);
cross_nested cn(t, [](const nex& n) {
std::cout << n << "\n";
auto nn = n;
nn.sort();
std::cout << "ordered version\n" << nn << "\n______________________\n";
} );
cn.run();
}
} }
} // end of namespace nla } // end of namespace nla