bundles/org.simantics.modeling.ui/src/org/simantics/modeling/ui/viewpoint/VariableChildren.java

   1 package org.simantics.modeling.ui.viewpoint;
   2
   3 import java.util.ArrayList;
   4 import java.util.Collection;
   5 import java.util.Collections;
   6 import java.util.Set;
   7
   8 import org.simantics.db.ReadGraph;
   9 import org.simantics.db.exception.DatabaseException;
  10 import org.simantics.db.layer0.variable.Variable;
  11
  12 /**
  13  * @author Tuukka Lehtonen
  14  */
  15 public class VariableChildren {
  16
  17     /**
  18      * @param graph
  19      * @param variables
  20      * @param requiredClassification
  21      * @return
  22      * @throws DatabaseException
  23      */
  24     public static Collection<Variable> filterByClassification(ReadGraph graph, Collection<Variable> variables, String requiredClassification) throws DatabaseException {
  25         if (variables.isEmpty())
  26             return Collections.emptyList();
  27         Collection<Variable> result = null;
  28         for (Variable v : variables) {
  29             //System.out.println("child : " + v.getURI(graph));
  30             Set<String> classifications = v.getClassifications(graph);
  31             //System.out.println("classifications: " + classifications);
  32             if (classifications != null && classifications.contains(requiredClassification)) {
  33                 if (result == null)
  34                     result = new ArrayList<Variable>(variables.size());
  35                 result.add(v);
  36             }
  37         }
  38         return result != null ? result : Collections.<Variable>emptyList();
  39     }
  40
  41 }