Fixed multiple issues causing dangling references to discarded queries

[simantics/platform.git] / bundles / org.simantics.diagram / src / org / simantics / diagram / adapter / DefinedElementHandler.java

diff --git a/bundles/org.simantics.diagram/src/org/simantics/diagram/adapter/DefinedElementHandler.java b/bundles/org.simantics.diagram/src/org/simantics/diagram/adapter/DefinedElementHandler.java
index 17f22cfcfae23d131531913ed526d9b1e5b84061..e3e1269f37e0b9e29feb577a1d82160da88e1dd4 100644 (file)
--- a/bundles/org.simantics.diagram/src/org/simantics/diagram/adapter/DefinedElementHandler.java
+++ b/bundles/org.simantics.diagram/src/org/simantics/diagram/adapter/DefinedElementHandler.java
@@ -18,6 +18,7 @@ import org.simantics.g2d.image.Image;
 import org.simantics.scenegraph.Node;
 import org.simantics.scenegraph.g2d.G2DParentNode;
 import org.simantics.scenegraph.g2d.IG2DNode;
+import org.simantics.scenegraph.g2d.nodes.SingleElementNode;
 import org.simantics.utils.datastructures.hints.IHintContext.Key;
 
 public class DefinedElementHandler implements SceneGraph, InternalSize, Resize, Outline {
@@ -52,6 +53,9 @@ public class DefinedElementHandler implements SceneGraph, InternalSize, Resize,
                        n.setTransform(at); // FIXME: not tested..
                }
         }
+        if(parent instanceof SingleElementNode) {
+            ((SingleElementNode)parent).setParameters(ElementUtils.getParameters(e));
+        }
     }
 
     public void cleanup(final IElement e) {