]> gerrit.simantics Code Review - simantics/district.git/blob - org.simantics.district.network.ui/src/org/simantics/district/network/ui/nodes/DistrictNetworkVertexNode.java
Make optional to calculate elevation for edge during CSV import
[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.Stroke;
8 import java.awt.geom.AffineTransform;
9 import java.awt.geom.Point2D;
10 import java.awt.geom.Rectangle2D;
11
12 import org.simantics.district.network.ModelledCRS;
13 import org.simantics.district.network.ui.adapters.DistrictNetworkVertex;
14 import org.simantics.scenegraph.ParentNode;
15 import org.simantics.scenegraph.g2d.G2DNode;
16 import org.simantics.scenegraph.utils.GeometryUtils;
17 import org.slf4j.Logger;
18 import org.slf4j.LoggerFactory;
19
20 public class DistrictNetworkVertexNode extends G2DNode {
21
22     private static final Logger LOGGER = LoggerFactory.getLogger(DistrictNetworkVertexNode.class);
23     
24     private static final long serialVersionUID = -2641639101400236719L;
25     private DistrictNetworkVertex vertex;
26
27     private static final double left = -0.25;
28     private static final double top = -0.25;
29     private static final double width = 0.5;
30     private static final double height = 0.5;
31     
32     private static final Rectangle2D NORMAL = new Rectangle2D.Double(left, top, width, height);
33     private static final Rectangle2D HOVERED = new Rectangle2D.Double(left * 3, top * 3, width * 3, height * 3);
34     
35     private Stroke stroke = new BasicStroke(2);
36     private boolean scaleStroke = true;
37     private boolean hover;
38
39     private Color color;
40
41     private Rectangle2D bounds;
42
43     private Double strokee;
44
45     @Override
46     public void init() {
47         setZIndex(2);
48     }
49     
50     @Override
51     public void render(Graphics2D g2d) {
52         AffineTransform ot = null;
53         AffineTransform t = getTransform();
54         if (t != null && !t.isIdentity()) {
55             ot = g2d.getTransform();
56             g2d.transform(getTransform());
57         }
58         
59         Object aaHint = g2d.getRenderingHint(RenderingHints.KEY_ANTIALIASING);
60         g2d.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_OFF);
61         
62         Color oldColor = g2d.getColor();
63         Stroke oldStroke = g2d.getStroke();
64         
65         g2d.setColor(color);
66         
67         double scaleRecip = 1;
68         if (scaleStroke) {
69             double scale = GeometryUtils.getScale(g2d.getTransform());
70             double str;
71             if (strokee != null)
72                 str = strokee;
73             else
74                 str = 1.0;
75             //System.out.println("scale: " + scale);
76             scaleRecip = 1.0 / scale;
77         }
78         scaleRecip = 8.0 * scaleRecip;
79         
80         // Translate lat and lon to X and Y
81         Point2D res = calculatePoint2D(vertex);
82         
83         Rectangle2D toDraw;
84         if (hover) {
85             toDraw = new Rectangle2D.Double(res.getX() - (HOVERED.getWidth() / 2 * scaleRecip), res.getY() - (HOVERED.getHeight() / 2 * scaleRecip), HOVERED.getWidth() * scaleRecip, HOVERED.getHeight() * scaleRecip);
86         } else {
87             toDraw = new Rectangle2D.Double(res.getX() - (NORMAL.getWidth() / 2 * scaleRecip), res.getY() - (NORMAL.getHeight() / 2 * scaleRecip), NORMAL.getWidth() * scaleRecip, NORMAL.getHeight() * scaleRecip);
88         }
89         // render
90         g2d.fill(toDraw);
91         
92         // Reset stats
93         g2d.setColor(oldColor);
94         g2d.setStroke(oldStroke);
95         
96         g2d.setRenderingHint(RenderingHints.KEY_ANTIALIASING, aaHint);
97         
98         if (ot != null)
99             g2d.setTransform(ot);
100     }
101
102     @Override
103     public Rectangle2D getBounds() {
104         return super.getBounds();
105     }
106     
107     @Override
108     public Rectangle2D getBoundsInLocal() {
109         return bounds;
110     }
111     
112     private void updateBounds() {
113         Rectangle2D oldBounds = bounds;
114         if (oldBounds == null)
115             oldBounds = new Rectangle2D.Double();
116         bounds = calculateBounds(oldBounds);
117     }
118
119     @Override
120     public void setTransform(AffineTransform transform) {
121         super.setTransform(transform);
122         // Update bounds
123         updateBounds();
124     }
125     
126     @Override
127     public AffineTransform getTransform() {
128         return super.getTransform();
129     }
130     
131     private Rectangle2D calculateBounds(Rectangle2D rect) {
132         Point2D calcPoint = calculatePoint2D(vertex);
133         AffineTransform at = getTransform();
134         return new Rectangle2D.Double(calcPoint.getX(), calcPoint.getY(), 1 / at.getScaleX(), 1 / at.getScaleY()).getBounds2D();
135     }
136
137     private static Point2D calculatePoint2D(DistrictNetworkVertex vertex) {
138         Point2D point= vertex.getPoint();
139         double x = ModelledCRS.longitudeToX(point.getX());
140         double y = ModelledCRS.latitudeToY(-point.getY()); // Inverse because Simantics Diagram is inverted
141
142         // Apply the scaling
143         Point2D res = new Point2D.Double(x, y);
144         return res;
145     }
146
147     public void setVertex(DistrictNetworkVertex vertex) {
148         this.vertex = vertex;
149         updateBounds();
150     }
151     
152     public boolean hover(boolean hover) {
153 //        if (hover && LOGGER.isDebugEnabled())
154 //            LOGGER.debug("Hovering " + this);
155         boolean changed = false;
156         if (this.hover != hover) {
157             this.hover = hover;
158             changed = true;
159         }
160         return changed;
161     }
162
163     public void setColor(Color color) {
164         this.color = color;
165     }
166     
167     public Color getColor() {
168         return color;
169     }
170
171     @PropertySetter(value = "stroke")
172     public void setStroke(Double stroke) {
173         this.strokee = stroke / 10;
174     }
175 }