+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