1 package org.simantics.district.network.ui;
4 import java.awt.geom.AffineTransform;
5 import java.util.Collections;
7 import java.util.Objects;
8 import java.util.concurrent.TimeUnit;
9 import java.util.function.Consumer;
10 import java.util.function.Supplier;
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.PossibleIndexRoot;
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.internal.Activator;
23 import org.simantics.district.network.ui.nodes.DistrictRenderingPreparationNode;
24 import org.simantics.district.network.ui.nodes.DistrictSelectionNode;
25 import org.simantics.district.network.ui.participants.DNPointerInteractor;
26 import org.simantics.district.network.ui.participants.DynamicVisualisationContributionsParticipant;
27 import org.simantics.district.network.ui.participants.MapRulerPainter;
28 import org.simantics.district.network.visualisations.DynamicVisualisations;
29 import org.simantics.district.network.visualisations.model.ColorBarOptions;
30 import org.simantics.district.network.visualisations.model.DynamicColorContribution;
31 import org.simantics.district.network.visualisations.model.DynamicSizeContribution;
32 import org.simantics.district.network.visualisations.model.SizeBarOptions;
33 import org.simantics.g2d.canvas.ICanvasContext;
34 import org.simantics.g2d.canvas.impl.CanvasContext;
35 import org.simantics.g2d.diagram.DiagramHints;
36 import org.simantics.g2d.diagram.handler.PickRequest.PickFilter;
37 import org.simantics.g2d.diagram.participant.DelayedBatchElementPainter;
38 import org.simantics.g2d.diagram.participant.ElementPainter;
39 import org.simantics.g2d.diagram.participant.ElementPainterConfiguration;
40 import org.simantics.g2d.diagram.participant.Selection;
41 import org.simantics.g2d.diagram.participant.ZOrderHandler;
42 import org.simantics.g2d.participant.BackgroundPainter;
43 import org.simantics.g2d.participant.GridPainter;
44 import org.simantics.g2d.participant.PanZoomRotateHandler;
45 import org.simantics.g2d.participant.RenderingQualityInteractor;
46 import org.simantics.g2d.participant.TransformUtil;
47 import org.simantics.g2d.participant.ZoomToAreaHandler;
48 import org.simantics.g2d.scenegraph.SceneGraphConstants;
49 import org.simantics.maps.MapScalingTransform;
50 import org.simantics.maps.eclipse.MapPainter;
51 import org.simantics.maps.sg.commands.MapCommands;
52 import org.simantics.modeling.ui.diagramEditor.DiagramViewer;
53 import org.simantics.scenegraph.g2d.G2DParentNode;
54 import org.simantics.scenegraph.g2d.events.command.Command;
55 import org.simantics.scenegraph.g2d.events.command.CommandEvent;
56 import org.simantics.scenegraph.g2d.events.command.Commands;
57 import org.simantics.utils.datastructures.hints.IHintContext;
58 import org.simantics.utils.datastructures.hints.IHintContext.Key;
59 import org.simantics.utils.datastructures.hints.IHintContext.KeyOf;
60 import org.slf4j.Logger;
61 import org.slf4j.LoggerFactory;
63 public class DistrictDiagramViewer extends DiagramViewer {
65 private static final Logger LOGGER = LoggerFactory.getLogger(DistrictDiagramViewer.class);
68 protected void addDiagramParticipants(ICanvasContext ctx) {
69 ctx.add(new ZOrderHandler());
70 ctx.add(new Selection());
71 ctx.add(new ElementPainter(new ElementPainterConfiguration().selectionNodeClass(DistrictSelectionNode.class)));
72 ctx.add(new DNPointerInteractor());
74 AffineTransform tr = new AffineTransform(MapScalingTransform.INSTANCE);
75 ctx.add(new MapPainter(tr));
77 DynamicVisualisationContributionsParticipant dynamicVisualisationContributionsParticipant = new DynamicVisualisationContributionsParticipant(tr);
78 ctx.add(new NetworkDrawingParticipant(dynamicVisualisationContributionsParticipant, tr));
79 ctx.add(dynamicVisualisationContributionsParticipant);
81 // Optimize AffineTransform memory allocations during district diagram rendering
82 G2DParentNode spatialRoot = (G2DParentNode) ctx.getSceneGraph().lookupNode(SceneGraphConstants.SPATIAL_ROOT_NODE_ID);
83 DistrictRenderingPreparationNode prepNode = new DistrictRenderingPreparationNode();
84 prepNode.setZIndex(Integer.MIN_VALUE / 2);
85 spatialRoot.addNode("districtRenderingPrepareNode", prepNode);
88 protected String getPopupId() {
89 return "#DistrictDiagramPopup";
93 protected void fillInitialDiagramHints(Resource diagram, IHintContext initialHints) throws DatabaseException {
94 super.fillInitialDiagramHints(diagram, initialHints);
99 public void initializeCanvasContext(CanvasContext ctx) {
100 super.initializeCanvasContext(ctx);
101 IHintContext h = ctx.getDefaultHintContext();
102 h.setHint(PanZoomRotateHandler.KEY_ZOOM_IN_LIMIT, 10000.0);
103 h.setHint(PanZoomRotateHandler.KEY_ZOOM_OUT_LIMIT, 0.01);
104 h.setHint(DiagramModelHints.KEY_DIAGRAM_RESOURCE, diagramResource);
105 h.removeHint(DiagramHints.SNAP_ADVISOR);
109 protected void addPainterParticipants(ICanvasContext ctx) {
110 ctx.add(new RenderingQualityInteractor());
111 ctx.add(new DelayedBatchElementPainter(PickFilter.FILTER_MONITORS, 500, TimeUnit.MILLISECONDS));
115 protected void addGridRulerBackgroundParticipants(CanvasContext ctx) {
116 ctx.add(new GridPainter());
117 ctx.add(new MapRulerPainter());
118 ctx.add(new BackgroundPainter());
121 protected void addViewManipulationParticipants(CanvasContext ctx) {
122 // Let's replace with our special util
123 TransformUtil util = ctx.getAtMostOneItemOfClass(TransformUtil.class);
126 ctx.add(new DistrictTransformUtil());
127 ctx.add(new DistrictPanZoomRotateHandler());
128 //ctx.add(new MousePanZoomInteractor());
129 //ctx.add(new MultitouchPanZoomRotateInteractor());
130 // ctx.add( new OrientationRestorer() );
131 ctx.add(new ZoomToAreaHandler());
135 protected void loadPageSettings(ICanvasContext ctx) {
136 super.loadPageSettings(ctx);
137 // this might be the wrong place to start such listening but at least
138 // super.loadPageSettings() does async-db-operations
139 setupDrawMapEnabled();
140 setupBackgroundColor();
141 setupColoringObjects();
142 setupColorBarOptions();
143 setupSizingObjects();
144 setupSizeBarOptions();
145 setupShowElevationServerBoundingBox();
148 DistrictDiagramViewerListener[] listeners = Activator.getInstance().getDistrictDiagramViewerListeners();
149 if (listeners != null) {
150 for (DistrictDiagramViewerListener listener : listeners) {
151 listener.diagramLoaded(getRuntime(), canvasContext);
157 public void dispose() {
158 DistrictDiagramViewerListener[] listeners = Activator.getInstance().getDistrictDiagramViewerListeners();
159 if (listeners != null) {
160 Resource runtime = getRuntime();
161 for (DistrictDiagramViewerListener listener : listeners) {
162 listener.diagramDisposed(runtime, canvasContext);
168 private void setupDrawMapEnabled() {
169 sessionContext.getSession().asyncRequest(new DrawMapEnabledRequest(getInputResource()), new DrawMapEnabledListener(
170 result -> canvasContext.getEventQueue().queueEvent(new CommandEvent(canvasContext, System.currentTimeMillis(), result ? Commands.MAP_ENABLE : Commands.MAP_DISABLE)),
171 () -> DistrictDiagramViewer.this.isDisposed()));
174 private void setupBackgroundColor() {
175 sessionContext.getSession().asyncRequest(new MapBackgroundColorRequest(getInputResource()), new MapBackgroundColorListener(
176 result -> queueBackgroundColorChangeEvent(result),
177 () -> DistrictDiagramViewer.this.isDisposed()));
180 private void queueBackgroundColorChangeEvent(RGB.Integer result) {
181 if (result != null) {
182 Color backgroundColor = new Color(result.red, result.green, result.blue);
183 canvasContext.getDefaultHintContext().setHint(MapCommands.KEY_MAP_BACKGROUND_COLOR, backgroundColor);
184 canvasContext.getEventQueue().queueEvent(new CommandEvent(canvasContext, System.currentTimeMillis(), MapCommands.MAP_BACKGROUND_COLOR_CHANGE));
188 private void setupColoringObjects() {
189 sessionContext.getSession().asyncRequest(new ColoringObjectsRequest(getInputResource()), new ColoringObjectsListener(
190 result -> queueColoringObjectsChangeEvent(result),
191 () -> DistrictDiagramViewer.this.isDisposed()));
195 private void setupColorBarOptions() {
196 sessionContext.getSession().asyncRequest(new ColorBarOptionsRequest(getInputResource()), new ColorBarOptionsListener(
197 result -> queueColorBarOptionsChangeEvent(result),
198 () -> DistrictDiagramViewer.this.isDisposed()));
201 private void setupSizingObjects() {
202 sessionContext.getSession().asyncRequest(new SizingObjectsRequest(getInputResource()), new SizingObjectsListener(
203 result -> queueSizingObjectsChangeEvent(result),
204 () -> DistrictDiagramViewer.this.isDisposed()));
207 private void setupSizeBarOptions() {
208 sessionContext.getSession().asyncRequest(new SizeBarOptionsRequest(getInputResource()), new SizeBarOptionsListener(
209 result -> queueSizeBarOptionsChangeEvent(result),
210 () -> DistrictDiagramViewer.this.isDisposed()));
213 private void setupShowElevationServerBoundingBox() {
214 sessionContext.getSession().asyncRequest(new ShowElevationServerRequest(getInputResource()), new ShowElevationServerListener(
215 result -> queueShowElevationServerChangeEvent(result),
216 () -> DistrictDiagramViewer.this.isDisposed()));
219 public static final Key KEY_MAP_COLOR_BAR_OPTIONS = new KeyOf(ColorBarOptions.class, "colorBarOptions");
220 public static final Command MAP_COLOR_BAR_OPTIONS_CHANGE = new Command("colorBarOptionsChange");
221 public static final Key KEY_MAP_SIZE_BAR_OPTIONS = new KeyOf(SizeBarOptions.class, "sizeBarOptions");
222 public static final Command MAP_SIZE_BAR_OPTIONS_CHANGE = new Command("sizeBarOptionsChange");
224 public static final Key KEY_MAP_COLORING_OBJECTS = new KeyOf(Map.class, "coloringObjects");
225 public static final Command MAP_COLORING_OBJECTS_CHANGE = new Command("coloringObjectsChange");
227 public static final Key KEY_MAP_SIZING_OBJECTS = new KeyOf(Map.class, "sizingObjects");
228 public static final Command MAP_SIZING_OBJECTS_CHANGE = new Command("sizingObjectsChange");
230 public static final Key KEY_SHOW_ELEVATION_SERVER = new KeyOf(Boolean.class, "showElevationServer");
231 public static final Command SHOW_ELEVATION_SERVER_CHANGE = new Command("showElevationServerChange");
233 private void queueColoringObjectsChangeEvent(Map<String, DynamicColorContribution> result) {
234 queueEventInternal(KEY_MAP_COLORING_OBJECTS, MAP_COLORING_OBJECTS_CHANGE, result);
237 private void queueColorBarOptionsChangeEvent(ColorBarOptions result) {
238 queueEventInternal(KEY_MAP_COLOR_BAR_OPTIONS, MAP_COLOR_BAR_OPTIONS_CHANGE, result);
241 private void queueSizingObjectsChangeEvent(Map<String, DynamicSizeContribution> result) {
242 queueEventInternal(KEY_MAP_SIZING_OBJECTS, MAP_SIZING_OBJECTS_CHANGE, result);
245 private void queueSizeBarOptionsChangeEvent(SizeBarOptions result) {
246 queueEventInternal(KEY_MAP_SIZE_BAR_OPTIONS, MAP_SIZE_BAR_OPTIONS_CHANGE, result);
249 private void queueShowElevationServerChangeEvent(Boolean result) {
250 queueEventInternal(KEY_SHOW_ELEVATION_SERVER, SHOW_ELEVATION_SERVER_CHANGE, result);
253 private void queueEventInternal(Key key, Command command, Object result) {
254 if (result != null && !canvasContext.isDisposed()) {
255 canvasContext.getThreadAccess().asyncExec(() -> {
256 canvasContext.getDefaultHintContext().setHint(key, result);
257 canvasContext.getEventQueue().queueEvent(new CommandEvent(canvasContext, System.currentTimeMillis(), command));
260 LOGGER.info("Result is either null or canvasContext is disposed", String.valueOf(result));
264 private static class DrawMapEnabledRequest extends UnaryRead<Resource, Boolean> {
266 public DrawMapEnabledRequest(Resource diagram) {
271 public Boolean perform(ReadGraph graph) throws DatabaseException {
272 return DistrictNetworkUtil.drawMapEnabled(graph, parameter);
276 private static class DrawMapEnabledListener implements Listener<Boolean> {
278 private static final Logger LOGGER = LoggerFactory.getLogger(DrawMapEnabledListener.class);
280 private Consumer<Boolean> callback;
281 private Supplier<Boolean> isDisposed;
283 private Boolean lastResult;
285 public DrawMapEnabledListener(Consumer<Boolean> callback, Supplier<Boolean> isDisposed) {
286 this.callback = callback;
287 this.isDisposed = isDisposed;
291 public void execute(Boolean result) {
292 // Minor optimization
293 if (!Objects.equals(lastResult, result)) {
295 callback.accept(result);
300 public void exception(Throwable t) {
301 LOGGER.error("Could not listen if draw map is enabled", t);
305 public boolean isDisposed() {
306 return isDisposed.get();
310 private static class MapBackgroundColorRequest extends UnaryRead<Resource, RGB.Integer> {
312 public MapBackgroundColorRequest(Resource diagram) {
317 public RGB.Integer perform(ReadGraph graph) throws DatabaseException {
318 return DistrictNetworkUtil.backgroundColor(graph, parameter);
322 private static class MapBackgroundColorListener implements Listener<RGB.Integer> {
324 private static final Logger LOGGER = LoggerFactory.getLogger(MapBackgroundColorListener.class);
326 private Consumer<RGB.Integer> callback;
327 private Supplier<Boolean> isDisposed;
329 private RGB.Integer lastResult;
331 public MapBackgroundColorListener(Consumer<RGB.Integer> callback, Supplier<Boolean> isDisposed) {
332 this.callback = callback;
333 this.isDisposed = isDisposed;
337 public void execute(RGB.Integer result) {
338 if (!Objects.equals(lastResult, result)) {
340 callback.accept(result);
345 public void exception(Throwable t) {
346 LOGGER.error("Could not listen map background color", t);
350 public boolean isDisposed() {
351 return isDisposed.get();
355 private static class ColorBarOptionsRequest extends UnaryRead<Resource, ColorBarOptions> {
357 public ColorBarOptionsRequest(Resource diagram) {
362 public ColorBarOptions perform(ReadGraph graph) throws DatabaseException {
363 DistrictNetworkResource DN = DistrictNetworkResource.getInstance(graph);
364 Resource model = graph.syncRequest(new PossibleIndexRoot(parameter));
366 Resource vf = DynamicVisualisations.getVisualisationFolder(graph, model);
368 Resource activeVisualisation = graph.getPossibleObject(vf, DN.Diagram_hasActiveVisualisation);
369 if (activeVisualisation != null) {
370 return DynamicVisualisations.colorBarOptions(graph, activeVisualisation);
373 LOGGER.debug("No visualisation folder available for model {}", model);
376 return ColorBarOptions.useDefault();
380 private static class ColoringObjectsRequest extends UnaryRead<Resource, Map<String,DynamicColorContribution>> {
382 public ColoringObjectsRequest(Resource diagram) {
387 public Map<String, DynamicColorContribution> perform(ReadGraph graph) throws DatabaseException {
388 DistrictNetworkResource DN = DistrictNetworkResource.getInstance(graph);
389 Resource model = graph.syncRequest(new PossibleIndexRoot(parameter));
391 Resource vf = DynamicVisualisations.getVisualisationFolder(graph, model);
393 Resource activeVisualisation = graph.getPossibleObject(vf, DN.Diagram_hasActiveVisualisation);
394 if (activeVisualisation != null) {
395 return DynamicVisualisations.colorContributions(graph, activeVisualisation);
398 LOGGER.debug("No visualisation folder available for model {}", model);
401 return Collections.emptyMap();
405 private static class ColoringObjectsListener implements Listener<Map<String,DynamicColorContribution>> {
407 private static final Logger LOGGER = LoggerFactory.getLogger(ColoringObjectsListener.class);
409 private Consumer<Map<String,DynamicColorContribution>> callback;
410 private Supplier<Boolean> isDisposed;
412 //private Map<String, DynamicColorContribution> lastResult
414 public ColoringObjectsListener(Consumer<Map<String,DynamicColorContribution>> callback, Supplier<Boolean> isDisposed) {
415 this.callback = callback;
416 this.isDisposed = isDisposed;
420 public void execute(Map<String,DynamicColorContribution> result) {
421 callback.accept(result);
425 public void exception(Throwable t) {
426 LOGGER.error("Could not listen ColoringObjects", t);
430 public boolean isDisposed() {
431 return isDisposed.get();
435 private static class ColorBarOptionsListener implements Listener<ColorBarOptions> {
437 private static final Logger LOGGER = LoggerFactory.getLogger(ColorBarOptionsListener.class);
439 private Consumer<ColorBarOptions> callback;
440 private Supplier<Boolean> isDisposed;
442 public ColorBarOptionsListener(Consumer<ColorBarOptions> callback, Supplier<Boolean> isDisposed) {
443 this.callback = callback;
444 this.isDisposed = isDisposed;
448 public void execute(ColorBarOptions result) {
449 callback.accept(result);
453 public void exception(Throwable t) {
454 LOGGER.error("Could not listen ColorBarOptions", t);
458 public boolean isDisposed() {
459 return isDisposed.get();
463 private static class SizeBarOptionsRequest extends UnaryRead<Resource, SizeBarOptions> {
465 public SizeBarOptionsRequest(Resource diagram) {
470 public SizeBarOptions perform(ReadGraph graph) throws DatabaseException {
471 DistrictNetworkResource DN = DistrictNetworkResource.getInstance(graph);
472 Resource model = graph.syncRequest(new PossibleIndexRoot(parameter));
474 Resource vf = DynamicVisualisations.getVisualisationFolder(graph, model);
476 Resource activeVisualisation = graph.getPossibleObject(vf, DN.Diagram_hasActiveVisualisation);
477 if (activeVisualisation != null) {
478 return DynamicVisualisations.sizeBarOptions(graph, activeVisualisation);
481 LOGGER.debug("No visualisation folder available for model {}", model);
484 return SizeBarOptions.useDefault();
488 private static class SizeBarOptionsListener implements Listener<SizeBarOptions> {
490 private static final Logger LOGGER = LoggerFactory.getLogger(SizeBarOptionsListener.class);
492 private Consumer<SizeBarOptions> callback;
493 private Supplier<Boolean> isDisposed;
495 public SizeBarOptionsListener(Consumer<SizeBarOptions> callback, Supplier<Boolean> isDisposed) {
496 this.callback = callback;
497 this.isDisposed = isDisposed;
501 public void execute(SizeBarOptions result) {
502 callback.accept(result);
506 public void exception(Throwable t) {
507 LOGGER.error("Could not listen SizeBarOptions", t);
511 public boolean isDisposed() {
512 return isDisposed.get();
516 private static class SizingObjectsRequest extends UnaryRead<Resource, Map<String, DynamicSizeContribution>> {
518 public SizingObjectsRequest(Resource diagram) {
523 public Map<String, DynamicSizeContribution> perform(ReadGraph graph) throws DatabaseException {
524 DistrictNetworkResource DN = DistrictNetworkResource.getInstance(graph);
525 Resource model = graph.syncRequest(new PossibleIndexRoot(parameter));
527 Resource vf = DynamicVisualisations.getVisualisationFolder(graph, model);
529 Resource activeVisualisation = graph.getPossibleObject(vf, DN.Diagram_hasActiveVisualisation);
530 if (activeVisualisation != null) {
531 return DynamicVisualisations.sizeContributions(graph, activeVisualisation);
534 LOGGER.debug("No visualisation folder available for model {}", model);
537 return Collections.emptyMap();
541 private static class SizingObjectsListener implements Listener<Map<String,DynamicSizeContribution>> {
543 private static final Logger LOGGER = LoggerFactory.getLogger(SizingObjectsListener.class);
545 private Consumer<Map<String,DynamicSizeContribution>> callback;
546 private Supplier<Boolean> isDisposed;
548 public SizingObjectsListener(Consumer<Map<String, DynamicSizeContribution>> callback, Supplier<Boolean> isDisposed) {
549 this.callback = callback;
550 this.isDisposed = isDisposed;
554 public void execute(Map<String, DynamicSizeContribution> result) {
555 callback.accept(result);
559 public void exception(Throwable t) {
560 LOGGER.error("Could not listen SizingObjectsOptions", t);
564 public boolean isDisposed() {
565 return isDisposed.get();
569 private static class ShowElevationServerRequest extends UnaryRead<Resource, Boolean> {
571 public ShowElevationServerRequest(Resource diagram) {
576 public Boolean perform(ReadGraph graph) throws DatabaseException {
577 DistrictNetworkResource DN = DistrictNetworkResource.getInstance(graph);
578 Resource model = graph.syncRequest(new PossibleIndexRoot(parameter));
580 Resource vf = DynamicVisualisations.getVisualisationFolder(graph, model);
582 Resource activeVisualisation = graph.getPossibleObject(vf, DN.Diagram_hasActiveVisualisation);
583 if (activeVisualisation != null) {
584 return DynamicVisualisations.showElevationServerBoundingBox(graph, activeVisualisation);
587 LOGGER.debug("No visualisation folder available for model {}", model);
594 private static class ShowElevationServerListener implements Listener<Boolean> {
596 private static final Logger LOGGER = LoggerFactory.getLogger(ShowElevationServerListener.class);
598 private Consumer<Boolean> callback;
599 private Supplier<Boolean> isDisposed;
601 public ShowElevationServerListener(Consumer<Boolean> callback, Supplier<Boolean> isDisposed) {
602 this.callback = callback;
603 this.isDisposed = isDisposed;
607 public void execute(Boolean result) {
608 callback.accept(result);
612 public void exception(Throwable t) {
613 LOGGER.error("Could not listen Show Elevation Server", t);
617 public boolean isDisposed() {
618 return isDisposed.get();