+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