Migrated source code from Simantics SVN

[simantics/platform.git] / bundles / org.simantics.scl.runtime / src / org / simantics / scl / runtime / unification / UMapUtils.java

diff --git a/bundles/org.simantics.scl.runtime/src/org/simantics/scl/runtime/unification/UMapUtils.java b/bundles/org.simantics.scl.runtime/src/org/simantics/scl/runtime/unification/UMapUtils.java
new file mode 100644 (file)
index 0000000..8bc8993
--- /dev/null
+++ b/bundles/org.simantics.scl.runtime/src/org/simantics/scl/runtime/unification/UMapUtils.java
@@ -0,0 +1,39 @@
+package org.simantics.scl.runtime.unification;
+
+import gnu.trove.map.hash.THashMap;
+
+import java.util.Map;
+
+import org.simantics.scl.runtime.function.Function;
+import org.simantics.scl.runtime.tuple.Tuple0;
+
+public class UMapUtils {
+    public static void put(Map<Object,Object> map, Object key, Object value) {
+        if(map.containsKey(key))
+            Unification.unify(map.get(key), value);
+        else
+            map.put(key, value);
+    }
+    
+    @SuppressWarnings({ "rawtypes", "unchecked" })
+    public static Object get(final Function def, THashMap<Object,Object> map, Object key) {
+        try {
+            Object value = map.get(key);
+            if(value == null) {
+                if(!map.containsKey(key)) {
+                    value = def.apply(Tuple0.INSTANCE);
+                    map.put(key, value);
+                }
+                return value;
+            }
+            else {
+                Object extrValue = Unification.extractWithDefault(def, value);
+                if(extrValue != value)
+                    map.put(key, extrValue);
+                return extrValue;
+            }
+        } catch(RuntimeException e) {
+            throw new RuntimeException("While generating value for key " + key + ".", e);
+        }
+    }
+}