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

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