diff options
author | Gereon Kremer <gereon.kremer@cs.rwth-aachen.de> | 2020-07-17 09:06:31 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2020-07-17 00:06:31 -0700 |
commit | 0a7e733a5cee4733ca8ca9fff1f6eab6fc22a549 (patch) | |
tree | 8df7919ddfa7e3d3b6207f61edddfac6ed204305 /src/util/poly_util.cpp | |
parent | 2ee5a2bcf5fd7aaf72d44553ebb85edd76fd06c8 (diff) |
Integration of libpoly (#4679)
This commit integrates LibPoly into CVC4. It adds `contrib/get-poly`, adds it to the configure script, cmake and places where CVC4 inspects its own build configuration.
Furthermore, it adds `CVC4::RealAlgebraicNumber` (which wraps `poly::AlgebraicNumber`) including some basic unit tests and some utilities.
Diffstat (limited to 'src/util/poly_util.cpp')
-rw-r--r-- | src/util/poly_util.cpp | 282 |
1 files changed, 282 insertions, 0 deletions
diff --git a/src/util/poly_util.cpp b/src/util/poly_util.cpp new file mode 100644 index 000000000..b4c5d1bf2 --- /dev/null +++ b/src/util/poly_util.cpp @@ -0,0 +1,282 @@ +/********************* */ +/*! \file poly_util.cpp + ** \verbatim + ** Top contributors (to current version): + ** Gereon Kremer + ** This file is part of the CVC4 project. + ** Copyright (c) 2009-2020 by the authors listed in the file AUTHORS + ** in the top-level source directory) and their institutional affiliations. + ** All rights reserved. See the file COPYING in the top-level source + ** directory for licensing information.\endverbatim + ** + ** This file is part of the CVC4 project. + ** Copyright (c) 2009-2020 by the authors listed in the file AUTHORS + ** in the top-level source directory) and their institutional affiliations. + ** All rights reserved. See the file COPYING in the top-level source + ** directory for licensing information.\endverbatim + ** + ** \brief Utilities for working with LibPoly. + ** + ** Utilities for working with LibPoly. + **/ + +#include "poly_util.h" + +#ifdef CVC4_POLY_IMP + +#include <poly/polyxx.h> + +#include <map> + +#include "base/check.h" +#include "maybe.h" +#include "util/integer.h" +#include "util/rational.h" +#include "util/real_algebraic_number.h" + +namespace CVC4 { +namespace poly_utils { + +namespace { +/** + * Convert arbitrary data using a string as intermediary. + * Assumes the existence of operator<<(std::ostream&, const From&) and To(const + * std::string&); should be the last resort for type conversions: it may not + * only yield bad performance, but is also dependent on compatible string + * representations. Use with care! + */ +template <typename To, typename From> +To cast_by_string(const From& f) +{ + std::stringstream s; + s << f; + return To(s.str()); +} +} // namespace + +Integer toInteger(const poly::Integer& i) +{ + const mpz_class& gi = *poly::detail::cast_to_gmp(&i); +#ifdef CVC4_GMP_IMP + return Integer(gi); +#endif +#ifdef CVC4_CLN_IMP + if (std::numeric_limits<long>::min() <= gi + && gi <= std::numeric_limits<long>::max()) + { + return Integer(gi.get_si()); + } + else + { + return cast_by_string<Integer, poly::Integer>(i); + } +#endif +} +Rational toRational(const poly::Integer& i) { return Rational(toInteger(i)); } +Rational toRational(const poly::Rational& r) +{ +#ifdef CVC4_GMP_IMP + return Rational(*poly::detail::cast_to_gmp(&r)); +#endif +#ifdef CVC4_CLN_IMP + return Rational(toInteger(numerator(r)), toInteger(denominator(r))); +#endif +} +Rational toRational(const poly::DyadicRational& dr) +{ + return Rational(toInteger(numerator(dr)), toInteger(denominator(dr))); +} + +poly::Integer toInteger(const Integer& i) +{ +#ifdef CVC4_GMP_IMP + return poly::Integer(i.getValue()); +#endif +#ifdef CVC4_CLN_IMP + if (std::numeric_limits<long>::min() <= i.getValue() + && i.getValue() <= std::numeric_limits<long>::max()) + { + return poly::Integer(cln::cl_I_to_long(i.getValue())); + } + else + { + return poly::Integer(cast_by_string<mpz_class, Integer>(i)); + } +#endif +} +std::vector<poly::Integer> toInteger(const std::vector<Integer>& vi) +{ + std::vector<poly::Integer> res; + for (const auto& i : vi) res.emplace_back(toInteger(i)); + return res; +} +poly::Rational toRational(const Rational& r) +{ +#ifdef CVC4_GMP_IMP + return poly::Rational(r.getValue()); +#endif +#ifdef CVC4_CLN_IMP + return poly::Rational(toInteger(r.getNumerator()), + toInteger(r.getDenominator())); +#endif +} + +Maybe<poly::DyadicRational> toDyadicRational(const Rational& r) +{ + Integer den = r.getDenominator(); + if (den.isOne()) + { // It's an integer anyway. + return poly::DyadicRational(toInteger(r.getNumerator())); + } + unsigned long exp = den.isPow2(); + if (exp > 0) + { + // It's a dyadic rational. + return div_2exp(poly::DyadicRational(toInteger(r.getNumerator())), exp - 1); + } + return Maybe<poly::DyadicRational>(); +} + +Maybe<poly::DyadicRational> toDyadicRational(const poly::Rational& r) +{ + poly::Integer den = denominator(r); + if (den == poly::Integer(1)) + { // It's an integer anyway. + return poly::DyadicRational(numerator(r)); + } + // Use bit_size as an estimate for the dyadic exponent. + unsigned long size = bit_size(den) - 1; + if (mul_pow2(poly::Integer(1), size) == den) + { + // It's a dyadic rational. + return div_2exp(poly::DyadicRational(numerator(r)), size); + } + return Maybe<poly::DyadicRational>(); +} + +poly::Rational approximateToDyadic(const poly::Rational& r, + const poly::Rational& original) +{ + // Multiply both numerator and denominator by two. + // Increase or decrease the numerator, depending on whether r is too small or + // too large. + poly::Integer n = mul_pow2(numerator(r), 1); + if (r < original) + { + ++n; + } + else if (r > original) + { + --n; + } + return poly::Rational(n, mul_pow2(denominator(r), 1)); +} + +poly::AlgebraicNumber toPolyRanWithRefinement(poly::UPolynomial&& p, + const Rational& lower, + const Rational& upper) +{ + Maybe<poly::DyadicRational> ml = toDyadicRational(lower); + Maybe<poly::DyadicRational> mu = toDyadicRational(upper); + if (ml && mu) + { + return poly::AlgebraicNumber(std::move(p), + poly::DyadicInterval(ml.value(), mu.value())); + } + // The encoded real algebraic number did not have dyadic rational endpoints. + poly::Rational origl = toRational(lower); + poly::Rational origu = toRational(upper); + poly::Rational l(floor(origl)); + poly::Rational u(ceil(origu)); + poly::RationalInterval ri(l, u); + while (count_real_roots(p, ri) != 1) + { + l = approximateToDyadic(l, origl); + u = approximateToDyadic(u, origu); + ri = poly::RationalInterval(l, u); + } + Assert(count_real_roots(p, poly::RationalInterval(l, u)) == 1); + ml = toDyadicRational(l); + mu = toDyadicRational(u); + Assert(ml && mu) << "Both bounds should be dyadic by now."; + return poly::AlgebraicNumber(std::move(p), + poly::DyadicInterval(ml.value(), mu.value())); +} + +RealAlgebraicNumber toRanWithRefinement(poly::UPolynomial&& p, + const Rational& lower, + const Rational& upper) +{ + return RealAlgebraicNumber( + toPolyRanWithRefinement(std::move(p), lower, upper)); +} + +std::size_t totalDegree(const poly::Polynomial& p) +{ + std::size_t tdeg = 0; + + lp_polynomial_traverse_f f = + [](const lp_polynomial_context_t* ctx, lp_monomial_t* m, void* data) { + std::size_t sum = 0; + for (std::size_t i = 0; i < m->n; ++i) + { + sum += m->p[i].d; + } + + std::size_t* td = static_cast<std::size_t*>(data); + *td = std::max(*td, sum); + }; + + lp_polynomial_traverse(p.get_internal(), f, &tdeg); + + return tdeg; +} + +struct GetVarInfo +{ + VariableInformation* info; + std::size_t cur_var_degree = 0; + std::size_t cur_lc_degree = 0; +}; +void getVariableInformation(VariableInformation& vi, + const poly::Polynomial& poly) +{ + GetVarInfo varinfo; + varinfo.info = &vi; + lp_polynomial_traverse_f f = + [](const lp_polynomial_context_t* ctx, lp_monomial_t* m, void* data) { + GetVarInfo* gvi = static_cast<GetVarInfo*>(data); + VariableInformation* info = gvi->info; + // Total degree of this term + std::size_t tdeg = 0; + // Degree of this variable within this term + std::size_t vardeg = 0; + for (std::size_t i = 0; i < m->n; ++i) + { + tdeg += m->p[i].d; + if (m->p[i].x == info->var) + { + info->max_degree = std::max(info->max_degree, m->p[i].d); + info->sum_degree += m->p[i].d; + ++info->num_terms; + vardeg = m->p[i].d; + } + } + if (vardeg > 0) + { + if (gvi->cur_var_degree < vardeg) + { + gvi->cur_lc_degree = tdeg - vardeg; + } + info->max_terms_tdegree = std::max(info->max_terms_tdegree, tdeg); + } + }; + + lp_polynomial_traverse(poly.get_internal(), f, &varinfo); + vi.max_lc_degree = std::max(vi.max_lc_degree, varinfo.cur_lc_degree); +} + +} // namespace poly_utils +} // namespace CVC4 + +#endif |