X-Git-Url: https://gerrit.simantics.org/r/gitweb?a=blobdiff_plain;f=org.simantics.district.network.ui%2Fsrc%2Forg%2Fsimantics%2Fdistrict%2Fnetwork%2Fui%2Fnodes%2FDistrictNetworkVertexNode.java;h=4ea8bbf1ccc5d366c7e2ca83dc5338d7d047bc2c;hb=05fb97a04278adf774281f05135eb6fc369dc3b5;hp=b1bb1468dd63653ee1c423a9653fbc307088bc8e;hpb=ff5441dd6a85ef6fb92c6b3b0cf709bbce878e5e;p=simantics%2Fdistrict.git

diff --git a/org.simantics.district.network.ui/src/org/simantics/district/network/ui/nodes/DistrictNetworkVertexNode.java b/org.simantics.district.network.ui/src/org/simantics/district/network/ui/nodes/DistrictNetworkVertexNode.java
index b1bb1468..4ea8bbf1 100644
--- a/org.simantics.district.network.ui/src/org/simantics/district/network/ui/nodes/DistrictNetworkVertexNode.java
+++ b/org.simantics.district.network.ui/src/org/simantics/district/network/ui/nodes/DistrictNetworkVertexNode.java
@@ -7,6 +7,7 @@ import java.awt.RenderingHints;
 import java.awt.geom.AffineTransform;
 import java.awt.geom.Point2D;
 import java.awt.geom.Rectangle2D;
+import java.util.List;
 import java.util.Optional;
 
 import org.simantics.district.network.ui.adapters.DistrictNetworkVertex;
@@ -257,7 +258,7 @@ public class DistrictNetworkVertexNode extends G2DParentNode implements ISelecti
             child.setInfo(null);
         }
     }
-    
+
     public void setInSimulation(Optional<Boolean> isInSimulation) {
         if (!isInSimulation.isPresent()) {
             removeNode(NotInSimulationNode.NODE_NAME);
@@ -267,4 +268,15 @@ public class DistrictNetworkVertexNode extends G2DParentNode implements ISelecti
             child.setIsInSimulation(isInSimulation.get());
         }
     }
+
+    public void setConnectionLinePoints(List<Point2D> points) {
+        if (points == null) {
+            removeNode(ConnectionLineNode.NODE_NAME);
+        } else {
+            ConnectionLineNode child = getOrCreateNode(ConnectionLineNode.NODE_NAME, ConnectionLineNode.class);
+            child.setZIndex(0);
+            child.setStrokeWidth(2.f);
+            child.setPoints(points);
+        }
+    }
 }