Migrated source code from Simantics SVN

[simantics/platform.git] / bundles / org.simantics.scl.compiler / src / org / simantics / scl / compiler / elaboration / expressions / Case.java

diff --git a/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/elaboration/expressions/Case.java b/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/elaboration/expressions/Case.java
new file mode 100755 (executable)
index 0000000..219cce4
--- /dev/null
+++ b/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/elaboration/expressions/Case.java
@@ -0,0 +1,112 @@
+package org.simantics.scl.compiler.elaboration.expressions;\r
+\r
+import gnu.trove.map.hash.TObjectIntHashMap;\r
+import gnu.trove.set.hash.THashSet;\r
+import gnu.trove.set.hash.TIntHashSet;\r
+\r
+import org.simantics.scl.compiler.elaboration.contexts.ReplaceContext;\r
+import org.simantics.scl.compiler.elaboration.contexts.SimplificationContext;\r
+import org.simantics.scl.compiler.elaboration.contexts.TranslationContext;\r
+import org.simantics.scl.compiler.elaboration.contexts.TypingContext;\r
+import org.simantics.scl.compiler.errors.Locations;\r
+import org.simantics.scl.compiler.internal.elaboration.utils.ExpressionDecorator;\r
+import org.simantics.scl.compiler.internal.parsing.Symbol;\r
+import org.simantics.scl.compiler.types.Type;\r
+\r
+public class Case extends Symbol {\r
+    public Expression[] patterns;\r
+    public Expression value;\r
+    \r
+    long lhs;\r
+    \r
+    public Case(Expression[] patterns, Expression value) {\r
+        this.patterns = patterns;\r
+        this.value = value;\r
+    }\r
+    \r
+    public Case(Expression pattern, Expression value) {\r
+        this(new Expression[] {pattern}, value);\r
+    }\r
+\r
+    public void setLhs(long lhs) {\r
+        this.lhs = lhs;\r
+    }\r
+    \r
+    public long getLhs() {\r
+        return lhs;\r
+    }\r
+\r
+       public void collectRefs(TObjectIntHashMap<Object> allRefs, TIntHashSet refs) {\r
+        value.collectRefs(allRefs, refs);\r
+    }\r
+\r
+    public void collectVars(TObjectIntHashMap<Variable> allVars,\r
+            TIntHashSet vars) {\r
+        value.collectVars(allVars, vars);\r
+    }\r
+\r
+    public void collectFreeVariables(THashSet<Variable> vars) {\r
+        value.collectFreeVariables(vars);\r
+        for(int i=patterns.length-1;i>=0;--i)\r
+            patterns[i].removeFreeVariables(vars);\r
+    }\r
+\r
+    public void resolve(TranslationContext context) {\r
+        context.pushFrame();\r
+        for(int i=0;i<patterns.length;++i)\r
+            patterns[i] = patterns[i].resolveAsPattern(context);\r
+        value = value.resolve(context);\r
+        context.popFrame();\r
+    }\r
+\r
+    public void simplify(SimplificationContext context) {\r
+        for(int i=0;i<patterns.length;++i)\r
+            patterns[i] = patterns[i].simplify(context);\r
+        value = value.simplify(context);\r
+    }\r
+\r
+    public void setLocationDeep(long loc) {\r
+        if(location == Locations.NO_LOCATION) {\r
+            location = loc;\r
+            for(Expression pattern : patterns)\r
+                pattern.setLocationDeep(loc);\r
+            value.setLocationDeep(loc);\r
+        }\r
+    }\r
+\r
+    public Case replace(ReplaceContext context) {\r
+        Expression[] newPatterns = new Expression[patterns.length];        \r
+        for(int i=0;i<patterns.length;++i)\r
+            newPatterns[i] = patterns[i].replaceInPattern(context);\r
+        Expression newValue = value.replace(context);\r
+        Case result = new Case(newPatterns, newValue);\r
+        result.setLhs(lhs);\r
+        return result;\r
+    }\r
+\r
+    public Expression[] getPatterns() {\r
+        return patterns;\r
+    }\r
+\r
+    public void checkType(TypingContext context, Type[] parameterTypes,\r
+            Type requiredType) {\r
+        if(patterns.length != parameterTypes.length) {\r
+            context.getErrorLog().log(location, "This case has different arity ("+patterns.length+\r
+                    ") than than the first case (+"+parameterTypes.length+"+).");\r
+            return;\r
+        }\r
+        for(int i=0;i<patterns.length;++i)\r
+            patterns[i] = patterns[i].checkTypeAsPattern(context, parameterTypes[i]);\r
+        value = value.checkType(context, requiredType);\r
+    }\r
+        \r
+    public void decorate(ExpressionDecorator decorator) {\r
+        for(int i=0;i<patterns.length;++i)\r
+            patterns[i] = patterns[i].decorate(decorator);\r
+        value = value.decorate(decorator);\r
+    }\r
+\r
+    public void forVariables(VariableProcedure procedure) {\r
+        value.forVariables(procedure);\r
+    }\r
+}\r