]> www.ginac.de Git - ginac.git/blobdiff - ginac/normal.cpp
- expairseq::to_rational(): Fixed thinko.
[ginac.git] / ginac / normal.cpp
index a4681122a662a0b03cb0e333f024aa360bc9d7b2..1126d27c2fffb738e3c6cf57decf92af6dc430f0 100644 (file)
@@ -93,7 +93,6 @@ static struct _stat_print {
  *  @param e  expression to search
  *  @param x  pointer to first symbol found (returned)
  *  @return "false" if no symbol was found, "true" otherwise */
-
 static bool get_first_symbol(const ex &e, const symbol *&x)
 {
     if (is_ex_exactly_of_type(e, symbol)) {
@@ -137,11 +136,11 @@ struct sym_desc {
     /** Lowest degree of symbol in polynomial "b" */
     int ldeg_b;
 
-    /** Minimum of ldeg_a and ldeg_b (Used for sorting) */
-    int min_deg;
+    /** Maximum of deg_a and deg_b (Used for sorting) */
+    int max_deg;
 
     /** Commparison operator for sorting */
-    bool operator<(const sym_desc &x) const {return min_deg < x.min_deg;}
+    bool operator<(const sym_desc &x) const {return max_deg < x.max_deg;}
 };
 
 // Vector of sym_desc structures
@@ -186,7 +185,6 @@ static void collect_symbols(const ex &e, sym_desc_vec &v)
  *  @param a  first multivariate polynomial
  *  @param b  second multivariate polynomial
  *  @param v  vector of sym_desc structs (filled in) */
-
 static void get_symbol_stats(const ex &a, const ex &b, sym_desc_vec &v)
 {
     collect_symbols(a.eval(), v);   // eval() to expand assigned symbols
@@ -197,12 +195,21 @@ static void get_symbol_stats(const ex &a, const ex &b, sym_desc_vec &v)
         int deg_b = b.degree(*(it->sym));
         it->deg_a = deg_a;
         it->deg_b = deg_b;
-        it->min_deg = min(deg_a, deg_b);
+        it->max_deg = max(deg_a, deg_b);
         it->ldeg_a = a.ldegree(*(it->sym));
         it->ldeg_b = b.ldegree(*(it->sym));
         it++;
     }
     sort(v.begin(), v.end());
+#if 0
+       clog << "Symbols:\n";
+       it = v.begin(); itend = v.end();
+       while (it != itend) {
+               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 << endl;
+               clog << "  lcoeff_a=" << a.lcoeff(*(it->sym)) << ", lcoeff_b=" << b.lcoeff(*(it->sym)) << endl;
+               it++;
+       }
+#endif
 }
 
 
@@ -238,7 +245,6 @@ static numeric lcmcoeff(const ex &e, const numeric &l)
  *
  *  @param e  multivariate polynomial (need not be expanded)
  *  @return LCM of denominators of coefficients */
-
 static numeric lcm_of_coefficients_denominators(const ex &e)
 {
     return lcmcoeff(e, _num1());
@@ -249,7 +255,6 @@ static numeric lcm_of_coefficients_denominators(const ex &e)
  *
  *  @param e  multivariate polynomial (need not be expanded)
  *  @param lcm  LCM to multiply in */
-
 static ex multiply_lcm(const ex &e, const numeric &lcm)
 {
        if (is_ex_exactly_of_type(e, mul)) {
@@ -279,7 +284,6 @@ static ex multiply_lcm(const ex &e, const numeric &lcm)
  *
  *  @param e  expanded polynomial
  *  @return integer content */
-
 numeric ex::integer_content(void) const
 {
     GINAC_ASSERT(bp!=0);
@@ -340,7 +344,6 @@ numeric mul::integer_content(void) const
  *  @param check_args  check whether a and b are polynomials with rational
  *         coefficients (defaults to "true")
  *  @return quotient of a and b in Q[x] */
-
 ex quo(const ex &a, const ex &b, const symbol &x, bool check_args)
 {
     if (b.is_zero())
@@ -391,7 +394,6 @@ ex quo(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 remainder of a(x) and b(x) in Q[x] */
-
 ex rem(const ex &a, const ex &b, const symbol &x, bool check_args)
 {
     if (b.is_zero())
@@ -443,7 +445,6 @@ ex rem(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 pseudo-remainder of a(x) and b(x) in Z[x] */
-
 ex prem(const ex &a, const ex &b, const symbol &x, bool check_args)
 {
     if (b.is_zero())
@@ -497,12 +498,13 @@ ex prem(const ex &a, const ex &b, const symbol &x, bool check_args)
  *         coefficients (defaults to "true")
  *  @return "true" when exact division succeeds (quotient returned in q),
  *          "false" otherwise */
-
 bool divide(const ex &a, const ex &b, ex &q, bool check_args)
 {
     q = _ex0();
     if (b.is_zero())
         throw(std::overflow_error("divide: division by zero"));
+    if (a.is_zero())
+        return true;
     if (is_ex_exactly_of_type(b, numeric)) {
         q = a / b;
         return true;
@@ -805,7 +807,6 @@ ex ex::primpart(const symbol &x) const
  *  @param x  variable in which to compute the primitive part
  *  @param c  previously computed content part
  *  @return primitive part */
-
 ex ex::primpart(const symbol &x, const ex &c) const
 {
     if (is_zero())
@@ -827,8 +828,47 @@ ex ex::primpart(const symbol &x, const ex &c) const
  *  GCD of multivariate polynomials
  */
 
-/** Compute GCD of multivariate polynomials using the subresultant PRS
- *  algorithm. This function is used internally gy gcd().
+/** Compute GCD of multivariate polynomials using the Euclidean PRS algorithm
+ *  (not really suited for multivariate GCDs). This function is only provided
+ *  for testing purposes.
+ *
+ *  @param a  first multivariate polynomial
+ *  @param b  second multivariate polynomial
+ *  @param x  pointer to symbol (main variable) in which to compute the GCD in
+ *  @return the GCD as a new expression
+ *  @see gcd */
+
+static ex eu_gcd(const ex &a, const ex &b, const symbol *x)
+{
+//clog << "eu_gcd(" << a << "," << b << ")\n";
+
+    // Sort c and d so that c has higher degree
+    ex c, d;
+    int adeg = a.degree(*x), bdeg = b.degree(*x);
+    if (adeg >= bdeg) {
+        c = a;
+        d = b;
+    } else {
+        c = b;
+        d = a;
+    }
+
+       // Euclidean algorithm
+    ex r;
+    for (;;) {
+//clog << " d = " << d << endl;
+        r = rem(c, d, *x, false);
+        if (r.is_zero())
+            return d.primpart(*x);
+        c = d;
+               d = r;
+    }
+}
+
+
+/** Compute GCD of multivariate polynomials using the Euclidean PRS algorithm
+ *  with pseudo-remainders ("World's Worst GCD Algorithm", staying in Z[X]).
+ *  This function is only provided for testing purposes.
  *
  *  @param a  first multivariate polynomial
  *  @param b  second multivariate polynomial
@@ -836,6 +876,159 @@ ex ex::primpart(const symbol &x, const ex &c) const
  *  @return the GCD as a new expression
  *  @see gcd */
 
+static ex euprem_gcd(const ex &a, const ex &b, const symbol *x)
+{
+//clog << "euprem_gcd(" << a << "," << b << ")\n";
+
+    // Sort c and d so that c has higher degree
+    ex c, d;
+    int adeg = a.degree(*x), bdeg = b.degree(*x);
+    if (adeg >= bdeg) {
+        c = a;
+        d = b;
+    } else {
+        c = b;
+        d = a;
+    }
+
+       // Euclidean algorithm with pseudo-remainders
+    ex r;
+    for (;;) {
+//clog << " d = " << d << endl;
+        r = prem(c, d, *x, false);
+        if (r.is_zero())
+            return d.primpart(*x);
+        c = d;
+               d = r;
+    }
+}
+
+
+/** Compute GCD of multivariate polynomials using the primitive Euclidean
+ *  PRS algorithm (complete content removal at each step). This function is
+ *  only provided for testing purposes.
+ *
+ *  @param a  first multivariate polynomial
+ *  @param b  second multivariate polynomial
+ *  @param x  pointer to symbol (main variable) in which to compute the GCD in
+ *  @return the GCD as a new expression
+ *  @see gcd */
+
+static ex peu_gcd(const ex &a, const ex &b, const symbol *x)
+{
+//clog << "peu_gcd(" << a << "," << b << ")\n";
+
+    // Sort c and d so that c has higher degree
+    ex c, d;
+    int adeg = a.degree(*x), bdeg = b.degree(*x);
+    int ddeg;
+    if (adeg >= bdeg) {
+        c = a;
+        d = b;
+        ddeg = bdeg;
+    } else {
+        c = b;
+        d = a;
+        ddeg = adeg;
+    }
+
+    // Remove content from c and d, to be attached to GCD later
+    ex cont_c = c.content(*x);
+    ex cont_d = d.content(*x);
+    ex gamma = gcd(cont_c, cont_d, NULL, NULL, false);
+    if (ddeg == 0)
+        return gamma;
+    c = c.primpart(*x, cont_c);
+    d = d.primpart(*x, cont_d);
+
+    // Euclidean algorithm with content removal
+       ex r;
+    for (;;) {
+//clog << " d = " << d << endl;
+        r = prem(c, d, *x, false);
+        if (r.is_zero())
+            return gamma * d;
+        c = d;
+               d = r.primpart(*x);
+    }
+}
+
+
+/** Compute GCD of multivariate polynomials using the reduced PRS algorithm.
+ *  This function is only provided for testing purposes.
+ *
+ *  @param a  first multivariate polynomial
+ *  @param b  second multivariate polynomial
+ *  @param x  pointer to symbol (main variable) in which to compute the GCD in
+ *  @return the GCD as a new expression
+ *  @see gcd */
+
+static ex red_gcd(const ex &a, const ex &b, const symbol *x)
+{
+//clog << "red_gcd(" << a << "," << b << ")\n";
+
+    // Sort c and d so that c has higher degree
+    ex c, d;
+    int adeg = a.degree(*x), bdeg = b.degree(*x);
+    int cdeg, ddeg;
+    if (adeg >= bdeg) {
+        c = a;
+        d = b;
+        cdeg = adeg;
+        ddeg = bdeg;
+    } else {
+        c = b;
+        d = a;
+        cdeg = bdeg;
+        ddeg = adeg;
+    }
+
+    // Remove content from c and d, to be attached to GCD later
+    ex cont_c = c.content(*x);
+    ex cont_d = d.content(*x);
+    ex gamma = gcd(cont_c, cont_d, NULL, NULL, false);
+    if (ddeg == 0)
+        return gamma;
+    c = c.primpart(*x, cont_c);
+    d = d.primpart(*x, cont_d);
+
+    // First element of subresultant sequence
+    ex r, ri = _ex1();
+    int delta = cdeg - ddeg;
+
+    for (;;) {
+        // Calculate polynomial pseudo-remainder
+//clog << " d = " << d << endl;
+        r = prem(c, d, *x, false);
+        if (r.is_zero())
+            return gamma * d.primpart(*x);
+        c = d;
+        cdeg = ddeg;
+
+        if (!divide(r, pow(ri, delta), d, false))
+            throw(std::runtime_error("invalid expression in red_gcd(), division failed"));
+        ddeg = d.degree(*x);
+        if (ddeg == 0) {
+            if (is_ex_exactly_of_type(r, numeric))
+                return gamma;
+            else
+                return gamma * r.primpart(*x);
+        }
+
+        ri = c.expand().lcoeff(*x);
+        delta = cdeg - ddeg;
+    }
+}
+
+
+/** Compute GCD of multivariate polynomials using the subresultant PRS
+ *  algorithm. This function is used internally by gcd().
+ *
+ *  @param a  first multivariate polynomial
+ *  @param b  second multivariate polynomial
+ *  @param x  pointer to symbol (main variable) in which to compute the GCD in
+ *  @return the GCD as a new expression
+ *  @see gcd */
 static ex sr_gcd(const ex &a, const ex &b, const symbol *x)
 {
 //clog << "sr_gcd(" << a << "," << b << ")\n";
@@ -876,13 +1069,14 @@ static ex sr_gcd(const ex &a, const ex &b, const symbol *x)
     for (;;) {
         // Calculate polynomial pseudo-remainder
 //clog << " start of loop, psi = " << psi << ", calculating pseudo-remainder...\n";
+//clog << " d = " << d << endl;
         r = prem(c, d, *x, false);
         if (r.is_zero())
             return gamma * d.primpart(*x);
         c = d;
         cdeg = ddeg;
 //clog << " dividing...\n";
-        if (!divide(r, ri * power(psi, delta), d, false))
+        if (!divide(r, ri * pow(psi, delta), d, false))
             throw(std::runtime_error("invalid expression in sr_gcd(), division failed"));
         ddeg = d.degree(*x);
         if (ddeg == 0) {
@@ -898,7 +1092,7 @@ static ex sr_gcd(const ex &a, const ex &b, const symbol *x)
         if (delta == 1)
             psi = ri;
         else if (delta)
-            divide(power(ri, delta), power(psi, delta-1), psi, false);
+            divide(pow(ri, delta), pow(psi, delta-1), psi, false);
         delta = cdeg - ddeg;
     }
 }
@@ -910,7 +1104,6 @@ static ex sr_gcd(const ex &a, const ex &b, const symbol *x)
  *  @param e  expanded multivariate polynomial
  *  @return maximum coefficient
  *  @see heur_gcd */
-
 numeric ex::max_coefficient(void) const
 {
     GINAC_ASSERT(bp!=0);
@@ -966,7 +1159,6 @@ numeric mul::max_coefficient(void) const
  *  @param xi  modulus
  *  @return mapped polynomial
  *  @see heur_gcd */
-
 ex ex::smod(const numeric &xi) const
 {
     GINAC_ASSERT(bp!=0);
@@ -1054,7 +1246,6 @@ class gcdheu_failed {};
  *  @return the GCD as a new expression
  *  @see gcd
  *  @exception gcdheu_failed() */
-
 static ex heur_gcd(const ex &a, const ex &b, ex *ca, ex *cb, sym_desc_vec::const_iterator var)
 {
 //clog << "heur_gcd(" << a << "," << b << ")\n";
@@ -1142,7 +1333,6 @@ static ex heur_gcd(const ex &a, const ex &b, ex *ca, ex *cb, sym_desc_vec::const
  *  @param check_args  check whether a and b are polynomials with rational
  *         coefficients (defaults to "true")
  *  @return the GCD as a new expression */
-
 ex gcd(const ex &a, const ex &b, ex *ca, ex *cb, bool check_args)
 {
 //clog << "gcd(" << a << "," << b << ")\n";
@@ -1317,8 +1507,9 @@ factored_b:
         return g;
     }
 
-    // Try heuristic algorithm first, fall back to PRS if that failed
     ex g;
+#if 1
+    // Try heuristic algorithm first, fall back to PRS if that failed
     try {
         g = heur_gcd(aex, bex, ca, cb, var);
     } catch (gcdheu_failed) {
@@ -1329,7 +1520,13 @@ factored_b:
 #if STATISTICS
                heur_gcd_failed++;
 #endif
-        g = sr_gcd(aex, bex, x);
+#endif
+//             g = heur_gcd(aex, bex, ca, cb, var);
+//             g = eu_gcd(aex, bex, x);
+//             g = euprem_gcd(aex, bex, x);
+//             g = peu_gcd(aex, bex, x);
+//             g = red_gcd(aex, bex, x);
+               g = sr_gcd(aex, bex, x);
                if (g.is_equal(_ex1())) {
                        // Keep cofactors factored if possible
                        if (ca)
@@ -1342,6 +1539,7 @@ factored_b:
                if (cb)
                    divide(bex, g, *cb, false);
                }
+#if 1
     } else {
                if (g.is_equal(_ex1())) {
                        // Keep cofactors factored if possible
@@ -1350,8 +1548,9 @@ factored_b:
                        if (cb)
                                *cb = b;
                }
-           return g;
        }
+#endif
+    return g;
 }
 
 
@@ -1468,7 +1667,7 @@ static ex replace_with_symbol(const ex &e, lst &sym_lst, lst &repl_lst)
     for (unsigned i=0; i<repl_lst.nops(); i++)
         if (repl_lst.op(i).is_equal(e))
             return sym_lst.op(i);
-
+    
     // Otherwise create new symbol and add to list, taking care that the
        // replacement expression doesn't contain symbols from the sym_lst
        // because subs() is not recursive
@@ -1490,7 +1689,7 @@ static ex replace_with_symbol(const ex &e, lst &repl_lst)
     for (unsigned i=0; i<repl_lst.nops(); i++)
         if (repl_lst.op(i).op(1).is_equal(e))
             return repl_lst.op(i).op(0);
-
+    
     // Otherwise create new symbol and add to list, taking care that the
        // replacement expression doesn't contain symbols from the sym_lst
        // because subs() is not recursive
@@ -1762,6 +1961,14 @@ ex pseries::normal(lst &sym_lst, lst &repl_lst, int level) const
 }
 
 
+/** 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
@@ -1853,19 +2060,17 @@ ex symbol::to_rational(lst &repl_lst) const
  *  @see ex::to_rational */
 ex numeric::to_rational(lst &repl_lst) const
 {
-       numeric num = numer();
-       ex numex = num;
-
-    if (num.is_real()) {
-        if (!num.is_integer())
-            numex = replace_with_symbol(numex, repl_lst);
+    if (is_real()) {
+        if (!is_integer())
+            return replace_with_symbol(*this, repl_lst);
     } else { // complex
-        numeric re = num.real(), im = num.imag();
+        numeric re = real();
+        numeric im = imag();
         ex re_ex = re.is_rational() ? re : replace_with_symbol(re, repl_lst);
         ex im_ex = im.is_rational() ? im : replace_with_symbol(im, repl_lst);
-        numex = re_ex + im_ex * replace_with_symbol(I, repl_lst);
+        return re_ex + im_ex * replace_with_symbol(I, repl_lst);
     }
-       return numex;
+       return *this;
 }
 
 
@@ -1881,6 +2086,24 @@ ex power::to_rational(lst &repl_lst) const
 }
 
 
+/** Implementation of ex::to_rational() for expairseqs.
+ *  @see ex::to_rational */
+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),
+    }
+    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()));
+    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,