package org.simantics.district.network.ui.adapters; import java.awt.geom.Path2D; import org.simantics.db.AsyncReadGraph; import org.simantics.db.ReadGraph; import org.simantics.db.Resource; import org.simantics.db.exception.DatabaseException; import org.simantics.db.procedure.AsyncProcedure; import org.simantics.diagram.adapter.SyncElementFactory; import org.simantics.diagram.stubs.DiagramResource; import org.simantics.diagram.synchronization.IModifiableSynchronizationContext; import org.simantics.diagram.synchronization.SynchronizationHints; import org.simantics.diagram.synchronization.graph.GraphSynchronizationHints; import org.simantics.diagram.synchronization.graph.layer.GraphLayerManager; import org.simantics.diagram.ui.DiagramModelHints; import org.simantics.district.network.ontology.DistrictNetworkResource; import org.simantics.district.network.ui.DistrictNetworkEdge; import org.simantics.g2d.canvas.ICanvasContext; import org.simantics.g2d.diagram.DiagramHints; import org.simantics.g2d.diagram.IDiagram; import org.simantics.g2d.element.ElementClass; import org.simantics.g2d.element.IElement; import org.simantics.g2d.element.handler.impl.StaticObjectAdapter; import org.simantics.g2d.layers.ILayersEditor; public class DistrictNetworkEdgeElementFactory extends SyncElementFactory { public static final ElementClass CLASS = DistrictNetworkEdgeElement.CLASS; private DistrictNetworkResource DN; private DiagramResource DIA; public DistrictNetworkEdgeElementFactory(ReadGraph graph) { this.DN = DistrictNetworkResource.getInstance(graph); this.DIA = DiagramResource.getInstance(graph); } @Override public void create(AsyncReadGraph graph, ICanvasContext canvas, IDiagram diagram, Resource elementType, AsyncProcedure procedure) { procedure.execute(graph, CLASS.newClassWith(false, new StaticObjectAdapter(elementType))); } @Override protected Resource getElementClassBaseType(AsyncReadGraph graph) { return DN.Edge; } @Override public void load(ReadGraph graph, ICanvasContext canvas, IDiagram diagram, Resource edgeResource, IElement element) throws DatabaseException { Resource diagramRuntime = diagram.getHint(DiagramModelHints.KEY_DIAGRAM_RUNTIME_RESOURCE); Resource startVertex = graph.getSingleObject(edgeResource, DN.HasStartVertex); Resource endVertex = graph.getSingleObject(edgeResource, DN.HasEndVertex); double[] startCoords = graph.getRelatedValue2(startVertex, DIA.HasLocation); double[] endCoords = graph.getRelatedValue2(endVertex, DIA.HasLocation); Path2D path = new Path2D.Double(); path.moveTo(startCoords[0], startCoords[1]); path.lineTo(endCoords[0], endCoords[1]); DistrictNetworkEdge edge = new DistrictNetworkEdge(path); element.setHint(DistrictNetworkEdgeElement.KEY_DN_EDGE, edge); IModifiableSynchronizationContext context = diagram.getHint(SynchronizationHints.CONTEXT); GraphLayerManager layerManager = context.get(GraphSynchronizationHints.GRAPH_LAYER_MANAGER); ILayersEditor layersEditor = diagram.getHint(DiagramHints.KEY_LAYERS_EDITOR); layerManager.loadLayersForElement(graph, layersEditor, element, edgeResource); } }