]> gerrit.simantics Code Review - simantics/district.git/blob - org.simantics.district.network.ui/src/org/simantics/district/network/ui/nodes/DistrictNetworkEdgeNode.java
First prototype of HSV color space based dynamic DN element coloring
[simantics/district.git] / org.simantics.district.network.ui / src / org / simantics / district / network / ui / nodes / DistrictNetworkEdgeNode.java
1 package org.simantics.district.network.ui.nodes;
2
3 import java.awt.BasicStroke;
4 import java.awt.Color;
5 import java.awt.Graphics2D;
6 import java.awt.RenderingHints;
7 import java.awt.geom.AffineTransform;
8 import java.awt.geom.Line2D;
9 import java.awt.geom.Path2D;
10 import java.awt.geom.Rectangle2D;
11
12 import org.simantics.district.network.ModelledCRS;
13 import org.simantics.district.network.ui.DistrictNetworkEdge;
14 import org.simantics.scenegraph.ISelectionPainterNode;
15 import org.simantics.scenegraph.g2d.G2DNode;
16 import org.simantics.scenegraph.utils.GeometryUtils;
17 import org.simantics.scenegraph.utils.NodeUtil;
18
19 public class DistrictNetworkEdgeNode extends G2DNode implements ISelectionPainterNode {
20
21     private static final long serialVersionUID = 8049769475036519806L;
22
23     private static final BasicStroke STROKE           = new BasicStroke(4, BasicStroke.CAP_ROUND, BasicStroke.JOIN_ROUND);
24     private static final Color       SELECTION_COLOR  = new Color(255, 0, 255, 96);
25
26     private DistrictNetworkEdge edge;
27     private Rectangle2D bounds;
28     private transient Line2D path;
29
30     private boolean scaleStroke = true;
31     private Color color;
32     private Double stroke;
33     private transient Color dynamicColor = null;
34
35     @Override
36     public void init() {
37     }
38
39     @Override
40     public void render(Graphics2D g2d) {
41         AffineTransform ot = null;
42         AffineTransform t = getTransform();
43         if (t != null && !t.isIdentity()) {
44             ot = g2d.getTransform();
45             g2d.transform(getTransform());
46         }
47
48         Object aaHint = g2d.getRenderingHint(RenderingHints.KEY_ANTIALIASING);
49         g2d.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON);
50
51         Color oldColor = g2d.getColor();
52         BasicStroke oldStroke = (BasicStroke) g2d.getStroke();
53
54         BasicStroke bs = null;
55         if (scaleStroke) {
56             double scale = GeometryUtils.getScale(g2d.getTransform());
57             scale = Math.max(10000, Math.min(scale, 50000));
58             double str = stroke != null ? Math.abs(stroke) : 1.0;
59             bs = GeometryUtils.scaleStroke(STROKE, (float) (str / scale));
60         } else {
61             bs = STROKE;
62         }
63
64         path = calculateLine(edge, path);
65
66         if (isSelected()) {
67             g2d.setColor(SELECTION_COLOR);
68             g2d.setStroke(GeometryUtils.scaleStroke(bs, 4f));
69             g2d.draw(path);
70         }
71
72         g2d.setColor(dynamicColor != null ? dynamicColor : color);
73         g2d.setStroke(bs);
74         g2d.draw(path);
75
76         // Reset
77         g2d.setStroke(oldStroke);
78         g2d.setColor(oldColor);
79         g2d.setRenderingHint(RenderingHints.KEY_ANTIALIASING, aaHint);
80
81         if (ot != null)
82             g2d.setTransform(ot);
83     }
84
85     public static Line2D calculateLine(DistrictNetworkEdge edge, Line2D result) {
86         // Convert to screen coordinates
87         double startX = ModelledCRS.longitudeToX(edge.getStartPoint().getX());
88         double startY = ModelledCRS.latitudeToY(-edge.getStartPoint().getY()); // Invert for Simantics
89         double endX = ModelledCRS.longitudeToX(edge.getEndPoint().getX());
90         double endY = ModelledCRS.latitudeToY(-edge.getEndPoint().getY());// Invert for Simantics
91
92         if (result == null)
93             result = new Line2D.Double();
94         result.setLine(startX, startY, endX, endY);
95         return result;
96     }
97
98     public static Path2D calculatePath(DistrictNetworkEdge edge, Path2D result) {
99         // Convert to screen coordinates
100         double startX = ModelledCRS.longitudeToX(edge.getStartPoint().getX());
101         double startY = ModelledCRS.latitudeToY(-edge.getStartPoint().getY()); // Invert for Simantics
102         double endX = ModelledCRS.longitudeToX(edge.getEndPoint().getX());
103         double endY = ModelledCRS.latitudeToY(-edge.getEndPoint().getY());// Invert for Simantics
104
105         if (result == null) {
106              result = new Path2D.Double();
107         } else {
108              result.reset();
109         }
110         result.moveTo(startX, startY);
111         result.lineTo(endX, endY);
112         return result;
113     }
114
115     private boolean isSelected() {
116         return NodeUtil.isSelected(this, 1);
117     }
118
119     @Override
120     public Rectangle2D getBoundsInLocal() {
121         return bounds;
122     }
123
124     private void updateBounds() {
125         Rectangle2D oldBounds = bounds;
126         if (oldBounds == null)
127             oldBounds = new Rectangle2D.Double();
128         bounds = calculateBounds(oldBounds);
129     }
130
131     private Rectangle2D calculateBounds(Rectangle2D rect) {
132         return calculatePath(edge, null).getBounds2D();
133     }
134
135     public void setDNEdge(DistrictNetworkEdge edge) {
136         this.edge = edge;
137         updateBounds();
138     }
139
140     public void setColor(Color color) {
141         this.color = color;
142     }
143
144     public Color getColor() {
145         return color;
146     }
147
148     @PropertySetter(value = "stroke")
149     public void setStroke(Double stroke) {
150         this.stroke = stroke;
151     }
152
153     @PropertySetter(value = "dynamicColor")
154     public void setDynamicColor(Color color) {
155         this.dynamicColor = color;
156     }
157
158 }