gerrit.simantics Code Review - simantics/platform.git/blob

   1 package org.simantics.scl.compiler.elaboration.expressions;
   2
   3 import org.simantics.scl.compiler.common.exceptions.InternalCompilerError;
   4 import org.simantics.scl.compiler.compilation.CompilationContext;
   5 import org.simantics.scl.compiler.elaboration.contexts.SimplificationContext;
   6 import org.simantics.scl.compiler.elaboration.contexts.TranslationContext;
   7 import org.simantics.scl.compiler.elaboration.contexts.TypingContext;
   8 import org.simantics.scl.compiler.errors.Locations;
   9 import org.simantics.scl.compiler.internal.codegen.references.IVal;
  10 import org.simantics.scl.compiler.internal.codegen.writer.CodeWriter;
  11 import org.simantics.scl.compiler.types.Type;
  12 import org.simantics.scl.compiler.types.Types;
  13 import org.simantics.scl.compiler.types.exceptions.MatchException;
  14 import org.simantics.scl.compiler.types.util.MultiFunction;
  15
  16 import gnu.trove.map.hash.TObjectIntHashMap;
  17 import gnu.trove.set.hash.THashSet;
  18 import gnu.trove.set.hash.TIntHashSet;
  19
  20 public class EViewPattern extends Expression {
  21     public Expression expression;
  22     public Expression pattern;
  23
  24     public EViewPattern(Expression expression, Expression pattern) {
  25         this.expression = expression;
  26         this.pattern = pattern;
  27     }
  28
  29     @Override
  30     public void collectRefs(TObjectIntHashMap<Object> allRefs, TIntHashSet refs) {
  31         expression.collectRefs(allRefs, refs);
  32         pattern.collectRefs(allRefs, refs);
  33     }
  34
  35     @Override
  36     public void collectVars(TObjectIntHashMap<Variable> allVars, TIntHashSet vars) {
  37         expression.collectVars(allVars, vars);
  38         pattern.collectVars(allVars, vars);
  39     }
  40
  41     @Override
  42     public void forVariables(VariableProcedure procedure) {
  43         expression.forVariables(procedure);
  44         pattern.forVariables(procedure);
  45     }
  46
  47     @Override
  48     public Expression inferType(TypingContext context) {
  49         context.setInPattern(false);
  50         expression = expression.inferType(context);
  51         context.setInPattern(true);
  52         MultiFunction mfun;
  53         try {
  54             mfun = Types.matchFunction(expression.getType(), 1);
  55         } catch (MatchException e) {
  56             context.getErrorLog().log(expression.location, "Expected a function as a transformation expression.");
  57             return new EError(location);
  58         }
  59         setType(mfun.parameterTypes[0]);
  60         pattern.checkType(context, mfun.returnType);
  61         return this;
  62     }
  63
  64     @Override
  65     protected void updateType() throws MatchException {
  66         MultiFunction mfun = Types.matchFunction(expression.getType(), 1);
  67         setType(mfun.parameterTypes[0]);
  68     }
  69
  70     @Override
  71     public IVal toVal(CompilationContext context, CodeWriter w) {
  72         throw new InternalCompilerError(location, "EViewPattern.toVal should not be invoked.");
  73     }
  74
  75     @Override
  76     public void collectFreeVariables(THashSet<Variable> vars) {
  77         throw new InternalCompilerError(location, "Cannot collect free variables for a pattern.");
  78     }
  79
  80     @Override
  81     public void removeFreeVariables(THashSet<Variable> vars) {
  82         expression.collectFreeVariables(vars);
  83         pattern.removeFreeVariables(vars);
  84     }
  85
  86     @Override
  87     public Expression resolve(TranslationContext context) {
  88         context.getErrorLog().log("View pattern cannot occur only in patterns. Maybe you are missing '\\' in front of a lambda experssion?");
  89         return new EError(location);
  90     }
  91
  92     @Override
  93     public Expression resolveAsPattern(TranslationContext context) {
  94         expression = expression.resolve(context);
  95         pattern = pattern.resolveAsPattern(context);
  96         return this;
  97     }
  98
  99     @Override
 100     public void setLocationDeep(long loc) {
 101         if(location == Locations.NO_LOCATION) {
 102             location = loc;
 103             expression.setLocationDeep(loc);
 104             pattern.setLocationDeep(loc);
 105         }
 106     }
 107
 108     @Override
 109     public void collectEffects(THashSet<Type> effects) {
 110         expression.collectEffects(effects);
 111     }
 112
 113     @Override
 114     public void accept(ExpressionVisitor visitor) {
 115         visitor.visit(this);
 116     }
 117
 118     @Override
 119     public Expression accept(ExpressionTransformer transformer) {
 120         return transformer.transform(this);
 121     }
 122
 123     @Override
 124     public Expression simplify(SimplificationContext context) {
 125         expression = expression.simplify(context);
 126         pattern = pattern.simplify(context);
 127         return this;
 128     }
 129
 130 }