Add connected components to visualisation from profiles

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

diff --git a/org.simantics.district.network.ui/src/org/simantics/district/network/ui/nodes/ConnectionLineNode.java b/org.simantics.district.network.ui/src/org/simantics/district/network/ui/nodes/ConnectionLineNode.java
new file mode 100644 (file)
index 0000000..12be4f7
--- /dev/null
+++ b/org.simantics.district.network.ui/src/org/simantics/district/network/ui/nodes/ConnectionLineNode.java
@@ -0,0 +1,75 @@
+package org.simantics.district.network.ui.nodes;
+
+import java.awt.BasicStroke;
+import java.awt.Color;
+import java.awt.Graphics2D;
+import java.awt.geom.Line2D;
+import java.awt.geom.Point2D;
+import java.awt.geom.Rectangle2D;
+import java.util.List;
+
+import org.simantics.g2d.utils.GeometryUtils;
+import org.simantics.scenegraph.g2d.G2DNode;
+
+public class ConnectionLineNode extends G2DNode {
+    
+    public static final String NODE_NAME = "districtNetworkConnection";
+    
+       private static final BasicStroke STROKE = new BasicStroke(1.f, BasicStroke.CAP_BUTT, BasicStroke.JOIN_MITER, 1.f, new float[] {4.f, 2.f}, 0.f);
+       private static final Color[] colors = { Color.RED, Color.GREEN, Color.BLUE, Color.ORANGE, Color.CYAN, Color.PINK };
+       
+       private float strokeWidth;
+       private Line2D[] lines;
+
+       public ConnectionLineNode() {
+               super();
+       }
+       
+       private static final long serialVersionUID = 1L;
+
+       @Override
+       public Rectangle2D getBoundsInLocal() {
+               return null;
+       }
+
+       @Override
+       public Rectangle2D getBoundsInLocal(boolean b) {
+               return null;
+       }
+
+       @Override
+       public Rectangle2D getBounds() {
+               return null;
+       }
+       
+       public void setStrokeWidth(float w) {
+               strokeWidth = w;
+       }
+       
+       public void setPoints(List<Point2D> result) {
+               Point2D p0 = DistrictNetworkNodeUtils.calculatePoint2D(result.get(0), null);
+               lines = new Line2D[result.size() - 1];
+               for (int i = 1; i < result.size(); i++) 
+               {
+                       Point2D p = result.get(i);
+                       lines[i-1] = p != null ? new Line2D.Double(p0, DistrictNetworkNodeUtils.calculatePoint2D(p, null)) : null;
+               }
+       }
+       
+       @Override
+       public void render(Graphics2D g2d) {
+               if (lines == null || lines.length == 0)
+                       return;
+               
+               // Keep fixed line width on screen
+               float scaleRecip = (float) GeometryUtils.getScale(g2d.getTransform());
+               g2d.setStroke(GeometryUtils.scaleStroke(STROKE, strokeWidth / scaleRecip));
+               
+               for (int i = 0; i < lines.length; i++) {
+                       if (lines[i] != null) {
+                               g2d.setColor(colors[i % colors.length]);
+                               g2d.draw(lines[i]);
+                       }
+               }
+       }
+}
\ No newline at end of file