marci@1097: // -*- c++ -*-
marci@1097: #ifndef LEMON_EXPRESSION_H
marci@1097: #define LEMON_EXPRESSION_H
marci@1097: 
marci@1097: #include <iostream>
marci@1097: #include <map>
marci@1097: 
marci@1097: namespace lemon {
marci@1097: 
marci@1097:   /*! \brief Linear expression
marci@1097: 
marci@1097:     \c Expr<_Col,_Value> implements a class of linear expressions with the 
marci@1097:     operations of addition and multiplication with scalar. 
marci@1097: 
marci@1097:     \author Marton Makai
marci@1097:    */
marci@1097:   template <typename _Col, typename _Value>
marci@1097:   class Expr;
marci@1097: 
marci@1097:   template <typename _Col, typename _Value>
marci@1097:   class Expr {
marci@1099: //  protected:
marci@1099:   public:
marci@1097:     typedef 
marci@1097:     typename std::map<_Col, _Value> Data; 
marci@1097:     Data data;
marci@1097:   public:
marci@1099:     void simplify() {
marci@1099:       for (typename Data::iterator i=data.begin(); 
marci@1099: 	   i!=data.end(); ++i) {
marci@1099: 	if ((*i).second==0) data.erase(i);
marci@1099:       }
marci@1099:     }
marci@1097:     Expr() { }
marci@1097:     Expr(_Col _col) { 
marci@1097:       data.insert(std::make_pair(_col, 1));
marci@1097:     }
marci@1097:     Expr& operator*=(_Value _value) {
marci@1097:       for (typename Data::iterator i=data.begin(); 
marci@1097: 	   i!=data.end(); ++i) {
marci@1097: 	(*i).second *= _value;
marci@1097:       }
marci@1099:       simplify();
marci@1097:       return *this;
marci@1097:     }
marci@1097:     Expr& operator+=(const Expr<_Col, _Value>& expr) {
marci@1097:       for (typename Data::const_iterator j=expr.data.begin(); 
marci@1097: 	   j!=expr.data.end(); ++j) {
marci@1097: 	typename Data::iterator i=data.find((*j).first);
marci@1097: 	if (i==data.end()) {
marci@1097: 	  data.insert(std::make_pair((*j).first, (*j).second));
marci@1097: 	} else {
marci@1097: 	  (*i).second+=(*j).second;
marci@1097: 	}
marci@1097:       }
marci@1099:       simplify();
marci@1099:       return *this;
marci@1099:     }
marci@1099:     Expr& operator-=(const Expr<_Col, _Value>& expr) {
marci@1099:       for (typename Data::const_iterator j=expr.data.begin(); 
marci@1099: 	   j!=expr.data.end(); ++j) {
marci@1099: 	typename Data::iterator i=data.find((*j).first);
marci@1099: 	if (i==data.end()) {
marci@1099: 	  data.insert(std::make_pair((*j).first, -(*j).second));
marci@1099: 	} else {
marci@1099: 	  (*i).second+=-(*j).second;
marci@1099: 	}
marci@1099:       }
marci@1099:       simplify();
marci@1097:       return *this;
marci@1097:     }
marci@1097:     template <typename _C, typename _V> 
marci@1097:     friend std::ostream& operator<<(std::ostream& os, 
marci@1097: 				    const Expr<_C, _V>& expr);
marci@1097:   };
marci@1097: 
marci@1097:   template <typename _Col, typename _Value>
marci@1097:   Expr<_Col, _Value> operator*(_Value _value, _Col _col) {
marci@1097:     Expr<_Col, _Value> tmp(_col);
marci@1097:     tmp*=_value;
marci@1099:     tmp.simplify();
marci@1097:     return tmp;
marci@1097:   }
marci@1097: 
marci@1097:   template <typename _Col, typename _Value>
marci@1097:   Expr<_Col, _Value> operator*(_Value _value, 
marci@1097: 			       const Expr<_Col, _Value>& expr) {
marci@1097:     Expr<_Col, _Value> tmp(expr);
marci@1097:     tmp*=_value;
marci@1099:     tmp.simplify();
marci@1097:     return tmp;
marci@1097:   }
marci@1097: 
marci@1097:   template <typename _Col, typename _Value>
marci@1097:   Expr<_Col, _Value> operator+(const Expr<_Col, _Value>& expr1, 
marci@1097: 			       const Expr<_Col, _Value>& expr2) {
marci@1097:     Expr<_Col, _Value> tmp(expr1);
marci@1097:     tmp+=expr2;
marci@1099:     tmp.simplify();
marci@1099:     return tmp;
marci@1099:   }
marci@1099: 
marci@1099:   template <typename _Col, typename _Value>
marci@1099:   Expr<_Col, _Value> operator-(const Expr<_Col, _Value>& expr1, 
marci@1099: 			       const Expr<_Col, _Value>& expr2) {
marci@1099:     Expr<_Col, _Value> tmp(expr1);
marci@1099:     tmp-=expr2;
marci@1099:     tmp.simplify();
marci@1097:     return tmp;
marci@1097:   }
marci@1097: 
marci@1097:   template <typename _Col, typename _Value>
marci@1097:   std::ostream& operator<<(std::ostream& os, 
marci@1097: 			   const Expr<_Col, _Value>& expr) {
marci@1097:     for (typename Expr<_Col, _Value>::Data::const_iterator i=
marci@1097: 	   expr.data.begin(); 
marci@1097: 	 i!=expr.data.end(); ++i) {
marci@1097:       os << (*i).second << "*" << (*i).first << " ";
marci@1097:     }
marci@1097:     return os;
marci@1097:   }
marci@1097:   
marci@1097: } //namespace lemon
marci@1097: 
marci@1097: #endif //LEMON_EXPRESSION_H