Added symbol rendering for edges.

[simantics/district.git] / org.simantics.district.network.ui / src / org / simantics / district / network / ui / adapters / DistrictNetworkEdgeElement.java

diff --git a/org.simantics.district.network.ui/src/org/simantics/district/network/ui/adapters/DistrictNetworkEdgeElement.java b/org.simantics.district.network.ui/src/org/simantics/district/network/ui/adapters/DistrictNetworkEdgeElement.java
index 9595f232debb6339459ff11203b5e2f56139bed4..d47580074a9d5a3ef23855bc78745d44be726d4d 100644 (file)
--- a/org.simantics.district.network.ui/src/org/simantics/district/network/ui/adapters/DistrictNetworkEdgeElement.java
+++ b/org.simantics.district.network.ui/src/org/simantics/district/network/ui/adapters/DistrictNetworkEdgeElement.java
@@ -28,6 +28,7 @@ import org.simantics.g2d.element.handler.impl.DefaultTransform;
 import org.simantics.g2d.element.handler.impl.SimpleElementLayers;
 import org.simantics.maps.MapScalingTransform;
 import org.simantics.scenegraph.g2d.G2DParentNode;
+import org.simantics.scenegraph.g2d.nodes.SVGNode;
 import org.simantics.utils.datastructures.hints.IHintContext.Key;
 import org.simantics.utils.datastructures.hints.IHintContext.KeyOf;
 import org.slf4j.Logger;
@@ -37,6 +38,7 @@ public class DistrictNetworkEdgeElement {
 
     public static final Key KEY_DN_EDGE = new KeyOf(DistrictNetworkEdge.class, "DN_EDGE");
     public static final Key KEY_DN_EDGE_NODE = new SceneGraphNodeKey(DistrictNetworkEdgeNode.class, "DN_EDGE_NODE");
+    public static final Key KEY_DN_EDGE_SYMBOL_NODE = new SceneGraphNodeKey(DistrictNetworkEdgeNode.class, "DN_EDGE_SYMBOL_NODE");
 
     public static final ElementClass CLASS =
             ElementClass.compile(
@@ -66,6 +68,9 @@ public class DistrictNetworkEdgeElement {
                 if (node == null) {
                     node = parent.addNode(ElementUtils.generateNodeId(edgeElement), DistrictNetworkEdgeNode.class);
                     edgeElement.setHint(KEY_DN_EDGE_NODE, node);
+                    
+                    SVGNode symbol = node.addNode(ElementUtils.generateNodeId(edgeElement), SVGNode.class);
+                    edgeElement.setHint(KEY_DN_EDGE_SYMBOL_NODE, symbol);
                 }
                 
                 node.setColor(ElementUtils.getAdditionalColor(edgeElement, Color.BLUE));
@@ -79,8 +84,10 @@ public class DistrictNetworkEdgeElement {
 
         @Override
         public void cleanup(IElement edge) {
+            ElementUtils.removePossibleNode(edge, KEY_DN_EDGE_SYMBOL_NODE);
             ElementUtils.removePossibleNode(edge, KEY_DN_EDGE_NODE);
             edge.removeHint(KEY_DN_EDGE_NODE);
+            edge.removeHint(KEY_DN_EDGE_SYMBOL_NODE);
         }
     }