Migrated source code from Simantics SVN

[simantics/platform.git] / bundles / org.simantics.scl.compiler / src / org / simantics / scl / compiler / internal / parsing / translation / ValueRepository.java

diff --git a/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/internal/parsing/translation/ValueRepository.java b/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/internal/parsing/translation/ValueRepository.java
new file mode 100644 (file)
index 0000000..45e8057
--- /dev/null
+++ b/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/internal/parsing/translation/ValueRepository.java
@@ -0,0 +1,88 @@
+package org.simantics.scl.compiler.internal.parsing.translation;\r
+\r
+import java.util.ArrayList;\r
+import java.util.Collection;\r
+\r
+import org.simantics.scl.compiler.elaboration.errors.NotPatternException;\r
+import org.simantics.scl.compiler.elaboration.expressions.EVar;\r
+import org.simantics.scl.compiler.errors.Locations;\r
+import org.simantics.scl.compiler.internal.parsing.declarations.DAnnotationAst;\r
+import org.simantics.scl.compiler.internal.parsing.declarations.DValueAst;\r
+\r
+import gnu.trove.impl.Constants;\r
+import gnu.trove.map.hash.THashMap;\r
+import gnu.trove.map.hash.TObjectLongHashMap;\r
+\r
+public class ValueRepository {\r
+    THashMap<String, ArrayList<DValueAst>> values = \r
+            new THashMap<String, ArrayList<DValueAst>>();\r
+    THashMap<String, ArrayList<DAnnotationAst>> annotations = \r
+            new THashMap<String, ArrayList<DAnnotationAst>>();\r
+    TObjectLongHashMap<String> locations = new TObjectLongHashMap<String>(Constants.DEFAULT_CAPACITY, Constants.DEFAULT_LOAD_FACTOR,\r
+            Locations.NO_LOCATION); \r
+            \r
+    public String add(DValueAst value) throws NotPatternException {\r
+        EVar name = value.lhs.getPatternHead();\r
+        ArrayList<DValueAst> vs = values.get(name.name);\r
+        if(vs == null) {\r
+            vs = new ArrayList<DValueAst>(2);\r
+            values.put(name.name, vs);\r
+        }\r
+        long oldLocation = locations.get(name.name);\r
+        if(oldLocation == Locations.NO_LOCATION || name.location < oldLocation)\r
+            locations.put(name.name, name.location);\r
+        vs.add(value);\r
+        return name.name;\r
+    }\r
+    \r
+    public void addDefinitions(String name, ArrayList<DValueAst> valueList) {\r
+        ArrayList<DValueAst> vs = values.get(name);\r
+        if(vs == null) {\r
+            vs = new ArrayList<DValueAst>(valueList.size());\r
+            values.put(name, vs);\r
+        }\r
+        vs.addAll(valueList);\r
+    }\r
+    \r
+    public void addAnnotation(String name, DAnnotationAst annotation) {\r
+        ArrayList<DAnnotationAst> as = annotations.get(name);\r
+        if(as == null) {\r
+            as = new ArrayList<DAnnotationAst>(2);\r
+            annotations.put(name, as);\r
+        }\r
+        as.add(annotation);\r
+    }\r
+    \r
+    public void addAnnotations(String name, ArrayList<DAnnotationAst> annotationList) {\r
+        ArrayList<DAnnotationAst> as = annotations.get(name);\r
+        if(as == null) {\r
+            as = new ArrayList<DAnnotationAst>(annotationList.size());\r
+            annotations.put(name, as);\r
+        }\r
+        as.addAll(annotationList);\r
+    }\r
+    \r
+    public Collection<String> getValueNames() {\r
+        return values.keySet();\r
+    }\r
+\r
+    public ArrayList<DValueAst> getDefinition(String name) {\r
+        return values.get(name);\r
+    }\r
+\r
+    public ArrayList<DAnnotationAst> getAnnotations(String name) {\r
+        return annotations.get(name);\r
+    }\r
+\r
+    public void addFrom(ValueRepository repo, String oldName, String newName) {\r
+        addDefinitions(newName, repo.getDefinition(oldName));\r
+        \r
+        ArrayList<DAnnotationAst> as = repo.getAnnotations(oldName);\r
+        if(as != null)\r
+            addAnnotations(newName, as);\r
+    }\r
+\r
+    public long getLocation(String name) {\r
+        return locations.get(name);\r
+    }\r
+}\r