X-Git-Url: https://www.ginac.de/ginac.git//ginac.git?a=blobdiff_plain;f=ginac%2Fnormal.cpp;h=e9117c4d163ecb88a676c0f1dd47ec3282ceed16;hb=d03664462aedcf6f300f26c6492b3ca3327a640f;hp=369e2c656449ff296385397b1bce19c1b1b356af;hpb=94c15f8b02a5a9e274d3aedcefd76565861b4219;p=ginac.git diff --git a/ginac/normal.cpp b/ginac/normal.cpp index 369e2c65..e9117c4d 100644 --- a/ginac/normal.cpp +++ b/ginac/normal.cpp @@ -36,10 +36,10 @@ #include "inifcns.h" #include "lst.h" #include "mul.h" -#include "ncmul.h" #include "numeric.h" #include "power.h" #include "relational.h" +#include "matrix.h" #include "pseries.h" #include "symbol.h" #include "utils.h" @@ -93,7 +93,7 @@ static struct _stat_print { static bool get_first_symbol(const ex &e, const symbol *&x) { if (is_ex_exactly_of_type(e, symbol)) { - x = static_cast(e.bp); + x = &ex_to(e); return true; } else if (is_ex_exactly_of_type(e, add) || is_ex_exactly_of_type(e, mul)) { for (unsigned i=0; i sym_desc_vec; // Add symbol the sym_desc_vec (used internally by get_symbol_stats()) static void add_symbol(const symbol *s, sym_desc_vec &v) { - sym_desc_vec::iterator it = v.begin(), itend = v.end(); + sym_desc_vec::const_iterator it = v.begin(), itend = v.end(); while (it != itend) { if (it->sym->compare(*s) == 0) // If it's already in there, don't add it a second time return; - it++; + ++it; } sym_desc d; d.sym = s; @@ -170,7 +170,7 @@ static void add_symbol(const symbol *s, sym_desc_vec &v) static void collect_symbols(const ex &e, sym_desc_vec &v) { if (is_ex_exactly_of_type(e, symbol)) { - add_symbol(static_cast(e.bp), v); + add_symbol(&ex_to(e), v); } else if (is_ex_exactly_of_type(e, add) || is_ex_exactly_of_type(e, mul)) { for (unsigned i=0; imax_lcnops = std::max(a.lcoeff(*(it->sym)).nops(), b.lcoeff(*(it->sym)).nops()); it->ldeg_a = a.ldegree(*(it->sym)); it->ldeg_b = b.ldegree(*(it->sym)); - it++; + ++it; } - sort(v.begin(), v.end()); + std::sort(v.begin(), v.end()); #if 0 std::clog << "Symbols:\n"; it = v.begin(); itend = v.end(); while (it != itend) { std::clog << " " << *it->sym << ": deg_a=" << it->deg_a << ", deg_b=" << it->deg_b << ", ldeg_a=" << it->ldeg_a << ", ldeg_b=" << it->ldeg_b << ", max_deg=" << it->max_deg << ", max_lcnops=" << it->max_lcnops << endl; std::clog << " lcoeff_a=" << a.lcoeff(*(it->sym)) << ", lcoeff_b=" << b.lcoeff(*(it->sym)) << endl; - it++; + ++it; } #endif } @@ -229,7 +229,7 @@ static void get_symbol_stats(const ex &a, const ex &b, sym_desc_vec &v) static numeric lcmcoeff(const ex &e, const numeric &l) { if (e.info(info_flags::rational)) - return lcm(ex_to_numeric(e).denom(), l); + return lcm(ex_to(e).denom(), l); else if (is_ex_exactly_of_type(e, add)) { numeric c = _num1(); for (unsigned i=0; i(e.op(1))); } return l; } @@ -269,25 +269,27 @@ static numeric lcm_of_coefficients_denominators(const ex &e) static ex multiply_lcm(const ex &e, const numeric &lcm) { if (is_ex_exactly_of_type(e, mul)) { - ex c = _ex1(); + unsigned num = e.nops(); + exvector v; v.reserve(num + 1); numeric lcm_accum = _num1(); for (unsigned i=0; isetflag(status_flags::dynallocated); } else if (is_ex_exactly_of_type(e, add)) { - ex c = _ex0(); - for (unsigned i=0; isetflag(status_flags::dynallocated); } else if (is_ex_exactly_of_type(e, power)) { if (is_ex_exactly_of_type(e.op(0), symbol)) return e * lcm; else - return pow(multiply_lcm(e.op(0), lcm.power(ex_to_numeric(e.op(1)).inverse())), e.op(1)); + return pow(multiply_lcm(e.op(0), lcm.power(ex_to(e.op(1)).inverse())), e.op(1)); } else return e * lcm; } @@ -320,13 +322,13 @@ numeric add::integer_content(void) const epvector::const_iterator itend = seq.end(); numeric c = _num0(); while (it != itend) { - GINAC_ASSERT(!is_ex_exactly_of_type(it->rest,numeric)); - GINAC_ASSERT(is_ex_exactly_of_type(it->coeff,numeric)); - c = gcd(ex_to_numeric(it->coeff), c); + GINAC_ASSERT(!is_exactly_a(it->rest)); + GINAC_ASSERT(is_exactly_a(it->coeff)); + c = gcd(ex_to(it->coeff), c); it++; } - GINAC_ASSERT(is_ex_exactly_of_type(overall_coeff,numeric)); - c = gcd(ex_to_numeric(overall_coeff),c); + GINAC_ASSERT(is_exactly_a(overall_coeff)); + c = gcd(ex_to(overall_coeff),c); return c; } @@ -336,12 +338,12 @@ numeric mul::integer_content(void) const epvector::const_iterator it = seq.begin(); epvector::const_iterator itend = seq.end(); while (it != itend) { - GINAC_ASSERT(!is_ex_exactly_of_type(recombine_pair_to_ex(*it),numeric)); + GINAC_ASSERT(!is_exactly_a(recombine_pair_to_ex(*it))); ++it; } #endif // def DO_GINAC_ASSERT - GINAC_ASSERT(is_ex_exactly_of_type(overall_coeff,numeric)); - return abs(ex_to_numeric(overall_coeff)); + GINAC_ASSERT(is_exactly_a(overall_coeff)); + return abs(ex_to(overall_coeff)); } @@ -372,7 +374,6 @@ ex quo(const ex &a, const ex &b, const symbol &x, bool check_args) throw(std::invalid_argument("quo: arguments must be polynomials over the rationals")); // Polynomial long division - ex q = _ex0(); ex r = a.expand(); if (r.is_zero()) return r; @@ -380,6 +381,7 @@ ex quo(const ex &a, const ex &b, const symbol &x, bool check_args) int rdeg = r.degree(x); ex blcoeff = b.expand().coeff(x, bdeg); bool blcoeff_is_numeric = is_ex_exactly_of_type(blcoeff, numeric); + exvector v; v.reserve(rdeg - bdeg + 1); while (rdeg >= bdeg) { ex term, rcoeff = r.coeff(x, rdeg); if (blcoeff_is_numeric) @@ -389,13 +391,13 @@ ex quo(const ex &a, const ex &b, const symbol &x, bool check_args) return (new fail())->setflag(status_flags::dynallocated); } term *= power(x, rdeg - bdeg); - q += term; + v.push_back(term); r -= (term * b).expand(); if (r.is_zero()) break; rdeg = r.degree(x); } - return q; + return (new add(v))->setflag(status_flags::dynallocated); } @@ -416,7 +418,7 @@ ex rem(const ex &a, const ex &b, const symbol &x, bool check_args) if (is_ex_exactly_of_type(b, numeric)) return _ex0(); else - return b; + return a; } #if FAST_COMPARE if (a.is_equal(b)) @@ -451,6 +453,24 @@ ex rem(const ex &a, const ex &b, const symbol &x, bool check_args) } +/** Decompose rational function a(x)=N(x)/D(x) into P(x)+n(x)/D(x) + * with degree(n, x) < degree(D, x). + * + * @param a rational function in x + * @param x a is a function of x + * @return decomposed function. */ +ex decomp_rational(const ex &a, const symbol &x) +{ + ex nd = numer_denom(a); + ex numer = nd.op(0), denom = nd.op(1); + ex q = quo(numer, denom, x); + if (is_ex_exactly_of_type(q, fail)) + return a; + else + return q + rem(numer, denom, x) / denom; +} + + /** Pseudo-remainder of polynomials a(x) and b(x) in Z[x]. * * @param a first polynomial in x (dividend) @@ -511,7 +531,6 @@ ex prem(const ex &a, const ex &b, const symbol &x, bool check_args) * @param check_args check whether a and b are polynomials with rational * coefficients (defaults to "true") * @return sparse pseudo-remainder of a(x) and b(x) in Z[x] */ - ex sprem(const ex &a, const ex &b, const symbol &x, bool check_args) { if (b.is_zero()) @@ -598,6 +617,7 @@ bool divide(const ex &a, const ex &b, ex &q, bool check_args) int rdeg = r.degree(*x); ex blcoeff = b.expand().coeff(*x, bdeg); bool blcoeff_is_numeric = is_ex_exactly_of_type(blcoeff, numeric); + exvector v; v.reserve(rdeg - bdeg + 1); while (rdeg >= bdeg) { ex term, rcoeff = r.coeff(*x, rdeg); if (blcoeff_is_numeric) @@ -606,10 +626,12 @@ bool divide(const ex &a, const ex &b, ex &q, bool check_args) if (!divide(rcoeff, blcoeff, term, false)) return false; term *= power(*x, rdeg - bdeg); - q += term; + v.push_back(term); r -= (term * b).expand(); - if (r.is_zero()) + if (r.is_zero()) { + q = (new add(v))->setflag(status_flags::dynallocated); return true; + } rdeg = r.degree(*x); } return false; @@ -756,14 +778,16 @@ static bool divide_in_z(const ex &a, const ex &b, ex &q, sym_desc_vec::const_ite int rdeg = adeg; ex eb = b.expand(); ex blcoeff = eb.coeff(*x, bdeg); + exvector v; v.reserve(rdeg - bdeg + 1); while (rdeg >= bdeg) { ex term, rcoeff = r.coeff(*x, rdeg); if (!divide_in_z(rcoeff, blcoeff, term, var+1)) break; term = (term * power(*x, rdeg - bdeg)).expand(); - q += term; + v.push_back(term); r -= (term * eb).expand(); if (r.is_zero()) { + q = (new add(v))->setflag(status_flags::dynallocated); #if USE_REMEMBER dr_remember[ex2(a, b)] = exbool(q, true); #endif @@ -1204,12 +1228,12 @@ numeric add::max_coefficient(void) const { epvector::const_iterator it = seq.begin(); epvector::const_iterator itend = seq.end(); - GINAC_ASSERT(is_ex_exactly_of_type(overall_coeff,numeric)); - numeric cur_max = abs(ex_to_numeric(overall_coeff)); + GINAC_ASSERT(is_exactly_a(overall_coeff)); + numeric cur_max = abs(ex_to(overall_coeff)); while (it != itend) { numeric a; - GINAC_ASSERT(!is_ex_exactly_of_type(it->rest,numeric)); - a = abs(ex_to_numeric(it->coeff)); + GINAC_ASSERT(!is_exactly_a(it->rest)); + a = abs(ex_to(it->coeff)); if (a > cur_max) cur_max = a; it++; @@ -1223,28 +1247,21 @@ numeric mul::max_coefficient(void) const epvector::const_iterator it = seq.begin(); epvector::const_iterator itend = seq.end(); while (it != itend) { - GINAC_ASSERT(!is_ex_exactly_of_type(recombine_pair_to_ex(*it),numeric)); + GINAC_ASSERT(!is_exactly_a(recombine_pair_to_ex(*it))); it++; } #endif // def DO_GINAC_ASSERT - GINAC_ASSERT(is_ex_exactly_of_type(overall_coeff,numeric)); - return abs(ex_to_numeric(overall_coeff)); + GINAC_ASSERT(is_exactly_a(overall_coeff)); + return abs(ex_to(overall_coeff)); } -/** Apply symmetric modular homomorphism to a multivariate polynomial. - * This function is used internally by heur_gcd(). +/** Apply symmetric modular homomorphism to an expanded multivariate + * polynomial. This function is usually used internally by heur_gcd(). * - * @param e expanded multivariate polynomial * @param xi modulus * @return mapped polynomial * @see heur_gcd */ -ex ex::smod(const numeric &xi) const -{ - GINAC_ASSERT(bp!=0); - return bp->smod(xi); -} - ex basic::smod(const numeric &xi) const { return *this; @@ -1262,14 +1279,14 @@ ex add::smod(const numeric &xi) const epvector::const_iterator it = seq.begin(); epvector::const_iterator itend = seq.end(); while (it != itend) { - GINAC_ASSERT(!is_ex_exactly_of_type(it->rest,numeric)); - numeric coeff = GiNaC::smod(ex_to_numeric(it->coeff), xi); + GINAC_ASSERT(!is_exactly_a(it->rest)); + numeric coeff = GiNaC::smod(ex_to(it->coeff), xi); if (!coeff.is_zero()) newseq.push_back(expair(it->rest, coeff)); it++; } - GINAC_ASSERT(is_ex_exactly_of_type(overall_coeff,numeric)); - numeric coeff = GiNaC::smod(ex_to_numeric(overall_coeff), xi); + GINAC_ASSERT(is_exactly_a(overall_coeff)); + numeric coeff = GiNaC::smod(ex_to(overall_coeff), xi); return (new add(newseq,coeff))->setflag(status_flags::dynallocated); } @@ -1279,13 +1296,13 @@ ex mul::smod(const numeric &xi) const epvector::const_iterator it = seq.begin(); epvector::const_iterator itend = seq.end(); while (it != itend) { - GINAC_ASSERT(!is_ex_exactly_of_type(recombine_pair_to_ex(*it),numeric)); + GINAC_ASSERT(!is_exactly_a(recombine_pair_to_ex(*it))); it++; } #endif // def DO_GINAC_ASSERT mul * mulcopyp = new mul(*this); - GINAC_ASSERT(is_ex_exactly_of_type(overall_coeff,numeric)); - mulcopyp->overall_coeff = GiNaC::smod(ex_to_numeric(overall_coeff),xi); + GINAC_ASSERT(is_exactly_a(overall_coeff)); + mulcopyp->overall_coeff = GiNaC::smod(ex_to(overall_coeff),xi); mulcopyp->clearflag(status_flags::evaluated); mulcopyp->clearflag(status_flags::hash_calculated); return mulcopyp->setflag(status_flags::dynallocated); @@ -1293,17 +1310,17 @@ ex mul::smod(const numeric &xi) const /** xi-adic polynomial interpolation */ -static ex interpolate(const ex &gamma, const numeric &xi, const symbol &x) +static ex interpolate(const ex &gamma, const numeric &xi, const symbol &x, int degree_hint = 1) { - ex g = _ex0(); + exvector g; g.reserve(degree_hint); ex e = gamma; numeric rxi = xi.inverse(); for (int i=0; !e.is_zero(); i++) { ex gi = e.smod(xi); - g += gi * power(x, i); + g.push_back(gi * power(x, i)); e = (e - gi) * rxi; } - return g; + return (new add(g))->setflag(status_flags::dynallocated); } /** Exception thrown by heur_gcd() to signal failure. */ @@ -1331,17 +1348,17 @@ static ex heur_gcd(const ex &a, const ex &b, ex *ca, ex *cb, sym_desc_vec::const heur_gcd_called++; #endif - // Algorithms only works for non-vanishing input polynomials + // Algorithm only works for non-vanishing input polynomials if (a.is_zero() || b.is_zero()) return (new fail())->setflag(status_flags::dynallocated); // GCD of two numeric values -> CLN if (is_ex_exactly_of_type(a, numeric) && is_ex_exactly_of_type(b, numeric)) { - numeric g = gcd(ex_to_numeric(a), ex_to_numeric(b)); + numeric g = gcd(ex_to(a), ex_to(b)); if (ca) - *ca = ex_to_numeric(a) / g; + *ca = ex_to(a) / g; if (cb) - *cb = ex_to_numeric(b) / g; + *cb = ex_to(b) / g; return g; } @@ -1353,7 +1370,7 @@ static ex heur_gcd(const ex &a, const ex &b, ex *ca, ex *cb, sym_desc_vec::const numeric rgc = gc.inverse(); ex p = a * rgc; ex q = b * rgc; - int maxdeg = std::max(p.degree(x),q.degree(x)); + int maxdeg = std::max(p.degree(x), q.degree(x)); // Find evaluation point numeric mp = p.max_coefficient(); @@ -1377,7 +1394,7 @@ static ex heur_gcd(const ex &a, const ex &b, ex *ca, ex *cb, sym_desc_vec::const if (!is_ex_exactly_of_type(gamma, fail)) { // Reconstruct polynomial from GCD of mapped polynomials - ex g = interpolate(gamma, xi, x); + ex g = interpolate(gamma, xi, x, maxdeg); // Remove integer content g /= g.integer_content(); @@ -1387,7 +1404,7 @@ static ex heur_gcd(const ex &a, const ex &b, ex *ca, ex *cb, sym_desc_vec::const if (divide_in_z(p, g, ca ? *ca : dummy, var) && divide_in_z(q, g, cb ? *cb : dummy, var)) { g *= gc; ex lc = g.lcoeff(x); - if (is_ex_exactly_of_type(lc, numeric) && ex_to_numeric(lc).is_negative()) + if (is_ex_exactly_of_type(lc, numeric) && ex_to(lc).is_negative()) return -g; else return g; @@ -1400,7 +1417,7 @@ static ex heur_gcd(const ex &a, const ex &b, ex *ca, ex *cb, sym_desc_vec::const if (ca) *ca = cp; ex lc = g.lcoeff(x); - if (is_ex_exactly_of_type(lc, numeric) && ex_to_numeric(lc).is_negative()) + if (is_ex_exactly_of_type(lc, numeric) && ex_to(lc).is_negative()) return -g; else return g; @@ -1413,7 +1430,7 @@ static ex heur_gcd(const ex &a, const ex &b, ex *ca, ex *cb, sym_desc_vec::const if (cb) *cb = cq; ex lc = g.lcoeff(x); - if (is_ex_exactly_of_type(lc, numeric) && ex_to_numeric(lc).is_negative()) + if (is_ex_exactly_of_type(lc, numeric) && ex_to(lc).is_negative()) return -g; else return g; @@ -1446,7 +1463,7 @@ ex gcd(const ex &a, const ex &b, ex *ca, ex *cb, bool check_args) // GCD of numerics -> CLN if (is_ex_exactly_of_type(a, numeric) && is_ex_exactly_of_type(b, numeric)) { - numeric g = gcd(ex_to_numeric(a), ex_to_numeric(b)); + numeric g = gcd(ex_to(a), ex_to(b)); if (ca || cb) { if (g.is_zero()) { if (ca) @@ -1455,9 +1472,9 @@ ex gcd(const ex &a, const ex &b, ex *ca, ex *cb, bool check_args) *cb = _ex0(); } else { if (ca) - *ca = ex_to_numeric(a) / g; + *ca = ex_to(a) / g; if (cb) - *cb = ex_to_numeric(b) / g; + *cb = ex_to(b) / g; } } return g; @@ -1473,38 +1490,40 @@ ex gcd(const ex &a, const ex &b, ex *ca, ex *cb, bool check_args) if (is_ex_exactly_of_type(b, mul) && b.nops() > a.nops()) goto factored_b; factored_a: - ex g = _ex1(); - ex acc_ca = _ex1(); + unsigned num = a.nops(); + exvector g; g.reserve(num); + exvector acc_ca; acc_ca.reserve(num); ex part_b = b; - for (unsigned i=0; isetflag(status_flags::dynallocated); if (cb) *cb = part_b; - return g; + return (new mul(g))->setflag(status_flags::dynallocated); } else if (is_ex_exactly_of_type(b, mul)) { if (is_ex_exactly_of_type(a, mul) && a.nops() > b.nops()) goto factored_a; factored_b: - ex g = _ex1(); - ex acc_cb = _ex1(); + unsigned num = b.nops(); + exvector g; g.reserve(num); + exvector acc_cb; acc_cb.reserve(num); ex part_a = a; - for (unsigned i=0; isetflag(status_flags::dynallocated); + return (new mul(g))->setflag(status_flags::dynallocated); } #if FAST_COMPARE @@ -1677,7 +1696,7 @@ factored_b: ex lcm(const ex &a, const ex &b, bool check_args) { if (is_ex_exactly_of_type(a, numeric) && is_ex_exactly_of_type(b, numeric)) - return lcm(ex_to_numeric(a), ex_to_numeric(b)); + return lcm(ex_to(a), ex_to(b)); if (check_args && (!a.info(info_flags::rational_polynomial) || !b.info(info_flags::rational_polynomial))) throw(std::invalid_argument("lcm: arguments must be polynomials over the rationals")); @@ -1718,6 +1737,7 @@ static exvector sqrfree_yun(const ex &a, const symbol &x) } while (!z.is_zero()); return res; } + /** Compute square-free factorization of multivariate polynomial in Q[X]. * * @param a multivariate polynomial over Q[X] @@ -1728,6 +1748,7 @@ ex sqrfree(const ex &a, const lst &l) if (is_ex_of_type(a,numeric) || // algorithm does not trap a==0 is_ex_of_type(a,symbol)) // shortcut return a; + // If no lst of variables to factorize in was specified we have to // invent one now. Maybe one can optimize here by reversing the order // or so, I don't know. @@ -1735,34 +1756,46 @@ ex sqrfree(const ex &a, const lst &l) if (l.nops()==0) { sym_desc_vec sdv; get_symbol_stats(a, _ex0(), sdv); - for (sym_desc_vec::iterator it=sdv.begin(); it!=sdv.end(); ++it) + sym_desc_vec::const_iterator it = sdv.begin(), itend = sdv.end(); + while (it != itend) { args.append(*it->sym); + ++it; + } } else { args = l; } + // Find the symbol to factor in at this stage if (!is_ex_of_type(args.op(0), symbol)) throw (std::runtime_error("sqrfree(): invalid factorization variable")); - const symbol x = ex_to_symbol(args.op(0)); + const symbol x = ex_to(args.op(0)); + // convert the argument from something in Q[X] to something in Z[X] numeric lcm = lcm_of_coefficients_denominators(a); ex tmp = multiply_lcm(a,lcm); + // find the factors exvector factors = sqrfree_yun(tmp,x); + // construct the next list of symbols with the first element popped - lst newargs; - for (int i=1; i0) { - for (exvector::iterator i=factors.begin(); i!=factors.end(); ++i) + exvector::iterator i = factors.begin(), end = factors.end(); + while (i != end) { *i = sqrfree(*i, newargs); + ++i; + } } + // Done with recursion, now construct the final result ex result = _ex1(); - exvector::iterator it = factors.begin(); - for (int p = 1; it!=factors.end(); ++it, ++p) + exvector::const_iterator it = factors.begin(), itend = factors.end(); + for (int p = 1; it!=itend; ++it, ++p) result *= power(*it, p); + // Yun's algorithm does not account for constant factors. (For // univariate polynomials it works only in the monic case.) We can // correct this by inserting what has been lost back into the result: @@ -1770,6 +1803,75 @@ ex sqrfree(const ex &a, const lst &l) return result * lcm.inverse(); } +/** Compute square-free partial fraction decomposition of rational function + * a(x). + * + * @param a rational function over Z[x], treated as univariate polynomial + * in x + * @param x variable to factor in + * @return decomposed rational function */ +ex sqrfree_parfrac(const ex & a, const symbol & x) +{ + // Find numerator and denominator + ex nd = numer_denom(a); + ex numer = nd.op(0), denom = nd.op(1); +//clog << "numer = " << numer << ", denom = " << denom << endl; + + // Convert N(x)/D(x) -> Q(x) + R(x)/D(x), so degree(R) < degree(D) + ex red_poly = quo(numer, denom, x), red_numer = rem(numer, denom, x).expand(); +//clog << "red_poly = " << red_poly << ", red_numer = " << red_numer << endl; + + // Factorize denominator and compute cofactors + exvector yun = sqrfree_yun(denom, x); +//clog << "yun factors: " << exprseq(yun) << endl; + unsigned num_yun = yun.size(); + exvector factor; factor.reserve(num_yun); + exvector cofac; cofac.reserve(num_yun); + for (unsigned i=0; isetflag(status_flags::dynallocated); + if (nops() == 0) + return (new lst(replace_with_symbol(*this, sym_lst, repl_lst), _ex1()))->setflag(status_flags::dynallocated); + else { + if (level == 1) + return (new lst(replace_with_symbol(*this, sym_lst, repl_lst), _ex1()))->setflag(status_flags::dynallocated); + else if (level == -max_recursion_level) + throw(std::runtime_error("max recursion level reached")); + else { + normal_map_function map_normal(level - 1); + return (new lst(replace_with_symbol(map(map_normal), sym_lst, repl_lst), _ex1()))->setflag(status_flags::dynallocated); + } + } } @@ -1908,8 +2030,8 @@ static ex frac_cancel(const ex &n, const ex &d) // as defined by get_first_symbol() is made positive) const symbol *x; if (get_first_symbol(den, x)) { - GINAC_ASSERT(is_ex_exactly_of_type(den.unit(*x),numeric)); - if (ex_to_numeric(den.unit(*x)).is_negative()) { + GINAC_ASSERT(is_exactly_a(den.unit(*x))); + if (ex_to(den.unit(*x)).is_negative()) { num *= _ex_1(); den *= _ex_1(); } @@ -1937,12 +2059,12 @@ ex add::normal(lst &sym_lst, lst &repl_lst, int level) const dens.reserve(seq.size()+1); epvector::const_iterator it = seq.begin(), itend = seq.end(); while (it != itend) { - ex n = recombine_pair_to_ex(*it).bp->normal(sym_lst, repl_lst, level-1); + ex n = ex_to(recombine_pair_to_ex(*it)).normal(sym_lst, repl_lst, level-1); nums.push_back(n.op(0)); dens.push_back(n.op(1)); it++; } - ex n = overall_coeff.bp->normal(sym_lst, repl_lst, level-1); + ex n = ex_to(overall_coeff).normal(sym_lst, repl_lst, level-1); nums.push_back(n.op(0)); dens.push_back(n.op(1)); GINAC_ASSERT(nums.size() == dens.size()); @@ -1991,22 +2113,23 @@ ex mul::normal(lst &sym_lst, lst &repl_lst, int level) const throw(std::runtime_error("max recursion level reached")); // Normalize children, separate into numerator and denominator - ex num = _ex1(); - ex den = _ex1(); + exvector num; num.reserve(seq.size()); + exvector den; den.reserve(seq.size()); ex n; epvector::const_iterator it = seq.begin(), itend = seq.end(); while (it != itend) { - n = recombine_pair_to_ex(*it).bp->normal(sym_lst, repl_lst, level-1); - num *= n.op(0); - den *= n.op(1); + n = ex_to(recombine_pair_to_ex(*it)).normal(sym_lst, repl_lst, level-1); + num.push_back(n.op(0)); + den.push_back(n.op(1)); it++; } - n = overall_coeff.bp->normal(sym_lst, repl_lst, level-1); - num *= n.op(0); - den *= n.op(1); + n = ex_to(overall_coeff).normal(sym_lst, repl_lst, level-1); + num.push_back(n.op(0)); + den.push_back(n.op(1)); // Perform fraction cancellation - return frac_cancel(num, den); + return frac_cancel((new mul(num))->setflag(status_flags::dynallocated), + (new mul(den))->setflag(status_flags::dynallocated)); } @@ -2022,8 +2145,8 @@ ex power::normal(lst &sym_lst, lst &repl_lst, int level) const throw(std::runtime_error("max recursion level reached")); // Normalize basis and exponent (exponent gets reassembled) - ex n_basis = basis.bp->normal(sym_lst, repl_lst, level-1); - ex n_exponent = exponent.bp->normal(sym_lst, repl_lst, level-1); + ex n_basis = ex_to(basis).normal(sym_lst, repl_lst, level-1); + ex n_exponent = ex_to(exponent).normal(sym_lst, repl_lst, level-1); n_exponent = n_exponent.op(0) / n_exponent.op(1); if (n_exponent.info(info_flags::integer)) { @@ -2074,24 +2197,18 @@ ex power::normal(lst &sym_lst, lst &repl_lst, int level) const ex pseries::normal(lst &sym_lst, lst &repl_lst, int level) const { epvector newseq; - for (epvector::const_iterator i=seq.begin(); i!=seq.end(); ++i) { + epvector::const_iterator i = seq.begin(), end = seq.end(); + while (i != end) { ex restexp = i->rest.normal(); if (!restexp.is_zero()) newseq.push_back(expair(restexp, i->coeff)); + ++i; } ex n = pseries(relational(var,point), newseq); return (new lst(replace_with_symbol(n, sym_lst, repl_lst), _ex1()))->setflag(status_flags::dynallocated); } -/** Implementation of ex::normal() for relationals. It normalizes both sides. - * @see ex::normal */ -ex relational::normal(lst &sym_lst, lst &repl_lst, int level) const -{ - return (new lst(relational(lh.normal(), rh.normal(), o), _ex1()))->setflag(status_flags::dynallocated); -} - - /** Normalization of rational functions. * This function converts an expression to its normal form * "numerator/denominator", where numerator and denominator are (relatively @@ -2109,7 +2226,7 @@ ex ex::normal(int level) const lst sym_lst, repl_lst; ex e = bp->normal(sym_lst, repl_lst, level); - GINAC_ASSERT(is_ex_of_type(e, lst)); + GINAC_ASSERT(is_a(e)); // Re-insert replaced symbols if (sym_lst.nops() > 0) @@ -2119,9 +2236,9 @@ ex ex::normal(int level) const return e.op(0) / e.op(1); } -/** Numerator of an expression. If the expression is not of the normal form - * "numerator/denominator", it is first converted to this form and then the - * numerator is returned. +/** Get numerator of an expression. If the expression is not of the normal + * form "numerator/denominator", it is first converted to this form and + * then the numerator is returned. * * @see ex::normal * @return numerator */ @@ -2130,7 +2247,7 @@ ex ex::numer(void) const lst sym_lst, repl_lst; ex e = bp->normal(sym_lst, repl_lst, 0); - GINAC_ASSERT(is_ex_of_type(e, lst)); + GINAC_ASSERT(is_a(e)); // Re-insert replaced symbols if (sym_lst.nops() > 0) @@ -2139,9 +2256,9 @@ ex ex::numer(void) const return e.op(0); } -/** Denominator of an expression. If the expression is not of the normal form - * "numerator/denominator", it is first converted to this form and then the - * denominator is returned. +/** Get denominator of an expression. If the expression is not of the normal + * form "numerator/denominator", it is first converted to this form and + * then the denominator is returned. * * @see ex::normal * @return denominator */ @@ -2150,7 +2267,7 @@ ex ex::denom(void) const lst sym_lst, repl_lst; ex e = bp->normal(sym_lst, repl_lst, 0); - GINAC_ASSERT(is_ex_of_type(e, lst)); + GINAC_ASSERT(is_a(e)); // Re-insert replaced symbols if (sym_lst.nops() > 0) @@ -2159,10 +2276,41 @@ ex ex::denom(void) const return e.op(1); } +/** Get numerator and denominator of an expression. If the expresison is not + * of the normal form "numerator/denominator", it is first converted to this + * form and then a list [numerator, denominator] is returned. + * + * @see ex::normal + * @return a list [numerator, denominator] */ +ex ex::numer_denom(void) const +{ + lst sym_lst, repl_lst; + + ex e = bp->normal(sym_lst, repl_lst, 0); + GINAC_ASSERT(is_a(e)); + + // Re-insert replaced symbols + if (sym_lst.nops() > 0) + return e.subs(sym_lst, repl_lst); + else + return e; +} + -/** Default implementation of ex::to_rational(). It replaces the object with a - * temporary symbol. - * @see ex::to_rational */ +/** Rationalization of non-rational functions. + * This function converts a general expression to a rational polynomial + * by replacing all non-rational subexpressions (like non-rational numbers, + * non-integer powers or functions like sin(), cos() etc.) to temporary + * symbols. This makes it possible to use functions like gcd() and divide() + * on non-rational functions by applying to_rational() on the arguments, + * calling the desired function and re-substituting the temporary symbols + * in the result. To make the last step possible, all temporary symbols and + * their associated expressions are collected in the list specified by the + * repl_lst parameter in the form {symbol == expression}, ready to be passed + * as an argument to ex::subs(). + * + * @param repl_lst collects a list of all temporary symbols and their replacements + * @return rationalized expression */ ex basic::to_rational(lst &repl_lst) const { return replace_with_symbol(*this, repl_lst); @@ -2170,8 +2318,7 @@ ex basic::to_rational(lst &repl_lst) const /** Implementation of ex::to_rational() for symbols. This returns the - * unmodified symbol. - * @see ex::to_rational */ + * unmodified symbol. */ ex symbol::to_rational(lst &repl_lst) const { return *this; @@ -2180,8 +2327,7 @@ ex symbol::to_rational(lst &repl_lst) const /** Implementation of ex::to_rational() for a numeric. It splits complex * numbers into re+I*im and replaces I and non-rational real numbers with a - * temporary symbol. - * @see ex::to_rational */ + * temporary symbol. */ ex numeric::to_rational(lst &repl_lst) const { if (is_real()) { @@ -2199,8 +2345,7 @@ ex numeric::to_rational(lst &repl_lst) const /** Implementation of ex::to_rational() for powers. It replaces non-integer - * powers by temporary symbols. - * @see ex::to_rational */ + * powers by temporary symbols. */ ex power::to_rational(lst &repl_lst) const { if (exponent.info(info_flags::integer)) @@ -2210,42 +2355,23 @@ ex power::to_rational(lst &repl_lst) const } -/** Implementation of ex::to_rational() for expairseqs. - * @see ex::to_rational */ +/** Implementation of ex::to_rational() for expairseqs. */ ex expairseq::to_rational(lst &repl_lst) const { epvector s; s.reserve(seq.size()); - for (epvector::const_iterator it=seq.begin(); it!=seq.end(); ++it) { - s.push_back(split_ex_to_pair(recombine_pair_to_ex(*it).to_rational(repl_lst))); - // s.push_back(combine_ex_with_coeff_to_pair((*it).rest.to_rational(repl_lst), + epvector::const_iterator i = seq.begin(), end = seq.end(); + while (i != end) { + s.push_back(split_ex_to_pair(recombine_pair_to_ex(*i).to_rational(repl_lst))); + ++i; } ex oc = overall_coeff.to_rational(repl_lst); if (oc.info(info_flags::numeric)) return thisexpairseq(s, overall_coeff); - else s.push_back(combine_ex_with_coeff_to_pair(oc,_ex1())); + else + s.push_back(combine_ex_with_coeff_to_pair(oc, _ex1())); return thisexpairseq(s, default_overall_coeff()); } -/** Rationalization of non-rational functions. - * This function converts a general expression to a rational polynomial - * by replacing all non-rational subexpressions (like non-rational numbers, - * non-integer powers or functions like sin(), cos() etc.) to temporary - * symbols. This makes it possible to use functions like gcd() and divide() - * on non-rational functions by applying to_rational() on the arguments, - * calling the desired function and re-substituting the temporary symbols - * in the result. To make the last step possible, all temporary symbols and - * their associated expressions are collected in the list specified by the - * repl_lst parameter in the form {symbol == expression}, ready to be passed - * as an argument to ex::subs(). - * - * @param repl_lst collects a list of all temporary symbols and their replacements - * @return rationalized expression */ -ex ex::to_rational(lst &repl_lst) const -{ - return bp->to_rational(repl_lst); -} - - } // namespace GiNaC