Add connected components to visualisation from profiles

[simantics/district.git] / org.simantics.district.network.ui / src / org / simantics / district / network / ui / nodes / DistrictNetworkVertexNode.java

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 b1bb1468dd63653ee1c423a9653fbc307088bc8e..4ea8bbf1ccc5d366c7e2ca83dc5338d7d047bc2c 100644 (file)
--- 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);
+        }
+    }
 }