Migrated source code from Simantics SVN

[simantics/platform.git] / bundles / org.simantics.modeling.ui / src / org / simantics / modeling / ui / viewpoint / VariableChildren.java

diff --git a/bundles/org.simantics.modeling.ui/src/org/simantics/modeling/ui/viewpoint/VariableChildren.java b/bundles/org.simantics.modeling.ui/src/org/simantics/modeling/ui/viewpoint/VariableChildren.java
new file mode 100644 (file)
index 0000000..4f244dc
--- /dev/null
+++ b/bundles/org.simantics.modeling.ui/src/org/simantics/modeling/ui/viewpoint/VariableChildren.java
@@ -0,0 +1,41 @@
+package org.simantics.modeling.ui.viewpoint;\r
+\r
+import java.util.ArrayList;\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.exception.DatabaseException;\r
+import org.simantics.db.layer0.variable.Variable;\r
+\r
+/**\r
+ * @author Tuukka Lehtonen\r
+ */\r
+public class VariableChildren {\r
+\r
+    /**\r
+     * @param graph\r
+     * @param variables\r
+     * @param requiredClassification\r
+     * @return\r
+     * @throws DatabaseException\r
+     */\r
+    public static Collection<Variable> filterByClassification(ReadGraph graph, Collection<Variable> variables, String requiredClassification) throws DatabaseException {\r
+        if (variables.isEmpty())\r
+            return Collections.emptyList();\r
+        Collection<Variable> result = null;\r
+        for (Variable v : variables) {\r
+            //System.out.println("child : " + v.getURI(graph));\r
+            Set<String> classifications = v.getClassifications(graph);\r
+            //System.out.println("classifications: " + classifications);\r
+            if (classifications != null && classifications.contains(requiredClassification)) {\r
+                if (result == null)\r
+                    result = new ArrayList<Variable>(variables.size());\r
+                result.add(v);\r
+            }\r
+        }\r
+        return result != null ? result : Collections.<Variable>emptyList();\r
+    }\r
+\r
+}\r