Access to mapping between object and resources via P3DRootNode

[simantics/3d.git] / org.simantics.plant3d / src / org / simantics / plant3d / scl / SCLUtil.java

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 cdbf6626b197548c624f65e57b6f2d7b58cbdff1..4c4b6d5760a6f40b68533dba92a67b741585fc27 100644 (file)
--- 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;
                        }
                });