org.simantics.district.network.ui/src/org/simantics/district/network/ui/adapters/DistrictNetworkEdgeElementFactory.java

   1 package org.simantics.district.network.ui.adapters;
   2
   3 import java.awt.geom.Point2D;
   4
   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;
  28
  29 public class DistrictNetworkEdgeElementFactory extends SyncElementFactory {
  30
  31     public static final ElementClass CLASS = DistrictNetworkEdgeElement.CLASS;
  32     public static final double[] EMPTY = new double[0];
  33
  34     private DistrictNetworkResource DN;
  35     private DiagramResource DIA;
  36
  37     public DistrictNetworkEdgeElementFactory(ReadGraph graph) {
  38         this.DN = DistrictNetworkResource.getInstance(graph);
  39         this.DIA = DiagramResource.getInstance(graph);
  40     }
  41
  42     @Override
  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)));
  45     }
  46
  47     @Override
  48     protected Resource getElementClassBaseType(AsyncReadGraph graph) {
  49         return DN.Edge;
  50     }
  51
  52     @Override
  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
  56
  57         if (!graph.hasStatement(edgeResource, DN.HasStartVertex))
  58             return; // already deleted
  59         Resource startVertex = graph.getSingleObject(edgeResource, DN.HasStartVertex);
  60
  61         if (!graph.hasStatement(edgeResource, DN.HasEndVertex))
  62             return; // already deleted
  63         Resource endVertex = graph.getSingleObject(edgeResource, DN.HasEndVertex);
  64
  65         double[] startCoords = graph.getRelatedValue2(startVertex, DIA.HasLocation, Bindings.DOUBLE_ARRAY);
  66         double[] endCoords = graph.getRelatedValue2(endVertex, DIA.HasLocation, Bindings.DOUBLE_ARRAY);
  67
  68         double[] geometry = EMPTY;
  69         try {
  70             geometry = graph.getPossibleRelatedValue2(edgeResource, DN.Edge_HasGeometry, Bindings.DOUBLE_ARRAY);
  71         } catch (Exception e) {
  72             // most likely no geometry available
  73         }
  74         DistrictNetworkEdge edge = new DistrictNetworkEdge(new Point2D.Double(startCoords[0], startCoords[1]), new Point2D.Double(endCoords[0], endCoords[1]), geometry);
  75
  76         Resource mapping = graph.getSingleObject(edgeResource, DistrictNetworkResource.getInstance(graph).HasMapping);
  77         element.setHint(DistrictNetworkAdditionalColor.KEY_DN_MAPPING_RESOURCE, mapping);
  78
  79         element.setHint(DistrictNetworkEdgeElement.KEY_DN_EDGE, edge);
  80
  81         // set scaling transform
  82         ElementUtils.setTransform(element, MapScalingTransform.INSTANCE);
  83
  84         IModifiableSynchronizationContext context = diagram.getHint(SynchronizationHints.CONTEXT);
  85         GraphLayerManager layerManager = context.get(GraphSynchronizationHints.GRAPH_LAYER_MANAGER);
  86
  87         ILayersEditor layersEditor = diagram.getHint(DiagramHints.KEY_LAYERS_EDITOR);
  88         layerManager.loadLayersForElement(graph, layersEditor, element, edgeResource);
  89
  90     }
  91 }