Migrated source code from Simantics SVN

[simantics/platform.git] / bundles / org.simantics.structural2 / src / org / simantics / structural2 / modelingRules / AllowedConnectionTypes.java

diff --git a/bundles/org.simantics.structural2/src/org/simantics/structural2/modelingRules/AllowedConnectionTypes.java b/bundles/org.simantics.structural2/src/org/simantics/structural2/modelingRules/AllowedConnectionTypes.java
new file mode 100644 (file)
index 0000000..da11286
--- /dev/null
+++ b/bundles/org.simantics.structural2/src/org/simantics/structural2/modelingRules/AllowedConnectionTypes.java
@@ -0,0 +1,30 @@
+package org.simantics.structural2.modelingRules;\r
+\r
+import java.util.Collection;\r
+import java.util.HashSet;\r
+import java.util.Set;\r
+\r
+import org.simantics.db.ReadGraph;\r
+import org.simantics.db.Resource;\r
+import org.simantics.db.common.request.ResourceRead;\r
+import org.simantics.db.exception.DatabaseException;\r
+import org.simantics.structural.stubs.StructuralResource2;\r
+\r
+public class AllowedConnectionTypes extends ResourceRead<Collection<Resource>> {\r
+\r
+       public AllowedConnectionTypes(Resource connectionPoint) {\r
+               super(connectionPoint);\r
+       }\r
+\r
+       @Override\r
+       public Collection<Resource> perform(ReadGraph graph) throws DatabaseException {\r
+               StructuralResource2 STR = StructuralResource2.getInstance(graph);\r
+               Collection<Resource> types = graph.getObjects(resource, STR.AllowsConnectionType);\r
+               Set<Resource> result = new HashSet<Resource>(types);\r
+               for(Resource type : types)\r
+                       for(Resource override : graph.getObjects(type, STR.OverridesConnectionType))\r
+                               result.remove(override);\r
+               return result;\r
+       }\r
+\r
+}\r