-package org.simantics.scl.compiler.elaboration.expressions;\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
-import gnu.trove.map.hash.TObjectIntHashMap;\r
-import gnu.trove.set.hash.THashSet;\r
-import gnu.trove.set.hash.TIntHashSet;\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
+package org.simantics.scl.compiler.elaboration.expressions;
+
+import org.simantics.scl.compiler.elaboration.contexts.SimplificationContext;
+import org.simantics.scl.compiler.elaboration.contexts.TranslationContext;
+import org.simantics.scl.compiler.elaboration.contexts.TypingContext;
+import org.simantics.scl.compiler.elaboration.expressions.list.CompiledQualifier;
+import org.simantics.scl.compiler.elaboration.expressions.list.ListQualifier;
+import org.simantics.scl.compiler.errors.Locations;
+import org.simantics.scl.compiler.types.Type;
+import org.simantics.scl.compiler.types.Types;
+import org.simantics.scl.compiler.types.exceptions.MatchException;
+
+import gnu.trove.map.hash.TObjectIntHashMap;
+import gnu.trove.set.hash.THashSet;
+import gnu.trove.set.hash.TIntHashSet;
+
+public class EListComprehension extends SimplifiableExpression {
+
+ public Expression head;
+ public ListQualifier qualifier;
+
+ public EListComprehension(Expression head, ListQualifier qualifier) {
+ this.head = head;
+ this.qualifier = qualifier;
+ }
+
+ public EListComprehension(long loc, Expression head,
+ ListQualifier qualifier) {
+ super(loc);
+ this.head = head;
+ this.qualifier = qualifier;
+ }
+
+ @Override
+ public void collectVars(TObjectIntHashMap<Variable> allVars,
+ TIntHashSet vars) {
+ head.collectVars(allVars, vars);
+ qualifier.collectVars(allVars, vars);
+ }
+
+ @Override
+ public Expression checkBasicType(TypingContext context, Type requiredType) {
+ qualifier.checkType(context);
+ Type componentType;
+ try {
+ componentType = Types.unifyApply(Types.LIST, requiredType);
+ } catch (MatchException e) {
+ context.getErrorLog().log(location, "Expected a value with type " + requiredType + " but got a list.");
+ return new EError(location);
+ }
+ head = head.checkType(context, componentType);
+ return this;
+ }
+
+ @Override
+ protected void updateType() throws MatchException {
+ setType(Types.list(head.getType()));
+ }
+
+ @Override
+ public void collectFreeVariables(THashSet<Variable> vars) {
+ head.collectFreeVariables(vars);
+ qualifier.collectFreeVariables(vars);
+ }
+
+ @Override
+ public Expression simplify(SimplificationContext context) {
+ context.pushLocation(location);
+ try {
+ CompiledQualifier cq = qualifier.compile(context);
+ Expression exp = context.mapList(
+ context.lambda(cq.pattern, head),
+ cq.value);
+ //System.out.println("simplified: " + exp);
+ /*try {
+ exp.validateType(context.getEnvironment());
+ } catch (TypeValidationException e) {
+ throw new InternalCompilerError(e);
+ }*/
+ exp = exp.simplify(context);
+ return exp;
+ } finally {
+ context.popLocation();
+ }
+ }
+
+ @Override
+ public Expression resolve(TranslationContext context) {
+ context.pushFrame();
+ qualifier.resolve(context);
+ head = head.resolve(context);
+ context.popFrame();
+ return this;
+ }
+
+ @Override
+ public void setLocationDeep(long loc) {
+ if(location == Locations.NO_LOCATION) {
+ location = loc;
+ head.setLocationDeep(loc);
+ qualifier.setLocationDeep(loc);
+ }
+ }
+
+ @Override
+ public void accept(ExpressionVisitor visitor) {
+ visitor.visit(this);
+ }
+
+ @Override
+ public Expression accept(ExpressionTransformer transformer) {
+ return transformer.transform(this);
+ }
+
+}