#ifndef MAX_EXPRESSION_HPP
#define MAX_EXPRESSION_HPP

#include <ostream>
#include "Expression.hpp"
#include "EquationSystem.hpp"
#include "IdSet.hpp"

template<typename Domain>
struct MaxStrategy {
  virtual ~MaxStrategy() { }
  virtual unsigned int get(const MaxExpression<Domain>& e) const = 0;
};

template<typename Domain>
struct ConcreteMaxStrategy : public MaxStrategy<Domain> {
  ConcreteMaxStrategy(const EquationSystem<Domain>& system)
    : _system(system),
      _strategy(system.maxExpressionCount(), 0) {
  }

  unsigned int get(const MaxExpression<Domain>& e) const {
    return _strategy[e];
  }
  unsigned int set(const MaxExpression<Domain>& e, unsigned int i) {
    _strategy[e] = i;
    return i;
  }

  void print(std::ostream& cout) const {
    for (unsigned int i = 0, length = _system.maxExpressionCount();
         i < length;
         ++i) {
      auto expr = _system.maxExpression(i);
      unsigned int subExpression = get(expr);
      cout << expr
           << " --[" << subExpression << "]-> "
           << *expr.arguments()[subExpression] << std::endl;
    }
  }

  private:
  const EquationSystem<Domain>& _system;
  IdMap<MaxExpression<Domain>,unsigned int> _strategy;
};

template<typename Domain>
std::ostream& operator<<(std::ostream& cout, const ConcreteMaxStrategy<Domain>& strat) {
  strat.print(cout);
  return cout;
}

#endif