X-Git-Url: https://gerrit.simantics.org/r/gitweb?a=blobdiff_plain;f=org.simantics.g3d%2Fsrc%2Forg%2Fsimantics%2Fg3d%2Fscl%2FScriptNodeMap.java;h=d532e1f0557adbf8b93099713bc56732a9e28a66;hb=refs%2Fchanges%2F20%2F3420%2F1;hp=37b124e3f49757fdf2461ac3a8b9ceb012221041;hpb=53d55c24c779745f188bdb18d32f71d20acb61b2;p=simantics%2F3d.git

diff --git a/org.simantics.g3d/src/org/simantics/g3d/scl/ScriptNodeMap.java b/org.simantics.g3d/src/org/simantics/g3d/scl/ScriptNodeMap.java
index 37b124e3..d532e1f0 100644
--- a/org.simantics.g3d/src/org/simantics/g3d/scl/ScriptNodeMap.java
+++ b/org.simantics.g3d/src/org/simantics/g3d/scl/ScriptNodeMap.java
@@ -1,8 +1,10 @@
 package org.simantics.g3d.scl;
 
+import java.util.ArrayDeque;
 import java.util.ArrayList;
 import java.util.Collection;
 import java.util.Collections;
+import java.util.Deque;
 import java.util.HashSet;
 import java.util.List;
 import java.util.Set;
@@ -40,12 +42,12 @@ import org.simantics.utils.datastructures.Pair;
  *
  * @param <E>
  */
-public abstract class ScriptNodeMap<E extends INode> implements NodeMap<Object,E>, IMappingListener, NodeListener {
+public abstract class ScriptNodeMap<DBObject,E extends INode> implements NodeMap<DBObject,Object,E>, IMappingListener, NodeListener {
 
 	private static final boolean DEBUG = false;
 	
 	protected Session session;
-	protected IMapping<Object,E> mapping;
+	protected IMapping<DBObject,E> mapping;
 	
 	protected ParentNode<E> rootNode;
 	
@@ -53,7 +55,7 @@ public abstract class ScriptNodeMap<E extends INode> implements NodeMap<Object,E
 	
 	private boolean dirty = false;
 	
-	public ScriptNodeMap(Session session, IMapping<Object,E> mapping, ParentNode<E> rootNode) {
+	public ScriptNodeMap(Session session, IMapping<DBObject,E> mapping, ParentNode<E> rootNode) {
 		this.session = session;
 		this.mapping = mapping;
 		this.rootNode = rootNode;
@@ -240,7 +242,7 @@ public abstract class ScriptNodeMap<E extends INode> implements NodeMap<Object,E
 		synchronized (syncMutex) {
 			graphUpdates = true;
 			mapping.getRangeModified().clear();
-			for (Object o : mapping.getDomain())
+			for (DBObject o : mapping.getDomain())
 				mapping.domainModified(o);
 			mapping.updateRange(graph);
 			graphModified.clear();
@@ -252,7 +254,7 @@ public abstract class ScriptNodeMap<E extends INode> implements NodeMap<Object,E
 		if (DEBUG) System.out.println("Graph update start");
 		synchronized (syncMutex) {
 			graphUpdates = true;
-			for (Object domainObject : mapping.getDomainModified()) {
+			for (DBObject domainObject : mapping.getDomainModified()) {
 				E rangeObject = mapping.get(domainObject);
 				if (rangeObject != null)
 					graphModified.add(rangeObject);
@@ -360,6 +362,27 @@ public abstract class ScriptNodeMap<E extends INode> implements NodeMap<Object,E
 		
 		
 		synchronized (syncMutex) {
+		    // Check for overlapping additions and deletions, prevent deleting objects that are also added.
+            Deque<E> stack = new ArrayDeque<E>();
+            for (Pair<E, String> n : added) {
+                stack.add(n.first);
+            }
+            while (!stack.isEmpty()) {
+                E n = stack.pop();
+                for (int i = removed.size()-1; i >= 0; i--) {
+                    if (removed.get(i).first == n) {
+                        removed.remove(i);
+                        break;
+                    }
+                }
+                if (n instanceof ParentNode) {
+                    ParentNode<INode> pn = (ParentNode<INode>)n;
+                    for (INode cn : pn.getNodes()) {
+                        stack.push((E)cn);
+                    }
+                }
+            }
+            
 			rem.addAll(removed);
 			add.addAll(added);
 			for (E e : updated.getKeys()) {
@@ -538,7 +561,7 @@ public abstract class ScriptNodeMap<E extends INode> implements NodeMap<Object,E
 		
 	}
 	
-	public IMapping<Object,E> getMapping() {
+	public IMapping<DBObject,E> getMapping() {
 		return mapping;
 	}