X-Git-Url: https://gerrit.simantics.org/r/gitweb?a=blobdiff_plain;f=bundles%2Forg.simantics.scl.runtime%2Fsrc%2Forg%2Fsimantics%2Fscl%2Fruntime%2Funification%2FUMapUtils.java;fp=bundles%2Forg.simantics.scl.runtime%2Fsrc%2Forg%2Fsimantics%2Fscl%2Fruntime%2Funification%2FUMapUtils.java;h=8bc899370bf341b59bf099b77683e4794174f939;hb=969bd23cab98a79ca9101af33334000879fb60c5;hp=0000000000000000000000000000000000000000;hpb=866dba5cd5a3929bbeae85991796acb212338a08;p=simantics%2Fplatform.git

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
index 000000000..8bc899370
--- /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);
+        }
+    }
+}