]> www.ginac.de Git - ginac.git/blobdiff - ginac/power.cpp
Fix pow(+(...),2).expand().
[ginac.git] / ginac / power.cpp
index d8c9b4efc9d4f692d5a725627532a26a9da0e3f2..b2460fda8661d234dc8c4b6fb30890c5d1920465 100644 (file)
@@ -3,7 +3,7 @@
  *  Implementation of GiNaC's symbolic exponentiation (basis^exponent). */
 
 /*
- *  GiNaC Copyright (C) 1999-2011 Johannes Gutenberg University Mainz, Germany
+ *  GiNaC Copyright (C) 1999-2015 Johannes Gutenberg University Mainz, Germany
  *
  *  This program is free software; you can redistribute it and/or modify
  *  it under the terms of the GNU General Public License as published by
@@ -242,6 +242,9 @@ bool power::info(unsigned inf) const
                        return (flags & status_flags::expanded);
                case info_flags::positive:
                        return basis.info(info_flags::positive) && exponent.info(info_flags::real);
+               case info_flags::nonnegative:
+                       return (basis.info(info_flags::positive) && exponent.info(info_flags::real)) ||
+                              (basis.info(info_flags::real) && exponent.info(info_flags::integer) && exponent.info(info_flags::even));
                case info_flags::has_indices: {
                        if (flags & status_flags::has_indices)
                                return true;
@@ -287,11 +290,16 @@ ex power::map(map_function & f) const
 
 bool power::is_polynomial(const ex & var) const
 {
-       if (exponent.has(var))
-               return false;
-       if (!exponent.info(info_flags::nonnegint))
-               return false;
-       return basis.is_polynomial(var);
+       if (basis.is_polynomial(var)) {
+               if (basis.has(var))
+                       // basis is non-constant polynomial in var
+                       return exponent.info(info_flags::nonnegint);
+               else
+                       // basis is constant in var
+                       return !exponent.has(var);
+       }
+       // basis is a non-polynomial function of var
+       return false;
 }
 
 int power::degree(const ex & s) const
@@ -362,7 +370,7 @@ ex power::coeff(const ex & s, int n) const
  *  - ^(1,x) -> 1
  *  - ^(c1,c2) -> *(c1^n,c1^(c2-n))  (so that 0<(c2-n)<1, try to evaluate roots, possibly in numerator and denominator of c1)
  *  - ^(^(x,c1),c2) -> ^(x,c1*c2)  if x is positive and c1 is real.
- *  - ^(^(x,c1),c2) -> ^(x,c1*c2)  (c2 integer or -1 < c1 <= 1, case c1=1 should not happen, see below!)
+ *  - ^(^(x,c1),c2) -> ^(x,c1*c2)  (c2 integer or -1 < c1 <= 1 or (c1=-1 and c2>0), case c1=1 should not happen, see below!)
  *  - ^(*(x,y,z),c) -> *(x^c,y^c,z^c)  (if c integer)
  *  - ^(*(x,c1),c2) -> ^(x,c2)*c1^c2  (c1>0)
  *  - ^(*(x,c1),c2) -> ^(-x,c2)*c1^c2  (c1<0)
@@ -480,7 +488,7 @@ ex power::eval(int level) const
                }
        
                // ^(^(x,c1),c2) -> ^(x,c1*c2)
-               // (c1, c2 numeric(), c2 integer or -1 < c1 <= 1,
+               // (c1, c2 numeric(), c2 integer or -1 < c1 <= 1 or (c1=-1 and c2>0),
                // case c1==1 should not happen, see below!)
                if (is_exactly_a<power>(ebasis)) {
                        const power & sub_power = ex_to<power>(ebasis);
@@ -489,7 +497,8 @@ ex power::eval(int level) const
                        if (is_exactly_a<numeric>(sub_exponent)) {
                                const numeric & num_sub_exponent = ex_to<numeric>(sub_exponent);
                                GINAC_ASSERT(num_sub_exponent!=numeric(1));
-                               if (num_exponent->is_integer() || (abs(num_sub_exponent) - (*_num1_p)).is_negative()) {
+                               if (num_exponent->is_integer() || (abs(num_sub_exponent) - (*_num1_p)).is_negative() 
+                                               || (num_sub_exponent == *_num_1_p && num_exponent->is_positive())) {
                                        return power(sub_basis,num_sub_exponent.mul(*num_exponent));
                                }
                        }
@@ -735,8 +744,7 @@ ex power::imag_part() const
        ex b=basis.imag_part();
        ex c=exponent.real_part();
        ex d=exponent.imag_part();
-       return
-               power(abs(basis),c)*exp(-d*atan2(b,a))*sin(c*atan2(b,a)+d*log(abs(basis)));
+       return power(abs(basis),c)*exp(-d*atan2(b,a))*sin(c*atan2(b,a)+d*log(abs(basis)));
 }
 
 // protected
@@ -792,6 +800,51 @@ ex power::expand(unsigned options) const
                return *this;
        }
 
+       // (x*p)^c -> x^c * p^c, if p>0
+       // makes sense before expanding the basis
+       if (is_exactly_a<mul>(basis) && !basis.info(info_flags::indefinite)) {
+               const mul &m = ex_to<mul>(basis);
+               exvector prodseq;
+               epvector powseq;
+               prodseq.reserve(m.seq.size() + 1);
+               powseq.reserve(m.seq.size() + 1);
+               epvector::const_iterator last = m.seq.end();
+               epvector::const_iterator cit = m.seq.begin();
+               bool possign = true;
+
+               // search for positive/negative factors
+               while (cit!=last) {
+                       ex e=m.recombine_pair_to_ex(*cit);
+                       if (e.info(info_flags::positive))
+                               prodseq.push_back(pow(e, exponent).expand(options));
+                       else if (e.info(info_flags::negative)) {
+                               prodseq.push_back(pow(-e, exponent).expand(options));
+                               possign = !possign;
+                       } else
+                               powseq.push_back(*cit);
+                       ++cit;
+               }
+
+               // take care on the numeric coefficient
+               ex coeff=(possign? _ex1 : _ex_1);
+               if (m.overall_coeff.info(info_flags::positive) && m.overall_coeff != _ex1)
+                       prodseq.push_back(power(m.overall_coeff, exponent));
+               else if (m.overall_coeff.info(info_flags::negative) && m.overall_coeff != _ex_1)
+                       prodseq.push_back(power(-m.overall_coeff, exponent));
+               else
+                       coeff *= m.overall_coeff;
+
+               // If positive/negative factors are found, then extract them.
+               // In either case we set a flag to avoid the second run on a part
+               // which does not have positive/negative terms.
+               if (prodseq.size() > 0) {
+                       ex newbasis = coeff*mul(powseq);
+                       ex_to<basic>(newbasis).setflag(status_flags::purely_indefinite);
+                       return ((new mul(prodseq))->setflag(status_flags::dynallocated)*(new power(newbasis, exponent))->setflag(status_flags::dynallocated).expand(options)).expand(options);
+               } else
+                       ex_to<basic>(basis).setflag(status_flags::purely_indefinite);
+       }
+
        const ex expanded_basis = basis.expand(options);
        const ex expanded_exponent = exponent.expand(options);
        
@@ -979,11 +1032,11 @@ ex power::expand_add_2(const add & a, unsigned options) const
                
                if (c.is_equal(_ex1)) {
                        if (is_exactly_a<mul>(r)) {
-                               sum.push_back(expair(expand_mul(ex_to<mul>(r), *_num2_p, options, true),
-                                                    _ex1));
+                               sum.push_back(a.combine_ex_with_coeff_to_pair(expand_mul(ex_to<mul>(r), *_num2_p, options, true),
+                                                                             _ex1));
                        } else {
-                               sum.push_back(expair((new power(r,_ex2))->setflag(status_flags::dynallocated),
-                                                    _ex1));
+                               sum.push_back(a.combine_ex_with_coeff_to_pair((new power(r,_ex2))->setflag(status_flags::dynallocated),
+                                                                             _ex1));
                        }
                } else {
                        if (is_exactly_a<mul>(r)) {