Enabling/Disabling map painting

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

diff --git a/org.simantics.district.network.ui/src/org/simantics/district/network/ui/DistrictDiagramViewer.java b/org.simantics.district.network.ui/src/org/simantics/district/network/ui/DistrictDiagramViewer.java
index f3e556718202027ce966b2bab9c1f1e86414461e..c6c510e0a59d3546743ce68c0bd603003f599d91 100644 (file)
--- a/org.simantics.district.network.ui/src/org/simantics/district/network/ui/DistrictDiagramViewer.java
+++ b/org.simantics.district.network.ui/src/org/simantics/district/network/ui/DistrictDiagramViewer.java
@@ -2,6 +2,11 @@ package org.simantics.district.network.ui;
 
 import java.util.concurrent.TimeUnit;
 
+import org.simantics.db.ReadGraph;
+import org.simantics.db.common.request.UniqueRead;
+import org.simantics.db.exception.DatabaseException;
+import org.simantics.db.procedure.Listener;
+import org.simantics.district.network.DistrictNetworkUtil;
 import org.simantics.district.network.ui.participants.DNPointerInteractor;
 import org.simantics.district.network.ui.participants.MapRulerPainter;
 import org.simantics.g2d.canvas.ICanvasContext;
@@ -19,10 +24,16 @@ import org.simantics.g2d.participant.ZoomToAreaHandler;
 import org.simantics.maps.MapScalingTransform;
 import org.simantics.maps.eclipse.MapPainter;
 import org.simantics.modeling.ui.diagramEditor.DiagramViewer;
+import org.simantics.scenegraph.g2d.events.command.CommandEvent;
+import org.simantics.scenegraph.g2d.events.command.Commands;
 import org.simantics.utils.datastructures.hints.IHintContext;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 
 public class DistrictDiagramViewer extends DiagramViewer {
 
+    private static final Logger LOGGER = LoggerFactory.getLogger(DistrictDiagramViewer.class);
+
     @Override
     protected void addDiagramParticipants(ICanvasContext ctx) {
         ctx.add(new ZOrderHandler());
@@ -68,4 +79,33 @@ public class DistrictDiagramViewer extends DiagramViewer {
         ctx.add(new ZoomToAreaHandler());
     }
 
+    @Override
+    protected void loadPageSettings(ICanvasContext ctx) {
+        super.loadPageSettings(ctx);
+        // this might be the wrong place to start such listening but at least
+        // super.loadPageSettings() does async-db-operations
+        sessionContext.getSession().asyncRequest(new UniqueRead<Boolean>() {
+
+            @Override
+            public Boolean perform(ReadGraph graph) throws DatabaseException {
+                return DistrictNetworkUtil.drawMapEnabled(graph, getInputResource());
+            }
+        }, new Listener<Boolean>() {
+
+            @Override
+            public void execute(Boolean result) {
+                canvasContext.getEventQueue().queueEvent(new CommandEvent(canvasContext, System.currentTimeMillis(), result ? Commands.MAP_ENABLE : Commands.MAP_DISABLE));
+            }
+
+            @Override
+            public void exception(Throwable t) {
+                LOGGER.error("Could not listen draw map for {}", getInputResource(), t);
+            }
+
+            @Override
+            public boolean isDisposed() {
+                return DistrictDiagramViewer.this.isDisposed();
+            }
+        });
+    }
 }