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.handler.PickRequest.PickFilter;
36 import org.simantics.g2d.diagram.participant.DelayedBatchElementPainter;
37 import org.simantics.g2d.diagram.participant.ElementPainter;
38 import org.simantics.g2d.diagram.participant.ElementPainterConfiguration;
39 import org.simantics.g2d.diagram.participant.Selection;
40 import org.simantics.g2d.diagram.participant.ZOrderHandler;
41 import org.simantics.g2d.participant.BackgroundPainter;
42 import org.simantics.g2d.participant.GridPainter;
43 import org.simantics.g2d.participant.PanZoomRotateHandler;
44 import org.simantics.g2d.participant.RenderingQualityInteractor;
45 import org.simantics.g2d.participant.TransformUtil;
46 import org.simantics.g2d.participant.ZoomToAreaHandler;
47 import org.simantics.g2d.scenegraph.SceneGraphConstants;
48 import org.simantics.maps.MapScalingTransform;
49 import org.simantics.maps.eclipse.MapPainter;
50 import org.simantics.maps.sg.commands.MapCommands;
51 import org.simantics.modeling.ui.diagramEditor.DiagramViewer;
52 import org.simantics.scenegraph.g2d.G2DParentNode;
53 import org.simantics.scenegraph.g2d.events.command.Command;
54 import org.simantics.scenegraph.g2d.events.command.CommandEvent;
55 import org.simantics.scenegraph.g2d.events.command.Commands;
56 import org.simantics.utils.datastructures.hints.IHintContext;
57 import org.simantics.utils.datastructures.hints.IHintContext.Key;
58 import org.simantics.utils.datastructures.hints.IHintContext.KeyOf;
59 import org.slf4j.Logger;
60 import org.slf4j.LoggerFactory;
62 public class DistrictDiagramViewer extends DiagramViewer {
64 private static final Logger LOGGER = LoggerFactory.getLogger(DistrictDiagramViewer.class);
67 protected void addDiagramParticipants(ICanvasContext ctx) {
68 ctx.add(new ZOrderHandler());
69 ctx.add(new Selection());
70 ctx.add(new ElementPainter(new ElementPainterConfiguration().selectionNodeClass(DistrictSelectionNode.class)));
71 ctx.add(new DNPointerInteractor());
73 AffineTransform tr = new AffineTransform(MapScalingTransform.INSTANCE);
74 ctx.add(new MapPainter(tr));
76 DynamicVisualisationContributionsParticipant dynamicVisualisationContributionsParticipant = new DynamicVisualisationContributionsParticipant(tr);
77 ctx.add(new NetworkDrawingParticipant(dynamicVisualisationContributionsParticipant, tr));
78 ctx.add(dynamicVisualisationContributionsParticipant);
80 // Optimize AffineTransform memory allocations during district diagram rendering
81 G2DParentNode spatialRoot = (G2DParentNode) ctx.getSceneGraph().lookupNode(SceneGraphConstants.SPATIAL_ROOT_NODE_ID);
82 DistrictRenderingPreparationNode prepNode = new DistrictRenderingPreparationNode();
83 prepNode.setZIndex(Integer.MIN_VALUE / 2);
84 spatialRoot.addNode("districtRenderingPrepareNode", prepNode);
87 protected String getPopupId() {
88 return "#DistrictDiagramPopup";
92 protected void fillInitialDiagramHints(Resource diagram, IHintContext initialHints) throws DatabaseException {
93 super.fillInitialDiagramHints(diagram, initialHints);
98 public void initializeCanvasContext(CanvasContext ctx) {
99 super.initializeCanvasContext(ctx);
100 IHintContext h = ctx.getDefaultHintContext();
101 h.setHint(PanZoomRotateHandler.KEY_ZOOM_IN_LIMIT, 10000.0);
102 h.setHint(PanZoomRotateHandler.KEY_ZOOM_OUT_LIMIT, 0.01);
103 h.setHint(DiagramModelHints.KEY_DIAGRAM_RESOURCE, diagramResource);
107 protected void addPainterParticipants(ICanvasContext ctx) {
108 ctx.add(new RenderingQualityInteractor());
109 ctx.add(new DelayedBatchElementPainter(PickFilter.FILTER_MONITORS, 500, TimeUnit.MILLISECONDS));
113 protected void addGridRulerBackgroundParticipants(CanvasContext ctx) {
114 ctx.add(new GridPainter());
115 ctx.add(new MapRulerPainter());
116 ctx.add(new BackgroundPainter());
119 protected void addViewManipulationParticipants(CanvasContext ctx) {
120 // Let's replace with our special util
121 TransformUtil util = ctx.getAtMostOneItemOfClass(TransformUtil.class);
124 ctx.add(new DistrictTransformUtil());
125 ctx.add(new DistrictPanZoomRotateHandler());
126 //ctx.add(new MousePanZoomInteractor());
127 //ctx.add(new MultitouchPanZoomRotateInteractor());
128 // ctx.add( new OrientationRestorer() );
129 ctx.add(new ZoomToAreaHandler());
133 protected void loadPageSettings(ICanvasContext ctx) {
134 super.loadPageSettings(ctx);
135 // this might be the wrong place to start such listening but at least
136 // super.loadPageSettings() does async-db-operations
137 setupDrawMapEnabled();
138 setupBackgroundColor();
139 setupColoringObjects();
140 setupColorBarOptions();
141 setupSizingObjects();
142 setupSizeBarOptions();
143 setupShowElevationServerBoundingBox();
146 DistrictDiagramViewerListener[] listeners = Activator.getInstance().getDistrictDiagramViewerListeners();
147 if (listeners != null) {
148 for (DistrictDiagramViewerListener listener : listeners) {
149 listener.diagramLoaded(getRuntime(), canvasContext);
155 public void dispose() {
156 DistrictDiagramViewerListener[] listeners = Activator.getInstance().getDistrictDiagramViewerListeners();
157 if (listeners != null) {
158 Resource runtime = getRuntime();
159 for (DistrictDiagramViewerListener listener : listeners) {
160 listener.diagramDisposed(runtime, canvasContext);
166 private void setupDrawMapEnabled() {
167 sessionContext.getSession().asyncRequest(new DrawMapEnabledRequest(getInputResource()), new DrawMapEnabledListener(
168 result -> canvasContext.getEventQueue().queueEvent(new CommandEvent(canvasContext, System.currentTimeMillis(), result ? Commands.MAP_ENABLE : Commands.MAP_DISABLE)),
169 () -> DistrictDiagramViewer.this.isDisposed()));
172 private void setupBackgroundColor() {
173 sessionContext.getSession().asyncRequest(new MapBackgroundColorRequest(getInputResource()), new MapBackgroundColorListener(
174 result -> queueBackgroundColorChangeEvent(result),
175 () -> DistrictDiagramViewer.this.isDisposed()));
178 private void queueBackgroundColorChangeEvent(RGB.Integer result) {
179 if (result != null) {
180 Color backgroundColor = new Color(result.red, result.green, result.blue);
181 canvasContext.getDefaultHintContext().setHint(MapCommands.KEY_MAP_BACKGROUND_COLOR, backgroundColor);
182 canvasContext.getEventQueue().queueEvent(new CommandEvent(canvasContext, System.currentTimeMillis(), MapCommands.MAP_BACKGROUND_COLOR_CHANGE));
186 private void setupColoringObjects() {
187 sessionContext.getSession().asyncRequest(new ColoringObjectsRequest(getInputResource()), new ColoringObjectsListener(
188 result -> queueColoringObjectsChangeEvent(result),
189 () -> DistrictDiagramViewer.this.isDisposed()));
193 private void setupColorBarOptions() {
194 sessionContext.getSession().asyncRequest(new ColorBarOptionsRequest(getInputResource()), new ColorBarOptionsListener(
195 result -> queueColorBarOptionsChangeEvent(result),
196 () -> DistrictDiagramViewer.this.isDisposed()));
199 private void setupSizingObjects() {
200 sessionContext.getSession().asyncRequest(new SizingObjectsRequest(getInputResource()), new SizingObjectsListener(
201 result -> queueSizingObjectsChangeEvent(result),
202 () -> DistrictDiagramViewer.this.isDisposed()));
205 private void setupSizeBarOptions() {
206 sessionContext.getSession().asyncRequest(new SizeBarOptionsRequest(getInputResource()), new SizeBarOptionsListener(
207 result -> queueSizeBarOptionsChangeEvent(result),
208 () -> DistrictDiagramViewer.this.isDisposed()));
211 private void setupShowElevationServerBoundingBox() {
212 sessionContext.getSession().asyncRequest(new ShowElevationServerRequest(getInputResource()), new ShowElevationServerListener(
213 result -> queueShowElevationServerChangeEvent(result),
214 () -> DistrictDiagramViewer.this.isDisposed()));
217 public static final Key KEY_MAP_COLOR_BAR_OPTIONS = new KeyOf(ColorBarOptions.class, "colorBarOptions");
218 public static final Command MAP_COLOR_BAR_OPTIONS_CHANGE = new Command("colorBarOptionsChange");
219 public static final Key KEY_MAP_SIZE_BAR_OPTIONS = new KeyOf(SizeBarOptions.class, "sizeBarOptions");
220 public static final Command MAP_SIZE_BAR_OPTIONS_CHANGE = new Command("sizeBarOptionsChange");
222 public static final Key KEY_MAP_COLORING_OBJECTS = new KeyOf(Map.class, "coloringObjects");
223 public static final Command MAP_COLORING_OBJECTS_CHANGE = new Command("coloringObjectsChange");
225 public static final Key KEY_MAP_SIZING_OBJECTS = new KeyOf(Map.class, "sizingObjects");
226 public static final Command MAP_SIZING_OBJECTS_CHANGE = new Command("sizingObjectsChange");
228 public static final Key KEY_SHOW_ELEVATION_SERVER = new KeyOf(Boolean.class, "showElevationServer");
229 public static final Command SHOW_ELEVATION_SERVER_CHANGE = new Command("showElevationServerChange");
231 private void queueColoringObjectsChangeEvent(Map<String, DynamicColorContribution> result) {
232 queueEventInternal(KEY_MAP_COLORING_OBJECTS, MAP_COLORING_OBJECTS_CHANGE, result);
235 private void queueColorBarOptionsChangeEvent(ColorBarOptions result) {
236 queueEventInternal(KEY_MAP_COLOR_BAR_OPTIONS, MAP_COLOR_BAR_OPTIONS_CHANGE, result);
239 private void queueSizingObjectsChangeEvent(Map<String, DynamicSizeContribution> result) {
240 queueEventInternal(KEY_MAP_SIZING_OBJECTS, MAP_SIZING_OBJECTS_CHANGE, result);
243 private void queueSizeBarOptionsChangeEvent(SizeBarOptions result) {
244 queueEventInternal(KEY_MAP_SIZE_BAR_OPTIONS, MAP_SIZE_BAR_OPTIONS_CHANGE, result);
247 private void queueShowElevationServerChangeEvent(Boolean result) {
248 queueEventInternal(KEY_SHOW_ELEVATION_SERVER, SHOW_ELEVATION_SERVER_CHANGE, result);
251 private void queueEventInternal(Key key, Command command, Object result) {
252 if (result != null && !canvasContext.isDisposed()) {
253 canvasContext.getThreadAccess().asyncExec(() -> {
254 canvasContext.getDefaultHintContext().setHint(key, result);
255 canvasContext.getEventQueue().queueEvent(new CommandEvent(canvasContext, System.currentTimeMillis(), command));
258 LOGGER.info("Result is either null or canvasContext is disposed", String.valueOf(result));
262 private static class DrawMapEnabledRequest extends UnaryRead<Resource, Boolean> {
264 public DrawMapEnabledRequest(Resource diagram) {
269 public Boolean perform(ReadGraph graph) throws DatabaseException {
270 return DistrictNetworkUtil.drawMapEnabled(graph, parameter);
274 private static class DrawMapEnabledListener implements Listener<Boolean> {
276 private static final Logger LOGGER = LoggerFactory.getLogger(DrawMapEnabledListener.class);
278 private Consumer<Boolean> callback;
279 private Supplier<Boolean> isDisposed;
281 private Boolean lastResult;
283 public DrawMapEnabledListener(Consumer<Boolean> callback, Supplier<Boolean> isDisposed) {
284 this.callback = callback;
285 this.isDisposed = isDisposed;
289 public void execute(Boolean result) {
290 // Minor optimization
291 if (!Objects.equals(lastResult, result)) {
293 callback.accept(result);
298 public void exception(Throwable t) {
299 LOGGER.error("Could not listen if draw map is enabled", t);
303 public boolean isDisposed() {
304 return isDisposed.get();
308 private static class MapBackgroundColorRequest extends UnaryRead<Resource, RGB.Integer> {
310 public MapBackgroundColorRequest(Resource diagram) {
315 public RGB.Integer perform(ReadGraph graph) throws DatabaseException {
316 return DistrictNetworkUtil.backgroundColor(graph, parameter);
320 private static class MapBackgroundColorListener implements Listener<RGB.Integer> {
322 private static final Logger LOGGER = LoggerFactory.getLogger(MapBackgroundColorListener.class);
324 private Consumer<RGB.Integer> callback;
325 private Supplier<Boolean> isDisposed;
327 private RGB.Integer lastResult;
329 public MapBackgroundColorListener(Consumer<RGB.Integer> callback, Supplier<Boolean> isDisposed) {
330 this.callback = callback;
331 this.isDisposed = isDisposed;
335 public void execute(RGB.Integer result) {
336 if (!Objects.equals(lastResult, result)) {
338 callback.accept(result);
343 public void exception(Throwable t) {
344 LOGGER.error("Could not listen map background color", t);
348 public boolean isDisposed() {
349 return isDisposed.get();
353 private static class ColorBarOptionsRequest extends UnaryRead<Resource, ColorBarOptions> {
355 public ColorBarOptionsRequest(Resource diagram) {
360 public ColorBarOptions perform(ReadGraph graph) throws DatabaseException {
361 DistrictNetworkResource DN = DistrictNetworkResource.getInstance(graph);
362 Resource model = graph.syncRequest(new PossibleIndexRoot(parameter));
364 Resource vf = DynamicVisualisations.getVisualisationFolder(graph, model);
366 Resource activeVisualisation = graph.getPossibleObject(vf, DN.Diagram_hasActiveVisualisation);
367 if (activeVisualisation != null) {
368 return DynamicVisualisations.colorBarOptions(graph, activeVisualisation);
371 LOGGER.debug("No visualisation folder available for model {}", model);
374 return ColorBarOptions.useDefault();
378 private static class ColoringObjectsRequest extends UnaryRead<Resource, Map<String,DynamicColorContribution>> {
380 public ColoringObjectsRequest(Resource diagram) {
385 public Map<String, DynamicColorContribution> perform(ReadGraph graph) throws DatabaseException {
386 DistrictNetworkResource DN = DistrictNetworkResource.getInstance(graph);
387 Resource model = graph.syncRequest(new PossibleIndexRoot(parameter));
389 Resource vf = DynamicVisualisations.getVisualisationFolder(graph, model);
391 Resource activeVisualisation = graph.getPossibleObject(vf, DN.Diagram_hasActiveVisualisation);
392 if (activeVisualisation != null) {
393 return DynamicVisualisations.colorContributions(graph, activeVisualisation);
396 LOGGER.debug("No visualisation folder available for model {}", model);
399 return Collections.emptyMap();
403 private static class ColoringObjectsListener implements Listener<Map<String,DynamicColorContribution>> {
405 private static final Logger LOGGER = LoggerFactory.getLogger(ColoringObjectsListener.class);
407 private Consumer<Map<String,DynamicColorContribution>> callback;
408 private Supplier<Boolean> isDisposed;
410 //private Map<String, DynamicColorContribution> lastResult
412 public ColoringObjectsListener(Consumer<Map<String,DynamicColorContribution>> callback, Supplier<Boolean> isDisposed) {
413 this.callback = callback;
414 this.isDisposed = isDisposed;
418 public void execute(Map<String,DynamicColorContribution> result) {
419 callback.accept(result);
423 public void exception(Throwable t) {
424 LOGGER.error("Could not listen ColoringObjects", t);
428 public boolean isDisposed() {
429 return isDisposed.get();
433 private static class ColorBarOptionsListener implements Listener<ColorBarOptions> {
435 private static final Logger LOGGER = LoggerFactory.getLogger(ColorBarOptionsListener.class);
437 private Consumer<ColorBarOptions> callback;
438 private Supplier<Boolean> isDisposed;
440 public ColorBarOptionsListener(Consumer<ColorBarOptions> callback, Supplier<Boolean> isDisposed) {
441 this.callback = callback;
442 this.isDisposed = isDisposed;
446 public void execute(ColorBarOptions result) {
447 callback.accept(result);
451 public void exception(Throwable t) {
452 LOGGER.error("Could not listen ColorBarOptions", t);
456 public boolean isDisposed() {
457 return isDisposed.get();
461 private static class SizeBarOptionsRequest extends UnaryRead<Resource, SizeBarOptions> {
463 public SizeBarOptionsRequest(Resource diagram) {
468 public SizeBarOptions perform(ReadGraph graph) throws DatabaseException {
469 DistrictNetworkResource DN = DistrictNetworkResource.getInstance(graph);
470 Resource model = graph.syncRequest(new PossibleIndexRoot(parameter));
472 Resource vf = DynamicVisualisations.getVisualisationFolder(graph, model);
474 Resource activeVisualisation = graph.getPossibleObject(vf, DN.Diagram_hasActiveVisualisation);
475 if (activeVisualisation != null) {
476 return DynamicVisualisations.sizeBarOptions(graph, activeVisualisation);
479 LOGGER.debug("No visualisation folder available for model {}", model);
482 return SizeBarOptions.useDefault();
486 private static class SizeBarOptionsListener implements Listener<SizeBarOptions> {
488 private static final Logger LOGGER = LoggerFactory.getLogger(SizeBarOptionsListener.class);
490 private Consumer<SizeBarOptions> callback;
491 private Supplier<Boolean> isDisposed;
493 public SizeBarOptionsListener(Consumer<SizeBarOptions> callback, Supplier<Boolean> isDisposed) {
494 this.callback = callback;
495 this.isDisposed = isDisposed;
499 public void execute(SizeBarOptions result) {
500 callback.accept(result);
504 public void exception(Throwable t) {
505 LOGGER.error("Could not listen SizeBarOptions", t);
509 public boolean isDisposed() {
510 return isDisposed.get();
514 private static class SizingObjectsRequest extends UnaryRead<Resource, Map<String, DynamicSizeContribution>> {
516 public SizingObjectsRequest(Resource diagram) {
521 public Map<String, DynamicSizeContribution> perform(ReadGraph graph) throws DatabaseException {
522 DistrictNetworkResource DN = DistrictNetworkResource.getInstance(graph);
523 Resource model = graph.syncRequest(new PossibleIndexRoot(parameter));
525 Resource vf = DynamicVisualisations.getVisualisationFolder(graph, model);
527 Resource activeVisualisation = graph.getPossibleObject(vf, DN.Diagram_hasActiveVisualisation);
528 if (activeVisualisation != null) {
529 return DynamicVisualisations.sizeContributions(graph, activeVisualisation);
532 LOGGER.debug("No visualisation folder available for model {}", model);
535 return Collections.emptyMap();
539 private static class SizingObjectsListener implements Listener<Map<String,DynamicSizeContribution>> {
541 private static final Logger LOGGER = LoggerFactory.getLogger(SizingObjectsListener.class);
543 private Consumer<Map<String,DynamicSizeContribution>> callback;
544 private Supplier<Boolean> isDisposed;
546 public SizingObjectsListener(Consumer<Map<String, DynamicSizeContribution>> callback, Supplier<Boolean> isDisposed) {
547 this.callback = callback;
548 this.isDisposed = isDisposed;
552 public void execute(Map<String, DynamicSizeContribution> result) {
553 callback.accept(result);
557 public void exception(Throwable t) {
558 LOGGER.error("Could not listen SizingObjectsOptions", t);
562 public boolean isDisposed() {
563 return isDisposed.get();
567 private static class ShowElevationServerRequest extends UnaryRead<Resource, Boolean> {
569 public ShowElevationServerRequest(Resource diagram) {
574 public Boolean perform(ReadGraph graph) throws DatabaseException {
575 DistrictNetworkResource DN = DistrictNetworkResource.getInstance(graph);
576 Resource model = graph.syncRequest(new PossibleIndexRoot(parameter));
578 Resource vf = DynamicVisualisations.getVisualisationFolder(graph, model);
580 Resource activeVisualisation = graph.getPossibleObject(vf, DN.Diagram_hasActiveVisualisation);
581 if (activeVisualisation != null) {
582 return DynamicVisualisations.showElevationServerBoundingBox(graph, activeVisualisation);
585 LOGGER.debug("No visualisation folder available for model {}", model);
592 private static class ShowElevationServerListener implements Listener<Boolean> {
594 private static final Logger LOGGER = LoggerFactory.getLogger(ShowElevationServerListener.class);
596 private Consumer<Boolean> callback;
597 private Supplier<Boolean> isDisposed;
599 public ShowElevationServerListener(Consumer<Boolean> callback, Supplier<Boolean> isDisposed) {
600 this.callback = callback;
601 this.isDisposed = isDisposed;
605 public void execute(Boolean result) {
606 callback.accept(result);
610 public void exception(Throwable t) {
611 LOGGER.error("Could not listen Show Elevation Server", t);
615 public boolean isDisposed() {
616 return isDisposed.get();