Enhancements to district functionalities and code

[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 12f26182dca15fe03cd4cf02ff6588a2a3954f3f..a7cd09ac63a522d3dfb6d740f9d67652fa527248 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
@@ -1,5 +1,7 @@
 package org.simantics.district.network.ui.adapters;
 
+import java.awt.Color;
+import java.awt.geom.AffineTransform;
 import java.awt.geom.Rectangle2D;
 import java.util.Collection;
 import java.util.Collections;
@@ -33,7 +35,8 @@ public class DistrictNetworkEdgeElement {
                     DNEdgeInternalSize.INSTANCE,
                     DNEdgeSceneGraph.INSTANCE,
                     DNEdgeConnectionHandler.INSTANCE,
-                    SimpleElementLayers.INSTANCE
+                    SimpleElementLayers.INSTANCE,
+                    DistrictNetworkAdditionalColor.INSTANCE
             ).setId(DistrictNetworkEdgeElement.class.getSimpleName());
     
     static final class DNEdgeSceneGraph implements SceneGraph {
@@ -53,6 +56,9 @@ public class DistrictNetworkEdgeElement {
                     node = parent.addNode(ElementUtils.generateNodeId(edgeElement), DistrictNetworkEdgeNode.class);
                     edgeElement.setHint(KEY_DN_EDGE_NODE, node);
                 }
+                
+                node.setColor(ElementUtils.getAdditionalColor(edgeElement, Color.BLUE));
+                
                 node.setDNEdge(edge);
             }
         }
@@ -84,7 +90,6 @@ public class DistrictNetworkEdgeElement {
 
             return size;
         }
-        
     }
     
     static class DNEdgeConnectionHandler implements ConnectionHandler {