bundles/org.simantics.diagram/src/org/simantics/diagram/symbolcontribution/SCLFilter.java

   1 package org.simantics.diagram.symbolcontribution;\r
   2 \r
   3 import org.simantics.db.ReadGraph;\r
   4 import org.simantics.db.Resource;\r
   5 import org.simantics.db.exception.DatabaseException;\r
   6 import org.simantics.db.layer0.variable.Variable;\r
   7 import org.simantics.db.layer0.variable.Variables;\r
   8 import org.simantics.diagram.stubs.DiagramResource;\r
   9 import org.simantics.scl.runtime.SCLContext;\r
  10 import org.simantics.scl.runtime.function.Function3;\r
  11 \r
  12 /**\r
  13  * @author Antti Villberg\r
  14  */\r
  15 public class SCLFilter implements SymbolContributionFilter {\r
  16         \r
  17         Resource filter;\r
  18         \r
  19         public SCLFilter(Resource filter) {\r
  20                 this.filter = filter;\r
  21         }\r
  22 \r
  23         @Override\r
  24         public boolean accept(ReadGraph graph, Resource contribution, Resource filter, Resource diagram) throws DatabaseException {\r
  25                 \r
  26         DiagramResource DIA = DiagramResource.getInstance(graph);\r
  27         \r
  28         Variable ruleVariable = Variables.getVariable(graph, filter);\r
  29 \r
  30         Function3<Resource,Resource, Resource, Boolean> fn = ruleVariable.getPossiblePropertyValue(graph, DIA.SCLFilter_filter);\r
  31         if(fn == null) return true;\r
  32 \r
  33         SCLContext sclContext = SCLContext.getCurrent();\r
  34         Object oldGraph = sclContext.get("graph");\r
  35         try {\r
  36             sclContext.put("graph", graph);\r
  37             return fn.apply(contribution, filter, diagram);\r
  38         } catch (Throwable t) {\r
  39                 throw new DatabaseException(t);\r
  40         } finally {\r
  41             sclContext.put("graph", oldGraph);\r
  42         }\r
  43                 \r
  44         }\r
  45 \r
  46 }\r