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