1 files changed, 53 insertions, 13 deletions

diff --git a/impl/MaxStrategy.hpp b/impl/MaxStrategy.hpp
index 96aeef4..96d0492 100644
--- a/impl/MaxStrategy.hpp
+++ b/impl/MaxStrategy.hpp
@@ -31,7 +31,8 @@ template<typename Domain>
 struct DynamicMaxStrategy : public MaxStrategy<Domain> {
   DynamicMaxStrategy(
     const EquationSystem<Domain>& system
-  ) : _system(system),
+  ) : _frozen(false),
+      _system(system),
       _rho(NULL),
       _values(system.maxExpressionCount(), 0),
       _stable(system.maxExpressionCount()),
@@ -46,23 +47,26 @@ struct DynamicMaxStrategy : public MaxStrategy<Domain> {
   }
 
   unsigned int get(const MaxExpression<Domain>& e) const {
-    solve(e);
+    if (!_frozen)
+      solve(e);
     return _values[e];
   }
 
   void invalidate(const Variable<Domain>& v) const {
     log::strategy << indent() << "Invalidating " << v << " - " << *_system[v] << std::endl;
-    _stable.filter(_var_influence[v]);
-
-    IdSet<MaxExpression<Domain> > infl = _var_influence[v];
-    _var_influence[v].clear();
-
-    for (typename IdSet<MaxExpression<Domain> >::iterator
-           it = infl.begin(),
-           end = infl.end();
-         it != end;
-         ++it) {
-      solve(_system.maxExpression(*it));
+    if (_system[v] && _stable.contains(*_system[v])) {
+      _stable.remove(*_system[v]);
+      _stable.filter(_var_influence[v]);
+      
+      IdSet<MaxExpression<Domain> > infl = _var_influence[v];
+      _var_influence[v].clear();
+      for (typename IdSet<MaxExpression<Domain> >::iterator
+             it = infl.begin(),
+             end = infl.end();
+           it != end;
+           ++it) {
+        solve(_system.maxExpression(*it));
+      }
     }
   }
 
@@ -78,6 +82,7 @@ private:
       stack_depth--;
 
       if (val != _values[x]) {
+
         log::strategy << x << " => " << *x.arguments()[val] << std::endl;
 
         IdSet<MaxExpression<Domain> > oldInfluence = _influence[x];
@@ -105,6 +110,16 @@ private:
     }
   }
 
+  const DynamicMaxStrategy& freeze() const {
+    _frozen = true;
+    return *this;
+  }
+
+  const DynamicMaxStrategy& thaw() const {
+    _frozen = false;
+    return *this;
+  }
+
   struct DependencyAssignment : public VariableAssignment<Domain>{
     DependencyAssignment(const DynamicMaxStrategy& strat,
                          VariableAssignment<Domain>& rho,
@@ -123,6 +138,30 @@ private:
     const MaxExpression<Domain>& _expr;
   };
 
+  struct InvalidatingAssignment : public VariableAssignment<Domain>{
+    InvalidatingAssignment(DynamicVariableAssignment<Domain>& rho)
+      : _rho(rho) {
+    }
+    ~InvalidatingAssignment() {
+      for (typename std::set<const Variable<Domain>*>::iterator
+             it = seen.begin(),
+             ei = seen.end();
+           it != ei;
+           ++it) {
+        if (!_old_stable.contains(**it))
+          _rho.invalidate(**it);
+      }
+    }
+    const Domain& operator[](const Variable<Domain>& var) const {
+      seen.insert(&var);
+      return _rho[var];
+    }
+  private:
+    DynamicVariableAssignment<Domain>& _rho;
+    IdSet<Variable<Domain> > _old_stable;
+    mutable std::set<const Variable<Domain>*> seen;
+  };
+
   struct DependencyStrategy : public MaxStrategy<Domain> {
     DependencyStrategy(const DynamicMaxStrategy& strat, const MaxExpression<Domain>& expr)
       : _strat(strat),
@@ -141,6 +180,7 @@ private:
   };
 
 private:
+  mutable bool _frozen;
   const EquationSystem<Domain>& _system;
   mutable DynamicVariableAssignment<Domain>* _rho;
   mutable IdMap<MaxExpression<Domain>,unsigned int> _values;