Migrated source code from Simantics SVN

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

diff --git a/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/internal/parsing/translation/RelationRepository.java b/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/internal/parsing/translation/RelationRepository.java
new file mode 100644 (file)
index 0000000..d94d6c1
--- /dev/null
+++ b/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/internal/parsing/translation/RelationRepository.java
@@ -0,0 +1,75 @@
+package org.simantics.scl.compiler.internal.parsing.translation;\r
+\r
+import gnu.trove.map.hash.THashMap;\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.internal.parsing.declarations.DAnnotationAst;\r
+import org.simantics.scl.compiler.internal.parsing.declarations.DRelationAst;\r
+\r
+public class RelationRepository {\r
+    THashMap<String, ArrayList<DRelationAst>> relations = \r
+            new THashMap<String, ArrayList<DRelationAst>>();\r
+    THashMap<String, ArrayList<DAnnotationAst>> annotations = \r
+            new THashMap<String, ArrayList<DAnnotationAst>>();\r
+            \r
+    public String add(DRelationAst relation) throws NotPatternException {\r
+        String name = relation.lhs.getPatternHead().name;\r
+        ArrayList<DRelationAst> vs = relations.get(name);\r
+        if(vs == null) {\r
+            vs = new ArrayList<DRelationAst>(2);\r
+            relations.put(name, vs);\r
+        }\r
+        vs.add(relation);\r
+        return name;\r
+    }\r
+    \r
+    public void addDefinitions(String name, ArrayList<DRelationAst> relationList) {\r
+        ArrayList<DRelationAst> vs = relations.get(name);\r
+        if(vs == null) {\r
+            vs = new ArrayList<DRelationAst>(relationList.size());\r
+            relations.put(name, vs);\r
+        }\r
+        vs.addAll(relationList);\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> getRelationNames() {\r
+        return relations.keySet();\r
+    }\r
+\r
+    public ArrayList<DRelationAst> getDefinition(String name) {\r
+        return relations.get(name);\r
+    }\r
+\r
+    public ArrayList<DAnnotationAst> getAnnotations(String name) {\r
+        return annotations.get(name);\r
+    }\r
+\r
+    public void addFrom(RelationRepository 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