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