Migrated source code from Simantics SVN

[simantics/platform.git] / bundles / org.simantics.diagram / src / org / simantics / diagram / symbolcontribution / SCLFilter.java

diff --git a/bundles/org.simantics.diagram/src/org/simantics/diagram/symbolcontribution/SCLFilter.java b/bundles/org.simantics.diagram/src/org/simantics/diagram/symbolcontribution/SCLFilter.java
new file mode 100644 (file)
index 0000000..d819de7
--- /dev/null
+++ b/bundles/org.simantics.diagram/src/org/simantics/diagram/symbolcontribution/SCLFilter.java
@@ -0,0 +1,46 @@
+package org.simantics.diagram.symbolcontribution;\r
+\r
+import org.simantics.db.ReadGraph;\r
+import org.simantics.db.Resource;\r
+import org.simantics.db.exception.DatabaseException;\r
+import org.simantics.db.layer0.variable.Variable;\r
+import org.simantics.db.layer0.variable.Variables;\r
+import org.simantics.diagram.stubs.DiagramResource;\r
+import org.simantics.scl.runtime.SCLContext;\r
+import org.simantics.scl.runtime.function.Function3;\r
+\r
+/**\r
+ * @author Antti Villberg\r
+ */\r
+public class SCLFilter implements SymbolContributionFilter {\r
+       \r
+       Resource filter;\r
+       \r
+       public SCLFilter(Resource filter) {\r
+               this.filter = filter;\r
+       }\r
+\r
+       @Override\r
+       public boolean accept(ReadGraph graph, Resource contribution, Resource filter, Resource diagram) throws DatabaseException {\r
+               \r
+        DiagramResource DIA = DiagramResource.getInstance(graph);\r
+        \r
+        Variable ruleVariable = Variables.getVariable(graph, filter);\r
+\r
+        Function3<Resource,Resource, Resource, Boolean> fn = ruleVariable.getPossiblePropertyValue(graph, DIA.SCLFilter_filter);\r
+        if(fn == null) return true;\r
+\r
+        SCLContext sclContext = SCLContext.getCurrent();\r
+        Object oldGraph = sclContext.get("graph");\r
+        try {\r
+            sclContext.put("graph", graph);\r
+            return fn.apply(contribution, filter, diagram);\r
+        } catch (Throwable t) {\r
+               throw new DatabaseException(t);\r
+        } finally {\r
+            sclContext.put("graph", oldGraph);\r
+        }\r
+               \r
+       }\r
+\r
+}\r