3 * Implementation of GiNaC's sums of expressions. */
6 * GiNaC Copyright (C) 1999-2001 Johannes Gutenberg University Mainz, Germany
8 * This program is free software; you can redistribute it and/or modify
9 * it under the terms of the GNU General Public License as published by
10 * the Free Software Foundation; either version 2 of the License, or
11 * (at your option) any later version.
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
18 * You should have received a copy of the GNU General Public License
19 * along with this program; if not, write to the Free Software
20 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
34 GINAC_IMPLEMENT_REGISTERED_CLASS(add, expairseq)
37 // default constructor, destructor, copy constructor assignment operator and helpers
42 debugmsg("add default constructor",LOGLEVEL_CONSTRUCT);
43 tinfo_key = TINFO_add;
55 add::add(const ex & lh, const ex & rh)
57 debugmsg("add constructor from ex,ex",LOGLEVEL_CONSTRUCT);
58 tinfo_key = TINFO_add;
59 overall_coeff = _ex0();
60 construct_from_2_ex(lh,rh);
61 GINAC_ASSERT(is_canonical());
64 add::add(const exvector & v)
66 debugmsg("add constructor from exvector",LOGLEVEL_CONSTRUCT);
67 tinfo_key = TINFO_add;
68 overall_coeff = _ex0();
69 construct_from_exvector(v);
70 GINAC_ASSERT(is_canonical());
73 add::add(const epvector & v)
75 debugmsg("add constructor from epvector",LOGLEVEL_CONSTRUCT);
76 tinfo_key = TINFO_add;
77 overall_coeff = _ex0();
78 construct_from_epvector(v);
79 GINAC_ASSERT(is_canonical());
82 add::add(const epvector & v, const ex & oc)
84 debugmsg("add constructor from epvector,ex",LOGLEVEL_CONSTRUCT);
85 tinfo_key = TINFO_add;
87 construct_from_epvector(v);
88 GINAC_ASSERT(is_canonical());
91 add::add(epvector * vp, const ex & oc)
93 debugmsg("add constructor from epvector *,ex",LOGLEVEL_CONSTRUCT);
94 tinfo_key = TINFO_add;
97 construct_from_epvector(*vp);
99 GINAC_ASSERT(is_canonical());
106 DEFAULT_ARCHIVING(add)
109 // functions overriding virtual functions from bases classes
114 void add::print(const print_context & c, unsigned level) const
116 debugmsg("add print", LOGLEVEL_PRINT);
118 if (is_of_type(c, print_tree)) {
120 inherited::print(c, level);
122 } else if (is_of_type(c, print_csrc)) {
124 if (precedence <= level)
127 // Print arguments, separated by "+"
128 epvector::const_iterator it = seq.begin(), itend = seq.end();
129 while (it != itend) {
131 // If the coefficient is -1, it is replaced by a single minus sign
132 if (it->coeff.compare(_num1()) == 0) {
133 it->rest.bp->print(c, precedence);
134 } else if (it->coeff.compare(_num_1()) == 0) {
136 it->rest.bp->print(c, precedence);
137 } else if (ex_to_numeric(it->coeff).numer().compare(_num1()) == 0) {
138 it->rest.bp->print(c, precedence);
140 ex_to_numeric(it->coeff).denom().print(c, precedence);
141 } else if (ex_to_numeric(it->coeff).numer().compare(_num_1()) == 0) {
143 it->rest.bp->print(c, precedence);
145 ex_to_numeric(it->coeff).denom().print(c, precedence);
147 it->coeff.bp->print(c, precedence);
149 it->rest.bp->print(c, precedence);
152 // Separator is "+", except if the following expression would have a leading minus sign
154 if (it != itend && !(it->coeff.compare(_num0()) < 0 || (it->coeff.compare(_num1()) == 0 && is_ex_exactly_of_type(it->rest, numeric) && it->rest.compare(_num0()) < 0)))
158 if (!overall_coeff.is_zero()) {
159 if (overall_coeff.info(info_flags::positive))
161 overall_coeff.bp->print(c, precedence);
164 if (precedence <= level)
169 if (precedence <= level)
175 // First print the overall numeric coefficient, if present
176 if (!overall_coeff.is_zero()) {
177 if (!is_of_type(c, print_tree))
178 overall_coeff.print(c, 0);
180 overall_coeff.print(c, precedence);
184 // Then proceed with the remaining factors
185 epvector::const_iterator it = seq.begin(), itend = seq.end();
186 while (it != itend) {
187 coeff = ex_to_numeric(it->coeff);
189 if (coeff.csgn() == -1) c.s << '-'; else c.s << '+';
191 if (coeff.csgn() == -1) c.s << '-';
194 if (!coeff.is_equal(_num1()) &&
195 !coeff.is_equal(_num_1())) {
196 if (coeff.is_rational()) {
197 if (coeff.is_negative())
202 if (coeff.csgn() == -1)
203 (-coeff).print(c, precedence);
205 coeff.print(c, precedence);
209 it->rest.print(c, precedence);
213 if (precedence <= level)
218 bool add::info(unsigned inf) const
221 case info_flags::polynomial:
222 case info_flags::integer_polynomial:
223 case info_flags::cinteger_polynomial:
224 case info_flags::rational_polynomial:
225 case info_flags::crational_polynomial:
226 case info_flags::rational_function: {
227 for (epvector::const_iterator i=seq.begin(); i!=seq.end(); ++i) {
228 if (!(recombine_pair_to_ex(*i).info(inf)))
231 return overall_coeff.info(inf);
233 case info_flags::algebraic: {
234 for (epvector::const_iterator i=seq.begin(); i!=seq.end(); ++i) {
235 if ((recombine_pair_to_ex(*i).info(inf)))
241 return inherited::info(inf);
244 int add::degree(const ex & s) const
247 if (!overall_coeff.is_equal(_ex0()))
251 for (epvector::const_iterator cit=seq.begin(); cit!=seq.end(); ++cit) {
252 cur_deg = (*cit).rest.degree(s);
259 int add::ldegree(const ex & s) const
262 if (!overall_coeff.is_equal(_ex0()))
266 for (epvector::const_iterator cit=seq.begin(); cit!=seq.end(); ++cit) {
267 cur_deg = (*cit).rest.ldegree(s);
268 if (cur_deg<deg) deg=cur_deg;
273 ex add::coeff(const ex & s, int n) const
276 coeffseq.reserve(seq.size());
278 epvector::const_iterator it=seq.begin();
279 while (it!=seq.end()) {
280 coeffseq.push_back(combine_ex_with_coeff_to_pair((*it).rest.coeff(s,n),
285 return (new add(coeffseq,overall_coeff))->setflag(status_flags::dynallocated);
287 return (new add(coeffseq))->setflag(status_flags::dynallocated);
290 ex add::eval(int level) const
292 // simplifications: +(;c) -> c
295 debugmsg("add eval",LOGLEVEL_MEMBER_FUNCTION);
297 epvector * evaled_seqp = evalchildren(level);
298 if (evaled_seqp!=0) {
299 // do more evaluation later
300 return (new add(evaled_seqp,overall_coeff))->
301 setflag(status_flags::dynallocated);
304 #ifdef DO_GINAC_ASSERT
305 for (epvector::const_iterator cit=seq.begin(); cit!=seq.end(); ++cit) {
306 GINAC_ASSERT(!is_ex_exactly_of_type((*cit).rest,add));
307 if (is_ex_exactly_of_type((*cit).rest,numeric))
309 GINAC_ASSERT(!is_ex_exactly_of_type((*cit).rest,numeric));
311 #endif // def DO_GINAC_ASSERT
313 if (flags & status_flags::evaluated) {
314 GINAC_ASSERT(seq.size()>0);
315 GINAC_ASSERT(seq.size()>1 || !overall_coeff.is_zero());
319 int seq_size = seq.size();
322 return overall_coeff;
323 } else if ((seq_size==1) && overall_coeff.is_equal(_ex0())) {
325 return recombine_pair_to_ex(*(seq.begin()));
330 ex add::simplify_ncmul(const exvector & v) const
333 return inherited::simplify_ncmul(v);
335 return (*seq.begin()).rest.simplify_ncmul(v);
340 /** Implementation of ex::diff() for a sum. It differentiates each term.
342 ex add::derivative(const symbol & s) const
344 // D(a+b+c)=D(a)+D(b)+D(c)
345 return (new add(diffchildren(s)))->setflag(status_flags::dynallocated);
348 int add::compare_same_type(const basic & other) const
350 return inherited::compare_same_type(other);
353 bool add::is_equal_same_type(const basic & other) const
355 return inherited::is_equal_same_type(other);
358 unsigned add::return_type(void) const
361 return return_types::commutative;
363 return (*seq.begin()).rest.return_type();
366 unsigned add::return_type_tinfo(void) const
371 return (*seq.begin()).rest.return_type_tinfo();
374 ex add::thisexpairseq(const epvector & v, const ex & oc) const
376 return (new add(v,oc))->setflag(status_flags::dynallocated);
379 ex add::thisexpairseq(epvector * vp, const ex & oc) const
381 return (new add(vp,oc))->setflag(status_flags::dynallocated);
384 expair add::split_ex_to_pair(const ex & e) const
386 if (is_ex_exactly_of_type(e,mul)) {
387 const mul &mulref = ex_to_mul(e);
388 ex numfactor = mulref.overall_coeff;
389 mul *mulcopyp = new mul(mulref);
390 mulcopyp->overall_coeff = _ex1();
391 mulcopyp->clearflag(status_flags::evaluated);
392 mulcopyp->clearflag(status_flags::hash_calculated);
393 mulcopyp->setflag(status_flags::dynallocated);
394 return expair(*mulcopyp,numfactor);
396 return expair(e,_ex1());
399 expair add::combine_ex_with_coeff_to_pair(const ex & e,
402 GINAC_ASSERT(is_ex_exactly_of_type(c, numeric));
403 if (is_ex_exactly_of_type(e, mul)) {
404 const mul &mulref = ex_to_mul(e);
405 ex numfactor = mulref.overall_coeff;
406 mul *mulcopyp = new mul(mulref);
407 mulcopyp->overall_coeff = _ex1();
408 mulcopyp->clearflag(status_flags::evaluated);
409 mulcopyp->clearflag(status_flags::hash_calculated);
410 mulcopyp->setflag(status_flags::dynallocated);
411 if (are_ex_trivially_equal(c, _ex1()))
412 return expair(*mulcopyp, numfactor);
413 else if (are_ex_trivially_equal(numfactor, _ex1()))
414 return expair(*mulcopyp, c);
416 return expair(*mulcopyp, ex_to_numeric(numfactor).mul_dyn(ex_to_numeric(c)));
417 } else if (is_ex_exactly_of_type(e, numeric)) {
418 if (are_ex_trivially_equal(c, _ex1()))
419 return expair(e, _ex1());
420 return expair(ex_to_numeric(e).mul_dyn(ex_to_numeric(c)), _ex1());
425 expair add::combine_pair_with_coeff_to_pair(const expair & p,
428 GINAC_ASSERT(is_ex_exactly_of_type(p.coeff,numeric));
429 GINAC_ASSERT(is_ex_exactly_of_type(c,numeric));
431 if (is_ex_exactly_of_type(p.rest,numeric)) {
432 GINAC_ASSERT(ex_to_numeric(p.coeff).is_equal(_num1())); // should be normalized
433 return expair(ex_to_numeric(p.rest).mul_dyn(ex_to_numeric(c)),_ex1());
436 return expair(p.rest,ex_to_numeric(p.coeff).mul_dyn(ex_to_numeric(c)));
439 ex add::recombine_pair_to_ex(const expair & p) const
441 if (ex_to_numeric(p.coeff).is_equal(_num1()))
444 return p.rest*p.coeff;
447 ex add::expand(unsigned options) const
449 if (flags & status_flags::expanded)
452 epvector * vp = expandchildren(options);
454 // the terms have not changed, so it is safe to declare this expanded
455 setflag(status_flags::expanded);
459 return (new add(vp,overall_coeff))->setflag(status_flags::expanded | status_flags::dynallocated);
463 // static member variables
468 unsigned add::precedence = 40;