1 files changed, 28 insertions, 3 deletions

diff --git a/impl/VariableAssignment.hpp b/impl/VariableAssignment.hpp
index e575d60..fff00b5 100644
--- a/impl/VariableAssignment.hpp
+++ b/impl/VariableAssignment.hpp
@@ -24,9 +24,11 @@ struct DynamicVariableAssignment : public VariableAssignment<Domain> {
   ) : _system(system),
       _strategy(strat),
       _values(system.variableCount(), value),
+      _old_values(system.variableCount(), value),
       _unstable(system.variableCount()),
       _influence(system.variableCount(),
-                 IdSet<Variable<Domain> >(system.variableCount()))
+                 IdSet<Variable<Domain> >(system.variableCount())),
+      _touched(system.variableCount())
   { }
 
   const Domain operator[](const Variable<Domain>& var) {
@@ -39,6 +41,8 @@ struct DynamicVariableAssignment : public VariableAssignment<Domain> {
       log::fixpoint << indent() << "Invalidating " << x << std::endl;
  
       _unstable.insert(x);
+      _old_values[x] = _values[x];
+      _touched.insert(x);
       _values[x] = infinity<Domain>();
 
       IdSet<Variable<Domain> > infl = _influence[x];
@@ -54,6 +58,27 @@ struct DynamicVariableAssignment : public VariableAssignment<Domain> {
     }
   }
 
+  IdSet<Variable<Domain> > get_changed() {
+    IdSet<Variable<Domain> > changed;
+    for (typename IdSet<Variable<Domain> >::iterator
+           it = _touched.begin(),
+           ei = _touched.end();
+         it != ei;
+         ++it) {
+      Variable<Domain>& var = _system.variable(*it);
+      //if (_old_values[var] != _values[var]) {
+        changed.insert(var);
+        _touched.remove(var);
+      //}
+    }
+    //_touched.clear();
+    return changed;
+  }
+
+  const IdMap<Variable<Domain>, Domain> get_old_values() const {
+    return _old_values;
+  }
+
 private:
 
   void solve(const Variable<Domain>& x) {
@@ -72,8 +97,6 @@ private:
       if (val != _values[x]) {
         log::fixpoint << x << " = " << val << std::endl;
  
-        _strategy.invalidate(x);
-
         IdSet<Variable<Domain> > oldInfluence = _influence[x];
         _influence[x].clear();
         _values[x] = val;
@@ -111,8 +134,10 @@ private:
   const EquationSystem<Domain>& _system;
   DynamicMaxStrategy<Domain>& _strategy;
   IdMap<Variable<Domain>, Domain> _values;
+  IdMap<Variable<Domain>, Domain> _old_values;
   IdSet<Variable<Domain> > _unstable;
   IdMap<Variable<Domain>,IdSet<Variable<Domain> > > _influence;
+  IdSet<Variable<Domain> > _touched;
 };
 
 #endif