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