Added support for dynamically typed data access and a Variable interface.

[simantics/python.git] / org.simantics.pythonlink / src / org / simantics / pythonlink / Python.java

diff --git a/org.simantics.pythonlink/src/org/simantics/pythonlink/Python.java b/org.simantics.pythonlink/src/org/simantics/pythonlink/Python.java
index 07921fd2c530b10a4850544db2f5720a8f814dff..9d00c447a1dc81b67a8349ca7f9129506ef4702f 100644 (file)
--- a/org.simantics.pythonlink/src/org/simantics/pythonlink/Python.java
+++ b/org.simantics.pythonlink/src/org/simantics/pythonlink/Python.java
@@ -1,8 +1,20 @@
 package org.simantics.pythonlink;\r
 \r
+import java.util.List;\r
+import java.util.concurrent.TimeUnit;\r
+\r
+import org.simantics.db.layer0.variable.NodeSupport;\r
+import org.simantics.db.layer0.variable.StandardGraphChildVariable;\r
+import org.simantics.db.layer0.variable.Variable;\r
+import org.simantics.db.layer0.variable.VariableNode;\r
+import org.simantics.pythonlink.PythonContext.Listener;\r
+import org.simantics.pythonlink.variable.PythonNode;\r
+import org.simantics.pythonlink.variable.PythonNodeManager;\r
 import org.simantics.scl.runtime.SCLContext;\r
 import org.simantics.scl.runtime.function.Function;\r
 import org.simantics.scl.runtime.tuple.Tuple0;\r
+import org.simantics.simulator.variable.NodeManager;\r
+import org.simantics.simulator.variable.exceptions.NodeManagerException;\r
 \r
 public class Python {\r
     private static final String PYTHON_CONTEXT = "Simantics/Python/Python";\r
@@ -11,6 +23,50 @@ public class Python {
         return new PythonContext();\r
     }\r
     \r
+    public static Variable getPythonContextVariable(PythonContext context) {\r
+       NodeManager<PythonNode> nodeManager = new PythonNodeManager(context);\r
+       PythonNode root;\r
+               try {\r
+                       root = nodeManager.getNode("/");\r
+               } catch (NodeManagerException e) {\r
+                       // Should not happen\r
+                       throw new RuntimeException("Getting root Python node failed");\r
+               }\r
+               \r
+               final NodeSupport<PythonNode> support = new NodeSupport<PythonNode>(nodeManager, 0, TimeUnit.NANOSECONDS);\r
+               \r
+               context.addListener(new Listener() {\r
+            @Override\r
+            public void updated(String variableName) {\r
+                try {\r
+                    PythonNode root = nodeManager.getNode("/");\r
+                    if (variableName != null) {\r
+                        PythonNode node = nodeManager.getNode(variableName);\r
+                        if (node != null) support.valueCache.removeListening(node);\r
+                        support.structureCache.removeListening(root);\r
+                    }\r
+                    else {\r
+                        List<PythonNode> props = nodeManager.getProperties(root);\r
+                        for (PythonNode p : props)\r
+                            support.valueCache.removeListening(p);\r
+                        support.structureCache.removeListening(root);\r
+                    }\r
+                    \r
+                    support.valueCache.clearExpired();\r
+                    support.structureCache.clearExpired();\r
+                } catch (NodeManagerException e) {\r
+                    e.printStackTrace();\r
+                }\r
+            }\r
+            \r
+            @Override\r
+            public void closed() {\r
+            }\r
+        });\r
+               \r
+       return new StandardGraphChildVariable(null, new VariableNode<PythonNode>(support, root), null);\r
+    }\r
+    \r
     @SuppressWarnings( { "unchecked", "rawtypes" } )\r
     public static Object runPythonF(Function f) {\r
         SCLContext sclContext = SCLContext.getCurrent();\r