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.Shape; import java.awt.Stroke; import java.awt.geom.AffineTransform; import java.awt.geom.Path2D; import java.awt.geom.Rectangle2D; import org.simantics.district.network.ui.DistrictNetworkEdge; import org.simantics.scenegraph.g2d.G2DNode; import org.simantics.scenegraph.utils.NodeUtil; public class DistrictNetworkEdgeNode extends G2DNode { private static final long serialVersionUID = 8049769475036519806L; private static final Stroke SELECTION_STROKE = new BasicStroke(1f, BasicStroke.CAP_BUTT, BasicStroke.JOIN_MITER); private static final Color SELECTION_COLOR = new Color(255, 0, 255, 96); private DistrictNetworkEdge edge; private Rectangle2D bounds; @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(); // boolean selected = isSelected(); // if (selected) { // Path2D selectionPath = edge.getPath(); // Shape selectionShape = SELECTION_STROKE.createStrokedShape(selectionPath); // g2d.setColor(SELECTION_COLOR); // g2d.fill(selectionShape); // } // g2d.setColor(Color.BLUE); // render g2d.draw(edge.getPath()); // Reset g2d.setColor(oldColor); g2d.setRenderingHint(RenderingHints.KEY_ANTIALIASING, aaHint); if (ot != null) g2d.setTransform(ot); } private boolean isSelected() { return NodeUtil.isSelected(this, 1); } @Override public Rectangle2D getBoundsInLocal() { return bounds; } private void updateBounds() { Rectangle2D oldBounds = bounds; if (oldBounds == null) oldBounds = new Rectangle2D.Double(); bounds = calculateBounds(oldBounds); } private Rectangle2D calculateBounds(Rectangle2D rect) { return edge.getBounds(rect); } public void setDNEdge(DistrictNetworkEdge edge) { this.edge = edge; updateBounds(); } }