1 /*******************************************************************************
\r
2 * Copyright (c) 2007, 2011 Association for Decentralized Information Management in
\r
4 * All rights reserved. This program and the accompanying materials
\r
5 * are made available under the terms of the Eclipse Public License v1.0
\r
6 * which accompanies this distribution, and is available at
\r
7 * http://www.eclipse.org/legal/epl-v10.html
\r
10 * VTT Technical Research Centre of Finland - initial API and implementation
\r
11 *******************************************************************************/
\r
12 package org.simantics.modeling.ui.diagramEditor;
\r
14 import java.util.Collections;
\r
15 import java.util.HashSet;
\r
16 import java.util.Set;
\r
18 import org.simantics.Simantics;
\r
19 import org.simantics.databoard.Bindings;
\r
20 import org.simantics.db.ReadGraph;
\r
21 import org.simantics.db.Resource;
\r
22 import org.simantics.db.Session;
\r
23 import org.simantics.db.VirtualGraph;
\r
24 import org.simantics.db.WriteGraph;
\r
25 import org.simantics.db.common.procedure.adapter.ListenerSupport;
\r
26 import org.simantics.db.common.request.ResourceRead;
\r
27 import org.simantics.db.common.request.WriteRequest;
\r
28 import org.simantics.db.exception.DatabaseException;
\r
29 import org.simantics.db.layer0.util.Layer0Utils;
\r
30 import org.simantics.db.layer0.util.RemoverUtil;
\r
31 import org.simantics.db.service.VirtualGraphSupport;
\r
32 import org.simantics.diagram.stubs.DiagramResource;
\r
33 import org.simantics.diagram.stubs.G2DResource;
\r
34 import org.simantics.diagram.synchronization.graph.DiagramGraphUtil;
\r
35 import org.simantics.g2d.canvas.Hints;
\r
36 import org.simantics.g2d.canvas.ICanvasContext;
\r
37 import org.simantics.g2d.canvas.IToolMode;
\r
38 import org.simantics.g2d.diagram.IDiagram;
\r
39 import org.simantics.g2d.diagram.handler.DataElementMap;
\r
40 import org.simantics.g2d.diagram.participant.Selection;
\r
41 import org.simantics.g2d.element.ElementUtils;
\r
42 import org.simantics.g2d.element.IElement;
\r
43 import org.simantics.layer0.Layer0;
\r
44 import org.simantics.utils.datastructures.Callback;
\r
47 * @author Tuukka Lehtonen
\r
49 public class DiagramEditorStates {
\r
51 public static EditorState toEditorState(ICanvasContext ctx) {
\r
52 return toEditorState(ctx, true, true, true);
\r
55 public static EditorState toEditorState(ICanvasContext ctx, boolean saveTransform, boolean saveSelection, boolean saveToolMode) {
\r
56 EditorState state = new EditorState();
\r
58 if (saveTransform) {
\r
59 state.viewTransform = ctx.getHintStack().getHint( Hints.KEY_CANVAS_TRANSFORM );
\r
60 if (state.viewTransform != null && state.viewTransform.getDeterminant() == 0)
\r
61 state.viewTransform = null;
\r
65 IToolMode toolMode = ctx.getHintStack().getHint( Hints.KEY_TOOL );
\r
66 if (toolMode != null)
\r
67 state.toolMode = toolMode.getId();
\r
70 if (saveSelection) {
\r
71 Set<IElement> selection = Collections.emptySet();
\r
72 for (Selection s : ctx.getItemsByClass( Selection.class ))
\r
73 selection = s.getSelection(0);
\r
74 state.selection = toResources( selection );
\r
80 public static void saveEditorState(String virtualGraph, Resource diagram, ICanvasContext ctx, ListenerSupport support) {
\r
81 saveEditorState( virtualGraph, diagram, toEditorState(ctx), support );
\r
84 public static Set<Resource> toResources(Set<IElement> elements) {
\r
85 Set<Resource> result = new HashSet<Resource>();
\r
86 for (IElement e : elements) {
\r
87 Object obj = ElementUtils.getObject(e);
\r
88 if (obj instanceof Resource)
\r
89 result.add((Resource) obj);
\r
94 public static Set<IElement> toElements(Set<Resource> selection, IDiagram diagram) {
\r
95 Set<IElement> result = new HashSet<IElement>();
\r
96 DataElementMap dem = diagram.getDiagramClass().getAtMostOneItemOfClass(DataElementMap.class);
\r
97 for (Resource selected : selection) {
\r
98 IElement e = dem.getElement(diagram, selected);
\r
105 public static void saveEditorState(final String virtualGraph, final Resource diagram, final EditorState editorState, final ListenerSupport support) {
\r
106 Session session = Simantics.getSession();
\r
108 final VirtualGraph vg = virtualGraph == null ? null
\r
109 : session.getService(VirtualGraphSupport.class).getWorkspacePersistent(virtualGraph);
\r
111 session.asyncRequest(new WriteRequest() {
\r
113 public void perform(WriteGraph graph) throws DatabaseException {
\r
114 // Prevent the state writing from failing in RemoverUtil.remove.
\r
115 if (Layer0Utils.isContainerPublished(graph, diagram))
\r
118 // Remove all previous editor states.
\r
119 DiagramResource DIA = DiagramResource.getInstance(graph);
\r
120 for (Resource state : graph.getObjects(diagram, DIA.HasEditorState))
\r
121 RemoverUtil.remove(graph, state);
\r
123 graph.syncRequest(new WriteRequest(vg) {
\r
125 public void perform(WriteGraph graph) throws DatabaseException {
\r
126 newEditorState(graph, diagram, editorState);
\r
130 }, new Callback<DatabaseException>() {
\r
132 public void run(DatabaseException parameter) {
\r
133 if (parameter != null)
\r
134 support.exception(parameter);
\r
139 private static Resource newEditorState(WriteGraph graph, Resource stateOf, EditorState editorState) throws DatabaseException {
\r
140 Layer0 L0 = Layer0.getInstance(graph);
\r
141 G2DResource G2D = G2DResource.getInstance(graph);
\r
142 DiagramResource DIA = DiagramResource.getInstance(graph);
\r
144 Resource state = graph.newResource();
\r
145 graph.claim(state, L0.InstanceOf, null, DIA.EditorState);
\r
146 graph.claim(stateOf, DIA.HasEditorState, state);
\r
148 if (editorState.viewTransform != null) {
\r
149 double[] matrix = new double[6];
\r
150 editorState.viewTransform.getMatrix(matrix);
\r
151 graph.claimLiteral(state, DIA.EditorState_ViewTransform, G2D.Transform, matrix, Bindings.DOUBLE_ARRAY);
\r
154 if (editorState.toolMode != null)
\r
155 graph.claimLiteral(state, DIA.EditorState_ToolMode, L0.String, editorState.toolMode, Bindings.STRING);
\r
157 for (Resource selected : editorState.selection)
\r
158 graph.claim(state, DIA.EditorState_Selection, null, selected);
\r
163 public static EditorState readEditorState(final Resource source) throws DatabaseException {
\r
164 return Simantics.getSession().syncRequest(new ResourceRead<EditorState>(source) {
\r
166 public EditorState perform(ReadGraph graph) throws DatabaseException {
\r
167 EditorState state = new EditorState();
\r
168 DiagramResource DIA = DiagramResource.getInstance(graph);
\r
169 for (Resource editorState : graph.getObjects(source, DIA.HasEditorState)) {
\r
170 state.viewTransform = DiagramGraphUtil.getAffineTransform(graph, editorState, DIA.EditorState_ViewTransform, false);
\r
171 state.toolMode = graph.getPossibleRelatedValue(editorState, DIA.EditorState_ToolMode, Bindings.STRING);
\r
172 state.selection = new HashSet<Resource>();
\r
173 for (Resource selected : graph.getObjects(editorState, DIA.EditorState_Selection)) {
\r
174 if (graph.isInstanceOf(selected, DIA.Element))
\r
175 state.selection.add(selected);
\r