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

   1 package org.simantics.modeling.ui.viewpoint;\r
   2 \r
   3 import java.util.Collection;\r
   4 import java.util.Collections;\r
   5 \r
   6 import org.simantics.browsing.ui.model.children.ChildRule;\r
   7 import org.simantics.db.ReadGraph;\r
   8 import org.simantics.db.exception.DatabaseException;\r
   9 import org.simantics.db.layer0.variable.Variable;\r
  10 import org.simantics.structural.stubs.StructuralResource2;\r
  11 \r
  12 public class VariableChildRule implements ChildRule {\r
  13 \r
  14     @Override\r
  15     public boolean isCompatible(Class<?> contentType) {\r
  16         return contentType.equals(Variable.class);\r
  17     }\r
  18 \r
  19     @Override\r
  20     public Collection<?> getChildren(ReadGraph graph, Object parent)\r
  21             throws DatabaseException {\r
  22         Collection<Variable> children = ((Variable)parent).getChildren(graph);\r
  23         return VariableChildren.filterByClassification(graph, children, StructuralResource2.URIs.Component);\r
  24     }\r
  25 \r
  26     @Override\r
  27     public Collection<?> getParents(ReadGraph graph, Object child)\r
  28             throws DatabaseException {\r
  29         Variable parent = ((Variable)child).getParent(graph);\r
  30         if(parent == null)\r
  31             return Collections.emptyList();\r
  32         else\r
  33             return Collections.singleton(parent);\r
  34     }\r
  35 \r
  36 \r
  37 }\r