#ifndef OPERATOR_HPP
#define OPERATOR_HPP

template<typename T>
T infinity() { }
template<>
float infinity() {
  return INFINITY;
}

#include <vector>
#include "IdSet.hpp"

template<typename T>
struct Variable;
template<typename T>
struct VariableSet : public IdSet<const Variable<T> > {};

template<typename T>
struct VariableAssignment;
template<typename T>
struct Expression;

template<typename T>
struct Operator {
  virtual ~Operator() { }
  T operator() (const std::vector< Expression<T>* >& args, const VariableAssignment<T>& rho) const {
    return (*this)(args, rho, (VariableSet<T>*) NULL);
  }
  virtual T operator() (const std::vector< Expression<T>* >&, const VariableAssignment<T>&, VariableSet<T>*) const = 0;
};
template<typename T>
struct Maximum : public Operator<T> {
  virtual T operator() (const std::vector< Expression<T>* >& args, const VariableAssignment<T>& assignment, VariableSet<T>* visited) const {
    T value = -infinity<T>();
    for (typename std::vector< Expression<T>* >::const_iterator it = args.begin();
         it != args.end();
         ++it) {
      T temporary = (**it)(assignment, visited);
      value = (temporary < value ? value : temporary);
    }
    return value;
  }
};
template<typename T>
struct Minimum : public Operator<T> {
  virtual T operator() (const std::vector< Expression<T>* >& args, const VariableAssignment<T>& assignment, VariableSet<T>* visited) const {
    T value = infinity<T>();
    for (typename std::vector< Expression<T>* >::const_iterator it = args.begin();
         it != args.end();
         ++it) {
      T temporary = (**it)(assignment, visited);
      value = (temporary < value ? temporary : value);
    }
    return value;
  }
};

template<typename T>
struct Constant : public Operator<T> {
  Constant(const T& val)
    : _value(val) { }
  T operator() (const std::vector< Expression<T>* >& args, const VariableAssignment<T>& ass, VariableSet<T>*) const {
    return _value;
  }
  private:
  const T _value;
};

template<typename T>
struct Addition: public Operator<T> {
  T operator() (const std::vector< Expression<T>* >& args, const VariableAssignment<T>& ass, VariableSet<T>* visited) const {
    T sum = (*args[0])(ass);
    for (unsigned int i = 1, size = args.size(); i < size; ++i) {
      sum += (*args[i])(ass, visited);
    }
    return sum;
  }
};

template<typename T>
struct Subtraction: public Operator<T> {
  T operator() (const std::vector< Expression<T>* >& args, const VariableAssignment<T>& ass, VariableSet<T>* visited) const {
    T sum = (*args[0])(ass);
    for (unsigned int i = 1, size = args.size(); i < size; ++i) {
      sum -= (*args[i])(ass, visited);
    }
    return sum;
  }
};

template<typename T>
struct Comma: public Operator<T> {
  T operator() (const std::vector< Expression<T>* >& args, const VariableAssignment<T>& ass, VariableSet<T>* visited) const {
    if ((*args[0])(ass, visited) == -infinity<T>()) {
      std::cout << "Comma - neg inf" << std::endl;
      return -infinity<T>();
    } else {
      std::cout << "Comma - finite" << std::endl;
      return (*args[1])(ass, visited);
    }
  }
};

template<typename T>
struct Guard: public Operator<T> {
  T operator() (const std::vector< Expression<T>* >& args, const VariableAssignment<T>& ass, VariableSet<T>* visited) const {
    if ((*args[0])(ass, visited) < (*args[1])(ass, visited)) {
      return -infinity<T>();
    } else {
      return (*args[2])(ass, visited);
    }
  }
};

#include "VariableAssignment.hpp"
#include "Expression.hpp"

#endif