]> gerrit.simantics Code Review - simantics/district.git/blob
63590990cd9ba421097cd65270be0b6cf8faaa4c
[simantics/district.git] /
1 package org.simantics.district.network.ui.adapters;
2
3 import java.awt.geom.Path2D;
4
5 import org.simantics.db.AsyncReadGraph;
6 import org.simantics.db.ReadGraph;
7 import org.simantics.db.Resource;
8 import org.simantics.db.exception.DatabaseException;
9 import org.simantics.db.procedure.AsyncProcedure;
10 import org.simantics.diagram.adapter.SyncElementFactory;
11 import org.simantics.diagram.stubs.DiagramResource;
12 import org.simantics.diagram.synchronization.IModifiableSynchronizationContext;
13 import org.simantics.diagram.synchronization.SynchronizationHints;
14 import org.simantics.diagram.synchronization.graph.GraphSynchronizationHints;
15 import org.simantics.diagram.synchronization.graph.layer.GraphLayerManager;
16 import org.simantics.diagram.ui.DiagramModelHints;
17 import org.simantics.district.network.ontology.DistrictNetworkResource;
18 import org.simantics.district.network.ui.DistrictNetworkEdge;
19 import org.simantics.g2d.canvas.ICanvasContext;
20 import org.simantics.g2d.diagram.DiagramHints;
21 import org.simantics.g2d.diagram.IDiagram;
22 import org.simantics.g2d.element.ElementClass;
23 import org.simantics.g2d.element.IElement;
24 import org.simantics.g2d.element.handler.impl.StaticObjectAdapter;
25 import org.simantics.g2d.layers.ILayersEditor;
26
27 public class DistrictNetworkEdgeElementFactory extends SyncElementFactory {
28
29     public static final ElementClass CLASS = DistrictNetworkEdgeElement.CLASS;
30     
31     private DistrictNetworkResource DN;
32     private DiagramResource DIA;
33     
34     public DistrictNetworkEdgeElementFactory(ReadGraph graph) {
35         this.DN = DistrictNetworkResource.getInstance(graph);
36         this.DIA = DiagramResource.getInstance(graph);
37     }
38     
39     @Override
40     public void create(AsyncReadGraph graph, ICanvasContext canvas, IDiagram diagram, Resource elementType, AsyncProcedure<ElementClass> procedure) {
41         procedure.execute(graph, CLASS.newClassWith(false, new StaticObjectAdapter(elementType)));
42     }
43     
44     @Override
45     protected Resource getElementClassBaseType(AsyncReadGraph graph) {
46         return DN.Edge;
47     }
48     
49     @Override
50     public void load(ReadGraph graph, ICanvasContext canvas, IDiagram diagram, Resource edgeResource, IElement element) throws DatabaseException {
51         
52         Resource diagramRuntime = diagram.getHint(DiagramModelHints.KEY_DIAGRAM_RUNTIME_RESOURCE);
53         
54         Resource startVertex = graph.getSingleObject(edgeResource, DN.HasStartVertex);
55         Resource endVertex = graph.getSingleObject(edgeResource, DN.HasEndVertex);
56         
57         double[] startCoords = graph.getRelatedValue2(startVertex, DIA.HasLocation);
58         double[] endCoords = graph.getRelatedValue2(endVertex, DIA.HasLocation);
59         
60         Path2D path = new Path2D.Double();
61         path.moveTo(startCoords[0], startCoords[1]);
62         path.lineTo(endCoords[0], endCoords[1]);
63         
64         DistrictNetworkEdge edge = new DistrictNetworkEdge(path);
65         
66         element.setHint(DistrictNetworkEdgeElement.KEY_DN_EDGE, edge);
67         
68         IModifiableSynchronizationContext context = diagram.getHint(SynchronizationHints.CONTEXT);
69         GraphLayerManager layerManager = context.get(GraphSynchronizationHints.GRAPH_LAYER_MANAGER);
70         
71         ILayersEditor layersEditor = diagram.getHint(DiagramHints.KEY_LAYERS_EDITOR);
72         layerManager.loadLayersForElement(graph, layersEditor, element, edgeResource);
73         
74     }
75 }