Migrated source code from Simantics SVN

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

diff --git a/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/elaboration/expressions/EListComprehension.java b/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/elaboration/expressions/EListComprehension.java
new file mode 100755 (executable)
index 0000000..a7e3624
--- /dev/null
+++ b/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/elaboration/expressions/EListComprehension.java
@@ -0,0 +1,142 @@
+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.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.elaboration.expressions.list.CompiledQualifier;\r
+import org.simantics.scl.compiler.elaboration.expressions.list.ListQualifier;\r
+import org.simantics.scl.compiler.errors.Locations;\r
+import org.simantics.scl.compiler.internal.elaboration.utils.ExpressionDecorator;\r
+import org.simantics.scl.compiler.types.Type;\r
+import org.simantics.scl.compiler.types.Types;\r
+import org.simantics.scl.compiler.types.exceptions.MatchException;\r
+\r
+public class EListComprehension extends SimplifiableExpression {\r
+\r
+    public Expression head;\r
+    public ListQualifier qualifier;\r
+    \r
+    public EListComprehension(Expression head, ListQualifier qualifier) {\r
+        this.head = head;\r
+        this.qualifier = qualifier;\r
+    }\r
+\r
+    public EListComprehension(long loc, Expression head,\r
+            ListQualifier qualifier) {\r
+        super(loc);\r
+        this.head = head;\r
+        this.qualifier = qualifier;\r
+    }\r
+\r
+    @Override\r
+    public void collectRefs(TObjectIntHashMap<Object> allRefs,\r
+            TIntHashSet refs) {\r
+        head.collectRefs(allRefs, refs);\r
+        qualifier.collectRefs(allRefs, refs);\r
+    }\r
+\r
+    @Override\r
+    public void collectVars(TObjectIntHashMap<Variable> allVars,\r
+            TIntHashSet vars) {\r
+        head.collectVars(allVars, vars);\r
+        qualifier.collectVars(allVars, vars);\r
+    }\r
+    \r
+    @Override\r
+    public Expression checkBasicType(TypingContext context, Type requiredType) {\r
+        qualifier.checkType(context);\r
+        Type componentType;\r
+        try {\r
+            componentType = Types.unifyApply(Types.LIST, requiredType);\r
+        } catch (MatchException e) {\r
+            context.getErrorLog().log(location, "Expected a value with type " + requiredType + " but got a list.");\r
+            return new EError(location);\r
+        }\r
+        head = head.checkType(context, componentType);\r
+        return this;\r
+    }\r
+    \r
+    @Override\r
+    protected void updateType() throws MatchException {\r
+        setType(Types.list(head.getType()));\r
+    }\r
+\r
+    @Override\r
+    public void collectFreeVariables(THashSet<Variable> vars) {\r
+        head.collectFreeVariables(vars);\r
+        qualifier.collectFreeVariables(vars);\r
+    }\r
+\r
+    @Override\r
+    public Expression simplify(SimplificationContext context) {\r
+        context.pushLocation(location);\r
+        try {\r
+            CompiledQualifier cq = qualifier.compile(context);            \r
+            Expression exp = context.mapList(\r
+                    context.lambda(cq.pattern, head), \r
+                    cq.value);\r
+            //System.out.println("simplified: " + exp);\r
+            /*try {\r
+                exp.validateType(context.getEnvironment());\r
+            } catch (TypeValidationException e) {\r
+                throw new InternalCompilerError(e);\r
+            }*/\r
+            exp = exp.simplify(context);\r
+            return exp;\r
+        } finally {\r
+            context.popLocation();\r
+        }\r
+    }\r
+\r
+    @Override\r
+    public Expression resolve(TranslationContext context) {\r
+        context.pushFrame();\r
+        qualifier.resolve(context);\r
+        head = head.resolve(context);\r
+        context.popFrame();\r
+        return this;\r
+    }\r
+\r
+    @Override\r
+    public Expression decorate(ExpressionDecorator decorator) {\r
+        head = head.decorate(decorator);\r
+        qualifier.decorate(decorator);\r
+        return decorator.decorate(this);\r
+    }\r
+\r
+    @Override\r
+    public void collectEffects(THashSet<Type> effects) {\r
+        head.collectEffects(effects);\r
+        qualifier.collectEffects(effects);\r
+    }\r
+\r
+    @Override\r
+    public void setLocationDeep(long loc) {\r
+        if(location == Locations.NO_LOCATION) {\r
+            location = loc;\r
+            head.setLocationDeep(loc);\r
+            qualifier.setLocationDeep(loc);\r
+        }\r
+    }\r
+    \r
+    @Override\r
+    public void accept(ExpressionVisitor visitor) {\r
+        visitor.visit(this);\r
+    }\r
+\r
+    @Override\r
+    public void forVariables(VariableProcedure procedure) {\r
+        head.forVariables(procedure);\r
+        qualifier.forVariables(procedure);\r
+    }\r
+    \r
+    @Override\r
+    public Expression accept(ExpressionTransformer transformer) {\r
+        return transformer.transform(this);\r
+    }\r
+\r
+}\r