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=db730b1e6fafcec231b54ecdde6afe73d36e0d57;hb=refs%2Fchanges%2F16%2F616%2F1;hp=8ec1430b83c4e0de2d1bbbb5206505f8323c6b9c;hpb=a2e0d3dfba1245a13dcc8aa23c0188926e2a03c5;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 8ec1430b..db730b1e 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 @@ -1,71 +1,124 @@ -package org.simantics.district.network.ui.adapters; - -import java.awt.geom.Rectangle2D; - -import org.simantics.district.network.ui.DistrictNetworkEdge; -import org.simantics.district.network.ui.nodes.DistrictNetworkEdgeNode; -import org.simantics.g2d.element.ElementClass; -import org.simantics.g2d.element.ElementUtils; -import org.simantics.g2d.element.IElement; -import org.simantics.g2d.element.SceneGraphNodeKey; -import org.simantics.g2d.element.handler.InternalSize; -import org.simantics.g2d.element.handler.SceneGraph; -import org.simantics.g2d.elementclass.connection.EdgeClass.FixedTransform; -import org.simantics.scenegraph.g2d.G2DParentNode; -import org.simantics.utils.datastructures.hints.IHintContext.Key; -import org.simantics.utils.datastructures.hints.IHintContext.KeyOf; - -public class DistrictNetworkEdgeElement { - - 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 ElementClass CLASS = - ElementClass.compile( - FixedTransform.INSTANCE, - DNEdgeInternalSize.INSTANCE, - DNEdgeSceneGraph.INSTANCE - ).setId(DistrictNetworkEdgeElement.class.getSimpleName()); - - static final class DNEdgeSceneGraph implements SceneGraph { - - public static final DNEdgeSceneGraph INSTANCE = new DNEdgeSceneGraph(); - - private static final long serialVersionUID = 8894367073815556871L; - - @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); - } - node.setDNEdge(edge); - } - } - - @Override - public void cleanup(IElement edge) { - ElementUtils.removePossibleNode(edge, KEY_DN_EDGE_NODE); - edge.removeHint(KEY_DN_EDGE_NODE); - } - } - - static final class DNEdgeInternalSize implements InternalSize { - - private static final long serialVersionUID = -2725017034692179676L; - - public static final DNEdgeInternalSize INSTANCE = new DNEdgeInternalSize(); - - @Override - public Rectangle2D getBounds(IElement e, Rectangle2D size) { - DistrictNetworkEdge edge = e.getHint(KEY_DN_EDGE); - return edge.getBounds(size); - } - - } -} +package org.simantics.district.network.ui.adapters; + +import java.awt.Color; +import java.awt.geom.AffineTransform; +import java.awt.geom.Rectangle2D; +import java.util.Collection; +import java.util.Collections; + +import org.simantics.district.network.ui.DistrictNetworkEdge; +import org.simantics.district.network.ui.nodes.DistrictNetworkEdgeNode; +import org.simantics.g2d.connection.handler.ConnectionHandler; +import org.simantics.g2d.diagram.handler.Topology.Connection; +import org.simantics.g2d.element.ElementClass; +import org.simantics.g2d.element.ElementUtils; +import org.simantics.g2d.element.IElement; +import org.simantics.g2d.element.SceneGraphNodeKey; +import org.simantics.g2d.element.handler.InternalSize; +import org.simantics.g2d.element.handler.SceneGraph; +import org.simantics.g2d.element.handler.impl.DefaultTransform; +import org.simantics.g2d.element.handler.impl.SimpleElementLayers; +import org.simantics.scenegraph.g2d.G2DParentNode; +import org.simantics.utils.datastructures.hints.IHintContext.Key; +import org.simantics.utils.datastructures.hints.IHintContext.KeyOf; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; + +public class DistrictNetworkEdgeElement { + + 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 ElementClass CLASS = + ElementClass.compile( + DefaultTransform.INSTANCE, + DNEdgeInternalSize.INSTANCE, + DNEdgeSceneGraph.INSTANCE, + DNEdgeConnectionHandler.INSTANCE, + SimpleElementLayers.INSTANCE, + DistrictNetworkAdditionalColor.INSTANCE + ).setId(DistrictNetworkEdgeElement.class.getSimpleName()); + + static final class DNEdgeSceneGraph implements SceneGraph { + + public static final DNEdgeSceneGraph INSTANCE = new DNEdgeSceneGraph(); + + private static final long serialVersionUID = 8894367073815556871L; + + @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); + } + + node.setColor(ElementUtils.getAdditionalColor(edgeElement, Color.BLUE)); + + node.setDNEdge(edge); + AffineTransform at = ElementUtils.getTransform(edgeElement); + if (at != null) + node.setTransform(at); + } + } + + @Override + public void cleanup(IElement edge) { + ElementUtils.removePossibleNode(edge, KEY_DN_EDGE_NODE); + edge.removeHint(KEY_DN_EDGE_NODE); + } + } + + static final class DNEdgeInternalSize implements InternalSize { + + private static final Logger LOGGER = LoggerFactory.getLogger(DNEdgeInternalSize.class); + + private static final long serialVersionUID = -2725017034692179676L; + + public static final DNEdgeInternalSize INSTANCE = new DNEdgeInternalSize(); + + @Override + public Rectangle2D getBounds(IElement e, Rectangle2D size) { + DistrictNetworkEdge edge = e.getHint(KEY_DN_EDGE); + if (size == null) + size = new Rectangle2D.Double(); + if (edge != null) + size.setFrame(DistrictNetworkEdgeNode.calculatePath(edge).getBounds2D()); + else + LOGGER.debug("Element {} does not have edge!", e); + + return size; + } + } + + static class DNEdgeConnectionHandler implements ConnectionHandler { + + private static final long serialVersionUID = -410377314637446238L; + + public static final DNEdgeConnectionHandler INSTANCE = new DNEdgeConnectionHandler(); + + @Override + public Collection getChildren(IElement connection, Collection result) { + return Collections.emptyList(); + } + + @Override + public Collection getBranchPoints(IElement connection, Collection result) { + return Collections.emptyList(); + } + + @Override + public Collection getSegments(IElement connection, Collection result) { + return Collections.emptyList(); + } + + @Override + public Collection getTerminalConnections(IElement connection, Collection result) { + return Collections.emptyList(); + } + } +}