Support pending values in NodeManager

[simantics/platform.git] / bundles / org.simantics.simulator.toolkit / src / org / simantics / simulator / toolkit / StandardNodeManager.java

diff --git a/bundles/org.simantics.simulator.toolkit/src/org/simantics/simulator/toolkit/StandardNodeManager.java b/bundles/org.simantics.simulator.toolkit/src/org/simantics/simulator/toolkit/StandardNodeManager.java
index 87ae4f32f3b7394d1e8168c705c2584ba0ae3f8c..e09b3136e6307c7750f81f2ab1271ed340e4a2b4 100644 (file)
--- a/bundles/org.simantics.simulator.toolkit/src/org/simantics/simulator/toolkit/StandardNodeManager.java
+++ b/bundles/org.simantics.simulator.toolkit/src/org/simantics/simulator/toolkit/StandardNodeManager.java
@@ -13,6 +13,7 @@
 package org.simantics.simulator.toolkit;
 
 import java.util.ArrayList;
+import java.util.Collections;
 import java.util.List;
 import java.util.Map;
 import java.util.Set;
@@ -43,15 +44,15 @@ import gnu.trove.set.hash.THashSet;
 /**
  * StandardNodeManager gives default implementations to some methods
  * of NodeManager.
- * 
+ *
  * @author Antti Villberg
  */
-public abstract class StandardNodeManager<Node, Engine extends StandardNodeManagerSupport<Node>> implements NodeManager<Node> {
+public class StandardNodeManager<Node, Engine extends StandardNodeManagerSupport<Node>> implements NodeManager<Node> {
 
     private static final Logger LOGGER = LoggerFactory.getLogger(StandardNodeManager.class);
 
-    private final Node root;
-    private final StandardRealm<Node,Engine> realm;
+    protected final Node root;
+    protected final StandardRealm<Node,Engine> realm;
 
     static final Binding NO_BINDING = new VariantBinding() {
 
@@ -108,15 +109,15 @@ public abstract class StandardNodeManager<Node, Engine extends StandardNodeManag
                     return  System.identityHashCode(o1);
                 } else {
                     if(o1.equals(o2)) return 0;
-                    return System.identityHashCode(o1) - System.identityHashCode(o2);  
+                    return System.identityHashCode(o1) - System.identityHashCode(o2);
                 }
             }
         }
 
     };
 
-    THashMap<Node, Variant> valueCache = new THashMap<>(); 
-    protected THashMap<Node, THashSet<Runnable>> listeners = new THashMap<>(); 
+    protected THashMap<Node, Variant> valueCache = new THashMap<>();
+    protected THashMap<Node, THashSet<Runnable>> listeners = new THashMap<>();
 
     AtomicBoolean fireNodeListenersScheduled = new AtomicBoolean(false);
     Runnable fireNodeListeners = new Runnable() {
@@ -136,7 +137,7 @@ public abstract class StandardNodeManager<Node, Engine extends StandardNodeManag
         }
     };
 
-    Runnable clearValueCache = () -> valueCache.clear(); 
+    Runnable clearValueCache = () -> valueCache.clear();
 
     public StandardNodeManager(StandardRealm<Node,Engine> realm, Node root) {
         assert(realm != null);
@@ -193,7 +194,10 @@ public abstract class StandardNodeManager<Node, Engine extends StandardNodeManag
     @Override
     public Object getValue(Node node, Binding binding) throws NodeManagerException, BindingException {
         try {
-            return getValue(node).getValue(binding);
+            Variant value = getValue(node);
+            if(NodeManager.PENDING_NODE_VALUE == value)
+                return value;
+            return value.getValue(binding);
         } catch (AdaptException e) {
             throw new BindingException(e);
         }
@@ -263,6 +267,20 @@ public abstract class StandardNodeManager<Node, Engine extends StandardNodeManag
         }
     }
 
+    public void refreshVariable(Node node) {
+        realm.asyncExec(() -> {
+            valueCache.remove(node);
+            synchronized(listeners) {
+                THashSet<Runnable> runnables = listeners.get(node);
+                if (runnables != null) {
+                    for (Runnable r : runnables) {
+                        r.run();
+                    }
+                }
+            }
+        });
+    }
+
     public void refreshVariables() {
         realm.asyncExec(clearValueCache);
         fireNodeListeners();
@@ -281,6 +299,8 @@ public abstract class StandardNodeManager<Node, Engine extends StandardNodeManag
         Variant variant = valueCache.get(node);
         if(variant == null) {
             Object value = realm.getEngine().getEngineValue(node);
+            if(NodeManager.PENDING_NODE_VALUE == value)
+                return (Variant)value;
             Binding binding = realm.getEngine().getEngineBinding(node);
             variant = new Variant(binding, value);
             valueCache.put(node, variant);
@@ -308,6 +328,23 @@ public abstract class StandardNodeManager<Node, Engine extends StandardNodeManag
     @Override
     public void setValue(Node node, Object value, Binding binding)
             throws NodeManagerException {
+       updateValueInner(node, value, binding);
+        refreshVariable(node);
+    }
+
+    //Update the value of the node and remove from valueCache only the references nodes
+    public void setValueAndFireSelectedListeners(Node node, Object value, Binding binding, Set<Node> references) throws NodeManagerException {
+       if(references.size() > 0) {
+               for(Node n : references) {
+                       valueCache.remove(n);
+               }
+       }
+       updateValueInner(node, value, binding);
+       fireNodeListenersSync();
+    }
+    
+    //Update the value of the node helper method
+    private void updateValueInner(Node node, Object value, Binding binding) throws NodeManagerException {
         checkThreadAccess();
         Binding targetBinding = realm.getEngine().getEngineBinding(node);
         if(binding.equals(targetBinding)) {
@@ -327,7 +364,6 @@ public abstract class StandardNodeManager<Node, Engine extends StandardNodeManag
                 throw new NodeManagerException(e);
             }
         }
-        refreshVariables();
     }
 
     @Override
@@ -336,7 +372,7 @@ public abstract class StandardNodeManager<Node, Engine extends StandardNodeManag
             String id = getRealmId();
             int lastSlash = id.lastIndexOf("/");
             if(lastSlash == -1) throw new IllegalStateException("Invalid realm id " + id);
-            String name = id.substring(lastSlash+1); 
+            String name = id.substring(lastSlash+1);
             return name;
         } else {
             return realm.getEngine().getName(node);
@@ -394,4 +430,9 @@ public abstract class StandardNodeManager<Node, Engine extends StandardNodeManag
         listeners.clear();
     }
 
+    @Override
+    public Set<String> getClassifications(Node node) throws NodeManagerException {
+        return Collections.emptySet();
+    }
+
 }
\ No newline at end of file