From 3f2597b685effc8f262143ea487f5391824930e1 Mon Sep 17 00:00:00 2001
From: Reino Ruusu <reino.ruusu@semantum.fi>
Date: Wed, 6 Nov 2019 12:53:38 +0200
Subject: [PATCH] Access to mapping between object and resources via
 P3DRootNode

gitlab #48

Change-Id: I8fd2a6c0adbce3a1b6cc8e7e3055b9023de009ba
---
 .../plant3d/scenegraph/P3DRootNode.java       | 24 +++++++++++++++----
 .../plant3d/scl/P3DScriptNodeMap.java         |  8 +++----
 .../org/simantics/plant3d/scl/SCLUtil.java    |  3 +++
 3 files changed, 25 insertions(+), 10 deletions(-)

diff --git a/org.simantics.plant3d/src/org/simantics/plant3d/scenegraph/P3DRootNode.java b/org.simantics.plant3d/src/org/simantics/plant3d/scenegraph/P3DRootNode.java
index 0fb7ed4f..5b7fe274 100644
--- a/org.simantics.plant3d/src/org/simantics/plant3d/scenegraph/P3DRootNode.java
+++ b/org.simantics.plant3d/src/org/simantics/plant3d/scenegraph/P3DRootNode.java
@@ -15,17 +15,17 @@ import org.simantics.g3d.scenegraph.NodeMapProvider;
 import org.simantics.g3d.scenegraph.base.INode;
 import org.simantics.g3d.scenegraph.base.NodeException;
 import org.simantics.g3d.scenegraph.base.ParentNode;
+import org.simantics.objmap.graph.IMapping;
 import org.simantics.objmap.graph.annotations.GraphType;
 import org.simantics.objmap.graph.annotations.RelatedElementsAdd;
 import org.simantics.objmap.graph.annotations.RelatedElementsGet;
 import org.simantics.objmap.graph.annotations.RelatedElementsRem;
-import org.simantics.plant3d.editor.P3DNodeMap;
 import org.simantics.plant3d.ontology.Plant3D;
 
 import vtk.vtkProp;
 
 @GraphType(Plant3D.URIs.Plant)
-public class P3DRootNode extends ParentNode<INode> implements IG3DNode, NodeMapProvider<Resource, vtkProp , INode> {
+public class P3DRootNode extends ParentNode<INode> implements IG3DNode, NodeMapProvider<Resource, vtkProp, INode> {
 	
 	
 	@RelatedElementsAdd(Plant3D.URIs.children)
@@ -45,17 +45,31 @@ public class P3DRootNode extends ParentNode<INode> implements IG3DNode, NodeMapP
 		removeNode(Plant3D.URIs.children, node);
 	}
 	
-	private P3DNodeMap nodeMap;
+	private NodeMap<Resource, vtkProp, INode> nodeMap;
+	private IMapping<Resource, INode> mapping;
 	
-	public void setNodeMap(P3DNodeMap nodeMap) {
+	public void setNodeMap(NodeMap<Resource, vtkProp, INode> nodeMap) {
 		this.nodeMap = nodeMap;
+		this.mapping = nodeMap.getMapping();
+	}
+	
+	public void setMapping(IMapping<Resource, INode> mapping) {
+		this.mapping = mapping;
 	}
 	
 	@Override
-	public NodeMap<Resource,vtkProp, INode> getNodeMap() {
+	public NodeMap<Resource, vtkProp, INode> getNodeMap() {
 		return nodeMap;
 	}
 	
+	public Resource getNodeResource(INode node) {
+		return mapping.inverseGet(node);
+	}
+	
+	public INode getResourceNode(Resource r) {
+		return mapping.get(r);
+	}
+	
 	@Override
 	public ParentNode<?> getParent() {
 		return null;
diff --git a/org.simantics.plant3d/src/org/simantics/plant3d/scl/P3DScriptNodeMap.java b/org.simantics.plant3d/src/org/simantics/plant3d/scl/P3DScriptNodeMap.java
index 4d1fb18c..21a77a6e 100644
--- a/org.simantics.plant3d/src/org/simantics/plant3d/scl/P3DScriptNodeMap.java
+++ b/org.simantics.plant3d/src/org/simantics/plant3d/scl/P3DScriptNodeMap.java
@@ -11,7 +11,6 @@ import org.simantics.g3d.ontology.G3D;
 import org.simantics.g3d.scenegraph.base.INode;
 import org.simantics.g3d.scenegraph.base.ParentNode;
 import org.simantics.g3d.scl.ScriptNodeMap;
-import org.simantics.g3d.vtk.common.VtkView;
 import org.simantics.objmap.graph.IMapping;
 import org.simantics.plant3d.ontology.Plant3D;
 import org.simantics.plant3d.scenegraph.IP3DNode;
@@ -23,15 +22,13 @@ import org.simantics.plant3d.scenegraph.PipeRun;
 import org.simantics.plant3d.scenegraph.controlpoint.PipeControlPoint;
 import org.simantics.plant3d.scenegraph.controlpoint.PipingRules;
 
-import vtk.vtkProp3D;
-
-public class P3DScriptNodeMap extends ScriptNodeMap<Resource,INode>{
+public class P3DScriptNodeMap extends ScriptNodeMap<Resource,INode> {
 	
 	private static final boolean DEBUG = false;
 	
 	public P3DScriptNodeMap(Session session, IMapping mapping, P3DRootNode rootNode) {
 		super(session,mapping,rootNode);
-		//rootNode.setNodeMap(this);
+		rootNode.setMapping(mapping);
 	}
 	
 	@Override
@@ -94,6 +91,7 @@ public class P3DScriptNodeMap extends ScriptNodeMap<Resource,INode>{
 
 		
 		if (node instanceof ParentNode<?>) {
+			@SuppressWarnings("unchecked")
 			ParentNode<IP3DNode> p = (ParentNode<IP3DNode>)node;
 			for (IP3DNode n : p.getNodes())
 				updateTransform(n);
diff --git a/org.simantics.plant3d/src/org/simantics/plant3d/scl/SCLUtil.java b/org.simantics.plant3d/src/org/simantics/plant3d/scl/SCLUtil.java
index cdbf6626..4c4b6d57 100644
--- a/org.simantics.plant3d/src/org/simantics/plant3d/scl/SCLUtil.java
+++ b/org.simantics.plant3d/src/org/simantics/plant3d/scl/SCLUtil.java
@@ -58,6 +58,7 @@ public class SCLUtil {
 				IMappingSchema<Resource, Object> schema = schemaProvider.get(graph);
 				IMapping<Resource, Object> mapping = Mappings.createWithListening(schema);
 				P3DRootNode rootNode = (P3DRootNode) mapping.map(graph, root);
+				rootNode.setMapping((IMapping)mapping);
 				try {
 					P3DUtil.finalizeDBLoad(rootNode);
 					// FIXME: Something needs to be done here...
@@ -83,12 +84,14 @@ public class SCLUtil {
 				IMapping<Resource, Object> mapping = Mappings.createWithoutListening(schema);
 				@SuppressWarnings("unchecked")
 				Root rootNode = (Root) mapping.map(graph, root);
+				rootNode.setMapping((IMapping)mapping);
 				try {
 					P3DUtil.finalizeDBLoad(rootNode);
 					P3DUtil.finalizeDBLoad2(rootNode);
 				} catch (Exception e) {
 					throw new DatabaseException(e);
 				}
+				
 				return rootNode;
 			}
 		});
-- 
2.45.2