X-Git-Url: https://gerrit.simantics.org/r/gitweb?a=blobdiff_plain;f=org.simantics.district.network.ui%2Fsrc%2Forg%2Fsimantics%2Fdistrict%2Fnetwork%2Fui%2Fadapters%2FDistrictNetworkVertexElementFactory.java;h=976abd48596ccf177065addfa8ffbc1914e1904e;hb=16ee01dc5a40981c58fd5b478b89552e5814e8bb;hp=c19716469d243e81ce7f35ead2b7799310d49c46;hpb=9dace20146d19dd0541480a80fd289de45b6cfc8;p=simantics%2Fdistrict.git

diff --git a/org.simantics.district.network.ui/src/org/simantics/district/network/ui/adapters/DistrictNetworkVertexElementFactory.java b/org.simantics.district.network.ui/src/org/simantics/district/network/ui/adapters/DistrictNetworkVertexElementFactory.java
index c1971646..976abd48 100644
--- a/org.simantics.district.network.ui/src/org/simantics/district/network/ui/adapters/DistrictNetworkVertexElementFactory.java
+++ b/org.simantics.district.network.ui/src/org/simantics/district/network/ui/adapters/DistrictNetworkVertexElementFactory.java
@@ -1,52 +1,64 @@
-package org.simantics.district.network.ui.adapters;
-
-import java.awt.geom.AffineTransform;
-
-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.district.network.ontology.DistrictNetworkResource;
-import org.simantics.g2d.canvas.ICanvasContext;
-import org.simantics.g2d.diagram.IDiagram;
-import org.simantics.g2d.element.ElementClass;
-import org.simantics.g2d.element.ElementUtils;
-import org.simantics.g2d.element.IElement;
-import org.simantics.g2d.element.handler.impl.StaticObjectAdapter;
-
-public class DistrictNetworkVertexElementFactory extends SyncElementFactory {
-
-    public static final ElementClass CLASS = DistrictNetworkVertexElement.CLASS;
-    private DistrictNetworkResource DN;
-    private DiagramResource DIA;
-
-    public DistrictNetworkVertexElementFactory(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<ElementClass> procedure) {
-        procedure.execute(graph, CLASS.newClassWith(false, new StaticObjectAdapter(elementType)));
-    }
-
-    @Override
-    protected Resource getElementClassBaseType(AsyncReadGraph graph) {
-        return DN.Vertex;
-    }
-
-    @Override
-    public void load(ReadGraph graph, ICanvasContext canvas, IDiagram diagram, Resource vertexResource, IElement element) throws DatabaseException {
-        
-        double[] coords = graph.getRelatedValue(vertexResource, DIA.HasLocation);
-        DistrictNetworkVertex vertex = new DistrictNetworkVertex(coords);
-        element.setHint(DistrictNetworkVertexElement.KEY_DN_VERTEX, vertex);
-        
-        // set element transform based on layer SRS and coords
-        AffineTransform at = new AffineTransform(1, 0, 0, 1, coords[0], coords[1]);
-        ElementUtils.setTransform(element, at);
-    }
-}
+package org.simantics.district.network.ui.adapters;
+
+import java.awt.geom.AffineTransform;
+
+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.district.network.ontology.DistrictNetworkResource;
+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.ElementUtils;
+import org.simantics.g2d.element.IElement;
+import org.simantics.g2d.element.handler.impl.StaticObjectAdapter;
+import org.simantics.g2d.layers.ILayersEditor;
+
+public class DistrictNetworkVertexElementFactory extends SyncElementFactory {
+
+    public static final ElementClass CLASS = DistrictNetworkVertexElement.CLASS;
+    private DistrictNetworkResource DN;
+    private DiagramResource DIA;
+
+    public DistrictNetworkVertexElementFactory(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<ElementClass> procedure) {
+        procedure.execute(graph, CLASS.newClassWith(false, new StaticObjectAdapter(elementType)));
+    }
+
+    @Override
+    protected Resource getElementClassBaseType(AsyncReadGraph graph) {
+        return DN.Vertex;
+    }
+
+    @Override
+    public void load(ReadGraph graph, ICanvasContext canvas, IDiagram diagram, Resource vertexResource, IElement element) throws DatabaseException {
+        
+        double[] coords = graph.getRelatedValue(vertexResource, DIA.HasLocation);
+        DistrictNetworkVertex vertex = new DistrictNetworkVertex(coords);
+        element.setHint(DistrictNetworkVertexElement.KEY_DN_VERTEX, vertex);
+        
+        // set element transform based on layer SRS and coords
+        AffineTransform at = new AffineTransform(1, 0, 0, 1, coords[0], coords[1]);
+        ElementUtils.setTransform(element, at);
+        
+        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, vertexResource);
+    }
+}