X-Git-Url: https://www.ginac.de/ginac.git//ginac.git?p=ginac.git;a=blobdiff_plain;f=ginac%2Fex.cpp;h=e8ce2da1eabce3e5ba12b0ad1e60f415ceb62d42;hp=789911258b43cfd57f7098675ad05cebda23f91a;hb=df5f8db62815995d87ebd4f97a5dbc0d1a327b94;hpb=956eeb82c513a723e864edefa857133282cf692b diff --git a/ginac/ex.cpp b/ginac/ex.cpp index 78991125..e8ce2da1 100644 --- a/ginac/ex.cpp +++ b/ginac/ex.cpp @@ -3,7 +3,7 @@ * Implementation of GiNaC's light-weight expression handles. */ /* - * GiNaC Copyright (C) 1999-2000 Johannes Gutenberg University Mainz, Germany + * GiNaC Copyright (C) 1999-2001 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 @@ -21,6 +21,7 @@ */ #include +#include #include "ex.h" #include "add.h" @@ -28,111 +29,20 @@ #include "ncmul.h" #include "numeric.h" #include "power.h" +#include "relational.h" +#include "indexed.h" +#include "lst.h" +#include "input_lexer.h" #include "debugmsg.h" #include "utils.h" -#ifndef NO_NAMESPACE_GINAC namespace GiNaC { -#endif // ndef NO_NAMESPACE_GINAC ////////// -// default constructor, destructor, copy constructor assignment operator and helpers +// other ctors ////////// -// public - -#ifndef INLINE_EX_CONSTRUCTORS - -ex::ex() : bp(ex0().bp) -{ - debugmsg("ex default constructor",LOGLEVEL_CONSTRUCT); - GINAC_ASSERT(ex0().bp!=0); - GINAC_ASSERT(ex0().bp->flags & status_flags::dynallocated); - GINAC_ASSERT(bp!=0); - ++bp->refcount; -} - -ex::~ex() -{ - debugmsg("ex destructor",LOGLEVEL_DESTRUCT); - GINAC_ASSERT(bp!=0); - GINAC_ASSERT(bp->flags & status_flags::dynallocated); - if (--bp->refcount == 0) { - delete bp; - } -} - -ex::ex(const ex & other) : bp(other.bp) -{ - debugmsg("ex copy constructor",LOGLEVEL_CONSTRUCT); - GINAC_ASSERT(bp!=0); - GINAC_ASSERT((bp->flags) & status_flags::dynallocated); - ++bp->refcount; -} - -const ex & ex::operator=(const ex & other) -{ - debugmsg("ex operator=",LOGLEVEL_ASSIGNMENT); - GINAC_ASSERT(bp!=0); - GINAC_ASSERT(bp->flags & status_flags::dynallocated); - GINAC_ASSERT(other.bp!=0); - GINAC_ASSERT(other.bp->flags & status_flags::dynallocated); - ++other.bp->refcount; - basic * tmpbp=other.bp; - if (--bp->refcount==0) { - delete bp; - } - bp=tmpbp; - return *this; -} - -#endif // ndef INLINE_EX_CONSTRUCTORS - -////////// -// other constructors -////////// - -// public - -#ifndef INLINE_EX_CONSTRUCTORS - -ex::ex(const basic & other) -{ - debugmsg("ex constructor from basic",LOGLEVEL_CONSTRUCT); - construct_from_basic(other); -} - -ex::ex(int i) -{ - debugmsg("ex constructor from int",LOGLEVEL_CONSTRUCT); - construct_from_int(i); -} - -ex::ex(unsigned int i) -{ - debugmsg("ex constructor from unsigned int",LOGLEVEL_CONSTRUCT); - construct_from_uint(i); -} - -ex::ex(long i) -{ - debugmsg("ex constructor from long",LOGLEVEL_CONSTRUCT); - construct_from_long(i); -} - -ex::ex(unsigned long i) -{ - debugmsg("ex constructor from unsigned long",LOGLEVEL_CONSTRUCT); - construct_from_ulong(i); -} - -ex::ex(double const d) -{ - debugmsg("ex constructor from double",LOGLEVEL_CONSTRUCT); - construct_from_double(d); -} - -#endif // ndef INLINE_EX_CONSTRUCTORS +// none (all inlined) ////////// // functions overriding virtual functions from bases classes @@ -152,521 +62,335 @@ ex::ex(double const d) // public -/** Swap the contents of two expressions. */ +/** Efficiently swap the contents of two expressions. */ void ex::swap(ex & other) { - debugmsg("ex swap",LOGLEVEL_MEMBER_FUNCTION); - - GINAC_ASSERT(bp!=0); - GINAC_ASSERT(bp->flags & status_flags::dynallocated); - GINAC_ASSERT(other.bp!=0); - GINAC_ASSERT(other.bp->flags & status_flags::dynallocated); - - basic * tmpbp=bp; - bp=other.bp; - other.bp=tmpbp; -} - -/** Output formatted to be useful as ginsh input. */ -void ex::print(ostream & os, unsigned upper_precedence) const -{ - debugmsg("ex print",LOGLEVEL_PRINT); - GINAC_ASSERT(bp!=0); - bp->print(os,upper_precedence); -} + debugmsg("ex swap",LOGLEVEL_MEMBER_FUNCTION); -void ex::printraw(ostream & os) const -{ - debugmsg("ex printraw",LOGLEVEL_PRINT); - GINAC_ASSERT(bp!=0); - os << "ex("; - bp->printraw(os); - os << ")"; + GINAC_ASSERT(bp!=0); + GINAC_ASSERT(bp->flags & status_flags::dynallocated); + GINAC_ASSERT(other.bp!=0); + GINAC_ASSERT(other.bp->flags & status_flags::dynallocated); + + basic * tmpbp = bp; + bp = other.bp; + other.bp = tmpbp; } -void ex::printtree(ostream & os, unsigned indent) const +/** Print expression to stream. The formatting of the output is determined + * by the kind of print_context object that is passed. Possible formattings + * include ginsh-parsable output (the default), tree-like output for + * debugging, and C++ source. + * @see print_context */ +void ex::print(const print_context & c, unsigned level) const { - debugmsg("ex printtree",LOGLEVEL_PRINT); - GINAC_ASSERT(bp!=0); - // os << "refcount=" << bp->refcount << " "; - bp->printtree(os,indent); + debugmsg("ex print", LOGLEVEL_PRINT); + GINAC_ASSERT(bp!=0); + bp->print(c, level); } -/** Print expression as a C++ statement. The output looks like - * " = ;". The "type" parameter has an effect - * on how number literals are printed. - * - * @param os output stream - * @param type variable type (one of the csrc_types) - * @param var_name variable name to be printed */ -void ex::printcsrc(ostream & os, unsigned type, const char *var_name) const +/** Print expression to stream in a tree-like format suitable for debugging. */ +void ex::printtree(std::ostream & os) const { - debugmsg("ex print csrc", LOGLEVEL_PRINT); - GINAC_ASSERT(bp!=0); - switch (type) { - case csrc_types::ctype_float: - os << "float "; - break; - case csrc_types::ctype_double: - os << "double "; - break; - case csrc_types::ctype_cl_N: - os << "cl_N "; - break; - } - os << var_name << " = "; - bp->printcsrc(os, type, 0); - os << ";\n"; + debugmsg("ex printtree", LOGLEVEL_PRINT); + GINAC_ASSERT(bp!=0); + bp->print(print_tree(os)); } /** Little wrapper arount print to be called within a debugger. */ void ex::dbgprint(void) const { - debugmsg("ex dbgprint",LOGLEVEL_PRINT); - GINAC_ASSERT(bp!=0); - bp->dbgprint(); + debugmsg("ex dbgprint", LOGLEVEL_PRINT); + GINAC_ASSERT(bp!=0); + bp->dbgprint(); } /** Little wrapper arount printtree to be called within a debugger. */ void ex::dbgprinttree(void) const { - debugmsg("ex dbgprinttree",LOGLEVEL_PRINT); - GINAC_ASSERT(bp!=0); - bp->dbgprinttree(); -} - -bool ex::info(unsigned inf) const -{ - if (inf == info_flags::normal_form) { - - // Polynomials are in normal form - if (info(info_flags::polynomial)) - return true; - - // polynomial^(-int) is in normal form - if (is_ex_exactly_of_type(*this, power)) - return op(1).info(info_flags::negint); - - // polynomial^(int) * polynomial^(int) * ... is in normal form - if (!is_ex_exactly_of_type(*this, mul)) - return false; - for (unsigned i=0; iinfo(inf); - } -} - -unsigned ex::nops() const -{ - GINAC_ASSERT(bp!=0); - return bp->nops(); + debugmsg("ex dbgprinttree", LOGLEVEL_PRINT); + GINAC_ASSERT(bp!=0); + bp->dbgprinttree(); } ex ex::expand(unsigned options) const { - GINAC_ASSERT(bp!=0); - return bp->expand(options); + GINAC_ASSERT(bp!=0); + if (bp->flags & status_flags::expanded) + return *bp; + else + return bp->expand(options); } -bool ex::has(const ex & other) const -{ - GINAC_ASSERT(bp!=0); - return bp->has(other); -} - -int ex::degree(const symbol & s) const -{ - GINAC_ASSERT(bp!=0); - return bp->degree(s); -} - -int ex::ldegree(const symbol & s) const -{ - GINAC_ASSERT(bp!=0); - return bp->ldegree(s); -} - -ex ex::coeff(const symbol & s, int n) const -{ - GINAC_ASSERT(bp!=0); - return bp->coeff(s,n); -} - -ex ex::numer(bool normalize) const -{ - ex n; - if (normalize && !info(info_flags::normal_form)) - n = normal(); - else - n = *this; - - // polynomial - if (n.info(info_flags::polynomial)) - return n; - - // something^(-int) - if (is_ex_exactly_of_type(n, power) && n.op(1).info(info_flags::negint)) - return _ex1(); - - // something^(int) * something^(int) * ... - if (!is_ex_exactly_of_type(n, mul)) - return n; - ex res = _ex1(); - for (unsigned i=0; icollect(s); -} - -ex ex::eval(int level) const -{ - GINAC_ASSERT(bp!=0); - return bp->eval(level); -} - -ex ex::evalf(int level) const +/** Compute partial derivative of an expression. + * + * @param s symbol by which the expression is derived + * @param nth order of derivative (default 1) + * @return partial derivative as a new expression */ +ex ex::diff(const symbol & s, unsigned nth) const { - GINAC_ASSERT(bp!=0); - return bp->evalf(level); -} + GINAC_ASSERT(bp!=0); -ex ex::subs(const lst & ls, const lst & lr) const -{ - GINAC_ASSERT(bp!=0); - return bp->subs(ls,lr); + if (!nth) + return *this; + else + return bp->diff(s, nth); } -ex ex::subs(const ex & e) const +/** Check whether expression matches a specified pattern. */ +bool ex::match(const ex & pattern) const { - GINAC_ASSERT(bp!=0); - return bp->subs(e); + lst repl_lst; + return bp->match(pattern, repl_lst); } -exvector ex::get_indices(void) const +/** Simplify/canonicalize expression containing indexed objects. This + * performs contraction of dummy indices where possible and checks whether + * the free indices in sums are consistent. + * + * @return simplified expression */ +ex ex::simplify_indexed(void) const { - GINAC_ASSERT(bp!=0); - return bp->get_indices(); + return GiNaC::simplify_indexed(*this); } -ex ex::simplify_ncmul(const exvector & v) const +/** Simplify/canonicalize expression containing indexed objects. This + * performs contraction of dummy indices where possible, checks whether + * the free indices in sums are consistent, and automatically replaces + * scalar products by known values if desired. + * + * @param sp Scalar products to be replaced automatically + * @return simplified expression */ +ex ex::simplify_indexed(const scalar_products & sp) const { - GINAC_ASSERT(bp!=0); - return bp->simplify_ncmul(v); + return GiNaC::simplify_indexed(*this, sp); } ex ex::operator[](const ex & index) const { - debugmsg("ex operator[ex]",LOGLEVEL_OPERATOR); - GINAC_ASSERT(bp!=0); - return (*bp)[index]; + debugmsg("ex operator[ex]",LOGLEVEL_OPERATOR); + GINAC_ASSERT(bp!=0); + return (*bp)[index]; } ex ex::operator[](int i) const { - debugmsg("ex operator[int]",LOGLEVEL_OPERATOR); - GINAC_ASSERT(bp!=0); - return (*bp)[i]; -} - -ex ex::op(int i) const -{ - debugmsg("ex op()",LOGLEVEL_MEMBER_FUNCTION); - GINAC_ASSERT(bp!=0); - return bp->op(i); + debugmsg("ex operator[int]",LOGLEVEL_OPERATOR); + GINAC_ASSERT(bp!=0); + return (*bp)[i]; } +/** Return modifyable operand/member at position i. */ ex & ex::let_op(int i) { - debugmsg("ex let_op()",LOGLEVEL_MEMBER_FUNCTION); - makewriteable(); - GINAC_ASSERT(bp!=0); - return bp->let_op(i); + debugmsg("ex let_op()",LOGLEVEL_MEMBER_FUNCTION); + makewriteable(); + GINAC_ASSERT(bp!=0); + return bp->let_op(i); } -#ifndef INLINE_EX_CONSTRUCTORS -int ex::compare(const ex & other) const +/** Left hand side of relational expression. */ +ex ex::lhs(void) const { - GINAC_ASSERT(bp!=0); - GINAC_ASSERT(other.bp!=0); - if (bp==other.bp) { - // special case: both expression point to same basic, trivially equal - return 0; - } - return bp->compare(*other.bp); + debugmsg("ex lhs()",LOGLEVEL_MEMBER_FUNCTION); + if (!is_ex_of_type(*this,relational)) + throw std::runtime_error("ex::lhs(): not a relation"); + return (*static_cast(bp)).lhs(); } -#endif // ndef INLINE_EX_CONSTRUCTORS -#ifndef INLINE_EX_CONSTRUCTORS -bool ex::is_equal(const ex & other) const +/** Right hand side of relational expression. */ +ex ex::rhs(void) const { - GINAC_ASSERT(bp!=0); - GINAC_ASSERT(other.bp!=0); - if (bp==other.bp) { - // special case: both expression point to same basic, trivially equal - return true; - } - return bp->is_equal(*other.bp); -} -#endif // ndef INLINE_EX_CONSTRUCTORS - -unsigned ex::return_type(void) const -{ - GINAC_ASSERT(bp!=0); - return bp->return_type(); -} - -unsigned ex::return_type_tinfo(void) const -{ - GINAC_ASSERT(bp!=0); - return bp->return_type_tinfo(); -} - -unsigned ex::gethash(void) const -{ - GINAC_ASSERT(bp!=0); - return bp->gethash(); -} - -ex ex::exadd(const ex & rh) const -{ - return (new add(*this,rh))->setflag(status_flags::dynallocated); -} - -ex ex::exmul(const ex & rh) const -{ - return (new mul(*this,rh))->setflag(status_flags::dynallocated); -} - -ex ex::exncmul(const ex & rh) const -{ - return (new ncmul(*this,rh))->setflag(status_flags::dynallocated); + debugmsg("ex rhs()",LOGLEVEL_MEMBER_FUNCTION); + if (!is_ex_of_type(*this,relational)) + throw std::runtime_error("ex::rhs(): not a relation"); + return (*static_cast(bp)).rhs(); } // private +/** Make this ex writable (if more than one ex handle the same basic) by + * unlinking the object and creating an unshared copy of it. */ void ex::makewriteable() { - debugmsg("ex makewriteable",LOGLEVEL_MEMBER_FUNCTION); - GINAC_ASSERT(bp!=0); - GINAC_ASSERT(bp->flags & status_flags::dynallocated); - if (bp->refcount > 1) { - basic * bp2 = bp->duplicate(); - ++bp2->refcount; - bp2->setflag(status_flags::dynallocated); - --bp->refcount; - bp = bp2; - } - GINAC_ASSERT(bp->refcount == 1); -} - + debugmsg("ex makewriteable",LOGLEVEL_MEMBER_FUNCTION); + GINAC_ASSERT(bp!=0); + GINAC_ASSERT(bp->flags & status_flags::dynallocated); + if (bp->refcount > 1) { + basic * bp2 = bp->duplicate(); + ++bp2->refcount; + bp2->setflag(status_flags::dynallocated); + --bp->refcount; + bp = bp2; + } + GINAC_ASSERT(bp->refcount==1); +} + +/** Ctor from basic implementation. + * @see ex::ex(const basic &) */ void ex::construct_from_basic(const basic & other) { - if ((other.flags & status_flags::evaluated)==0) { - // cf. copy constructor - const ex & tmpex = other.eval(1); // evaluate only one (top) level - bp = tmpex.bp; - GINAC_ASSERT(bp!=0); - GINAC_ASSERT(bp->flags & status_flags::dynallocated); - ++bp->refcount; - if ((other.flags & status_flags::dynallocated)&&(other.refcount==0)) { - delete &const_cast(other); - } - } else { - if (other.flags & status_flags::dynallocated) { - bp = &const_cast(other); - } else { - bp = other.duplicate(); - bp->setflag(status_flags::dynallocated); - } - GINAC_ASSERT(bp!=0); - // bp->clearflag(status_flags::evaluated); - ++bp->refcount; - } - GINAC_ASSERT(bp!=0); - GINAC_ASSERT(bp->flags & status_flags::dynallocated); + if ((other.flags & status_flags::evaluated)==0) { + // cf. copy ctor + const ex & tmpex = other.eval(1); // evaluate only one (top) level + bp = tmpex.bp; + GINAC_ASSERT(bp!=0); + GINAC_ASSERT(bp->flags & status_flags::dynallocated); + ++bp->refcount; + if ((other.flags & status_flags::dynallocated)&&(other.refcount==0)) + delete &const_cast(other); + } else { + if (other.flags & status_flags::dynallocated) { + // ok, it is already on the heap, so just copy bp: + bp = &const_cast(other); + } else { + // create a duplicate on the heap: + bp = other.duplicate(); + bp->setflag(status_flags::dynallocated); + } + GINAC_ASSERT(bp!=0); + ++bp->refcount; + } + GINAC_ASSERT(bp!=0); + GINAC_ASSERT(bp->flags & status_flags::dynallocated); } void ex::construct_from_int(int i) { - switch (i) { // some tiny efficiency-hack - case -2: - bp = _ex_2().bp; - ++bp->refcount; - break; - case -1: - bp = _ex_1().bp; - ++bp->refcount; - break; - case 0: - bp = _ex0().bp; - ++bp->refcount; - break; - case 1: - bp = _ex1().bp; - ++bp->refcount; - break; - case 2: - bp = _ex2().bp; - ++bp->refcount; - break; - default: - bp = new numeric(i); - bp->setflag(status_flags::dynallocated); - ++bp->refcount; - GINAC_ASSERT((bp->flags) & status_flags::dynallocated); - GINAC_ASSERT(bp->refcount=1); - } -} - + switch (i) { // some tiny efficiency-hack + case -2: + bp = _ex_2().bp; + ++bp->refcount; + break; + case -1: + bp = _ex_1().bp; + ++bp->refcount; + break; + case 0: + bp = _ex0().bp; + ++bp->refcount; + break; + case 1: + bp = _ex1().bp; + ++bp->refcount; + break; + case 2: + bp = _ex2().bp; + ++bp->refcount; + break; + default: + bp = new numeric(i); + bp->setflag(status_flags::dynallocated); + ++bp->refcount; + GINAC_ASSERT((bp->flags) & status_flags::dynallocated); + GINAC_ASSERT(bp->refcount==1); + } +} + void ex::construct_from_uint(unsigned int i) { - switch (i) { // some tiny efficiency-hack - case -2: - bp = _ex_2().bp; - ++bp->refcount; - break; - case -1: - bp = _ex_1().bp; - ++bp->refcount; - break; - case 0: - bp = _ex0().bp; - ++bp->refcount; - break; - case 1: - bp = _ex1().bp; - ++bp->refcount; - break; - case 2: - bp = _ex2().bp; - ++bp->refcount; - break; - default: - bp = new numeric(i); - bp->setflag(status_flags::dynallocated); - ++bp->refcount; - GINAC_ASSERT((bp->flags) & status_flags::dynallocated); - GINAC_ASSERT(bp->refcount=1); - } -} - + switch (i) { // some tiny efficiency-hack + case 0: + bp = _ex0().bp; + ++bp->refcount; + break; + case 1: + bp = _ex1().bp; + ++bp->refcount; + break; + case 2: + bp = _ex2().bp; + ++bp->refcount; + break; + default: + bp = new numeric(i); + bp->setflag(status_flags::dynallocated); + ++bp->refcount; + GINAC_ASSERT((bp->flags) & status_flags::dynallocated); + GINAC_ASSERT(bp->refcount==1); + } +} + void ex::construct_from_long(long i) { - switch (i) { // some tiny efficiency-hack - case -2: - bp = _ex_2().bp; - ++bp->refcount; - break; - case -1: - bp = _ex_1().bp; - ++bp->refcount; - break; - case 0: - bp = _ex0().bp; - ++bp->refcount; - break; - case 1: - bp = _ex1().bp; - ++bp->refcount; - break; - case 2: - bp = _ex2().bp; - ++bp->refcount; - break; - default: - bp = new numeric(i); - bp->setflag(status_flags::dynallocated); - ++bp->refcount; - GINAC_ASSERT((bp->flags) & status_flags::dynallocated); - GINAC_ASSERT(bp->refcount=1); - } -} - + switch (i) { // some tiny efficiency-hack + case -2: + bp = _ex_2().bp; + ++bp->refcount; + break; + case -1: + bp = _ex_1().bp; + ++bp->refcount; + break; + case 0: + bp = _ex0().bp; + ++bp->refcount; + break; + case 1: + bp = _ex1().bp; + ++bp->refcount; + break; + case 2: + bp = _ex2().bp; + ++bp->refcount; + break; + default: + bp = new numeric(i); + bp->setflag(status_flags::dynallocated); + ++bp->refcount; + GINAC_ASSERT((bp->flags) & status_flags::dynallocated); + GINAC_ASSERT(bp->refcount==1); + } +} + void ex::construct_from_ulong(unsigned long i) { - switch (i) { // some tiny efficiency-hack - case -2: - bp = _ex_2().bp; - ++bp->refcount; - break; - case -1: - bp = _ex_1().bp; - ++bp->refcount; - break; - case 0: - bp = _ex0().bp; - ++bp->refcount; - break; - case 1: - bp = _ex1().bp; - ++bp->refcount; - break; - case 2: - bp = _ex2().bp; - ++bp->refcount; - break; - default: - bp = new numeric(i); - bp->setflag(status_flags::dynallocated); - ++bp->refcount; - GINAC_ASSERT((bp->flags) & status_flags::dynallocated); - GINAC_ASSERT(bp->refcount=1); - } -} - + switch (i) { // some tiny efficiency-hack + case 0: + bp = _ex0().bp; + ++bp->refcount; + break; + case 1: + bp = _ex1().bp; + ++bp->refcount; + break; + case 2: + bp = _ex2().bp; + ++bp->refcount; + break; + default: + bp = new numeric(i); + bp->setflag(status_flags::dynallocated); + ++bp->refcount; + GINAC_ASSERT((bp->flags) & status_flags::dynallocated); + GINAC_ASSERT(bp->refcount==1); + } +} + void ex::construct_from_double(double d) { - bp = new numeric(d); - bp->setflag(status_flags::dynallocated); - ++bp->refcount; - GINAC_ASSERT((bp->flags) & status_flags::dynallocated); - GINAC_ASSERT(bp->refcount=1); + bp = new numeric(d); + bp->setflag(status_flags::dynallocated); + ++bp->refcount; + GINAC_ASSERT((bp->flags) & status_flags::dynallocated); + GINAC_ASSERT(bp->refcount==1); +} + +void ex::construct_from_string_and_lst(const std::string &s, const ex &l) +{ + set_lexer_string(s); + set_lexer_symbols(l); + ginac_yyrestart(NULL); + if (ginac_yyparse()) + throw (std::runtime_error(get_parser_error())); + else { + bp = parsed_ex.bp; + GINAC_ASSERT(bp!=0); + GINAC_ASSERT((bp->flags) & status_flags::dynallocated); + ++bp->refcount; + } } - + ////////// // static member variables ////////// @@ -686,6 +410,4 @@ void ex::construct_from_double(double d) // none -#ifndef NO_NAMESPACE_GINAC } // namespace GiNaC -#endif // ndef NO_NAMESPACE_GINAC