* Implementation of GiNaC's overloaded operators. */
/*
- * GiNaC Copyright (C) 1999-2001 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
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
*/
#include "operators.h"
#include "ncmul.h"
#include "relational.h"
#include "print.h"
-#include "debugmsg.h"
#include "utils.h"
+#include <iomanip>
+#include <iostream>
+
namespace GiNaC {
/** Used internally by operator+() to add two ex objects together. */
// ncmul::eval()'s rule to pull out commutative elements we need to check
// only one of the elements.
if (rh.return_type()==return_types::commutative ||
- lh.return_type()==return_types::commutative)
+ lh.return_type()==return_types::commutative) {
return (new mul(lh,rh))->setflag(status_flags::dynallocated);
- else
+ } else {
return (new ncmul(lh,rh))->setflag(status_flags::dynallocated);
+ }
}
/** Used internally by operator-() and friends to change the sign of an argument. */
static inline const ex exminus(const ex & lh)
{
- return (new mul(lh,_ex_1()))->setflag(status_flags::dynallocated);
+ return (new mul(lh,_ex_1))->setflag(status_flags::dynallocated);
}
// binary arithmetic operators ex with ex
const ex operator+(const ex & lh, const ex & rh)
{
- debugmsg("operator+(ex,ex)",LOGLEVEL_OPERATOR);
return exadd(lh, rh);
}
const ex operator-(const ex & lh, const ex & rh)
{
- debugmsg("operator-(ex,ex)",LOGLEVEL_OPERATOR);
return exadd(lh, exminus(rh));
}
const ex operator*(const ex & lh, const ex & rh)
{
- debugmsg("operator*(ex,ex)",LOGLEVEL_OPERATOR);
return exmul(lh, rh);
}
const ex operator/(const ex & lh, const ex & rh)
{
- debugmsg("operator/(ex,ex)",LOGLEVEL_OPERATOR);
- return exmul(lh, power(rh,_ex_1()));
+ return exmul(lh, power(rh,_ex_1));
}
const numeric operator+(const numeric & lh, const numeric & rh)
{
- debugmsg("operator+(numeric,numeric)",LOGLEVEL_OPERATOR);
return lh.add(rh);
}
const numeric operator-(const numeric & lh, const numeric & rh)
{
- debugmsg("operator-(numeric,numeric)",LOGLEVEL_OPERATOR);
return lh.sub(rh);
}
const numeric operator*(const numeric & lh, const numeric & rh)
{
- debugmsg("operator*(numeric,numeric)",LOGLEVEL_OPERATOR);
return lh.mul(rh);
}
const numeric operator/(const numeric & lh, const numeric & rh)
{
- debugmsg("operator/(numeric,ex)",LOGLEVEL_OPERATOR);
return lh.div(rh);
}
ex & operator+=(ex & lh, const ex & rh)
{
- debugmsg("operator+=(ex,ex)",LOGLEVEL_OPERATOR);
return lh = exadd(lh, rh);
}
ex & operator-=(ex & lh, const ex & rh)
{
- debugmsg("operator-=(ex,ex)",LOGLEVEL_OPERATOR);
return lh = exadd(lh, exminus(rh));
}
ex & operator*=(ex & lh, const ex & rh)
{
- debugmsg("operator*=(ex,ex)",LOGLEVEL_OPERATOR);
return lh = exmul(lh, rh);
}
ex & operator/=(ex & lh, const ex & rh)
{
- debugmsg("operator/=(ex,ex)",LOGLEVEL_OPERATOR);
- return lh = exmul(lh, power(rh,_ex_1()));
+ return lh = exmul(lh, power(rh,_ex_1));
}
numeric & operator+=(numeric & lh, const numeric & rh)
{
- debugmsg("operator+=(numeric,numeric)",LOGLEVEL_OPERATOR);
lh = lh.add(rh);
return lh;
}
numeric & operator-=(numeric & lh, const numeric & rh)
{
- debugmsg("operator-=(numeric,numeric)",LOGLEVEL_OPERATOR);
lh = lh.sub(rh);
return lh;
}
numeric & operator*=(numeric & lh, const numeric & rh)
{
- debugmsg("operator*=(numeric,numeric)",LOGLEVEL_OPERATOR);
lh = lh.mul(rh);
return lh;
}
numeric & operator/=(numeric & lh, const numeric & rh)
{
- debugmsg("operator/=(numeric,numeric)",LOGLEVEL_OPERATOR);
lh = lh.div(rh);
return lh;
}
const ex operator+(const ex & lh)
{
- debugmsg("operator+(ex)",LOGLEVEL_OPERATOR);
return lh;
}
const ex operator-(const ex & lh)
{
- debugmsg("operator-(ex)",LOGLEVEL_OPERATOR);
return exminus(lh);
}
const numeric operator+(const numeric & lh)
{
- debugmsg("operator+(numeric)",LOGLEVEL_OPERATOR);
return lh;
}
const numeric operator-(const numeric & lh)
{
- debugmsg("operator-(numeric)",LOGLEVEL_OPERATOR);
- return _num_1().mul(lh);
+ return _num_1_p->mul(lh);
}
/** Expression prefix increment. Adds 1 and returns incremented ex. */
ex & operator++(ex & rh)
{
- debugmsg("operator++(ex)",LOGLEVEL_OPERATOR);
- return rh = exadd(rh, _ex1());
+ return rh = exadd(rh, _ex1);
}
/** Expression prefix decrement. Subtracts 1 and returns decremented ex. */
ex & operator--(ex & rh)
{
- debugmsg("operator--(ex)",LOGLEVEL_OPERATOR);
- return rh = exadd(rh, _ex_1());
+ return rh = exadd(rh, _ex_1);
}
/** Expression postfix increment. Returns the ex and leaves the original
* incremented by 1. */
const ex operator++(ex & lh, int)
{
- debugmsg("operator++(ex,int)",LOGLEVEL_OPERATOR);
ex tmp(lh);
- lh = exadd(lh, _ex1());
+ lh = exadd(lh, _ex1);
return tmp;
}
-/** Expression Postfix decrement. Returns the ex and leaves the original
+/** Expression postfix decrement. Returns the ex and leaves the original
* decremented by 1. */
const ex operator--(ex & lh, int)
{
- debugmsg("operator--(ex,int)",LOGLEVEL_OPERATOR);
ex tmp(lh);
- lh = exadd(lh, _ex_1());
+ lh = exadd(lh, _ex_1);
return tmp;
}
/** Numeric prefix increment. Adds 1 and returns incremented number. */
numeric& operator++(numeric & rh)
{
- debugmsg("operator++(numeric)",LOGLEVEL_OPERATOR);
- rh = rh.add(_num1());
+ rh = rh.add(*_num1_p);
return rh;
}
/** Numeric prefix decrement. Subtracts 1 and returns decremented number. */
numeric& operator--(numeric & rh)
{
- debugmsg("operator--(numeric)",LOGLEVEL_OPERATOR);
- rh = rh.add(_num_1());
+ rh = rh.add(*_num_1_p);
return rh;
}
* incremented by 1. */
const numeric operator++(numeric & lh, int)
{
- debugmsg("operator++(numeric,int)",LOGLEVEL_OPERATOR);
numeric tmp(lh);
- lh = lh.add(_num1());
+ lh = lh.add(*_num1_p);
return tmp;
}
-/** Numeric Postfix decrement. Returns the number and leaves the original
+/** Numeric postfix decrement. Returns the number and leaves the original
* decremented by 1. */
const numeric operator--(numeric & lh, int)
{
- debugmsg("operator--(numeric,int)",LOGLEVEL_OPERATOR);
numeric tmp(lh);
- lh = lh.add(_num_1());
+ lh = lh.add(*_num_1_p);
return tmp;
}
// binary relational operators ex with ex
-relational operator==(const ex & lh, const ex & rh)
+const relational operator==(const ex & lh, const ex & rh)
{
- debugmsg("operator==(ex,ex)",LOGLEVEL_OPERATOR);
return relational(lh,rh,relational::equal);
}
-relational operator!=(const ex & lh, const ex & rh)
+const relational operator!=(const ex & lh, const ex & rh)
{
- debugmsg("operator!=(ex,ex)",LOGLEVEL_OPERATOR);
return relational(lh,rh,relational::not_equal);
}
-relational operator<(const ex & lh, const ex & rh)
+const relational operator<(const ex & lh, const ex & rh)
{
- debugmsg("operator<(ex,ex)",LOGLEVEL_OPERATOR);
return relational(lh,rh,relational::less);
}
-relational operator<=(const ex & lh, const ex & rh)
+const relational operator<=(const ex & lh, const ex & rh)
{
- debugmsg("operator<=(ex,ex)",LOGLEVEL_OPERATOR);
return relational(lh,rh,relational::less_or_equal);
}
-relational operator>(const ex & lh, const ex & rh)
+const relational operator>(const ex & lh, const ex & rh)
{
- debugmsg("operator>(ex,ex)",LOGLEVEL_OPERATOR);
return relational(lh,rh,relational::greater);
}
-relational operator>=(const ex & lh, const ex & rh)
+const relational operator>=(const ex & lh, const ex & rh)
{
- debugmsg("operator>=(ex,ex)",LOGLEVEL_OPERATOR);
return relational(lh,rh,relational::greater_or_equal);
}
-// input/output stream operators
+// input/output stream operators and manipulators
+
+static int my_ios_index()
+{
+ static int i = std::ios_base::xalloc();
+ return i;
+}
+
+// Stream format gets copied or destroyed
+static void my_ios_callback(std::ios_base::event ev, std::ios_base & s, int i)
+{
+ print_context *p = static_cast<print_context *>(s.pword(i));
+ if (ev == std::ios_base::erase_event) {
+ delete p;
+ s.pword(i) = 0;
+ } else if (ev == std::ios_base::copyfmt_event && p != 0)
+ s.pword(i) = p->duplicate();
+}
+
+enum {
+ callback_registered = 1
+};
+
+// Get print_context associated with stream, may return 0 if no context has
+// been associated yet
+static inline print_context *get_print_context(std::ios_base & s)
+{
+ return static_cast<print_context *>(s.pword(my_ios_index()));
+}
+
+// Set print_context associated with stream, retain options
+static void set_print_context(std::ios_base & s, const print_context & c)
+{
+ int i = my_ios_index();
+ long flags = s.iword(i);
+ if (!(flags & callback_registered)) {
+ s.register_callback(my_ios_callback, i);
+ s.iword(i) = flags | callback_registered;
+ }
+ print_context *p = static_cast<print_context *>(s.pword(i));
+ unsigned options = p ? p->options : c.options;
+ delete p;
+ p = c.duplicate();
+ p->options = options;
+ s.pword(i) = p;
+}
+
+// Get options for print_context associated with stream
+static inline unsigned get_print_options(std::ios_base & s)
+{
+ print_context *p = get_print_context(s);
+ return p ? p->options : 0;
+}
+
+// Set options for print_context associated with stream
+static void set_print_options(std::ostream & s, unsigned options)
+{
+ print_context *p = get_print_context(s);
+ if (p == 0)
+ set_print_context(s, print_dflt(s, options));
+ else
+ p->options = options;
+}
std::ostream & operator<<(std::ostream & os, const ex & e)
{
- e.print(print_context(os));
+ print_context *p = get_print_context(os);
+ if (p == 0)
+ e.print(print_dflt(os));
+ else
+ e.print(*p);
+ return os;
+}
+
+std::ostream & operator<<(std::ostream & os, const exvector & e)
+{
+ print_context *p = get_print_context(os);
+ exvector::const_iterator i = e.begin();
+ exvector::const_iterator vend = e.end();
+
+ if (i==vend) {
+ os << "[]";
+ return os;
+ }
+
+ os << "[";
+ while (true) {
+ if (p == 0)
+ i -> print(print_dflt(os));
+ else
+ i -> print(*p);
+ ++i;
+ if (i==vend)
+ break;
+ os << ",";
+ }
+ os << "]";
+
+ return os;
+}
+
+std::ostream & operator<<(std::ostream & os, const exset & e)
+{
+ print_context *p = get_print_context(os);
+ exset::const_iterator i = e.begin();
+ exset::const_iterator send = e.end();
+
+ if (i==send) {
+ os << "<>";
+ return os;
+ }
+
+ os << "<";
+ while (true) {
+ if (p == 0)
+ i->print(print_dflt(os));
+ else
+ i->print(*p);
+ ++i;
+ if (i == send)
+ break;
+ os << ",";
+ }
+ os << ">";
+
+ return os;
+}
+
+std::ostream & operator<<(std::ostream & os, const exmap & e)
+{
+ print_context *p = get_print_context(os);
+ exmap::const_iterator i = e.begin();
+ exmap::const_iterator mend = e.end();
+
+ if (i==mend) {
+ os << "{}";
+ return os;
+ }
+
+ os << "{";
+ while (true) {
+ if (p == 0)
+ i->first.print(print_dflt(os));
+ else
+ i->first.print(*p);
+ os << "==";
+ if (p == 0)
+ i->second.print(print_dflt(os));
+ else
+ i->second.print(*p);
+ ++i;
+ if( i==mend )
+ break;
+ os << ",";
+ }
+ os << "}";
+
return os;
}
throw (std::logic_error("expression input from streams not implemented"));
}
+std::ostream & dflt(std::ostream & os)
+{
+ set_print_context(os, print_dflt(os));
+ set_print_options(os, 0);
+ return os;
+}
+
+std::ostream & latex(std::ostream & os)
+{
+ set_print_context(os, print_latex(os));
+ return os;
+}
+
+std::ostream & python(std::ostream & os)
+{
+ set_print_context(os, print_python(os));
+ return os;
+}
+
+std::ostream & python_repr(std::ostream & os)
+{
+ set_print_context(os, print_python_repr(os));
+ return os;
+}
+
+std::ostream & tree(std::ostream & os)
+{
+ set_print_context(os, print_tree(os));
+ return os;
+}
+
+std::ostream & csrc(std::ostream & os)
+{
+ set_print_context(os, print_csrc_double(os));
+ return os;
+}
+
+std::ostream & csrc_float(std::ostream & os)
+{
+ set_print_context(os, print_csrc_float(os));
+ return os;
+}
+
+std::ostream & csrc_double(std::ostream & os)
+{
+ set_print_context(os, print_csrc_double(os));
+ return os;
+}
+
+std::ostream & csrc_cl_N(std::ostream & os)
+{
+ set_print_context(os, print_csrc_cl_N(os));
+ return os;
+}
+
+std::ostream & index_dimensions(std::ostream & os)
+{
+ set_print_options(os, get_print_options(os) | print_options::print_index_dimensions);
+ return os;
+}
+
+std::ostream & no_index_dimensions(std::ostream & os)
+{
+ set_print_options(os, get_print_options(os) & ~print_options::print_index_dimensions);
+ return os;
+}
+
} // namespace GiNaC