-package org.simantics.district.network.ui.adapters;\r
-\r
-import java.awt.Dimension;\r
-import java.awt.geom.AffineTransform;\r
-import java.awt.geom.Rectangle2D;\r
-\r
-import org.simantics.district.network.ui.nodes.DistrictNetworkVertexNode;\r
-import org.simantics.g2d.element.ElementClass;\r
-import org.simantics.g2d.element.ElementUtils;\r
-import org.simantics.g2d.element.IElement;\r
-import org.simantics.g2d.element.SceneGraphNodeKey;\r
-import org.simantics.g2d.element.handler.InternalSize;\r
-import org.simantics.g2d.element.handler.SceneGraph;\r
-import org.simantics.g2d.element.handler.impl.DefaultTransform;\r
-import org.simantics.scenegraph.g2d.G2DParentNode;\r
-import org.simantics.utils.datastructures.hints.IHintContext.Key;\r
-import org.simantics.utils.datastructures.hints.IHintContext.KeyOf;\r
-\r
-public class DistrictNetworkVertexElement {\r
-\r
- public static final Key KEY_DN_VERTEX = new KeyOf(DistrictNetworkVertex.class, "DN_VERTEX");\r
- public static final Key KEY_DN_VERTEX_NODE = new SceneGraphNodeKey(DistrictNetworkVertexNode.class, "DN_VERTEX_NODE");\r
- \r
- public static final ElementClass CLASS =\r
- ElementClass.compile(\r
- DefaultTransform.INSTANCE,\r
- DNVertexInternalSize.INSTANCE,\r
- DNVertexSceneGraph.INSTANCE\r
- ).setId(DistrictNetworkVertexElement.class.getSimpleName());\r
- \r
- static final class DNVertexSceneGraph implements SceneGraph {\r
- \r
- public static final DNVertexSceneGraph INSTANCE = new DNVertexSceneGraph();\r
-\r
- private static final long serialVersionUID = 8894367073815556871L;\r
-\r
- @Override\r
- public void init(IElement edgeElement, G2DParentNode parent) {\r
- DistrictNetworkVertex vertex = edgeElement.getHint(KEY_DN_VERTEX);\r
- if (vertex == null) {\r
- cleanup(edgeElement);\r
- } else {\r
- DistrictNetworkVertexNode node = edgeElement.getHint(KEY_DN_VERTEX_NODE);\r
- if (node == null) {\r
- node = parent.addNode(ElementUtils.generateNodeId(edgeElement), DistrictNetworkVertexNode.class);\r
- edgeElement.setHint(KEY_DN_VERTEX_NODE, node);\r
- }\r
- node.setVertex(vertex);\r
- \r
- AffineTransform at = ElementUtils.getTransform(edgeElement);\r
- if(at != null)\r
- node.setTransform(at);\r
- }\r
- }\r
-\r
- @Override\r
- public void cleanup(IElement edge) {\r
- ElementUtils.removePossibleNode(edge, KEY_DN_VERTEX_NODE);\r
- edge.removeHint(KEY_DN_VERTEX_NODE);\r
- }\r
- }\r
- \r
- static final class DNVertexInternalSize implements InternalSize {\r
-\r
- public static final DNVertexInternalSize INSTANCE = new DNVertexInternalSize();\r
- \r
- private static final long serialVersionUID = 2625702114278956461L;\r
-\r
- @Override\r
- public Rectangle2D getBounds(IElement e, Rectangle2D size) {\r
- if (size == null)\r
- size = new Rectangle2D.Double();\r
- size.setFrame(-0.5, -0.5, 1, 1);\r
- return size;\r
- }\r
- \r
- }\r
-}\r
+package org.simantics.district.network.ui.adapters;
+
+import java.awt.Color;
+import java.awt.geom.AffineTransform;
+import java.awt.geom.Rectangle2D;
+
+import org.simantics.district.network.ui.nodes.DistrictNetworkVertexNode;
+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;
+
+public class DistrictNetworkVertexElement {
+
+ public static final Key KEY_DN_VERTEX = new KeyOf(DistrictNetworkVertex.class, "DN_VERTEX");
+ public static final Key KEY_DN_VERTEX_NODE = new SceneGraphNodeKey(DistrictNetworkVertexNode.class, "DN_VERTEX_NODE");
+
+ public static final ElementClass CLASS =
+ ElementClass.compile(
+ DefaultTransform.INSTANCE,
+ DNVertexInternalSize.INSTANCE,
+ DNVertexSceneGraph.INSTANCE,
+ SimpleElementLayers.INSTANCE,
+ DistrictNetworkAdditionalColor.INSTANCE
+ ).setId(DistrictNetworkVertexElement.class.getSimpleName());
+
+ static final class DNVertexSceneGraph implements SceneGraph {
+
+ public static final DNVertexSceneGraph INSTANCE = new DNVertexSceneGraph();
+
+ private static final long serialVersionUID = 8894367073815556871L;
+
+ @Override
+ public void init(IElement vertexElement, G2DParentNode parent) {
+ DistrictNetworkVertex vertex = vertexElement.getHint(KEY_DN_VERTEX);
+ if (vertex == null) {
+ cleanup(vertexElement);
+ } else {
+ DistrictNetworkVertexNode node = vertexElement.getHint(KEY_DN_VERTEX_NODE);
+ if (node == null) {
+ node = parent.addNode(ElementUtils.generateNodeId(vertexElement), DistrictNetworkVertexNode.class);
+ vertexElement.setHint(KEY_DN_VERTEX_NODE, node);
+ }
+ node.setVertex(vertex);
+
+ node.setColor(ElementUtils.getAdditionalColor(vertexElement, Color.BLUE));
+
+ AffineTransform at = ElementUtils.getTransform(vertexElement);
+ if(at != null)
+ node.setTransform(at);
+ }
+ }
+
+ @Override
+ public void cleanup(IElement edge) {
+ ElementUtils.removePossibleNode(edge, KEY_DN_VERTEX_NODE);
+ edge.removeHint(KEY_DN_VERTEX_NODE);
+ }
+ }
+
+ static final class DNVertexInternalSize implements InternalSize {
+
+ public static final DNVertexInternalSize INSTANCE = new DNVertexInternalSize();
+
+ private static final long serialVersionUID = 2625702114278956461L;
+
+ @Override
+ public Rectangle2D getBounds(IElement e, Rectangle2D size) {
+ DistrictNetworkVertexNode node = e.getHint(KEY_DN_VERTEX_NODE);
+ Rectangle2D boundsInLocal = node.getBoundsInLocal();
+ size.setFrame(boundsInLocal);
+ return size;
+ }
+
+ }
+}