Migrated source code from Simantics SVN

[simantics/platform.git] / bundles / org.simantics.diagram / src / org / simantics / diagram / adapter / ConnectionInfoRequest.java

diff --git a/bundles/org.simantics.diagram/src/org/simantics/diagram/adapter/ConnectionInfoRequest.java b/bundles/org.simantics.diagram/src/org/simantics/diagram/adapter/ConnectionInfoRequest.java
new file mode 100644 (file)
index 0000000..6b27eb2
--- /dev/null
+++ b/bundles/org.simantics.diagram/src/org/simantics/diagram/adapter/ConnectionInfoRequest.java
@@ -0,0 +1,129 @@
+/*******************************************************************************\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.adapter;\r
+\r
+import java.util.concurrent.atomic.AtomicInteger;\r
+\r
+import org.simantics.db.AsyncReadGraph;\r
+import org.simantics.db.Resource;\r
+import org.simantics.db.common.request.UnaryAsyncRead;\r
+import org.simantics.db.procedure.AsyncProcedure;\r
+import org.simantics.diagram.connection.ConnectionSegmentEnd;\r
+import org.simantics.diagram.connection.ConnectionSegmentType;\r
+import org.simantics.diagram.content.EdgeResource;\r
+import org.simantics.diagram.stubs.DiagramResource;\r
+\r
+/**\r
+ * @author Antti Villberg\r
+ *\r
+ */\r
+public class ConnectionInfoRequest extends UnaryAsyncRead<EdgeResource, ConnectionInfo> {\r
+\r
+    static class SingleConnectionInfo {\r
+        public Resource connection;\r
+        public ConnectionSegmentEnd end;\r
+    }\r
+\r
+    final DiagramResource dr;\r
+\r
+    private SingleConnectionInfo first;\r
+    private SingleConnectionInfo second;\r
+\r
+    public ConnectionInfoRequest(DiagramResource dr, EdgeResource resource) {\r
+        super(resource);\r
+        this.dr = dr;\r
+    }\r
+\r
+    public void fillPossibleType(AsyncReadGraph graph, Resource resource, final AsyncProcedure<ConnectionInfo> procedure, final ConnectionInfo result) {\r
+        if (resource == null) {\r
+            procedure.execute(graph, result);\r
+        } else {\r
+            graph.forSingleType(resource, dr.Connection, new AsyncProcedure<Resource>() {\r
+                @Override\r
+                public void exception(AsyncReadGraph graph, Throwable throwable) {\r
+                    procedure.exception(graph, throwable);\r
+                }\r
+\r
+                @Override\r
+                public void execute(AsyncReadGraph graph, Resource type) {\r
+                    result.connectionType = type;\r
+                    procedure.execute(graph, result);\r
+                }\r
+            });\r
+        }\r
+    }\r
+\r
+    ConnectionInfo createConnectionInfo(Resource connection) {\r
+        ConnectionInfo info = new ConnectionInfo();\r
+        info.connection = connection;\r
+        info.firstEnd = first.end;\r
+        info.secondEnd = second.end;\r
+        info.segmentType = ConnectionSegmentType.toSegmentType(first.end, second.end);\r
+        return info;\r
+    }\r
+\r
+    public void isBranchPointOf(AsyncReadGraph graph, Resource resource, final AsyncProcedure<ConnectionInfo> procedure, final SingleConnectionInfo result, final AtomicInteger ready) {\r
+\r
+        graph.forPossibleObject(resource, dr.IsBranchPointOf, new AsyncProcedure<Resource>() {\r
+            @Override\r
+            public void exception(AsyncReadGraph graph, Throwable throwable) {\r
+                procedure.exception(graph, throwable);\r
+            }\r
+\r
+            @Override\r
+            public void execute(AsyncReadGraph graph, Resource connection) {\r
+                result.end = ConnectionSegmentEnd.BRANCH;\r
+                result.connection = connection;\r
+                if (ready.decrementAndGet() == 0) {\r
+                    ConnectionInfo info = createConnectionInfo(connection);\r
+                    fillPossibleType(graph, connection, procedure, info);\r
+                }\r
+            }\r
+        });\r
+\r
+    }\r
+\r
+    public void isConnectorOf(AsyncReadGraph graph, final Resource resource, final AsyncProcedure<ConnectionInfo> procedure, final SingleConnectionInfo result, final AtomicInteger ready) {\r
+\r
+        graph.forPossibleObject(resource, dr.IsConnectorOf, new AsyncProcedure<Resource>() {\r
+            @Override\r
+            public void exception(AsyncReadGraph graph, Throwable throwable) {\r
+                procedure.exception(graph, throwable);\r
+            }\r
+\r
+            @Override\r
+            public void execute(AsyncReadGraph graph, Resource connection) {\r
+                if (connection != null) {\r
+                    result.end = ConnectionSegmentEnd.CONNECTOR;\r
+                    result.connection = connection;\r
+                    if (ready.decrementAndGet() == 0) {\r
+                        ConnectionInfo info = createConnectionInfo(connection);\r
+                        fillPossibleType(graph, connection, procedure, info);\r
+                    }\r
+                } else {\r
+                    isBranchPointOf(graph, resource, procedure, result, ready);\r
+                }\r
+            }\r
+        });\r
+\r
+    }\r
+\r
+    @Override\r
+    public void perform(AsyncReadGraph graph, AsyncProcedure<ConnectionInfo> procedure) {\r
+        AtomicInteger ready = new AtomicInteger(2);\r
+        first = new SingleConnectionInfo();\r
+        second = new SingleConnectionInfo();\r
+        isConnectorOf(graph, parameter.first(), procedure, first, ready);\r
+        isConnectorOf(graph, parameter.second(), procedure, second, ready);\r
+    }\r
+\r
+}\r