org.simantics.district.network.ui/src/org/simantics/district/network/ui/DistrictDiagramViewer.java

   1 package org.simantics.district.network.ui;
   2
   3 import java.awt.Color;
   4 import java.awt.geom.AffineTransform;
   5 import java.util.concurrent.TimeUnit;
   6 import java.util.function.Consumer;
   7 import java.util.function.Supplier;
   8
   9 import org.simantics.datatypes.literal.RGB;
  10 import org.simantics.db.ReadGraph;
  11 import org.simantics.db.Resource;
  12 import org.simantics.db.common.request.UnaryRead;
  13 import org.simantics.db.exception.DatabaseException;
  14 import org.simantics.db.procedure.Listener;
  15 import org.simantics.diagram.ui.DiagramModelHints;
  16 import org.simantics.district.network.DistrictNetworkUtil;
  17 import org.simantics.district.network.ui.participants.DNPointerInteractor;
  18 import org.simantics.district.network.ui.participants.ElevationServerParticipant;
  19 import org.simantics.district.network.ui.participants.MapRulerPainter;
  20 import org.simantics.g2d.canvas.ICanvasContext;
  21 import org.simantics.g2d.canvas.impl.CanvasContext;
  22 import org.simantics.g2d.diagram.handler.PickRequest.PickFilter;
  23 import org.simantics.g2d.diagram.participant.DelayedBatchElementPainter;
  24 import org.simantics.g2d.diagram.participant.ElementPainter;
  25 import org.simantics.g2d.diagram.participant.Selection;
  26 import org.simantics.g2d.diagram.participant.ZOrderHandler;
  27 import org.simantics.g2d.participant.BackgroundPainter;
  28 import org.simantics.g2d.participant.GridPainter;
  29 import org.simantics.g2d.participant.PanZoomRotateHandler;
  30 import org.simantics.g2d.participant.RenderingQualityInteractor;
  31 import org.simantics.g2d.participant.TransformUtil;
  32 import org.simantics.g2d.participant.ZoomToAreaHandler;
  33 import org.simantics.maps.MapScalingTransform;
  34 import org.simantics.maps.eclipse.MapPainter;
  35 import org.simantics.maps.sg.commands.MapCommands;
  36 import org.simantics.modeling.ui.diagramEditor.DiagramViewer;
  37 import org.simantics.scenegraph.g2d.events.command.CommandEvent;
  38 import org.simantics.scenegraph.g2d.events.command.Commands;
  39 import org.simantics.utils.datastructures.hints.IHintContext;
  40 import org.slf4j.Logger;
  41 import org.slf4j.LoggerFactory;
  42
  43 public class DistrictDiagramViewer extends DiagramViewer {
  44
  45     @SuppressWarnings("unused")
  46         private static final Logger LOGGER = LoggerFactory.getLogger(DistrictDiagramViewer.class);
  47
  48     @Override
  49     protected void addDiagramParticipants(ICanvasContext ctx) {
  50         ctx.add(new ZOrderHandler());
  51         ctx.add(new Selection());
  52         ctx.add(new ElementPainter());
  53         ctx.add(new DNPointerInteractor());
  54
  55         AffineTransform tr = new AffineTransform(MapScalingTransform.INSTANCE);
  56         ctx.add(new MapPainter(tr));
  57
  58         ctx.add(new NetworkDrawingParticipant(tr));
  59         ctx.add(new ElevationServerParticipant(tr));
  60     }
  61
  62     protected String getPopupId() {
  63         return "#DistrictDiagramPopup";
  64     }
  65
  66     @Override
  67     public void initializeCanvasContext(CanvasContext ctx) {
  68         super.initializeCanvasContext(ctx);
  69         IHintContext h = ctx.getDefaultHintContext();
  70         h.setHint(PanZoomRotateHandler.KEY_ZOOM_IN_LIMIT, 10000.0);
  71         h.setHint(PanZoomRotateHandler.KEY_ZOOM_OUT_LIMIT, 0.01);
  72         h.setHint(DiagramModelHints.KEY_DIAGRAM_RESOURCE, diagramResource);
  73     }
  74
  75     @Override
  76     protected void addPainterParticipants(ICanvasContext ctx) {
  77         ctx.add(new RenderingQualityInteractor());
  78         ctx.add(new DelayedBatchElementPainter(PickFilter.FILTER_MONITORS, 500, TimeUnit.MILLISECONDS));
  79     }
  80
  81     @Override
  82     protected void addGridRulerBackgroundParticipants(CanvasContext ctx) {
  83         ctx.add(new GridPainter());
  84         ctx.add(new MapRulerPainter());
  85         ctx.add(new BackgroundPainter());
  86     }
  87
  88     protected void addViewManipulationParticipants(CanvasContext ctx) {
  89         // Let's replace with our special util
  90         TransformUtil util = ctx.getAtMostOneItemOfClass(TransformUtil.class);
  91         if (util != null)
  92             ctx.remove(util);
  93         ctx.add(new DistrictTransformUtil());
  94         ctx.add(new DistrictPanZoomRotateHandler());
  95         //ctx.add(new MousePanZoomInteractor());
  96         //ctx.add(new MultitouchPanZoomRotateInteractor());
  97         // ctx.add( new OrientationRestorer() );
  98         ctx.add(new ZoomToAreaHandler());
  99     }
 100
 101     @Override
 102     protected void loadPageSettings(ICanvasContext ctx) {
 103         super.loadPageSettings(ctx);
 104         // this might be the wrong place to start such listening but at least
 105         // super.loadPageSettings() does async-db-operations
 106         setupDrawMapEnabled();
 107         setupBackgroundColor();
 108     }
 109
 110     private void setupDrawMapEnabled() {
 111         sessionContext.getSession().asyncRequest(new DrawMapEnabledRequest(getInputResource()), new DrawMapEnabledListener(
 112                 result -> canvasContext.getEventQueue().queueEvent(new CommandEvent(canvasContext, System.currentTimeMillis(), result ? Commands.MAP_ENABLE : Commands.MAP_DISABLE)),
 113                 () -> DistrictDiagramViewer.this.isDisposed()));
 114     }
 115
 116     private void setupBackgroundColor() {
 117         sessionContext.getSession().asyncRequest(new MapBackgroundColorRequest(getInputResource()), new MapBackgroundColorListener(
 118                 result -> queueBackgroundColorChangeEvent(result),
 119                 () -> DistrictDiagramViewer.this.isDisposed()));
 120     }
 121
 122     private void queueBackgroundColorChangeEvent(RGB.Integer result) {
 123         if (result != null) {
 124             Color backgroundColor = new Color(result.red, result.green, result.blue);
 125             canvasContext.getDefaultHintContext().setHint(MapCommands.KEY_MAP_BACKGROUND_COLOR, backgroundColor);
 126             canvasContext.getEventQueue().queueEvent(new CommandEvent(canvasContext, System.currentTimeMillis(), MapCommands.MAP_BACKGROUND_COLOR_CHANGE));
 127         }
 128     }
 129
 130     private static class DrawMapEnabledRequest extends UnaryRead<Resource, Boolean> {
 131
 132         public DrawMapEnabledRequest(Resource diagram) {
 133             super(diagram);
 134         }
 135
 136         @Override
 137         public Boolean perform(ReadGraph graph) throws DatabaseException {
 138             return DistrictNetworkUtil.drawMapEnabled(graph, parameter);
 139         }
 140     }
 141
 142     private static class DrawMapEnabledListener implements Listener<Boolean> {
 143
 144         private static final Logger LOGGER = LoggerFactory.getLogger(DrawMapEnabledListener.class);
 145
 146         private Consumer<Boolean> callback;
 147         private Supplier<Boolean> isDisposed;
 148
 149         public DrawMapEnabledListener(Consumer<Boolean> callback, Supplier<Boolean> isDisposed) {
 150             this.callback = callback;
 151             this.isDisposed = isDisposed;
 152         }
 153
 154         @Override
 155         public void execute(Boolean result) {
 156             callback.accept(result);
 157         }
 158
 159         @Override
 160         public void exception(Throwable t) {
 161             LOGGER.error("Could not listen if draw map is enabled", t);
 162         }
 163
 164         @Override
 165         public boolean isDisposed() {
 166             return isDisposed.get();
 167         }
 168     }
 169
 170     private static class MapBackgroundColorRequest extends UnaryRead<Resource, RGB.Integer> {
 171
 172         public MapBackgroundColorRequest(Resource diagram) {
 173             super(diagram);
 174         }
 175
 176         @Override
 177         public RGB.Integer perform(ReadGraph graph) throws DatabaseException {
 178             return DistrictNetworkUtil.backgroundColor(graph, parameter);
 179         }
 180     }
 181
 182     private static class MapBackgroundColorListener implements Listener<RGB.Integer> {
 183
 184         private static final Logger LOGGER = LoggerFactory.getLogger(DrawMapEnabledListener.class);
 185
 186         private Consumer<RGB.Integer> callback;
 187         private Supplier<Boolean> isDisposed;
 188
 189         public MapBackgroundColorListener(Consumer<RGB.Integer> callback, Supplier<Boolean> isDisposed) {
 190             this.callback = callback;
 191             this.isDisposed = isDisposed;
 192         }
 193
 194         @Override
 195         public void execute(RGB.Integer result) {
 196             callback.accept(result);
 197         }
 198
 199         @Override
 200         public void exception(Throwable t) {
 201             LOGGER.error("Could not listen map background color", t);
 202         }
 203
 204         @Override
 205         public boolean isDisposed() {
 206             return isDisposed.get();
 207         }
 208     }
 209 }