Some more added functionality to simantics district editor etc

[simantics/district.git] / org.simantics.district.network.ui / src / org / simantics / district / network / ui / adapters / DistrictDiagramClassAdapter.java

diff --git a/org.simantics.district.network.ui/src/org/simantics/district/network/ui/adapters/DistrictDiagramClassAdapter.java b/org.simantics.district.network.ui/src/org/simantics/district/network/ui/adapters/DistrictDiagramClassAdapter.java
new file mode 100644 (file)
index 0000000..fe3ed93
--- /dev/null
+++ b/org.simantics.district.network.ui/src/org/simantics/district/network/ui/adapters/DistrictDiagramClassAdapter.java
@@ -0,0 +1,89 @@
+package org.simantics.district.network.ui.adapters;
+
+import org.simantics.Simantics;
+import org.simantics.db.AsyncReadGraph;
+import org.simantics.db.Resource;
+import org.simantics.db.WriteGraph;
+import org.simantics.db.common.request.WriteRequest;
+import org.simantics.db.exception.DatabaseException;
+import org.simantics.db.procedure.AsyncProcedure;
+import org.simantics.diagram.adapter.DiagramClassAdapter;
+import org.simantics.diagram.synchronization.IModifiableSynchronizationContext;
+import org.simantics.diagram.synchronization.SynchronizationHints;
+import org.simantics.diagram.synchronization.graph.DiagramGraphUtil;
+import org.simantics.diagram.synchronization.graph.GraphSynchronizationHints;
+import org.simantics.diagram.synchronization.graph.layer.GraphLayer;
+import org.simantics.diagram.synchronization.graph.layer.GraphLayerManager;
+import org.simantics.g2d.diagram.DiagramClass;
+import org.simantics.g2d.diagram.IDiagram;
+import org.simantics.g2d.diagram.handler.SubstituteElementClass;
+import org.simantics.g2d.element.ElementClass;
+import org.simantics.g2d.element.ElementHints;
+import org.simantics.g2d.element.IElement;
+import org.simantics.g2d.element.handler.ElementLayerListener;
+import org.simantics.g2d.layers.ILayer;
+
+public class DistrictDiagramClassAdapter extends DiagramClassAdapter {
+
+    static final SubstituteElementClass DINSTANCE = new DistrictDiagramSubstituteElementClass();
+
+    @Override
+    public void adapt(AsyncReadGraph g, Resource source, Resource r, AsyncProcedure<DiagramClass> procedure) {
+        procedure.execute(g, INSTANCE.newClassWith(DINSTANCE));
+    }
+
+    static class DistrictDiagramElementLayerListenerImpl implements ElementLayerListener {
+
+        private static final long serialVersionUID = 7723435000661141433L;
+
+        @Override
+        public void visibilityChanged(IElement e, ILayer layer, boolean visible) {
+            IModifiableSynchronizationContext context = e.getDiagram().getHint(SynchronizationHints.CONTEXT);
+            GraphLayerManager layerManager = context.get(GraphSynchronizationHints.GRAPH_LAYER_MANAGER);
+            GraphLayer gl = layerManager.getGraphLayer(layer.getName());
+            if (gl != null) {
+                changeTag(e, gl.getVisible(), visible);
+            }
+        }
+
+        @Override
+        public void focusabilityChanged(IElement e, ILayer layer, boolean focusable) {
+            IModifiableSynchronizationContext context = e.getDiagram().getHint(SynchronizationHints.CONTEXT);
+            GraphLayerManager layerManager = context.get(GraphSynchronizationHints.GRAPH_LAYER_MANAGER);
+            GraphLayer gl = layerManager.getGraphLayer(layer.getName());
+            if (gl != null) {
+                changeTag(e, gl.getFocusable(), focusable);
+            }
+        }
+
+        void changeTag(IElement e, Resource tag, boolean set) {
+            Object object = e.getHint(ElementHints.KEY_OBJECT);
+            Resource tagged;
+            if (object instanceof Resource)
+                tagged = (Resource) object;
+            else
+                tagged = null;
+            if (tagged == null)
+                return;
+            
+            Simantics.getSession().asyncRequest(new WriteRequest() {
+                
+                @Override
+                public void perform(WriteGraph graph) throws DatabaseException {
+                    DiagramGraphUtil.tag(graph, tagged, tag, set);
+                }
+            });
+        }
+    }
+
+    static class DistrictDiagramSubstituteElementClass implements SubstituteElementClass {
+
+        static final ElementLayerListener LAYER_LISTENER = new DistrictDiagramElementLayerListenerImpl();
+
+        @Override
+        public ElementClass substitute(IDiagram d, ElementClass ec) {
+            return ec.newClassWith(LAYER_LISTENER);
+        }
+
+    }
+}