X-Git-Url: https://gerrit.simantics.org/r/gitweb?a=blobdiff_plain;f=org.simantics.interop.update%2Fsrc%2Forg%2Fsimantics%2Finterop%2Fupdate%2Fmodel%2FUpdateTree.java;h=aa6d7a3f65226a63344c9884bef9a481ab08a50f;hb=fa5a13d197b32fc9d3ed623ef9c16ca7fd8ceb23;hp=516ccd3f4194853238f5f4839b4e9c7aa5d8c019;hpb=2665fb66a324e6d08f28215a4734b95b001927f0;p=simantics%2Finterop.git

diff --git a/org.simantics.interop.update/src/org/simantics/interop/update/model/UpdateTree.java b/org.simantics.interop.update/src/org/simantics/interop/update/model/UpdateTree.java
index 516ccd3..aa6d7a3 100644
--- a/org.simantics.interop.update/src/org/simantics/interop/update/model/UpdateTree.java
+++ b/org.simantics.interop.update/src/org/simantics/interop/update/model/UpdateTree.java
@@ -7,10 +7,9 @@ import org.simantics.db.ReadGraph;
 import org.simantics.db.Resource;
 import org.simantics.db.exception.DatabaseException;
 import org.simantics.interop.test.GraphChanges;
-import org.simantics.interop.update.model.UpdateNode.Status;
 
 
-public abstract class UpdateTree {
+public class UpdateTree {
 	
 	private UpdateNode rootNode;
 	private Map<Resource,UpdateNode> nodes;
@@ -21,7 +20,7 @@ public abstract class UpdateTree {
 	public UpdateTree(ReadGraph g, GraphChanges changes, UpdateOperations updateOps) throws DatabaseException {
 		this.changes = changes;
 		this.nodes = new HashMap<Resource, UpdateNode>();
-		this.rootNode = createNode(Status.EXIST, changes.getResource1());
+		this.rootNode = createNode(g, UpdateStatus.EXIST, changes.getResource1());
 		nodes.put(changes.getResource1(), rootNode);
 		nodes.put(changes.getResource2(), rootNode);
 		this.updateOps = updateOps;
@@ -37,53 +36,64 @@ public abstract class UpdateTree {
 		return rootNode;
 	}
 	
-	protected abstract UpdateNode createNode(Status staus, Resource r);
-	protected abstract UpdateNode createNode(Status staus, UpdateOp op);
+	public GraphChanges getChanges() {
+		return changes;
+	}
+	
+	protected UpdateNode createNode(ReadGraph g, UpdateStatus status, Resource r) throws DatabaseException {
+		return new UpdateNode(g,status, r);
+	}
+	
+	protected UpdateNode createNode(ReadGraph g, UpdateStatus status, UpdateOp op) throws DatabaseException{
+		return new UpdateNode(g,status, op);
+	}
 	
-	private UpdateNode createNode(Resource r1, Resource r2) {
+	private UpdateNode createNode(ReadGraph g, Resource r1, Resource r2) throws DatabaseException {
 		UpdateNode node = null;
 		if (r1 != null && r2 != null) {
-			node =  createNode(Status.EXIST, r1);
+		    UpdateOp op = updateOps.getUpdateOp(r1);
+		    if (op == null)
+		        op = updateOps.getUpdateOp(r2);
+            if (op == null)		        
+                node = createNode(g, UpdateStatus.EXIST, r1);
+            else
+                node = createNode(g, UpdateStatus.EXIST, op);
 			nodes.put(r1, node);
 			nodes.put(r2, node);
 		} else if (r1 != null) {
-			node = createNode(Status.DELETED ,updateOps.getUpdateOp(r1));
+			node = createNode(g,UpdateStatus.DELETED ,updateOps.getUpdateOp(r1));
 			nodes.put(r1, node);
 		} else if (r2 != null) {
-			node = createNode(Status.NEW, updateOps.getUpdateOp(r2));
+			node = createNode(g,UpdateStatus.NEW, updateOps.getUpdateOp(r2));
 			nodes.put(r2, node);
 		}
 		return node;
 	}
 	
 	public UpdateNode addNode(ReadGraph g, Resource r1, Resource r2) throws DatabaseException {
-		if (r1 != null && r2 != null) {
-			return null;
-		}
 		if (nodes.containsKey(r1))
 			return nodes.get(r1);
 		if (nodes.containsKey(r2))
 			return nodes.get(r2);
 		
-		UpdateNode node = createNode(r1, r2);
+		UpdateNode node = createNode(g, r1, r2);
 		connectParent(g,node);
 		return node;
-		
 	}
 	
-	private boolean connectParent(ReadGraph g, UpdateNode node) throws DatabaseException {
+	public UpdateNode getNode(Resource r) {
+		return nodes.get(r);
+	}
+	
+	protected boolean connectParent(ReadGraph g, UpdateNode node) throws DatabaseException {
 		UpdateNode parent = null;
 		while (true) {
 			Resource parentResource = node.getParentResource(g);
 			parent = nodes.get(parentResource);
 			if (parent == null) {
-				if (changes.getComparable().containsLeft(parentResource)) {
-					parent = createNode(parentResource, changes.getComparable().getRight(parentResource));
-				} else if (changes.getComparable().containsRight(parentResource)) {
-					parent = createNode(changes.getComparable().getLeft(parentResource) ,parentResource);
-				} else {
+				parent = getOrCreate(g, parentResource);
+				if (parent == null)
 					return false;
-				}
 				//parent.setStatus(Status.CONTAINS);
 				parent.addChild(node);
 				node = parent;
@@ -96,7 +106,26 @@ public abstract class UpdateTree {
 		}
 	}
 	
-	protected abstract boolean handleCustom(ReadGraph g, UpdateOp op) throws DatabaseException;
+	protected UpdateNode getOrCreate(ReadGraph g, Resource parentResource) throws DatabaseException {
+		UpdateNode parent = nodes.get(parentResource);
+		if (parent == null) {
+			if (changes.getComparable().containsLeft(parentResource)) {
+				parent = createNode(g, parentResource, changes.getComparable().getRight(parentResource));
+			} else if (changes.getComparable().containsRight(parentResource)) {
+				parent = createNode(g, changes.getComparable().getLeft(parentResource) ,parentResource);
+			} else {
+				return null;
+			}
+			//parent.setStatus(Status.CONTAINS
+		} 
+		return parent;
+	}
+	
+	
+	
+	protected boolean handleCustom(ReadGraph g, UpdateOp op) throws DatabaseException {
+		return false;
+	}
 	
 	private void populate(ReadGraph g) throws DatabaseException{
 
@@ -104,8 +133,12 @@ public abstract class UpdateTree {
 			if (!handleCustom(g, op)) {
 				if (op.isAdd()) {
 					addNode(g, null,op.getResource());
-				} else {
+				} else if (op.isDelete()){
 					addNode(g, op.getResource(), null);
+				} else if (op.isChange()) {
+				    Resource l = getChanges().getComparable().containsLeft(op.getResource()) ? op.getResource() :getChanges().getComparable().getLeft(op.getResource());
+				    Resource r = getChanges().getComparable().containsRight(op.getResource()) ? op.getResource() :getChanges().getComparable().getRight(op.getResource());
+				    addNode(g, l, r);
 				}
 			}
 		}