]> gerrit.simantics Code Review - simantics/district.git/blob - org.simantics.district.network.ui/src/org/simantics/district/network/ui/nodes/DistrictNetworkVertexNode.java
Increase baseline size of vertices 10-fold to improve visibility.
[simantics/district.git] / org.simantics.district.network.ui / src / org / simantics / district / network / ui / nodes / DistrictNetworkVertexNode.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.Point2D;
9 import java.awt.geom.Rectangle2D;
10
11 import org.simantics.district.network.ModelledCRS;
12 import org.simantics.district.network.ui.adapters.DistrictNetworkVertex;
13 import org.simantics.scenegraph.INode;
14 import org.simantics.scenegraph.ISelectionPainterNode;
15 import org.simantics.scenegraph.g2d.G2DNode;
16 import org.simantics.scenegraph.g2d.G2DParentNode;
17 import org.simantics.scenegraph.g2d.nodes.SVGNode;
18 import org.simantics.scenegraph.utils.GeometryUtils;
19 import org.simantics.scenegraph.utils.NodeUtil;
20
21 public class DistrictNetworkVertexNode extends G2DParentNode implements ISelectionPainterNode {
22
23     //private static final Logger LOGGER = LoggerFactory.getLogger(DistrictNetworkVertexNode.class);
24
25     private static final long serialVersionUID = -2641639101400236719L;
26
27     private static final BasicStroke STROKE           = new BasicStroke(4, BasicStroke.CAP_ROUND, BasicStroke.JOIN_ROUND);
28     private static final Color       SELECTION_COLOR  = new Color(255, 0, 255, 96);
29
30     private static final double left = -0.25;
31     private static final double top = -0.25;
32     private static final double width = 0.5;
33     private static final double height = 0.5;
34
35     private static final Rectangle2D NORMAL = new Rectangle2D.Double(left, top, width, height);
36     private static final Rectangle2D HOVERED = new Rectangle2D.Double(left * 3, top * 3, width * 3, height * 3);
37
38     private DistrictNetworkVertex vertex;
39
40     private boolean scaleStroke = true;
41     private boolean hover;
42
43     private Color color;
44     private transient Color dynamicColor;
45
46     private Rectangle2D bounds;
47     private transient Point2D point;
48     private transient Rectangle2D rect;
49
50     private double nodeSize = 1;
51
52     @Override
53     public void init() {
54         setZIndex(2);
55     }
56
57     @Override
58     public void render(Graphics2D g2d) {
59         if (nodeSize <= 0.0)
60             return;
61
62         AffineTransform ot = null;
63         AffineTransform t = getTransform();
64         if (t != null && !t.isIdentity()) {
65             ot = g2d.getTransform();
66             g2d.transform(getTransform());
67         }
68
69         Object oaaHint = null;
70         Object aaHint = g2d.getRenderingHint(RenderingHints.KEY_ANTIALIASING);
71         if (aaHint != RenderingHints.VALUE_ANTIALIAS_OFF) {
72             oaaHint = aaHint;
73             g2d.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_OFF);
74         }
75
76         Color oldColor = g2d.getColor();
77         Color newColor = dynamicColor != null ? dynamicColor : color;
78         boolean changeColor = !oldColor.equals(newColor);
79
80         double viewScaleRecip = 10;
81         if (scaleStroke) {
82             double scale = GeometryUtils.getScale(g2d.getTransform());
83             scale = Math.max(10000, Math.min(scale, 50000));
84             viewScaleRecip /= scale;
85         }
86         double scaleRecip = viewScaleRecip * nodeSize;
87
88         // Translate lat and lon to X and Y
89         Point2D p = point = calculatePoint2D(vertex, point);
90         Rectangle2D toDraw = calculateDrawnGeometry(p, hover ? HOVERED : NORMAL, rect, scaleRecip);
91
92         if (NodeUtil.isSelected(this, 1)) {
93             changeColor = true;
94             g2d.setColor(SELECTION_COLOR);
95             BasicStroke ss = GeometryUtils.scaleStroke(STROKE, (float) (viewScaleRecip*0.5));
96             g2d.setStroke(ss);
97             g2d.draw(toDraw);
98         }
99
100         // render
101         if (changeColor)
102             g2d.setColor(newColor);
103         g2d.fill(toDraw);
104         
105         // Reset settings
106         if (changeColor)
107             g2d.setColor(oldColor);
108         if (oaaHint != null)
109             g2d.setRenderingHint(RenderingHints.KEY_ANTIALIASING, aaHint);
110
111         // Render SVG symbol
112         for (INode nn : getNodes()) {
113             ((G2DNode)nn).setTransform(new AffineTransform(toDraw.getWidth(), 0.0, 0.0, toDraw.getHeight(), toDraw.getCenterX(), toDraw.getCenterY()));
114             ((G2DNode)nn).render(g2d);
115         }
116         
117         if (ot != null)
118             g2d.setTransform(ot);
119     }
120
121     private Rectangle2D calculateDrawnGeometry(Point2D p, Rectangle2D margin, Rectangle2D result, double scaleRecip) {
122         if (result == null)
123             result = new Rectangle2D.Double();
124         double mw = margin.getWidth();
125         double mh = margin.getHeight();
126         result.setFrame(p.getX() - (mw / 2 * scaleRecip), p.getY() - (mh / 2 * scaleRecip), mw * scaleRecip, mh * scaleRecip);
127         return result;
128     }
129
130     @Override
131     public Rectangle2D getBounds() {
132         return super.getBounds();
133     }
134
135     @Override
136     public Rectangle2D getBoundsInLocal() {
137         return bounds;
138     }
139
140     private void updateBounds() {
141         Rectangle2D oldBounds = bounds;
142         if (oldBounds == null)
143             oldBounds = new Rectangle2D.Double();
144         bounds = calculateBounds(oldBounds);
145     }
146
147     @Override
148     public void setTransform(AffineTransform transform) {
149         super.setTransform(transform);
150         // Update bounds
151         updateBounds();
152     }
153
154     @Override
155     public AffineTransform getTransform() {
156         return super.getTransform();
157     }
158
159     private Rectangle2D calculateBounds(Rectangle2D rect) {
160         Point2D calcPoint = calculatePoint2D(vertex, point);
161         AffineTransform at = getTransform();
162         return new Rectangle2D.Double(calcPoint.getX(), calcPoint.getY(), width / at.getScaleX(), height / at.getScaleY()).getBounds2D();
163     }
164
165     private static Point2D calculatePoint2D(DistrictNetworkVertex vertex, Point2D result) {
166         Point2D point = vertex.getPoint();
167         double x = ModelledCRS.longitudeToX(point.getX());
168         double y = ModelledCRS.latitudeToY(-point.getY()); // Inverse because Simantics Diagram is inverted
169         if (result == null)
170             result = new Point2D.Double(x, y);
171         else
172             result.setLocation(x, y);
173         return result;
174     }
175
176     public void setVertex(DistrictNetworkVertex vertex) {
177         this.vertex = vertex;
178         updateBounds();
179     }
180
181     public boolean hover(boolean hover) {
182 //        if (hover && LOGGER.isDebugEnabled())
183 //            LOGGER.debug("Hovering " + this);
184         boolean changed = false;
185         if (this.hover != hover) {
186             this.hover = hover;
187             changed = true;
188         }
189         return changed;
190     }
191
192     public void setColor(Color color) {
193         this.color = color;
194     }
195
196     public Color getColor() {
197         return color;
198     }
199
200     @PropertySetter(value = "SVG")
201     public void setSVG(String value) {
202         for (INode nn : this.getNodes())
203             if (nn instanceof SVGNode)
204                 ((SVGNode)nn).setData(value);
205     }
206     
207     
208     @PropertySetter(value = "size")
209     public void setSize(Double size) {
210         boolean changed = false;
211         if (size != null) {
212             changed = size != this.nodeSize;
213             this.nodeSize = size;
214         } else {
215             changed = this.nodeSize != 1.0;
216             this.nodeSize = 1.0;
217         }
218         if (changed)
219             updateBounds();
220     }
221
222     @PropertySetter(value = "dynamicColor")
223     public void setDynamicColor(Color color) {
224         this.dynamicColor = color;
225     }
226
227 }