gerrit.simantics Code Review - simantics/district.git/blob

   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 }