1 package org.simantics.scl.compiler.elaboration.expressions;
3 import org.simantics.scl.compiler.elaboration.contexts.SimplificationContext;
4 import org.simantics.scl.compiler.elaboration.contexts.TranslationContext;
5 import org.simantics.scl.compiler.elaboration.contexts.TypingContext;
6 import org.simantics.scl.compiler.elaboration.expressions.list.CompiledQualifier;
7 import org.simantics.scl.compiler.elaboration.expressions.list.ListQualifier;
8 import org.simantics.scl.compiler.errors.Locations;
9 import org.simantics.scl.compiler.types.Type;
10 import org.simantics.scl.compiler.types.Types;
11 import org.simantics.scl.compiler.types.exceptions.MatchException;
13 public class EListComprehension extends SimplifiableExpression {
15 public Expression head;
16 public ListQualifier qualifier;
18 public EListComprehension(Expression head, ListQualifier qualifier) {
20 this.qualifier = qualifier;
23 public EListComprehension(long loc, Expression head,
24 ListQualifier qualifier) {
27 this.qualifier = qualifier;
31 public Expression checkBasicType(TypingContext context, Type requiredType) {
32 qualifier.checkType(context);
35 componentType = Types.unifyApply(Types.LIST, requiredType);
36 } catch (MatchException e) {
37 context.getErrorLog().log(location, "Expected a value with type " + requiredType + " but got a list.");
38 return new EError(location);
40 head = head.checkType(context, componentType);
45 protected void updateType() throws MatchException {
46 setType(Types.list(head.getType()));
50 public Expression simplify(SimplificationContext context) {
51 //System.out.println("--------");
52 context.pushLocation(location);
54 CompiledQualifier cq = qualifier.compile(context);
56 System.out.println("cq.pattern = " + cq.pattern);
57 System.out.println("cq.value = " + cq.value);
58 System.out.println("head = " + head);
60 Expression exp = context.mapList(
61 context.lambda(cq.pattern, head),
63 //System.out.println("simplified = " + exp);
65 exp.validateType(context.getEnvironment());
66 } catch (TypeValidationException e) {
67 throw new InternalCompilerError(e);
69 exp = exp.simplify(context);
72 context.popLocation();
77 public Expression resolve(TranslationContext context) {
79 qualifier.resolve(context);
80 head = head.resolve(context);
86 public void setLocationDeep(long loc) {
87 if(location == Locations.NO_LOCATION) {
89 head.setLocationDeep(loc);
90 qualifier.setLocationDeep(loc);
95 public void accept(ExpressionVisitor visitor) {
100 public Expression accept(ExpressionTransformer transformer) {
101 return transformer.transform(this);