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.INode;
import org.simantics.scenegraph.g2d.G2DParentNode;
+import org.simantics.scenegraph.g2d.nodes.SVGNode;
import org.simantics.utils.datastructures.hints.IHintContext.Key;
import org.simantics.utils.datastructures.hints.IHintContext.KeyOf;
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 Key KEY_DN_VERTEX_SYMBOL_NODE = new SceneGraphNodeKey(SVGNode.class, "DN_VERTEX_SYMBOL_NODE");
public static final ElementClass CLASS =
ElementClass.compile(
if (node == null) {
node = parent.addNode(ElementUtils.generateNodeId(vertexElement), DistrictNetworkVertexNode.class);
vertexElement.setHint(KEY_DN_VERTEX_NODE, node);
+
+ SVGNode symbol = node.addNode(ElementUtils.generateNodeId(vertexElement), SVGNode.class);
+ vertexElement.setHint(KEY_DN_VERTEX_SYMBOL_NODE, symbol);
}
+
node.setVertex(vertex);
node.setColor(ElementUtils.getAdditionalColor(vertexElement, Color.BLUE));
@Override
public void cleanup(IElement edge) {
- ElementUtils.removePossibleNode(edge, KEY_DN_VERTEX_NODE);
+ ElementUtils.removePossibleNode(edge, KEY_DN_VERTEX_SYMBOL_NODE);
+ INode node = ElementUtils.removePossibleNode(edge, KEY_DN_VERTEX_NODE);
+ if (node != null)
+ node.getParent().remove();
+
edge.removeHint(KEY_DN_VERTEX_NODE);
+ edge.removeHint(KEY_DN_VERTEX_SYMBOL_NODE);
}
}