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.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.nodes.DistrictRenderingPreparationNode;
23 import org.simantics.district.network.ui.participants.DNPointerInteractor;
24 import org.simantics.district.network.ui.participants.DynamicVisualisationContributionsParticipant;
25 import org.simantics.district.network.ui.participants.ElevationServerParticipant;
26 import org.simantics.district.network.ui.participants.MapRulerPainter;
27 import org.simantics.district.network.visualisations.DynamicVisualisations;
28 import org.simantics.district.network.visualisations.model.ColorBarOptions;
29 import org.simantics.district.network.visualisations.model.DynamicColorContribution;
30 import org.simantics.district.network.visualisations.model.DynamicSizeContribution;
31 import org.simantics.district.network.visualisations.model.SizeBarOptions;
32 import org.simantics.g2d.canvas.ICanvasContext;
33 import org.simantics.g2d.canvas.impl.CanvasContext;
34 import org.simantics.g2d.diagram.handler.PickRequest.PickFilter;
35 import org.simantics.g2d.diagram.participant.DelayedBatchElementPainter;
36 import org.simantics.g2d.diagram.participant.ElementPainter;
37 import org.simantics.g2d.diagram.participant.Selection;
38 import org.simantics.g2d.diagram.participant.ZOrderHandler;
39 import org.simantics.g2d.participant.BackgroundPainter;
40 import org.simantics.g2d.participant.GridPainter;
41 import org.simantics.g2d.participant.PanZoomRotateHandler;
42 import org.simantics.g2d.participant.RenderingQualityInteractor;
43 import org.simantics.g2d.participant.TransformUtil;
44 import org.simantics.g2d.participant.ZoomToAreaHandler;
45 import org.simantics.g2d.scenegraph.SceneGraphConstants;
46 import org.simantics.maps.MapScalingTransform;
47 import org.simantics.maps.eclipse.MapPainter;
48 import org.simantics.maps.sg.commands.MapCommands;
49 import org.simantics.modeling.ui.diagramEditor.DiagramViewer;
50 import org.simantics.scenegraph.g2d.G2DParentNode;
51 import org.simantics.scenegraph.g2d.events.command.Command;
52 import org.simantics.scenegraph.g2d.events.command.CommandEvent;
53 import org.simantics.scenegraph.g2d.events.command.Commands;
54 import org.simantics.utils.datastructures.hints.IHintContext;
55 import org.simantics.utils.datastructures.hints.IHintContext.Key;
56 import org.simantics.utils.datastructures.hints.IHintContext.KeyOf;
57 import org.slf4j.Logger;
58 import org.slf4j.LoggerFactory;
60 public class DistrictDiagramViewer extends DiagramViewer {
62 private static final Logger LOGGER = LoggerFactory.getLogger(DistrictDiagramViewer.class);
65 protected void addDiagramParticipants(ICanvasContext ctx) {
66 ctx.add(new ZOrderHandler());
67 ctx.add(new Selection());
68 ctx.add(new ElementPainter());
69 ctx.add(new DNPointerInteractor());
71 AffineTransform tr = new AffineTransform(MapScalingTransform.INSTANCE);
72 ctx.add(new MapPainter(tr));
74 ctx.add(new NetworkDrawingParticipant(tr));
75 ctx.add(new ElevationServerParticipant(tr));
76 ctx.add(new DynamicVisualisationContributionsParticipant(tr));
78 // Optimize AffineTransform memory allocations during district diagram rendering
79 G2DParentNode spatialRoot = (G2DParentNode) ctx.getSceneGraph().lookupNode(SceneGraphConstants.SPATIAL_ROOT_NODE_ID);
80 DistrictRenderingPreparationNode prepNode = new DistrictRenderingPreparationNode();
81 prepNode.setZIndex(Integer.MIN_VALUE / 2);
82 spatialRoot.addNode("districtRenderingPrepareNode", prepNode);
85 protected String getPopupId() {
86 return "#DistrictDiagramPopup";
90 protected void fillInitialDiagramHints(Resource diagram, IHintContext initialHints) throws DatabaseException {
91 super.fillInitialDiagramHints(diagram, initialHints);
96 public void initializeCanvasContext(CanvasContext ctx) {
97 super.initializeCanvasContext(ctx);
98 IHintContext h = ctx.getDefaultHintContext();
99 h.setHint(PanZoomRotateHandler.KEY_ZOOM_IN_LIMIT, 10000.0);
100 h.setHint(PanZoomRotateHandler.KEY_ZOOM_OUT_LIMIT, 0.01);
101 h.setHint(DiagramModelHints.KEY_DIAGRAM_RESOURCE, diagramResource);
105 protected void addPainterParticipants(ICanvasContext ctx) {
106 ctx.add(new RenderingQualityInteractor());
107 ctx.add(new DelayedBatchElementPainter(PickFilter.FILTER_MONITORS, 500, TimeUnit.MILLISECONDS));
111 protected void addGridRulerBackgroundParticipants(CanvasContext ctx) {
112 ctx.add(new GridPainter());
113 ctx.add(new MapRulerPainter());
114 ctx.add(new BackgroundPainter());
117 protected void addViewManipulationParticipants(CanvasContext ctx) {
118 // Let's replace with our special util
119 TransformUtil util = ctx.getAtMostOneItemOfClass(TransformUtil.class);
122 ctx.add(new DistrictTransformUtil());
123 ctx.add(new DistrictPanZoomRotateHandler());
124 //ctx.add(new MousePanZoomInteractor());
125 //ctx.add(new MultitouchPanZoomRotateInteractor());
126 // ctx.add( new OrientationRestorer() );
127 ctx.add(new ZoomToAreaHandler());
131 protected void loadPageSettings(ICanvasContext ctx) {
132 super.loadPageSettings(ctx);
133 // this might be the wrong place to start such listening but at least
134 // super.loadPageSettings() does async-db-operations
135 setupDrawMapEnabled();
136 setupBackgroundColor();
137 setupColoringObjects();
138 setupColorBarOptions();
139 setupSizingObjects();
140 setupSizeBarOptions();
143 private void setupDrawMapEnabled() {
144 sessionContext.getSession().asyncRequest(new DrawMapEnabledRequest(getInputResource()), new DrawMapEnabledListener(
145 result -> canvasContext.getEventQueue().queueEvent(new CommandEvent(canvasContext, System.currentTimeMillis(), result ? Commands.MAP_ENABLE : Commands.MAP_DISABLE)),
146 () -> DistrictDiagramViewer.this.isDisposed()));
149 private void setupBackgroundColor() {
150 sessionContext.getSession().asyncRequest(new MapBackgroundColorRequest(getInputResource()), new MapBackgroundColorListener(
151 result -> queueBackgroundColorChangeEvent(result),
152 () -> DistrictDiagramViewer.this.isDisposed()));
155 private void queueBackgroundColorChangeEvent(RGB.Integer result) {
156 if (result != null) {
157 Color backgroundColor = new Color(result.red, result.green, result.blue);
158 canvasContext.getDefaultHintContext().setHint(MapCommands.KEY_MAP_BACKGROUND_COLOR, backgroundColor);
159 canvasContext.getEventQueue().queueEvent(new CommandEvent(canvasContext, System.currentTimeMillis(), MapCommands.MAP_BACKGROUND_COLOR_CHANGE));
163 private void setupColoringObjects() {
164 sessionContext.getSession().asyncRequest(new ColoringObjectsRequest(getInputResource()), new ColoringObjectsListener(
165 result -> queueColoringObjectsChangeEvent(result),
166 () -> DistrictDiagramViewer.this.isDisposed()));
170 private void setupColorBarOptions() {
171 sessionContext.getSession().asyncRequest(new ColorBarOptionsRequest(getInputResource()), new ColorBarOptionsListener(
172 result -> queueColorBarOptionsChangeEvent(result),
173 () -> DistrictDiagramViewer.this.isDisposed()));
176 private void setupSizingObjects() {
177 sessionContext.getSession().asyncRequest(new SizingObjectsRequest(getInputResource()), new SizingObjectsListener(
178 result -> queueSizingObjectsChangeEvent(result),
179 () -> DistrictDiagramViewer.this.isDisposed()));
183 private void setupSizeBarOptions() {
184 sessionContext.getSession().asyncRequest(new SizeBarOptionsRequest(getInputResource()), new SizeBarOptionsListener(
185 result -> queueSizeBarOptionsChangeEvent(result),
186 () -> DistrictDiagramViewer.this.isDisposed()));
189 public static final Key KEY_MAP_COLOR_BAR_OPTIONS = new KeyOf(ColorBarOptions.class, "colorBarOptions");
190 public static final Command MAP_COLOR_BAR_OPTIONS_CHANGE = new Command("colorBarOptionsChange");
191 public static final Key KEY_MAP_SIZE_BAR_OPTIONS = new KeyOf(SizeBarOptions.class, "sizeBarOptions");
192 public static final Command MAP_SIZE_BAR_OPTIONS_CHANGE = new Command("sizeBarOptionsChange");
194 public static final Key KEY_MAP_COLORING_OBJECTS = new KeyOf(Map.class, "coloringObjects");
195 public static final Command MAP_COLORING_OBJECTS_CHANGE = new Command("coloringObjectsChange");
197 public static final Key KEY_MAP_SIZING_OBJECTS = new KeyOf(Map.class, "sizingObjects");
198 public static final Command MAP_SIZING_OBJECTS_CHANGE = new Command("sizingObjectsChange");
201 private void queueColoringObjectsChangeEvent(Map<String, DynamicColorContribution> result) {
202 queueEventInternal(KEY_MAP_COLORING_OBJECTS, MAP_COLORING_OBJECTS_CHANGE, result);
205 private void queueColorBarOptionsChangeEvent(ColorBarOptions result) {
206 queueEventInternal(KEY_MAP_COLOR_BAR_OPTIONS, MAP_COLOR_BAR_OPTIONS_CHANGE, result);
209 private void queueSizingObjectsChangeEvent(Map<String, DynamicSizeContribution> result) {
210 queueEventInternal(KEY_MAP_SIZING_OBJECTS, MAP_SIZING_OBJECTS_CHANGE, result);
213 private void queueSizeBarOptionsChangeEvent(SizeBarOptions result) {
214 queueEventInternal(KEY_MAP_SIZE_BAR_OPTIONS, MAP_SIZE_BAR_OPTIONS_CHANGE, result);
217 private void queueEventInternal(Key key, Command command, Object result) {
218 if (result != null && !canvasContext.isDisposed()) {
219 canvasContext.getThreadAccess().asyncExec(() -> {
220 canvasContext.getDefaultHintContext().setHint(key, result);
221 canvasContext.getEventQueue().queueEvent(new CommandEvent(canvasContext, System.currentTimeMillis(), command));
224 LOGGER.info("Result is either null or canvasContext is disposed", String.valueOf(result));
228 private static class DrawMapEnabledRequest extends UnaryRead<Resource, Boolean> {
230 public DrawMapEnabledRequest(Resource diagram) {
235 public Boolean perform(ReadGraph graph) throws DatabaseException {
236 return DistrictNetworkUtil.drawMapEnabled(graph, parameter);
240 private static class DrawMapEnabledListener implements Listener<Boolean> {
242 private static final Logger LOGGER = LoggerFactory.getLogger(DrawMapEnabledListener.class);
244 private Consumer<Boolean> callback;
245 private Supplier<Boolean> isDisposed;
247 private Boolean lastResult;
249 public DrawMapEnabledListener(Consumer<Boolean> callback, Supplier<Boolean> isDisposed) {
250 this.callback = callback;
251 this.isDisposed = isDisposed;
255 public void execute(Boolean result) {
256 // Minor optimization
257 if (!Objects.equals(lastResult, result)) {
259 callback.accept(result);
264 public void exception(Throwable t) {
265 LOGGER.error("Could not listen if draw map is enabled", t);
269 public boolean isDisposed() {
270 return isDisposed.get();
274 private static class MapBackgroundColorRequest extends UnaryRead<Resource, RGB.Integer> {
276 public MapBackgroundColorRequest(Resource diagram) {
281 public RGB.Integer perform(ReadGraph graph) throws DatabaseException {
282 return DistrictNetworkUtil.backgroundColor(graph, parameter);
286 private static class MapBackgroundColorListener implements Listener<RGB.Integer> {
288 private static final Logger LOGGER = LoggerFactory.getLogger(MapBackgroundColorListener.class);
290 private Consumer<RGB.Integer> callback;
291 private Supplier<Boolean> isDisposed;
293 private RGB.Integer lastResult;
295 public MapBackgroundColorListener(Consumer<RGB.Integer> callback, Supplier<Boolean> isDisposed) {
296 this.callback = callback;
297 this.isDisposed = isDisposed;
301 public void execute(RGB.Integer result) {
302 if (!Objects.equals(lastResult, result)) {
304 callback.accept(result);
309 public void exception(Throwable t) {
310 LOGGER.error("Could not listen map background color", t);
314 public boolean isDisposed() {
315 return isDisposed.get();
319 private static class ColorBarOptionsRequest extends UnaryRead<Resource, ColorBarOptions> {
321 public ColorBarOptionsRequest(Resource diagram) {
326 public ColorBarOptions perform(ReadGraph graph) throws DatabaseException {
327 DistrictNetworkResource DN = DistrictNetworkResource.getInstance(graph);
328 Resource model = graph.syncRequest(new IndexRoot(parameter));
329 Resource vf = DynamicVisualisations.getVisualisationFolder(graph, model);
331 Resource activeVisualisation = graph.getPossibleObject(vf, DN.Diagram_hasActiveVisualisation);
332 if (activeVisualisation != null) {
333 return DynamicVisualisations.colorBarOptions(graph, activeVisualisation);
336 LOGGER.debug("No visualisation folder available for model {}", model);
338 return ColorBarOptions.useDefault();
342 private static class ColoringObjectsRequest extends UnaryRead<Resource, Map<String,DynamicColorContribution>> {
344 public ColoringObjectsRequest(Resource diagram) {
349 public Map<String, DynamicColorContribution> perform(ReadGraph graph) throws DatabaseException {
350 DistrictNetworkResource DN = DistrictNetworkResource.getInstance(graph);
351 Resource model = graph.syncRequest(new IndexRoot(parameter));
352 Resource vf = DynamicVisualisations.getVisualisationFolder(graph, model);
354 Resource activeVisualisation = graph.getPossibleObject(vf, DN.Diagram_hasActiveVisualisation);
355 if (activeVisualisation != null) {
356 return DynamicVisualisations.colorContributions(graph, activeVisualisation);
359 LOGGER.debug("No visualisation folder available for model {}", model);
361 return Collections.emptyMap();
365 private static class ColoringObjectsListener implements Listener<Map<String,DynamicColorContribution>> {
367 private static final Logger LOGGER = LoggerFactory.getLogger(ColoringObjectsListener.class);
369 private Consumer<Map<String,DynamicColorContribution>> callback;
370 private Supplier<Boolean> isDisposed;
372 //private Map<String, DynamicColorContribution> lastResult
374 public ColoringObjectsListener(Consumer<Map<String,DynamicColorContribution>> callback, Supplier<Boolean> isDisposed) {
375 this.callback = callback;
376 this.isDisposed = isDisposed;
380 public void execute(Map<String,DynamicColorContribution> result) {
381 callback.accept(result);
385 public void exception(Throwable t) {
386 LOGGER.error("Could not listen ColorBarOptions", t);
390 public boolean isDisposed() {
391 return isDisposed.get();
395 private static class ColorBarOptionsListener implements Listener<ColorBarOptions> {
397 private static final Logger LOGGER = LoggerFactory.getLogger(ColorBarOptionsListener.class);
399 private Consumer<ColorBarOptions> callback;
400 private Supplier<Boolean> isDisposed;
402 public ColorBarOptionsListener(Consumer<ColorBarOptions> callback, Supplier<Boolean> isDisposed) {
403 this.callback = callback;
404 this.isDisposed = isDisposed;
408 public void execute(ColorBarOptions result) {
409 callback.accept(result);
413 public void exception(Throwable t) {
414 LOGGER.error("Could not listen ColorBarOptions", t);
418 public boolean isDisposed() {
419 return isDisposed.get();
423 private static class SizeBarOptionsRequest extends UnaryRead<Resource, SizeBarOptions> {
425 public SizeBarOptionsRequest(Resource diagram) {
430 public SizeBarOptions perform(ReadGraph graph) throws DatabaseException {
431 DistrictNetworkResource DN = DistrictNetworkResource.getInstance(graph);
432 Resource model = graph.syncRequest(new IndexRoot(parameter));
433 Resource vf = DynamicVisualisations.getVisualisationFolder(graph, model);
435 Resource activeVisualisation = graph.getPossibleObject(vf, DN.Diagram_hasActiveVisualisation);
436 if (activeVisualisation != null) {
437 return DynamicVisualisations.sizeBarOptions(graph, activeVisualisation);
440 LOGGER.debug("No visualisation folder available for model {}", model);
442 return SizeBarOptions.useDefault();
446 private static class SizeBarOptionsListener implements Listener<SizeBarOptions> {
448 private static final Logger LOGGER = LoggerFactory.getLogger(SizeBarOptionsListener.class);
450 private Consumer<SizeBarOptions> callback;
451 private Supplier<Boolean> isDisposed;
453 public SizeBarOptionsListener(Consumer<SizeBarOptions> callback, Supplier<Boolean> isDisposed) {
454 this.callback = callback;
455 this.isDisposed = isDisposed;
459 public void execute(SizeBarOptions result) {
460 callback.accept(result);
464 public void exception(Throwable t) {
465 LOGGER.error("Could not listen SizeBarOptions", t);
469 public boolean isDisposed() {
470 return isDisposed.get();
474 private static class SizingObjectsRequest extends UnaryRead<Resource, Map<String, DynamicSizeContribution>> {
476 public SizingObjectsRequest(Resource diagram) {
481 public Map<String, DynamicSizeContribution> perform(ReadGraph graph) throws DatabaseException {
482 DistrictNetworkResource DN = DistrictNetworkResource.getInstance(graph);
483 Resource model = graph.syncRequest(new IndexRoot(parameter));
484 Resource vf = DynamicVisualisations.getVisualisationFolder(graph, model);
486 Resource activeVisualisation = graph.getPossibleObject(vf, DN.Diagram_hasActiveVisualisation);
487 if (activeVisualisation != null) {
488 return DynamicVisualisations.sizeContributions(graph, activeVisualisation);
491 LOGGER.debug("No visualisation folder available for model {}", model);
493 return Collections.emptyMap();
497 private static class SizingObjectsListener implements Listener<Map<String,DynamicSizeContribution>> {
499 private static final Logger LOGGER = LoggerFactory.getLogger(SizingObjectsListener.class);
501 private Consumer<Map<String,DynamicSizeContribution>> callback;
502 private Supplier<Boolean> isDisposed;
504 public SizingObjectsListener(Consumer<Map<String, DynamicSizeContribution>> callback, Supplier<Boolean> isDisposed) {
505 this.callback = callback;
506 this.isDisposed = isDisposed;
510 public void execute(Map<String, DynamicSizeContribution> result) {
511 callback.accept(result);
515 public void exception(Throwable t) {
516 LOGGER.error("Could not listen ColorBarOptions", t);
520 public boolean isDisposed() {
521 return isDisposed.get();