1 files changed, 26 insertions, 0 deletions
diff --git a/impl/Expression.hpp b/impl/Expression.hpp
index 01921b6..63f732d 100644
--- a/impl/Expression.hpp
+++ b/impl/Expression.hpp
@@ -14,6 +14,12 @@ template<typename Domain>
 struct MaxStrategy;
 
 template<typename Domain>
+struct Variable;
+
+template<typename Domain>
+struct MaxExpression;
+
+template<typename Domain>
 struct Expression {
   virtual ~Expression() { }
 
@@ -23,6 +29,9 @@ struct Expression {
     return eval(rho);
   }
 
+  virtual void mapTo(Variable<Domain>&, IdMap<MaxExpression<Domain>, Variable<Domain>*>&) const {
+  }
+
   virtual void print(std::ostream&) const = 0;
 };
 
@@ -101,6 +110,14 @@ struct OperatorExpression : public Expression<Domain> {
     return _operator;
   }
 
+  virtual void mapTo(Variable<Domain>& v, IdMap<MaxExpression<Domain>, Variable<Domain>*>& m) const {
+    for (unsigned int i = 0, length = _arguments.size();
+         i < length;
+         ++i) {
+      _arguments[i]->mapTo(v, m);
+    }
+  }
+
   void print(std::ostream& cout) const {
     cout << _operator << "(";
     for (unsigned int i = 0, length = _arguments.size();
@@ -144,6 +161,15 @@ struct MaxExpression : public OperatorExpression<Domain> {
     return bestIndex;
   }
 
+  virtual void mapTo(Variable<Domain>& v, IdMap<MaxExpression<Domain>, Variable<Domain>*>& m) const {
+    m[*this] = &v;
+    for (unsigned int i = 0, length = this->_arguments.size();
+         i < length;
+         ++i) {
+      this->_arguments[i]->mapTo(v, m);
+    }
+  }
+
   unsigned int id() const {
     return _id;
   }