bundles/org.simantics.modeling.ui/src/org/simantics/modeling/ui/diagramEditor/DiagramEditorStates.java

   1 /*******************************************************************************\r
   2  * Copyright (c) 2007, 2011 Association for Decentralized Information Management in\r
   3  * Industry THTH ry.\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
   8  *\r
   9  * Contributors:\r
  10  *     VTT Technical Research Centre of Finland - initial API and implementation\r
  11  *******************************************************************************/\r
  12 package org.simantics.modeling.ui.diagramEditor;\r
  13 \r
  14 import java.util.Collections;\r
  15 import java.util.HashSet;\r
  16 import java.util.Set;\r
  17 \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
  45 \r
  46 /**\r
  47  * @author Tuukka Lehtonen\r
  48  */\r
  49 public class DiagramEditorStates {\r
  50 \r
  51     public static EditorState toEditorState(ICanvasContext ctx) {\r
  52         return toEditorState(ctx, true, true, true);\r
  53     }\r
  54 \r
  55     public static EditorState toEditorState(ICanvasContext ctx, boolean saveTransform, boolean saveSelection, boolean saveToolMode) {\r
  56         EditorState state = new EditorState();\r
  57 \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
  62         }\r
  63 \r
  64         if (saveToolMode) {\r
  65             IToolMode toolMode = ctx.getHintStack().getHint( Hints.KEY_TOOL );\r
  66             if (toolMode != null)\r
  67                 state.toolMode = toolMode.getId();\r
  68         }\r
  69 \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
  75         }\r
  76 \r
  77         return state;\r
  78     }\r
  79 \r
  80     public static void saveEditorState(String virtualGraph, Resource diagram, ICanvasContext ctx, ListenerSupport support) {\r
  81         saveEditorState( virtualGraph, diagram, toEditorState(ctx), support );\r
  82     }\r
  83 \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
  90         }\r
  91         return result;\r
  92     }\r
  93 \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
  99             if (e != null)\r
 100                 result.add(e);\r
 101         }\r
 102         return result;\r
 103     }\r
 104 \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
 107 \r
 108         final VirtualGraph vg = virtualGraph == null ? null\r
 109                 : session.getService(VirtualGraphSupport.class).getWorkspacePersistent(virtualGraph);\r
 110 \r
 111         session.asyncRequest(new WriteRequest() {\r
 112             @Override\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
 116                     return;\r
 117 \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
 122 \r
 123                 graph.syncRequest(new WriteRequest(vg) {\r
 124                     @Override\r
 125                     public void perform(WriteGraph graph) throws DatabaseException {\r
 126                         newEditorState(graph, diagram, editorState);\r
 127                     }\r
 128                 });\r
 129             }\r
 130         }, new Callback<DatabaseException>() {\r
 131             @Override\r
 132             public void run(DatabaseException parameter) {\r
 133                 if (parameter != null)\r
 134                     support.exception(parameter);\r
 135             }\r
 136         });\r
 137     }\r
 138 \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
 143 \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
 147 \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
 152         }\r
 153 \r
 154         if (editorState.toolMode != null)\r
 155             graph.claimLiteral(state, DIA.EditorState_ToolMode, L0.String, editorState.toolMode, Bindings.STRING);\r
 156 \r
 157         for (Resource selected : editorState.selection)\r
 158             graph.claim(state, DIA.EditorState_Selection, null, selected);\r
 159 \r
 160         return state;\r
 161     }\r
 162 \r
 163     public static EditorState readEditorState(final Resource source) throws DatabaseException {\r
 164         return Simantics.getSession().syncRequest(new ResourceRead<EditorState>(source) {\r
 165             @Override\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
 176                     }\r
 177                     break;\r
 178                 }\r
 179                 return state;\r
 180             }\r
 181         });\r
 182     }\r
 183 \r
 184 }\r