Migrated source code from Simantics SVN

[simantics/platform.git] / bundles / org.simantics.structural2 / src / org / simantics / structural2 / queries / ConnectionSet.java

diff --git a/bundles/org.simantics.structural2/src/org/simantics/structural2/queries/ConnectionSet.java b/bundles/org.simantics.structural2/src/org/simantics/structural2/queries/ConnectionSet.java
new file mode 100644 (file)
index 0000000..9591d8d
--- /dev/null
+++ b/bundles/org.simantics.structural2/src/org/simantics/structural2/queries/ConnectionSet.java
@@ -0,0 +1,102 @@
+/*******************************************************************************\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.structural2.queries;\r
+\r
+import gnu.trove.set.hash.THashSet;\r
+\r
+import java.util.Collection;\r
+import java.util.Collections;\r
+import java.util.Set;\r
+\r
+import org.simantics.db.ReadGraph;\r
+import org.simantics.db.Resource;\r
+import org.simantics.db.exception.DatabaseException;\r
+import org.simantics.db.service.CollectionSupport;\r
+import org.simantics.structural.stubs.StructuralResource2;\r
+\r
+public class ConnectionSet {\r
+       \r
+    final Set<Resource> connections;\r
+    THashSet<Resource> joins;\r
+    THashSet<Resource> excluded;\r
+    THashSet<Resource> excludedJoins;\r
+    \r
+    StructuralResource2 sr;\r
+    \r
+//    public ConnectionSet() {\r
+//     \r
+//    }\r
+    \r
+    public ConnectionSet(ReadGraph graph) {\r
+       sr = StructuralResource2.getInstance(graph);\r
+       connections = graph.getService(CollectionSupport.class).createSet();\r
+    }\r
+\r
+    public boolean excludeConnection(Resource connection) {\r
+        if (excluded == null)\r
+            excluded = new THashSet<Resource>();\r
+        return excluded.add(connection);\r
+    }\r
+\r
+    public boolean excludeJoin(Resource join) {\r
+        if (excludedJoins == null)\r
+               excludedJoins = new THashSet<Resource>();\r
+        return excludedJoins.add(join);\r
+    }\r
+    \r
+    public boolean excludeConnections(Collection<Resource> connections) {\r
+        if (connections.isEmpty())\r
+            return false;\r
+        if (excluded == null)\r
+            excluded = new THashSet<Resource>();\r
+        return excluded.addAll(connections);\r
+    }\r
+\r
+    public boolean isExcluded(Resource connection) {\r
+        return excluded != null && excluded.contains(connection);\r
+    }\r
+\r
+    public boolean isJoinExcluded(Resource join) {\r
+        return excludedJoins != null && excludedJoins.contains(join);\r
+    }\r
+\r
+    public void addConnection(ReadGraph g, Resource connection) throws DatabaseException {\r
+        if(isExcluded(connection))\r
+            return;\r
+        if(connections.add(connection)) {\r
+               if(sr == null) sr = StructuralResource2.getInstance(g);\r
+            for(Resource join : g.getObjects(connection, sr.IsJoinedBy))\r
+                addJoin(g, join);\r
+        }\r
+    }\r
+\r
+    public void addJoin(ReadGraph g, Resource join) throws DatabaseException {\r
+        if(isJoinExcluded(join))\r
+            return;\r
+        if (joins == null)\r
+               joins = new THashSet<Resource>();\r
+        if(joins.add(join)) {\r
+               if(sr == null) sr = StructuralResource2.getInstance(g);\r
+            for(Resource connection : g.getObjects(join, sr.Joins))\r
+                addConnection(g, connection);\r
+        }\r
+    }\r
+\r
+    public Set<Resource> getConnections() {\r
+        return connections;\r
+    }\r
+\r
+    public Set<Resource> getJoins() {\r
+       if(joins == null) return Collections.emptySet();\r
+        return joins;\r
+    }\r
+}\r