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.Map;
   6 import java.util.concurrent.TimeUnit;
   7 import java.util.function.Consumer;
   8 import java.util.function.Supplier;
   9
  10 import org.simantics.datatypes.literal.RGB;
  11 import org.simantics.db.ReadGraph;
  12 import org.simantics.db.Resource;
  13 import org.simantics.db.common.request.UnaryRead;
  14 import org.simantics.db.exception.DatabaseException;
  15 import org.simantics.db.procedure.Listener;
  16 import org.simantics.diagram.ui.DiagramModelHints;
  17 import org.simantics.district.network.DistrictNetworkUtil;
  18 import org.simantics.district.network.ui.participants.DNPointerInteractor;
  19 import org.simantics.district.network.ui.participants.DynamicVisualisationContributionsParticipant;
  20 import org.simantics.district.network.ui.participants.ElevationServerParticipant;
  21 import org.simantics.district.network.ui.participants.MapRulerPainter;
  22 import org.simantics.district.network.visualisations.model.ColorBarOptions;
  23 import org.simantics.district.network.visualisations.model.DynamicColorContribution;
  24 import org.simantics.district.network.visualisations.model.DynamicSizeContribution;
  25 import org.simantics.district.network.visualisations.model.SizeBarOptions;
  26 import org.simantics.g2d.canvas.ICanvasContext;
  27 import org.simantics.g2d.canvas.impl.CanvasContext;
  28 import org.simantics.g2d.diagram.handler.PickRequest.PickFilter;
  29 import org.simantics.g2d.diagram.participant.DelayedBatchElementPainter;
  30 import org.simantics.g2d.diagram.participant.ElementPainter;
  31 import org.simantics.g2d.diagram.participant.Selection;
  32 import org.simantics.g2d.diagram.participant.ZOrderHandler;
  33 import org.simantics.g2d.participant.BackgroundPainter;
  34 import org.simantics.g2d.participant.GridPainter;
  35 import org.simantics.g2d.participant.PanZoomRotateHandler;
  36 import org.simantics.g2d.participant.RenderingQualityInteractor;
  37 import org.simantics.g2d.participant.TransformUtil;
  38 import org.simantics.g2d.participant.ZoomToAreaHandler;
  39 import org.simantics.maps.MapScalingTransform;
  40 import org.simantics.maps.eclipse.MapPainter;
  41 import org.simantics.maps.sg.commands.MapCommands;
  42 import org.simantics.modeling.ui.diagramEditor.DiagramViewer;
  43 import org.simantics.scenegraph.g2d.events.command.Command;
  44 import org.simantics.scenegraph.g2d.events.command.CommandEvent;
  45 import org.simantics.scenegraph.g2d.events.command.Commands;
  46 import org.simantics.utils.datastructures.hints.IHintContext;
  47 import org.simantics.utils.datastructures.hints.IHintContext.Key;
  48 import org.simantics.utils.datastructures.hints.IHintContext.KeyOf;
  49 import org.slf4j.Logger;
  50 import org.slf4j.LoggerFactory;
  51
  52 public class DistrictDiagramViewer extends DiagramViewer {
  53
  54     @SuppressWarnings("unused")
  55         private static final Logger LOGGER = LoggerFactory.getLogger(DistrictDiagramViewer.class);
  56
  57     @Override
  58     protected void addDiagramParticipants(ICanvasContext ctx) {
  59         ctx.add(new ZOrderHandler());
  60         ctx.add(new Selection());
  61         ctx.add(new ElementPainter());
  62         ctx.add(new DNPointerInteractor());
  63
  64         AffineTransform tr = new AffineTransform(MapScalingTransform.INSTANCE);
  65         ctx.add(new MapPainter(tr));
  66
  67         ctx.add(new NetworkDrawingParticipant(tr));
  68         ctx.add(new ElevationServerParticipant(tr));
  69         ctx.add(new DynamicVisualisationContributionsParticipant(tr));
  70     }
  71
  72     protected String getPopupId() {
  73         return "#DistrictDiagramPopup";
  74     }
  75
  76     @Override
  77     public void initializeCanvasContext(CanvasContext ctx) {
  78         super.initializeCanvasContext(ctx);
  79         IHintContext h = ctx.getDefaultHintContext();
  80         h.setHint(PanZoomRotateHandler.KEY_ZOOM_IN_LIMIT, 10000.0);
  81         h.setHint(PanZoomRotateHandler.KEY_ZOOM_OUT_LIMIT, 0.01);
  82         h.setHint(DiagramModelHints.KEY_DIAGRAM_RESOURCE, diagramResource);
  83     }
  84
  85     @Override
  86     protected void addPainterParticipants(ICanvasContext ctx) {
  87         ctx.add(new RenderingQualityInteractor());
  88         ctx.add(new DelayedBatchElementPainter(PickFilter.FILTER_MONITORS, 500, TimeUnit.MILLISECONDS));
  89     }
  90
  91     @Override
  92     protected void addGridRulerBackgroundParticipants(CanvasContext ctx) {
  93         ctx.add(new GridPainter());
  94         ctx.add(new MapRulerPainter());
  95         ctx.add(new BackgroundPainter());
  96     }
  97
  98     protected void addViewManipulationParticipants(CanvasContext ctx) {
  99         // Let's replace with our special util
 100         TransformUtil util = ctx.getAtMostOneItemOfClass(TransformUtil.class);
 101         if (util != null)
 102             ctx.remove(util);
 103         ctx.add(new DistrictTransformUtil());
 104         ctx.add(new DistrictPanZoomRotateHandler());
 105         //ctx.add(new MousePanZoomInteractor());
 106         //ctx.add(new MultitouchPanZoomRotateInteractor());
 107         // ctx.add( new OrientationRestorer() );
 108         ctx.add(new ZoomToAreaHandler());
 109     }
 110
 111     @Override
 112     protected void loadPageSettings(ICanvasContext ctx) {
 113         super.loadPageSettings(ctx);
 114         // this might be the wrong place to start such listening but at least
 115         // super.loadPageSettings() does async-db-operations
 116         setupDrawMapEnabled();
 117         setupBackgroundColor();
 118         setupColoringObjects();
 119         setupColorBarOptions();
 120         setupSizingObjects();
 121         setupSizeBarOptions();
 122     }
 123
 124     private void setupDrawMapEnabled() {
 125         sessionContext.getSession().asyncRequest(new DrawMapEnabledRequest(getInputResource()), new DrawMapEnabledListener(
 126                 result -> canvasContext.getEventQueue().queueEvent(new CommandEvent(canvasContext, System.currentTimeMillis(), result ? Commands.MAP_ENABLE : Commands.MAP_DISABLE)),
 127                 () -> DistrictDiagramViewer.this.isDisposed()));
 128     }
 129
 130     private void setupBackgroundColor() {
 131         sessionContext.getSession().asyncRequest(new MapBackgroundColorRequest(getInputResource()), new MapBackgroundColorListener(
 132                 result -> queueBackgroundColorChangeEvent(result),
 133                 () -> DistrictDiagramViewer.this.isDisposed()));
 134     }
 135
 136     private void queueBackgroundColorChangeEvent(RGB.Integer result) {
 137         if (result != null) {
 138             Color backgroundColor = new Color(result.red, result.green, result.blue);
 139             canvasContext.getDefaultHintContext().setHint(MapCommands.KEY_MAP_BACKGROUND_COLOR, backgroundColor);
 140             canvasContext.getEventQueue().queueEvent(new CommandEvent(canvasContext, System.currentTimeMillis(), MapCommands.MAP_BACKGROUND_COLOR_CHANGE));
 141         }
 142     }
 143
 144     private void setupColoringObjects() {
 145         sessionContext.getSession().asyncRequest(new ColoringObjectsRequest(getInputResource()), new ColoringObjectsListener(
 146                 result -> queueColoringObjectsChangeEvent(result),
 147                 () -> DistrictDiagramViewer.this.isDisposed()));
 148     }
 149
 150
 151     private void setupColorBarOptions() {
 152         sessionContext.getSession().asyncRequest(new ColorBarOptionsRequest(getInputResource()), new ColorBarOptionsListener(
 153                 result -> queueColorBarOptionsChangeEvent(result),
 154                 () -> DistrictDiagramViewer.this.isDisposed()));
 155     }
 156
 157     private void setupSizingObjects() {
 158         sessionContext.getSession().asyncRequest(new SizingObjectsRequest(getInputResource()), new SizingObjectsListener(
 159                 result -> queueSizingObjectsChangeEvent(result),
 160                 () -> DistrictDiagramViewer.this.isDisposed()));
 161     }
 162
 163
 164     private void setupSizeBarOptions() {
 165         sessionContext.getSession().asyncRequest(new SizeBarOptionsRequest(getInputResource()), new SizeBarOptionsListener(
 166                 result -> queueSizeBarOptionsChangeEvent(result),
 167                 () -> DistrictDiagramViewer.this.isDisposed()));
 168     }
 169
 170     public static final Key KEY_MAP_COLOR_BAR_OPTIONS = new KeyOf(ColorBarOptions.class, "colorBarOptions");
 171     public static final Command MAP_COLOR_BAR_OPTIONS_CHANGE = new Command("colorBarOptionsChange");
 172     public static final Key KEY_MAP_SIZE_BAR_OPTIONS = new KeyOf(SizeBarOptions.class, "sizeBarOptions");
 173     public static final Command MAP_SIZE_BAR_OPTIONS_CHANGE = new Command("sizeBarOptionsChange");
 174
 175     public static final Key KEY_MAP_COLORING_OBJECTS = new KeyOf(Map.class, "coloringObjects");
 176     public static final Command MAP_COLORING_OBJECTS_CHANGE = new Command("coloringObjectsChange");
 177
 178     public static final Key KEY_MAP_SIZING_OBJECTS = new KeyOf(Map.class, "sizingObjects");
 179     public static final Command MAP_SIZING_OBJECTS_CHANGE = new Command("sizingObjectsChange");
 180
 181
 182     private void queueColoringObjectsChangeEvent(Map<String, DynamicColorContribution> result) {
 183         if (result != null) {
 184             canvasContext.getDefaultHintContext().setHint(KEY_MAP_COLORING_OBJECTS, result);
 185             canvasContext.getEventQueue().queueEvent(new CommandEvent(canvasContext, System.currentTimeMillis(), MAP_COLORING_OBJECTS_CHANGE));
 186         }
 187     }
 188
 189     private void queueColorBarOptionsChangeEvent(ColorBarOptions result) {
 190         if (result != null) {
 191             canvasContext.getDefaultHintContext().setHint(KEY_MAP_COLOR_BAR_OPTIONS, result);
 192             canvasContext.getEventQueue().queueEvent(new CommandEvent(canvasContext, System.currentTimeMillis(), MAP_COLOR_BAR_OPTIONS_CHANGE));
 193         }
 194     }
 195
 196     private void queueSizingObjectsChangeEvent(Map<String, DynamicSizeContribution> result) {
 197         if (result != null) {
 198             canvasContext.getDefaultHintContext().setHint(KEY_MAP_SIZING_OBJECTS, result);
 199             canvasContext.getEventQueue().queueEvent(new CommandEvent(canvasContext, System.currentTimeMillis(), MAP_SIZING_OBJECTS_CHANGE));
 200         }
 201     }
 202
 203     private void queueSizeBarOptionsChangeEvent(SizeBarOptions result) {
 204         if (result != null) {
 205             canvasContext.getDefaultHintContext().setHint(KEY_MAP_SIZE_BAR_OPTIONS, result);
 206             canvasContext.getEventQueue().queueEvent(new CommandEvent(canvasContext, System.currentTimeMillis(), MAP_SIZE_BAR_OPTIONS_CHANGE));
 207         }
 208     }
 209
 210     private static class DrawMapEnabledRequest extends UnaryRead<Resource, Boolean> {
 211
 212         public DrawMapEnabledRequest(Resource diagram) {
 213             super(diagram);
 214         }
 215
 216         @Override
 217         public Boolean perform(ReadGraph graph) throws DatabaseException {
 218             return DistrictNetworkUtil.drawMapEnabled(graph, parameter);
 219         }
 220     }
 221
 222     private static class DrawMapEnabledListener implements Listener<Boolean> {
 223
 224         private static final Logger LOGGER = LoggerFactory.getLogger(DrawMapEnabledListener.class);
 225
 226         private Consumer<Boolean> callback;
 227         private Supplier<Boolean> isDisposed;
 228
 229         public DrawMapEnabledListener(Consumer<Boolean> callback, Supplier<Boolean> isDisposed) {
 230             this.callback = callback;
 231             this.isDisposed = isDisposed;
 232         }
 233
 234         @Override
 235         public void execute(Boolean result) {
 236             callback.accept(result);
 237         }
 238
 239         @Override
 240         public void exception(Throwable t) {
 241             LOGGER.error("Could not listen if draw map is enabled", t);
 242         }
 243
 244         @Override
 245         public boolean isDisposed() {
 246             return isDisposed.get();
 247         }
 248     }
 249
 250     private static class MapBackgroundColorRequest extends UnaryRead<Resource, RGB.Integer> {
 251
 252         public MapBackgroundColorRequest(Resource diagram) {
 253             super(diagram);
 254         }
 255
 256         @Override
 257         public RGB.Integer perform(ReadGraph graph) throws DatabaseException {
 258             return DistrictNetworkUtil.backgroundColor(graph, parameter);
 259         }
 260     }
 261
 262     private static class MapBackgroundColorListener implements Listener<RGB.Integer> {
 263
 264         private static final Logger LOGGER = LoggerFactory.getLogger(DrawMapEnabledListener.class);
 265
 266         private Consumer<RGB.Integer> callback;
 267         private Supplier<Boolean> isDisposed;
 268
 269         public MapBackgroundColorListener(Consumer<RGB.Integer> callback, Supplier<Boolean> isDisposed) {
 270             this.callback = callback;
 271             this.isDisposed = isDisposed;
 272         }
 273
 274         @Override
 275         public void execute(RGB.Integer result) {
 276             callback.accept(result);
 277         }
 278
 279         @Override
 280         public void exception(Throwable t) {
 281             LOGGER.error("Could not listen map background color", t);
 282         }
 283
 284         @Override
 285         public boolean isDisposed() {
 286             return isDisposed.get();
 287         }
 288     }
 289
 290     private static class ColorBarOptionsRequest extends UnaryRead<Resource, ColorBarOptions> {
 291
 292         public ColorBarOptionsRequest(Resource diagram) {
 293             super(diagram);
 294         }
 295
 296         @Override
 297         public ColorBarOptions perform(ReadGraph graph) throws DatabaseException {
 298             return DistrictNetworkUtil.colorBarOptions(graph, parameter);
 299         }
 300     }
 301
 302     private static class ColoringObjectsRequest extends UnaryRead<Resource, Map<String,DynamicColorContribution>> {
 303
 304         public ColoringObjectsRequest(Resource diagram) {
 305             super(diagram);
 306         }
 307
 308         @Override
 309         public Map<String, DynamicColorContribution> perform(ReadGraph graph) throws DatabaseException {
 310             return DistrictNetworkUtil.colorContributions(graph, parameter);
 311         }
 312     }
 313
 314     private static class ColoringObjectsListener implements Listener<Map<String,DynamicColorContribution>> {
 315
 316         private static final Logger LOGGER = LoggerFactory.getLogger(DrawMapEnabledListener.class);
 317
 318         private Consumer<Map<String,DynamicColorContribution>> callback;
 319         private Supplier<Boolean> isDisposed;
 320
 321         public ColoringObjectsListener(Consumer<Map<String,DynamicColorContribution>> callback, Supplier<Boolean> isDisposed) {
 322             this.callback = callback;
 323             this.isDisposed = isDisposed;
 324         }
 325
 326         @Override
 327         public void execute(Map<String,DynamicColorContribution> result) {
 328             callback.accept(result);
 329         }
 330
 331         @Override
 332         public void exception(Throwable t) {
 333             LOGGER.error("Could not listen ColorBarOptions", t);
 334         }
 335
 336         @Override
 337         public boolean isDisposed() {
 338             return isDisposed.get();
 339         }
 340     }
 341
 342     private static class ColorBarOptionsListener implements Listener<ColorBarOptions> {
 343
 344         private static final Logger LOGGER = LoggerFactory.getLogger(DrawMapEnabledListener.class);
 345
 346         private Consumer<ColorBarOptions> callback;
 347         private Supplier<Boolean> isDisposed;
 348
 349         public ColorBarOptionsListener(Consumer<ColorBarOptions> callback, Supplier<Boolean> isDisposed) {
 350             this.callback = callback;
 351             this.isDisposed = isDisposed;
 352         }
 353
 354         @Override
 355         public void execute(ColorBarOptions result) {
 356             callback.accept(result);
 357         }
 358
 359         @Override
 360         public void exception(Throwable t) {
 361             LOGGER.error("Could not listen ColorBarOptions", t);
 362         }
 363
 364         @Override
 365         public boolean isDisposed() {
 366             return isDisposed.get();
 367         }
 368     }
 369
 370     private static class SizeBarOptionsRequest extends UnaryRead<Resource, SizeBarOptions> {
 371
 372         public SizeBarOptionsRequest(Resource diagram) {
 373             super(diagram);
 374         }
 375
 376         @Override
 377         public SizeBarOptions perform(ReadGraph graph) throws DatabaseException {
 378             return DistrictNetworkUtil.sizeBarOptions(graph, parameter);
 379         }
 380     }
 381
 382     private static class SizeBarOptionsListener implements Listener<SizeBarOptions> {
 383
 384         private static final Logger LOGGER = LoggerFactory.getLogger(DrawMapEnabledListener.class);
 385
 386         private Consumer<SizeBarOptions> callback;
 387         private Supplier<Boolean> isDisposed;
 388
 389         public SizeBarOptionsListener(Consumer<SizeBarOptions> callback, Supplier<Boolean> isDisposed) {
 390             this.callback = callback;
 391             this.isDisposed = isDisposed;
 392         }
 393
 394         @Override
 395         public void execute(SizeBarOptions result) {
 396             callback.accept(result);
 397         }
 398
 399         @Override
 400         public void exception(Throwable t) {
 401             LOGGER.error("Could not listen SizeBarOptions", t);
 402         }
 403
 404         @Override
 405         public boolean isDisposed() {
 406             return isDisposed.get();
 407         }
 408     }
 409
 410     private static class SizingObjectsRequest extends UnaryRead<Resource, Map<String, DynamicSizeContribution>> {
 411
 412         public SizingObjectsRequest(Resource diagram) {
 413             super(diagram);
 414         }
 415
 416         @Override
 417         public Map<String, DynamicSizeContribution> perform(ReadGraph graph) throws DatabaseException {
 418             return DistrictNetworkUtil.sizeContributions(graph, parameter);
 419         }
 420     }
 421
 422     private static class SizingObjectsListener implements Listener<Map<String,DynamicSizeContribution>> {
 423
 424         private static final Logger LOGGER = LoggerFactory.getLogger(DrawMapEnabledListener.class);
 425
 426         private Consumer<Map<String,DynamicSizeContribution>> callback;
 427         private Supplier<Boolean> isDisposed;
 428
 429         public SizingObjectsListener(Consumer<Map<String, DynamicSizeContribution>> callback, Supplier<Boolean> isDisposed) {
 430             this.callback = callback;
 431             this.isDisposed = isDisposed;
 432         }
 433
 434         @Override
 435         public void execute(Map<String, DynamicSizeContribution> result) {
 436             callback.accept(result);
 437         }
 438
 439         @Override
 440         public void exception(Throwable t) {
 441             LOGGER.error("Could not listen ColorBarOptions", t);
 442         }
 443
 444         @Override
 445         public boolean isDisposed() {
 446             return isDisposed.get();
 447         }
 448     }
 449 }