bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/elaboration/expressions/Case.java

   1 package org.simantics.scl.compiler.elaboration.expressions;
   2
   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.errors.Locations;
   8 import org.simantics.scl.compiler.internal.parsing.Symbol;
   9 import org.simantics.scl.compiler.types.Type;
  10
  11 import gnu.trove.map.hash.TObjectIntHashMap;
  12 import gnu.trove.set.hash.TIntHashSet;
  13
  14 public class Case extends Symbol {
  15     public Expression[] patterns;
  16     public Expression value;
  17
  18     long lhs;
  19
  20     public Case(Expression[] patterns, Expression value) {
  21         this.patterns = patterns;
  22         this.value = value;
  23     }
  24
  25     public Case(Expression pattern, Expression value) {
  26         this(new Expression[] {pattern}, value);
  27     }
  28
  29     public void setLhs(long lhs) {
  30         this.lhs = lhs;
  31     }
  32
  33     public long getLhs() {
  34         return lhs;
  35     }
  36
  37     public void collectVars(TObjectIntHashMap<Variable> allVars,
  38             TIntHashSet vars) {
  39         value.collectVars(allVars, vars);
  40     }
  41
  42     public void resolve(TranslationContext context) {
  43         context.pushFrame();
  44         for(int i=0;i<patterns.length;++i)
  45             patterns[i] = patterns[i].resolveAsPattern(context);
  46         value = value.resolve(context);
  47         context.popFrame();
  48     }
  49
  50     public void simplify(SimplificationContext context) {
  51         for(int i=0;i<patterns.length;++i)
  52             patterns[i] = patterns[i].simplify(context);
  53         value = value.simplify(context);
  54     }
  55
  56     public void setLocationDeep(long loc) {
  57         if(location == Locations.NO_LOCATION) {
  58             location = loc;
  59             for(Expression pattern : patterns)
  60                 pattern.setLocationDeep(loc);
  61             value.setLocationDeep(loc);
  62         }
  63     }
  64
  65     public Case replace(ReplaceContext context) {
  66         Expression[] newPatterns = new Expression[patterns.length];
  67         for(int i=0;i<patterns.length;++i)
  68             newPatterns[i] = patterns[i].replaceInPattern(context);
  69         Expression newValue = value.replace(context);
  70         Case result = new Case(newPatterns, newValue);
  71         result.setLhs(lhs);
  72         return result;
  73     }
  74
  75     public Expression[] getPatterns() {
  76         return patterns;
  77     }
  78
  79     public void checkType(TypingContext context, Type[] parameterTypes,
  80             Type requiredType) {
  81         if(patterns.length != parameterTypes.length) {
  82             context.getErrorLog().log(location, "This case has different arity ("+patterns.length+
  83                     ") than than the first case (+"+parameterTypes.length+"+).");
  84             return;
  85         }
  86         for(int i=0;i<patterns.length;++i)
  87             patterns[i] = patterns[i].checkTypeAsPattern(context, parameterTypes[i]);
  88         value = value.checkType(context, requiredType);
  89     }
  90
  91     public void checkIgnoredType(TypingContext context, Type[] parameterTypes) {
  92         if(patterns.length != parameterTypes.length) {
  93             context.getErrorLog().log(location, "This case has different arity ("+patterns.length+
  94                     ") than than the first case (+"+parameterTypes.length+"+).");
  95             return;
  96         }
  97         for(int i=0;i<patterns.length;++i)
  98             patterns[i] = patterns[i].checkTypeAsPattern(context, parameterTypes[i]);
  99         value = value.checkIgnoredType(context);
 100     }
 101 }