From 5f3cb0bf02ed32976370cf6c7f656d000b4d7694 Mon Sep 17 00:00:00 2001
From: Carlo Zancanaro <carlo@pc-4w14-0.cs.usyd.edu.au>
Date: Fri, 15 Jun 2012 11:26:32 +1000
Subject: Re-write heaps of code to work better.

---
 impl/VariableAssignment.hpp | 30 +++++++++++++++++++++---------
 1 file changed, 21 insertions(+), 9 deletions(-)

(limited to 'impl/VariableAssignment.hpp')

diff --git a/impl/VariableAssignment.hpp b/impl/VariableAssignment.hpp
index 9d8137a..e0f7dc8 100644
--- a/impl/VariableAssignment.hpp
+++ b/impl/VariableAssignment.hpp
@@ -1,17 +1,29 @@
 #ifndef VARIABLE_ASSIGNMENT_HPP
 #define VARIABLE_ASSIGNMENT_HPP
 
-#include "Variable.hpp"
+#include "Expression.hpp"
 #include "IdMap.hpp"
 
-template<typename T>
-struct VariableAssignment : public IdMap<Variable<T>,T> {
-  VariableAssignment(unsigned int length)
-    : IdMap<Variable<T>,T>(length, -infinity<T>()) { }
-  VariableAssignment(unsigned int length, const T& initial)
-    : IdMap<Variable<T>,T>(length, initial) { }
-  VariableAssignment(const VariableAssignment& other)
-    : IdMap<Variable<T>,T>(other) { }
+template<typename Domain>
+struct VariableAssignment {
+  virtual ~VariableAssignment() { }
+  virtual const Domain& operator[](const Variable<Domain>&) const = 0;
+};
+
+template<typename Domain>
+struct StableVariableAssignment
+: public VariableAssignment<Domain>, public IdMap<Variable<Domain>, Domain> {
+  StableVariableAssignment(unsigned int length)
+    : IdMap<Variable<Domain>,Domain>(length, -infinity<Domain>()) { }
+  StableVariableAssignment(unsigned int length, const Domain& value)
+    : IdMap<Variable<Domain>,Domain>(length, value) { }
+
+  const Domain& operator[](const Variable<Domain>& var) const {
+    return IdMap<Variable<Domain>,Domain>::operator[](var);
+  }
+  Domain& operator[](const Variable<Domain>& var) {
+    return IdMap<Variable<Domain>,Domain>::operator[](var);
+  }
 };
 
 #endif
-- 
cgit v1.2.3