X-Git-Url: https://gerrit.simantics.org/r/gitweb?a=blobdiff_plain;f=org.simantics.district.network.ui%2Fsrc%2Forg%2Fsimantics%2Fdistrict%2Fnetwork%2Fui%2Fadapters%2FDistrictNetworkEdgeElement.java;h=726c0c85d561c2ac85c340bc3d5e6b52e537fbee;hb=e6b35994e05232dd536af4da224a342c1bd090af;hp=cbc6a578638e9a1d679915613af759de27b751be;hpb=29914be09d4a237840e5c793bdb562ec83093b8d;p=simantics%2Fdistrict.git diff --git a/org.simantics.district.network.ui/src/org/simantics/district/network/ui/adapters/DistrictNetworkEdgeElement.java b/org.simantics.district.network.ui/src/org/simantics/district/network/ui/adapters/DistrictNetworkEdgeElement.java index cbc6a578..726c0c85 100644 --- a/org.simantics.district.network.ui/src/org/simantics/district/network/ui/adapters/DistrictNetworkEdgeElement.java +++ b/org.simantics.district.network.ui/src/org/simantics/district/network/ui/adapters/DistrictNetworkEdgeElement.java @@ -4,6 +4,9 @@ import java.awt.Color; import java.awt.Shape; import java.awt.geom.AffineTransform; import java.awt.geom.Line2D; +import java.awt.geom.Path2D; +import java.awt.geom.PathIterator; +import java.awt.geom.Point2D; import java.awt.geom.Rectangle2D; import java.util.Collection; import java.util.Collections; @@ -22,12 +25,13 @@ import org.simantics.g2d.element.handler.InternalSize; import org.simantics.g2d.element.handler.Outline; import org.simantics.g2d.element.handler.Pick; import org.simantics.g2d.element.handler.SceneGraph; -import org.simantics.g2d.element.handler.SelectionOutline; import org.simantics.g2d.element.handler.impl.ConnectionSelectionOutline; import org.simantics.g2d.element.handler.impl.DefaultTransform; import org.simantics.g2d.element.handler.impl.SimpleElementLayers; import org.simantics.maps.MapScalingTransform; import org.simantics.scenegraph.g2d.G2DParentNode; +import org.simantics.scenegraph.g2d.nodes.SVGNode; +import org.simantics.scenegraph.utils.NodeUtil; import org.simantics.utils.datastructures.hints.IHintContext.Key; import org.simantics.utils.datastructures.hints.IHintContext.KeyOf; import org.slf4j.Logger; @@ -35,8 +39,11 @@ import org.slf4j.LoggerFactory; public class DistrictNetworkEdgeElement { + private static final Logger LOGGER = LoggerFactory.getLogger(DistrictNetworkEdgeElement.class); + public static final Key KEY_DN_EDGE = new KeyOf(DistrictNetworkEdge.class, "DN_EDGE"); public static final Key KEY_DN_EDGE_NODE = new SceneGraphNodeKey(DistrictNetworkEdgeNode.class, "DN_EDGE_NODE"); + public static final Key KEY_DN_EDGE_SYMBOL_NODE = new SceneGraphNodeKey(DistrictNetworkEdgeNode.class, "DN_EDGE_SYMBOL_NODE"); public static final ElementClass CLASS = ElementClass.compile( @@ -58,29 +65,38 @@ public class DistrictNetworkEdgeElement { @Override public void init(IElement edgeElement, G2DParentNode parent) { - DistrictNetworkEdge edge = edgeElement.getHint(KEY_DN_EDGE); - if (edge == null) { - cleanup(edgeElement); - } else { - DistrictNetworkEdgeNode node = edgeElement.getHint(KEY_DN_EDGE_NODE); - if (node == null) { - node = parent.addNode(ElementUtils.generateNodeId(edgeElement), DistrictNetworkEdgeNode.class); - edgeElement.setHint(KEY_DN_EDGE_NODE, node); + try { + DistrictNetworkEdge edge = edgeElement.getHint(KEY_DN_EDGE); + if (edge == null) { + cleanup(edgeElement); + } else { + DistrictNetworkEdgeNode node = edgeElement.getHint(KEY_DN_EDGE_NODE); + if (node == null) { + node = parent.addNode(ElementUtils.generateNodeId(edgeElement), DistrictNetworkEdgeNode.class); + edgeElement.setHint(KEY_DN_EDGE_NODE, node); + + SVGNode symbol = node.addNode(ElementUtils.generateNodeId(edgeElement), SVGNode.class); + edgeElement.setHint(KEY_DN_EDGE_SYMBOL_NODE, symbol); + } + + node.setColor(ElementUtils.getAdditionalColor(edgeElement, Color.BLUE)); + + node.setDNEdge(edge); + AffineTransform at = ElementUtils.getTransform(edgeElement); + if (at != null) + node.setTransform(at); } - - node.setColor(ElementUtils.getAdditionalColor(edgeElement, Color.BLUE)); - - node.setDNEdge(edge); - AffineTransform at = ElementUtils.getTransform(edgeElement); - if (at != null) - node.setTransform(at); + } catch (Exception e) { + LOGGER.error("Could not initialize element {} for parent {}", edgeElement, parent, e); } } @Override public void cleanup(IElement edge) { + ElementUtils.removePossibleNode(edge, KEY_DN_EDGE_SYMBOL_NODE); ElementUtils.removePossibleNode(edge, KEY_DN_EDGE_NODE); edge.removeHint(KEY_DN_EDGE_NODE); + edge.removeHint(KEY_DN_EDGE_SYMBOL_NODE); } } @@ -94,11 +110,11 @@ public class DistrictNetworkEdgeElement { @Override public Rectangle2D getBounds(IElement e, Rectangle2D size) { - DistrictNetworkEdge edge = e.getHint(KEY_DN_EDGE); + DistrictNetworkEdgeNode edgeNode = e.getHint(KEY_DN_EDGE_NODE); if (size == null) size = new Rectangle2D.Double(); - if (edge != null) - size.setFrame(DistrictNetworkEdgeNode.calculatePath(edge, null).getBounds2D()); + if (edgeNode != null) + size.setFrame(edgeNode.getBoundsInLocal()); else LOGGER.debug("Element {} does not have edge!", e); @@ -107,60 +123,101 @@ public class DistrictNetworkEdgeElement { @Override public Shape getElementShape(IElement e) { - DistrictNetworkEdge edge = e.getHint(KEY_DN_EDGE); - if (edge != null) { - return DistrictNetworkEdgeNode.calculatePath(edge, null); + DistrictNetworkEdgeNode edgeNode = e.getHint(KEY_DN_EDGE_NODE); + if (edgeNode != null) { + return edgeNode.getPath(); } else { return getBounds(e, null); } } - private Shape getSelectionShape(IElement e) { - for (SelectionOutline so : e.getElementClass().getItemsByClass(SelectionOutline.class)) { - Shape shape = so.getSelectionShape(e); - if (shape != null) - return shape; - } - // Using on-diagram coordinates because neither connections nor - // edges have a non-identity transform which means that - // coordinates are always absolute. Therefore branch point - // shape also needs to be calculated in absolute coordinates. - Shape shape = ElementUtils.getElementShapeOrBoundsOnDiagram(e); - return shape; - } - @Override public boolean pickTest(IElement e, Shape s, PickPolicy policy) { DistrictNetworkEdge edge = e.getHint(KEY_DN_EDGE); if (edge != null) { Rectangle2D bounds = getBounds(s); - switch (policy) { - case PICK_CONTAINED_OBJECTS: - Shape selectionShape = getSelectionShape(e); - return bounds.contains(selectionShape.getBounds2D()); - - case PICK_INTERSECTING_OBJECTS: - double tolerance = (bounds.getHeight() + bounds.getHeight()) * 0.25 / MapScalingTransform.getScaleX(); - Line2D line = new Line2D.Double(edge.getStartPoint(), edge.getEndPoint()); - double sx = bounds.getCenterX() / MapScalingTransform.getScaleX(); - double sy = bounds.getCenterY() / MapScalingTransform.getScaleY(); - double ssx = ModelledCRS.xToLongitude(sx); - double ssy = ModelledCRS.yToLatitude(-sy); // Invert for Simantics diagram coordinate system - double distSq = line.ptSegDistSq(ssx, ssy); + case PICK_CONTAINED_OBJECTS: return pickContainedObjects(edge, bounds); + case PICK_INTERSECTING_OBJECTS: return pickIntersectingObjects(e, edge, bounds); + } + return false; + } + + return false; + } + + private boolean pickContainedObjects(DistrictNetworkEdge edge, Rectangle2D bounds) { + double bminx = bounds.getMinX() / MapScalingTransform.getScaleX(); + double bminy = bounds.getMinY() / MapScalingTransform.getScaleY(); + double bmaxx = bounds.getMaxX() / MapScalingTransform.getScaleX(); + double bmaxy = bounds.getMaxY() / MapScalingTransform.getScaleY(); + + double bsminx = ModelledCRS.xToLongitude(bminx); + double bsminy = ModelledCRS.yToLatitude(-bminy); // Invert for Simantics diagram coordinate system + double bsmaxx = ModelledCRS.xToLongitude(bmaxx); + double bsmaxy = ModelledCRS.yToLatitude(-bmaxy); // Invert for Simantics diagram coordinate system + + double boundsMinY = Math.min(bsminy, bsmaxy); + double boundsMaxY = Math.max(bsminy, bsmaxy); + + Point2D start = edge.getStartPoint(); + Point2D end = edge.getEndPoint(); + + double eminx = Math.min(start.getX(), end.getX()); + double eminy = Math.min(start.getY(), end.getY()); + double emaxx = Math.max(start.getX(), end.getX()); + double emaxy = Math.max(start.getY(), end.getY()); + + return eminx >= bsminx && eminy >= boundsMinY && emaxx <= bsmaxx && emaxy <= boundsMaxY; + } + + private boolean pickIntersectingObjects(IElement e, DistrictNetworkEdge edge, Rectangle2D bounds) { + double dx = bounds.getWidth() / MapScalingTransform.getScaleX(); + double dy = bounds.getHeight() / MapScalingTransform.getScaleY(); + + // Half the diagonal + half of the line width + DistrictNetworkEdgeNode node = e.getHint(KEY_DN_EDGE_NODE); + AffineTransform at = NodeUtil.getLocalToGlobalTransform(node); + + Path2D path = node.getPath(); + if (path == null) + return false; + + double lineWidth = node.getStrokeWidth(at, true); + double tolerance = Math.sqrt(dx * dx + dy * dy) / 2 + lineWidth / 2; + + double sx = bounds.getCenterX() / MapScalingTransform.getScaleX(); + double sy = bounds.getCenterY() / MapScalingTransform.getScaleY(); + + double coords[] = new double[6]; + Point2D prevPoint = new Point2D.Double(), curPoint = new Point2D.Double(); + Line2D line = new Line2D.Double(); + for (PathIterator it = path.getPathIterator(null); !it.isDone(); it.next()) { + int type = it.currentSegment(coords); + switch (type) { + case PathIterator.SEG_MOVETO: + curPoint.setLocation(coords[0], coords[1]); + break; + case PathIterator.SEG_LINETO: + prevPoint.setLocation(curPoint); + curPoint.setLocation(coords[0], coords[1]); + line.setLine(prevPoint, curPoint); + double distSq = line.ptSegDistSq(sx, sy); // System.out.println("s: " + sx + ", " + sy); // System.out.println("ss: " + ssx + ", " + ssy); // System.out.println("p1: " + edge.getStartPoint()); // System.out.println("p2: " + edge.getEndPoint()); // System.out.println("line: " + "(" + line.getX1() + ", " + line.getY1() + ", " + line.getX2() + ", " + line.getY2() + ")"); // System.out.println("distance from line is " + Math.sqrt(distSq) + " with tolerance " + tolerance); - if (distSq <= tolerance * tolerance) { - return true; - } + if (distSq <= tolerance * tolerance) + return true; + break; + default: + LOGGER.error("Invalid edge path", new IllegalStateException()); + return false; } - return false; } - + return false; }