1 package org.simantics.district.network.ui.adapters;
4 import java.awt.geom.AffineTransform;
5 import java.awt.geom.Rectangle2D;
6 import java.util.Collection;
7 import java.util.Collections;
9 import org.simantics.district.network.ui.DistrictNetworkEdge;
10 import org.simantics.district.network.ui.nodes.DistrictNetworkEdgeNode;
11 import org.simantics.g2d.connection.handler.ConnectionHandler;
12 import org.simantics.g2d.diagram.handler.Topology.Connection;
13 import org.simantics.g2d.element.ElementClass;
14 import org.simantics.g2d.element.ElementUtils;
15 import org.simantics.g2d.element.IElement;
16 import org.simantics.g2d.element.SceneGraphNodeKey;
17 import org.simantics.g2d.element.handler.InternalSize;
18 import org.simantics.g2d.element.handler.SceneGraph;
19 import org.simantics.g2d.element.handler.impl.DefaultTransform;
20 import org.simantics.g2d.element.handler.impl.SimpleElementLayers;
21 import org.simantics.scenegraph.g2d.G2DParentNode;
22 import org.simantics.utils.datastructures.hints.IHintContext.Key;
23 import org.simantics.utils.datastructures.hints.IHintContext.KeyOf;
24 import org.slf4j.Logger;
25 import org.slf4j.LoggerFactory;
27 public class DistrictNetworkEdgeElement {
29 public static final Key KEY_DN_EDGE = new KeyOf(DistrictNetworkEdge.class, "DN_EDGE");
30 public static final Key KEY_DN_EDGE_NODE = new SceneGraphNodeKey(DistrictNetworkEdgeNode.class, "DN_EDGE_NODE");
32 public static final ElementClass CLASS =
34 DefaultTransform.INSTANCE,
35 DNEdgeInternalSize.INSTANCE,
36 DNEdgeSceneGraph.INSTANCE,
37 DNEdgeConnectionHandler.INSTANCE,
38 SimpleElementLayers.INSTANCE,
39 DistrictNetworkAdditionalColor.INSTANCE
40 ).setId(DistrictNetworkEdgeElement.class.getSimpleName());
42 static final class DNEdgeSceneGraph implements SceneGraph {
44 public static final DNEdgeSceneGraph INSTANCE = new DNEdgeSceneGraph();
46 private static final long serialVersionUID = 8894367073815556871L;
49 public void init(IElement edgeElement, G2DParentNode parent) {
50 DistrictNetworkEdge edge = edgeElement.getHint(KEY_DN_EDGE);
54 DistrictNetworkEdgeNode node = edgeElement.getHint(KEY_DN_EDGE_NODE);
56 node = parent.addNode(ElementUtils.generateNodeId(edgeElement), DistrictNetworkEdgeNode.class);
57 edgeElement.setHint(KEY_DN_EDGE_NODE, node);
60 node.setColor(ElementUtils.getAdditionalColor(edgeElement, Color.BLUE));
67 public void cleanup(IElement edge) {
68 ElementUtils.removePossibleNode(edge, KEY_DN_EDGE_NODE);
69 edge.removeHint(KEY_DN_EDGE_NODE);
73 static final class DNEdgeInternalSize implements InternalSize {
75 private static final Logger LOGGER = LoggerFactory.getLogger(DNEdgeInternalSize.class);
77 private static final long serialVersionUID = -2725017034692179676L;
79 public static final DNEdgeInternalSize INSTANCE = new DNEdgeInternalSize();
82 public Rectangle2D getBounds(IElement e, Rectangle2D size) {
83 DistrictNetworkEdge edge = e.getHint(KEY_DN_EDGE);
85 size = new Rectangle2D.Double();
89 LOGGER.debug("Element {} does not have edge!", e);
95 static class DNEdgeConnectionHandler implements ConnectionHandler {
97 private static final long serialVersionUID = -410377314637446238L;
99 public static final DNEdgeConnectionHandler INSTANCE = new DNEdgeConnectionHandler();
102 public Collection<IElement> getChildren(IElement connection, Collection<IElement> result) {
103 return Collections.emptyList();
107 public Collection<IElement> getBranchPoints(IElement connection, Collection<IElement> result) {
108 return Collections.emptyList();
112 public Collection<IElement> getSegments(IElement connection, Collection<IElement> result) {
113 return Collections.emptyList();
117 public Collection<Connection> getTerminalConnections(IElement connection, Collection<Connection> result) {
118 return Collections.emptyList();