1 files changed, 42 insertions, 16 deletions

diff --git a/Solvers/VIATRA-Solver/org.eclipse.viatra.solver.data/src/org/eclipse/viatra/solver/data/map/VersionedMapStoreImpl.java b/Solvers/VIATRA-Solver/org.eclipse.viatra.solver.data/src/org/eclipse/viatra/solver/data/map/VersionedMapStoreImpl.java
index 0c00108b..b6a615ed 100644
--- a/Solvers/VIATRA-Solver/org.eclipse.viatra.solver.data/src/org/eclipse/viatra/solver/data/map/VersionedMapStoreImpl.java
+++ b/Solvers/VIATRA-Solver/org.eclipse.viatra.solver.data/src/org/eclipse/viatra/solver/data/map/VersionedMapStoreImpl.java
@@ -5,6 +5,7 @@ import java.util.Map;
 import java.util.Set;
 
 import org.eclipse.viatra.solver.data.map.internal.ImmutableNode;
+import org.eclipse.viatra.solver.data.map.internal.MapDiffCursor;
 import org.eclipse.viatra.solver.data.map.internal.Node;
 import org.eclipse.viatra.solver.data.map.internal.VersionedMapImpl;
 
@@ -25,7 +26,7 @@ public class VersionedMapStoreImpl<KEY,VALUE> implements VersionedMapStore<KEY,
                 nodeCache = new HashMap<>();
         }
         
-        Set<Long> getStates() {
+        synchronized Set<Long> getStates() {
                 return states.keySet();
         }
         
@@ -34,29 +35,54 @@ public class VersionedMapStoreImpl<KEY,VALUE> implements VersionedMapStore<KEY,
                 return new VersionedMapImpl<KEY,VALUE>(this,hashProvider,defaultValue);
         }
         @Override
-        public VersionedMap<KEY,VALUE> createMap(long state) throws IllegalAccessException {
-                ImmutableNode<KEY, VALUE> data = getStateData(state);
+        public VersionedMap<KEY,VALUE> createMap(long state) {
+                ImmutableNode<KEY, VALUE> data = revert(state);
                 if(data != null) {
                         return new VersionedMapImpl<KEY,VALUE>(this,hashProvider,defaultValue,data); 
                 } else {
-                        throw new IllegalAccessException("Store does not contain state " + state + "!");
+                        throw new IllegalArgumentException("Store does not contain state " + state + "!");
                 }
         }
-
-        private long getNextID() {
-                if(nextID == Long.MAX_VALUE) throw new IllegalStateException("Map store run out of Id-s");
-                return nextID++;
-        }
         
-        public ImmutableNode<KEY,VALUE> getStateData(long state) {
+        synchronized public ImmutableNode<KEY,VALUE> revert(long state) {
                 return states.get(state);
         }
-        public Map<Node<KEY,VALUE>, ImmutableNode<KEY,VALUE>> getStore() {
-                return this.nodeCache;
-        }
-        public long addState(ImmutableNode<KEY,VALUE> data) {
-                long id = getNextID();
-                states.put(id,data);
+        synchronized public long commit(Node<KEY,VALUE> data, VersionedMapImpl<KEY, VALUE> mapToUpdateRoot) {
+                ImmutableNode<KEY,VALUE> immutable;
+                if(data != null) {
+                        if(this.nodeCache != null) {
+                                immutable = data.toImmutable(this.nodeCache);
+                        } else {
+                                immutable = data.toImmutable();
+                        }
+                } else {
+                        immutable = null;
+                }
+                
+                
+                if(nextID == Long.MAX_VALUE) throw new IllegalStateException(
+                        "Map store run out of Id-s");
+                long id = nextID++;
+                this.states.put(id, immutable);
+                //mapToUpdateRoot.setRoot(immutable);
                 return id;
         }
+        
+//      public Map<Node<KEY,VALUE>, ImmutableNode<KEY,VALUE>> getStore() {
+//              return this.nodeCache;
+//      }
+//      public long addState(ImmutableNode<KEY,VALUE> data) {
+//              
+//              states.put(id,data);
+//              return id;
+//      }
+
+        @Override
+        public DiffCursor<KEY, VALUE> getDiffCursor(long fromState, long toState) {
+                VersionedMap<KEY, VALUE> map1 = createMap(fromState);
+                VersionedMap<KEY, VALUE> map2 = createMap(toState);
+                Cursor<KEY, VALUE> cursor1 = map1.getCursor();
+                Cursor<KEY, VALUE> cursor2 = map2.getCursor();
+                return new MapDiffCursor<KEY, VALUE>(this.defaultValue,cursor1,cursor2);
+        }
 }