X-Git-Url: https://gerrit.simantics.org/r/gitweb?a=blobdiff_plain;f=bundles%2Forg.simantics.modeling.ui%2Fsrc%2Forg%2Fsimantics%2Fmodeling%2Fui%2Fviewpoint%2FVariableChildren.java;fp=bundles%2Forg.simantics.modeling.ui%2Fsrc%2Forg%2Fsimantics%2Fmodeling%2Fui%2Fviewpoint%2FVariableChildren.java;h=4f244dcee57ced756337f559e9cc5e33720cf1c6;hb=969bd23cab98a79ca9101af33334000879fb60c5;hp=0000000000000000000000000000000000000000;hpb=866dba5cd5a3929bbeae85991796acb212338a08;p=simantics%2Fplatform.git

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
index 000000000..4f244dcee
--- /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;
+
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.Collections;
+import java.util.Set;
+
+import org.simantics.db.ReadGraph;
+import org.simantics.db.exception.DatabaseException;
+import org.simantics.db.layer0.variable.Variable;
+
+/**
+ * @author Tuukka Lehtonen
+ */
+public class VariableChildren {
+
+    /**
+     * @param graph
+     * @param variables
+     * @param requiredClassification
+     * @return
+     * @throws DatabaseException
+     */
+    public static Collection<Variable> filterByClassification(ReadGraph graph, Collection<Variable> variables, String requiredClassification) throws DatabaseException {
+        if (variables.isEmpty())
+            return Collections.emptyList();
+        Collection<Variable> result = null;
+        for (Variable v : variables) {
+            //System.out.println("child : " + v.getURI(graph));
+            Set<String> classifications = v.getClassifications(graph);
+            //System.out.println("classifications: " + classifications);
+            if (classifications != null && classifications.contains(requiredClassification)) {
+                if (result == null)
+                    result = new ArrayList<Variable>(variables.size());
+                result.add(v);
+            }
+        }
+        return result != null ? result : Collections.<Variable>emptyList();
+    }
+
+}