1 package org.simantics.scl.compiler.elaboration.expressions;
3 import org.simantics.scl.compiler.elaboration.contexts.ReplaceContext;
4 import org.simantics.scl.compiler.elaboration.contexts.SimplificationContext;
5 import org.simantics.scl.compiler.elaboration.contexts.TranslationContext;
6 import org.simantics.scl.compiler.elaboration.contexts.TypingContext;
7 import org.simantics.scl.compiler.elaboration.query.QExists;
8 import org.simantics.scl.compiler.elaboration.query.Query;
9 import org.simantics.scl.compiler.elaboration.query.compilation.QueryCompilationContext;
10 import org.simantics.scl.compiler.elaboration.query.compilation.QueryCompilationMode;
11 import org.simantics.scl.compiler.elaboration.query.compilation.UnsolvableQueryException;
12 import org.simantics.scl.compiler.errors.Locations;
13 import org.simantics.scl.compiler.types.Type;
14 import org.simantics.scl.compiler.types.Types;
15 import org.simantics.scl.compiler.types.exceptions.MatchException;
16 import org.simantics.scl.compiler.types.exceptions.UnificationException;
17 import org.simantics.scl.compiler.types.kinds.Kinds;
19 public class EWhen extends SimplifiableExpression {
22 public Expression action;
23 public Variable[] variables;
25 public EWhen(Query query, Expression action) {
30 public EWhen(long location, Query query, Expression action, Variable[] variables) {
32 throw new NullPointerException();
33 if(location == Locations.NO_LOCATION)
34 new Exception().printStackTrace();
35 this.location = location;
38 this.variables = variables;
42 protected void updateType() throws MatchException {
47 public Expression checkBasicType(TypingContext context, Type requiredType) {
49 Types.unify(requiredType, Types.UNIT);
50 } catch (UnificationException e) {
51 context.typeError(location, requiredType, Types.UNIT);
54 for(Variable variable : variables)
55 variable.setType(Types.metaVar(Kinds.STAR));
56 query.checkType(context);
57 action.checkIgnoredType(context);
60 return compile(context);
63 public Expression compile(TypingContext context) {
65 QueryCompilationContext queryCompilationContext =
66 new QueryCompilationContext(context, QueryCompilationMode.ITERATE, null, action);
67 new QExists(variables, query).generate(queryCompilationContext);
68 return queryCompilationContext.getContinuation();
69 } catch(UnsolvableQueryException e) {
70 context.getErrorLog().log(getLocation(), "Failed to compile the query.\n" + e.getMessage());
71 return new EError(getLocation());
76 public Expression resolve(TranslationContext context) {
77 context.pushExistentialFrame();
78 query = query.resolve(context);
79 action = action.resolve(context);
80 variables = context.popExistentialFrame();
85 public Expression replace(ReplaceContext context) {
86 Variable[] newVariables = new Variable[variables.length];
87 for(int i=0;i<variables.length;++i) {
88 Variable variable = variables[i];
89 Variable newVariable = new Variable(variable.name, variable.getType());
90 newVariables[i] = newVariable;
91 context.varMap.put(variable, new EVariable(newVariable));
93 EWhen result = new EWhen(location, query.replace(context), action.replace(context), newVariables);
94 for(int i=0;i<variables.length;++i)
95 context.varMap.remove(variables[i]);
100 public void setLocationDeep(long loc) {
101 if(location == Locations.NO_LOCATION) {
103 query.setLocationDeep(loc);
104 action.setLocationDeep(loc);
109 public void accept(ExpressionVisitor visitor) {
113 public Query getQuery() {
117 public Expression getAction() {
122 public Expression simplify(SimplificationContext context) {
123 throw new UnsupportedOperationException();
127 public Expression accept(ExpressionTransformer transformer) {
128 return transformer.transform(this);