1 package org.simantics.district.network.ui.adapters;
3 import java.awt.geom.Point2D;
5 import org.simantics.databoard.Bindings;
6 import org.simantics.db.AsyncReadGraph;
7 import org.simantics.db.ReadGraph;
8 import org.simantics.db.Resource;
9 import org.simantics.db.exception.DatabaseException;
10 import org.simantics.db.procedure.AsyncProcedure;
11 import org.simantics.diagram.adapter.SyncElementFactory;
12 import org.simantics.diagram.stubs.DiagramResource;
13 import org.simantics.diagram.synchronization.IModifiableSynchronizationContext;
14 import org.simantics.diagram.synchronization.SynchronizationHints;
15 import org.simantics.diagram.synchronization.graph.GraphSynchronizationHints;
16 import org.simantics.diagram.synchronization.graph.layer.GraphLayerManager;
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.ElementUtils;
24 import org.simantics.g2d.element.IElement;
25 import org.simantics.g2d.element.handler.impl.StaticObjectAdapter;
26 import org.simantics.g2d.layers.ILayersEditor;
27 import org.simantics.maps.MapScalingTransform;
29 public class DistrictNetworkEdgeElementFactory extends SyncElementFactory {
31 public static final ElementClass CLASS = DistrictNetworkEdgeElement.CLASS;
32 public static final double[] EMPTY = new double[0];
34 private DistrictNetworkResource DN;
35 private DiagramResource DIA;
37 public DistrictNetworkEdgeElementFactory(ReadGraph graph) {
38 this.DN = DistrictNetworkResource.getInstance(graph);
39 this.DIA = DiagramResource.getInstance(graph);
43 public void create(AsyncReadGraph graph, ICanvasContext canvas, IDiagram diagram, Resource elementType, AsyncProcedure<ElementClass> procedure) {
44 procedure.execute(graph, CLASS.newClassWith(false, new StaticObjectAdapter(elementType)));
48 protected Resource getElementClassBaseType(AsyncReadGraph graph) {
53 public void load(ReadGraph graph, ICanvasContext canvas, IDiagram diagram, Resource edgeResource, IElement element) throws DatabaseException {
54 if (!graph.hasStatement(edgeResource))
55 return; // already deleted
57 if (!graph.hasStatement(edgeResource, DN.HasStartVertex))
58 return; // already deleted
59 Resource startVertex = graph.getSingleObject(edgeResource, DN.HasStartVertex);
61 if (!graph.hasStatement(edgeResource, DN.HasEndVertex))
62 return; // already deleted
63 Resource endVertex = graph.getSingleObject(edgeResource, DN.HasEndVertex);
65 double[] startCoords = graph.getRelatedValue2(startVertex, DIA.HasLocation, Bindings.DOUBLE_ARRAY);
66 double[] endCoords = graph.getRelatedValue2(endVertex, DIA.HasLocation, Bindings.DOUBLE_ARRAY);
68 double[] geometry = EMPTY;
70 geometry = graph.getPossibleRelatedValue2(edgeResource, DN.Edge_HasGeometry, Bindings.DOUBLE_ARRAY);
71 } catch (Exception e) {
72 // most likely no geometry available
74 DistrictNetworkEdge edge = new DistrictNetworkEdge(new Point2D.Double(startCoords[0], startCoords[1]), new Point2D.Double(endCoords[0], endCoords[1]), geometry);
76 Resource mapping = graph.getSingleObject(edgeResource, DistrictNetworkResource.getInstance(graph).HasMapping);
77 element.setHint(DistrictNetworkAdditionalColor.KEY_DN_MAPPING_RESOURCE, mapping);
79 element.setHint(DistrictNetworkEdgeElement.KEY_DN_EDGE, edge);
81 // set scaling transform
82 ElementUtils.setTransform(element, MapScalingTransform.INSTANCE);
84 IModifiableSynchronizationContext context = diagram.getHint(SynchronizationHints.CONTEXT);
85 GraphLayerManager layerManager = context.get(GraphSynchronizationHints.GRAPH_LAYER_MANAGER);
87 ILayersEditor layersEditor = diagram.getHint(DiagramHints.KEY_LAYERS_EDITOR);
88 layerManager.loadLayersForElement(graph, layersEditor, element, edgeResource);