X-Git-Url: https://gerrit.simantics.org/r/gitweb?a=blobdiff_plain;ds=inline;f=org.simantics.district.network.ui%2Fsrc%2Forg%2Fsimantics%2Fdistrict%2Fnetwork%2Fui%2Fnodes%2FDistrictNetworkVertexNode.java;h=58e47f4bcf285dff9f5585aafc7d6a8e5f0eb49e;hb=00803fe68b14b5dce310a5a88b9c2a55bd1c85e6;hp=7365060b0254bc8c6f62d5c4cb4aeef8e1dcd6d5;hpb=8034693a5c86d690649a70a62709d29c3fa197aa;p=simantics%2Fdistrict.git diff --git a/org.simantics.district.network.ui/src/org/simantics/district/network/ui/nodes/DistrictNetworkVertexNode.java b/org.simantics.district.network.ui/src/org/simantics/district/network/ui/nodes/DistrictNetworkVertexNode.java index 7365060b..58e47f4b 100644 --- a/org.simantics.district.network.ui/src/org/simantics/district/network/ui/nodes/DistrictNetworkVertexNode.java +++ b/org.simantics.district.network.ui/src/org/simantics/district/network/ui/nodes/DistrictNetworkVertexNode.java @@ -1,54 +1,198 @@ -package org.simantics.district.network.ui.nodes; - -import java.awt.Color; -import java.awt.Graphics2D; -import java.awt.RenderingHints; -import java.awt.geom.AffineTransform; -import java.awt.geom.Rectangle2D; - -import org.simantics.district.network.ui.adapters.DistrictNetworkVertex; -import org.simantics.scenegraph.g2d.G2DNode; - -public class DistrictNetworkVertexNode extends G2DNode { - - private static final long serialVersionUID = -2641639101400236719L; - private DistrictNetworkVertex vertex; - - @Override - public void render(Graphics2D g2d) { - AffineTransform ot = null; - AffineTransform t = getTransform(); - if (t != null && !t.isIdentity()) { - ot = g2d.getTransform(); - g2d.transform(getTransform()); - } - - Object aaHint = g2d.getRenderingHint(RenderingHints.KEY_ANTIALIASING); - g2d.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_OFF); - - Color oldColor = g2d.getColor(); - g2d.setColor(Color.RED); - // render - - Rectangle2D.Double rect = new Rectangle2D.Double(-0.5, -0.5, 1, 1); - g2d.draw(rect); - - // Reset stats - g2d.setColor(oldColor); - - g2d.setRenderingHint(RenderingHints.KEY_ANTIALIASING, aaHint); - - if (ot != null) - g2d.setTransform(ot); - } - - @Override - public Rectangle2D getBoundsInLocal() { - return new Rectangle2D.Double(-0.5, -0.5, 1, 1); - } - - public void setVertex(DistrictNetworkVertex vertex) { - this.vertex = vertex; - } - -} +package org.simantics.district.network.ui.nodes; + +import java.awt.BasicStroke; +import java.awt.Color; +import java.awt.Graphics2D; +import java.awt.RenderingHints; +import java.awt.geom.AffineTransform; +import java.awt.geom.Point2D; +import java.awt.geom.Rectangle2D; + +import org.simantics.district.network.ModelledCRS; +import org.simantics.district.network.ui.adapters.DistrictNetworkVertex; +import org.simantics.scenegraph.ISelectionPainterNode; +import org.simantics.scenegraph.g2d.G2DNode; +import org.simantics.scenegraph.utils.GeometryUtils; +import org.simantics.scenegraph.utils.NodeUtil; + +public class DistrictNetworkVertexNode extends G2DNode implements ISelectionPainterNode { + + //private static final Logger LOGGER = LoggerFactory.getLogger(DistrictNetworkVertexNode.class); + + private static final long serialVersionUID = -2641639101400236719L; + + private static final BasicStroke STROKE = new BasicStroke(4, BasicStroke.CAP_ROUND, BasicStroke.JOIN_ROUND); + private static final Color SELECTION_COLOR = new Color(255, 0, 255, 96); + + private static final double left = -0.25; + private static final double top = -0.25; + private static final double width = 0.5; + private static final double height = 0.5; + + private static final Rectangle2D NORMAL = new Rectangle2D.Double(left, top, width, height); + private static final Rectangle2D HOVERED = new Rectangle2D.Double(left * 3, top * 3, width * 3, height * 3); + + private DistrictNetworkVertex vertex; + + private boolean scaleStroke = true; + private boolean hover; + + private Color color; + private transient Color dynamicColor; + + private Rectangle2D bounds; + + private double nodeSize = 1; + + @Override + public void init() { + setZIndex(2); + } + + @Override + public void render(Graphics2D g2d) { + if (nodeSize <= 0.0) + return; + + AffineTransform ot = null; + AffineTransform t = getTransform(); + if (t != null && !t.isIdentity()) { + ot = g2d.getTransform(); + g2d.transform(getTransform()); + } + + Object oaaHint = null; + Object aaHint = g2d.getRenderingHint(RenderingHints.KEY_ANTIALIASING); + if (aaHint != RenderingHints.VALUE_ANTIALIAS_OFF) { + oaaHint = aaHint; + g2d.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_OFF); + } + + Color oldColor = g2d.getColor(); + + double scaleRecip = 1; + if (scaleStroke) { + double scale = GeometryUtils.getScale(g2d.getTransform()); + scale = Math.max(10000, Math.min(scale, 50000)); + scaleRecip = 1.0 / scale; + } + scaleRecip = scaleRecip * nodeSize; + + // Translate lat and lon to X and Y + Point2D res = calculatePoint2D(vertex); + + Rectangle2D toDraw; + if (hover) { + toDraw = new Rectangle2D.Double(res.getX() - (HOVERED.getWidth() / 2 * scaleRecip), res.getY() - (HOVERED.getHeight() / 2 * scaleRecip), HOVERED.getWidth() * scaleRecip, HOVERED.getHeight() * scaleRecip); + } else { + toDraw = new Rectangle2D.Double(res.getX() - (NORMAL.getWidth() / 2 * scaleRecip), res.getY() - (NORMAL.getHeight() / 2 * scaleRecip), NORMAL.getWidth() * scaleRecip, NORMAL.getHeight() * scaleRecip); + } + + if (NodeUtil.isSelected(this, 1)) { + g2d.setColor(SELECTION_COLOR); + BasicStroke ss = GeometryUtils.scaleStroke(STROKE, (float) scaleRecip*2); + g2d.setStroke(ss); + g2d.draw(toDraw); + } + + // render + g2d.setColor(dynamicColor != null ? dynamicColor : color); + g2d.fill(toDraw); + + // Reset settings + g2d.setColor(oldColor); + if (oaaHint != null) + g2d.setRenderingHint(RenderingHints.KEY_ANTIALIASING, aaHint); + if (ot != null) + g2d.setTransform(ot); + } + + @Override + public Rectangle2D getBounds() { + return super.getBounds(); + } + + @Override + public Rectangle2D getBoundsInLocal() { + return bounds; + } + + private void updateBounds() { + Rectangle2D oldBounds = bounds; + if (oldBounds == null) + oldBounds = new Rectangle2D.Double(); + bounds = calculateBounds(oldBounds); + } + + @Override + public void setTransform(AffineTransform transform) { + super.setTransform(transform); + // Update bounds + updateBounds(); + } + + @Override + public AffineTransform getTransform() { + return super.getTransform(); + } + + private Rectangle2D calculateBounds(Rectangle2D rect) { + Point2D calcPoint = calculatePoint2D(vertex); + AffineTransform at = getTransform(); + return new Rectangle2D.Double(calcPoint.getX(), calcPoint.getY(), width / at.getScaleX(), height / at.getScaleY()).getBounds2D(); + } + + private static Point2D calculatePoint2D(DistrictNetworkVertex vertex) { + Point2D point= vertex.getPoint(); + double x = ModelledCRS.longitudeToX(point.getX()); + double y = ModelledCRS.latitudeToY(-point.getY()); // Inverse because Simantics Diagram is inverted + + // Apply the scaling + Point2D res = new Point2D.Double(x, y); + return res; + } + + public void setVertex(DistrictNetworkVertex vertex) { + this.vertex = vertex; + updateBounds(); + } + + public boolean hover(boolean hover) { +// if (hover && LOGGER.isDebugEnabled()) +// LOGGER.debug("Hovering " + this); + boolean changed = false; + if (this.hover != hover) { + this.hover = hover; + changed = true; + } + return changed; + } + + public void setColor(Color color) { + this.color = color; + } + + public Color getColor() { + return color; + } + + @PropertySetter(value = "size") + public void setSize(Double size) { + boolean changed = false; + if (size != null) { + changed = size != this.nodeSize; + this.nodeSize = size; + } else { + changed = this.nodeSize != 1.0; + this.nodeSize = 1.0; + } + if (changed) + updateBounds(); + } + + @PropertySetter(value = "dynamicColor") + public void setDynamicColor(Color color) { + this.dynamicColor = color; + } + +}