X-Git-Url: https://www.ginac.de/ginac.git//ginac.git?p=ginac.git;a=blobdiff_plain;f=ginac%2Foperators.cpp;h=c5972c6ce9acc946a393a9aa59b75548b4f604b4;hp=31c4cf03c3994f39a83b0bb4ffec8197083cf193;hb=aa1afbbdb118846cfc266a1d7a8d5188ac214e0e;hpb=28e5e38935cc32a4e5f8995645f92f12fedcdaf5

diff --git a/ginac/operators.cpp b/ginac/operators.cpp
index 31c4cf03..c5972c6c 100644
--- a/ginac/operators.cpp
+++ b/ginac/operators.cpp
@@ -3,7 +3,7 @@
  *  Implementation of GiNaC's overloaded operators. */
 
 /*
- *  GiNaC Copyright (C) 1999-2001 Johannes Gutenberg University Mainz, Germany
+ *  GiNaC Copyright (C) 1999-2009 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
@@ -17,7 +17,7 @@
  *
  *  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"
@@ -28,9 +28,11 @@
 #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. */
@@ -46,42 +48,39 @@ static inline const ex exmul(const ex & lh, const ex & rh)
 	// 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));
 }
 
 
@@ -89,25 +88,21 @@ const ex operator/(const ex & lh, const ex & rh)
 
 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);
 }
 
@@ -116,26 +111,22 @@ const numeric operator/(const numeric & lh, const numeric & 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));
 }
 
 
@@ -143,28 +134,24 @@ ex & operator/=(ex & lh, const ex & rh)
 
 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;
 }
@@ -174,26 +161,22 @@ numeric & operator/=(numeric & lh, const numeric & rh)
 
 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);
 }
 
 
@@ -202,24 +185,21 @@ const numeric operator-(const numeric & 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;
 }
 
@@ -227,25 +207,22 @@ const ex operator++(ex & lh, int)
  *  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;
 }
 
@@ -253,9 +230,8 @@ numeric& operator--(numeric & 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;
 }
 
@@ -263,9 +239,8 @@ const numeric operator++(numeric & lh, int)
  *  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;
 }
 
@@ -273,45 +248,191 @@ const numeric operator--(numeric & lh, int)
 
 const relational operator==(const ex & lh, const ex & rh)
 {
-	debugmsg("operator==(ex,ex)",LOGLEVEL_OPERATOR);
 	return relational(lh,rh,relational::equal);
 }
 
 const relational operator!=(const ex & lh, const ex & rh)
 {
-	debugmsg("operator!=(ex,ex)",LOGLEVEL_OPERATOR);
 	return relational(lh,rh,relational::not_equal);
 }
 
 const relational operator<(const ex & lh, const ex & rh)
 {
-	debugmsg("operator<(ex,ex)",LOGLEVEL_OPERATOR);
 	return relational(lh,rh,relational::less);
 }
 
 const relational operator<=(const ex & lh, const ex & rh)
 {
-	debugmsg("operator<=(ex,ex)",LOGLEVEL_OPERATOR);
 	return relational(lh,rh,relational::less_or_equal);
 }
 
 const relational operator>(const ex & lh, const ex & rh)
 {
-	debugmsg("operator>(ex,ex)",LOGLEVEL_OPERATOR);
 	return relational(lh,rh,relational::greater);
 }
 
 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;
 }
 
@@ -320,4 +441,71 @@ std::istream & operator>>(std::istream & is, ex & e)
 	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