Migrated source code from Simantics SVN

[simantics/platform.git] / bundles / org.simantics.diagram / src / org / simantics / diagram / synchronization / graph / AddConnection.java

diff --git a/bundles/org.simantics.diagram/src/org/simantics/diagram/synchronization/graph/AddConnection.java b/bundles/org.simantics.diagram/src/org/simantics/diagram/synchronization/graph/AddConnection.java
new file mode 100644 (file)
index 0000000..5dcbd23
--- /dev/null
+++ b/bundles/org.simantics.diagram/src/org/simantics/diagram/synchronization/graph/AddConnection.java
@@ -0,0 +1,106 @@
+/*******************************************************************************\r
+ * Copyright (c) 2007, 2010 Association for Decentralized Information Management\r
+ * in Industry THTH ry.\r
+ * All rights reserved. This program and the accompanying materials\r
+ * are made available under the terms of the Eclipse Public License v1.0\r
+ * which accompanies this distribution, and is available at\r
+ * http://www.eclipse.org/legal/epl-v10.html\r
+ *\r
+ * Contributors:\r
+ *     VTT Technical Research Centre of Finland - initial API and implementation\r
+ *******************************************************************************/\r
+package org.simantics.diagram.synchronization.graph;\r
+\r
+import org.simantics.db.Resource;\r
+import org.simantics.db.WriteGraph;\r
+import org.simantics.db.common.utils.OrderedSetUtils;\r
+import org.simantics.db.exception.DatabaseException;\r
+import org.simantics.diagram.content.ConnectionUtil;\r
+import org.simantics.diagram.stubs.DiagramResource;\r
+import org.simantics.diagram.synchronization.CopyAdvisor;\r
+import org.simantics.diagram.synchronization.IModifiableSynchronizationContext;\r
+import org.simantics.diagram.synchronization.ModificationAdapter;\r
+import org.simantics.diagram.synchronization.SynchronizationHints;\r
+import org.simantics.diagram.synchronization.graph.layer.GraphLayerManager;\r
+import org.simantics.diagram.ui.DiagramModelHints;\r
+import org.simantics.g2d.diagram.DiagramHints;\r
+import org.simantics.g2d.diagram.DiagramMutator;\r
+import org.simantics.g2d.diagram.IDiagram;\r
+import org.simantics.g2d.element.ElementHints;\r
+import org.simantics.g2d.element.ElementUtils;\r
+import org.simantics.g2d.element.IElement;\r
+import org.simantics.layer0.Layer0;\r
+import org.simantics.structural.stubs.StructuralResource2;\r
+\r
+/**\r
+ * @author Tuukka Lehtonen\r
+ */\r
+public class AddConnection extends ModificationAdapter {\r
+\r
+    IModifiableSynchronizationContext context;\r
+    IDiagram                          diagram;\r
+    Resource                          diagramResource;\r
+    IElement                          element;\r
+    Resource                          copyOf;\r
+    Resource                          connectionClassResource;\r
+\r
+    public AddConnection(IModifiableSynchronizationContext context, IDiagram diagram, IElement element) {\r
+        super(ADD_CONNECTION_PRIORITY);\r
+        this.context = context;\r
+        this.diagram = diagram;\r
+        this.diagramResource = diagram.getHint(DiagramModelHints.KEY_DIAGRAM_RESOURCE);\r
+        assert this.diagramResource != null;\r
+        this.element = element;\r
+        this.copyOf = element.getHint(ElementHints.KEY_COPY_OF_OBJECT);\r
+\r
+        this.connectionClassResource = ElementUtils.checkedAdapt(element.getElementClass(), Resource.class);\r
+    }\r
+\r
+    @Override\r
+    public void perform(WriteGraph graph) throws DatabaseException {\r
+        ConnectionUtil cu = new ConnectionUtil(graph);\r
+        DiagramMutator mutator = diagram.getHint(DiagramHints.KEY_MUTATOR);\r
+\r
+        Resource resource = null;\r
+\r
+        // 3. Try to copy the element from an existing element if requested.\r
+        if (copyOf != null) {\r
+            CopyAdvisor ca = diagram.getHint(SynchronizationHints.COPY_ADVISOR);\r
+            if (ca != null) {\r
+                Resource sourceDiagram = graph.getPossibleObject(copyOf, Layer0.getInstance(graph).PartOf);\r
+                if (sourceDiagram == null)\r
+                    sourceDiagram = OrderedSetUtils.getSingleOwnerList(graph, copyOf, DiagramResource.getInstance(graph).Diagram);\r
+                resource = CopyAdvisorUtil.copy(context, graph, ca, copyOf, sourceDiagram, diagramResource);\r
+                if (resource != null) {\r
+                    OrderedSetUtils.addFirst(graph, diagramResource, resource);\r
+                }\r
+            }\r
+        }\r
+\r
+        if (resource == null)\r
+            resource = cu.newConnection(diagramResource, connectionClassResource);\r
+\r
+        if (copyOf != null)\r
+            copyConnectionType(graph, copyOf, resource);\r
+\r
+        element.setHint(ElementHints.KEY_OBJECT, resource);\r
+        mutator.register(element, resource);\r
+\r
+        GraphLayerManager glm = context.get(GraphSynchronizationHints.GRAPH_LAYER_MANAGER);\r
+        if (glm != null) {\r
+            glm.removeFromAllLayers(graph, resource);\r
+            glm.putElementOnVisibleLayers(diagram, graph, resource);\r
+        }\r
+    }\r
+\r
+    public static void copyConnectionType(WriteGraph graph, Resource source, Resource target) throws DatabaseException {\r
+        // Copy HasConnectionType relation\r
+        StructuralResource2 sr = StructuralResource2.getInstance(graph);\r
+        Resource connectionType = graph.getPossibleObject(source, sr.HasConnectionType);\r
+        if (connectionType != null) {\r
+            graph.deny(target, sr.HasConnectionType);\r
+            graph.claim(target, sr.HasConnectionType, null, connectionType);\r
+        }\r
+    }\r
+\r
+}\r