1 package org.simantics.scl.compiler.elaboration.expressions;
\r
3 import org.simantics.scl.compiler.elaboration.contexts.ReplaceContext;
\r
4 import org.simantics.scl.compiler.elaboration.contexts.SimplificationContext;
\r
5 import org.simantics.scl.compiler.elaboration.contexts.TranslationContext;
\r
6 import org.simantics.scl.compiler.elaboration.contexts.TypingContext;
\r
7 import org.simantics.scl.compiler.errors.Locations;
\r
8 import org.simantics.scl.compiler.internal.elaboration.utils.ExpressionDecorator;
\r
9 import org.simantics.scl.compiler.internal.parsing.Symbol;
\r
10 import org.simantics.scl.compiler.types.Type;
\r
12 import gnu.trove.map.hash.TObjectIntHashMap;
\r
13 import gnu.trove.set.hash.THashSet;
\r
14 import gnu.trove.set.hash.TIntHashSet;
\r
16 public class Case extends Symbol {
\r
17 public Expression[] patterns;
\r
18 public Expression value;
\r
22 public Case(Expression[] patterns, Expression value) {
\r
23 this.patterns = patterns;
\r
27 public Case(Expression pattern, Expression value) {
\r
28 this(new Expression[] {pattern}, value);
\r
31 public void setLhs(long lhs) {
\r
35 public long getLhs() {
\r
39 public void collectRefs(TObjectIntHashMap<Object> allRefs, TIntHashSet refs) {
\r
40 value.collectRefs(allRefs, refs);
\r
43 public void collectVars(TObjectIntHashMap<Variable> allVars,
\r
45 value.collectVars(allVars, vars);
\r
48 public void collectFreeVariables(THashSet<Variable> vars) {
\r
49 value.collectFreeVariables(vars);
\r
50 for(int i=patterns.length-1;i>=0;--i)
\r
51 patterns[i].removeFreeVariables(vars);
\r
54 public void resolve(TranslationContext context) {
\r
55 context.pushFrame();
\r
56 for(int i=0;i<patterns.length;++i)
\r
57 patterns[i] = patterns[i].resolveAsPattern(context);
\r
58 value = value.resolve(context);
\r
62 public void simplify(SimplificationContext context) {
\r
63 for(int i=0;i<patterns.length;++i)
\r
64 patterns[i] = patterns[i].simplify(context);
\r
65 value = value.simplify(context);
\r
68 public void setLocationDeep(long loc) {
\r
69 if(location == Locations.NO_LOCATION) {
\r
71 for(Expression pattern : patterns)
\r
72 pattern.setLocationDeep(loc);
\r
73 value.setLocationDeep(loc);
\r
77 public Case replace(ReplaceContext context) {
\r
78 Expression[] newPatterns = new Expression[patterns.length];
\r
79 for(int i=0;i<patterns.length;++i)
\r
80 newPatterns[i] = patterns[i].replaceInPattern(context);
\r
81 Expression newValue = value.replace(context);
\r
82 Case result = new Case(newPatterns, newValue);
\r
87 public Expression[] getPatterns() {
\r
91 public void checkType(TypingContext context, Type[] parameterTypes,
\r
92 Type requiredType) {
\r
93 if(patterns.length != parameterTypes.length) {
\r
94 context.getErrorLog().log(location, "This case has different arity ("+patterns.length+
\r
95 ") than than the first case (+"+parameterTypes.length+"+).");
\r
98 for(int i=0;i<patterns.length;++i)
\r
99 patterns[i] = patterns[i].checkTypeAsPattern(context, parameterTypes[i]);
\r
100 value = value.checkType(context, requiredType);
\r
103 public void checkIgnoredType(TypingContext context, Type[] parameterTypes) {
\r
104 if(patterns.length != parameterTypes.length) {
\r
105 context.getErrorLog().log(location, "This case has different arity ("+patterns.length+
\r
106 ") than than the first case (+"+parameterTypes.length+"+).");
\r
109 for(int i=0;i<patterns.length;++i)
\r
110 patterns[i] = patterns[i].checkTypeAsPattern(context, parameterTypes[i]);
\r
111 value = value.checkIgnoredType(context);
\r
114 public void decorate(ExpressionDecorator decorator) {
\r
115 for(int i=0;i<patterns.length;++i)
\r
116 patterns[i] = patterns[i].decorate(decorator);
\r
117 value = value.decorate(decorator);
\r
120 public void forVariables(VariableProcedure procedure) {
\r
121 value.forVariables(procedure);
\r