bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/elaboration/expressions/EEquations.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.elaboration.equation.Equation;
   8 import org.simantics.scl.compiler.errors.Locations;
   9 import org.simantics.scl.compiler.internal.elaboration.utils.ExpressionDecorator;
  10 import org.simantics.scl.compiler.types.Type;
  11 import org.simantics.scl.compiler.types.Types;
  12 import org.simantics.scl.compiler.types.exceptions.MatchException;
  13
  14 import gnu.trove.map.hash.TObjectIntHashMap;
  15 import gnu.trove.set.hash.THashSet;
  16 import gnu.trove.set.hash.TIntHashSet;
  17
  18 public class EEquations extends SimplifiableExpression {
  19
  20     public Equation[] equations;
  21     public Type effect = Types.NO_EFFECTS;
  22
  23     public EEquations(Equation[] equations) {
  24         this.equations = equations;
  25     }
  26
  27     public EEquations(long location, Equation[] equations) {
  28         this(equations);
  29         this.location = location;
  30     }
  31
  32     @Override
  33     public Expression resolve(TranslationContext context) {
  34         for(Equation equation : equations)
  35             equation.resolve(context);
  36         return this;
  37     }
  38
  39     @Override
  40     public void setLocationDeep(long loc) {
  41         if(location == Locations.NO_LOCATION) {
  42             location = loc;
  43             for(Equation equation : equations)
  44                 equation.setLocationDeep(loc);
  45         }
  46     }
  47
  48     @Override
  49     public Expression accept(ExpressionTransformer transformer) {
  50         return transformer.transform(this);
  51     }
  52
  53     @Override
  54     public void collectRefs(TObjectIntHashMap<Object> allRefs, TIntHashSet refs) {
  55         for(Equation equation : equations)
  56             equation.collectRefs(allRefs, refs);
  57     }
  58
  59     @Override
  60     public void collectVars(TObjectIntHashMap<Variable> allVars, TIntHashSet vars) {
  61         for(Equation equation : equations)
  62             equation.collectVars(allVars, vars);
  63     }
  64
  65     @Override
  66     public void forVariables(VariableProcedure procedure) {
  67         for(Equation equation : equations)
  68             equation.forVariables(procedure);
  69     }
  70
  71     @Override
  72     protected void updateType() throws MatchException {
  73         setType(Types.UNIT);
  74     }
  75
  76     @Override
  77     public void collectFreeVariables(THashSet<Variable> vars) {
  78         for(Equation equation : equations)
  79             equation.collectFreeVariables(vars);
  80     }
  81
  82     @Override
  83     public Expression decorate(ExpressionDecorator decorator) {
  84         if(decorator.decorateSubstructure(this)) {
  85             for(Equation equation : equations)
  86                  equation.decorate(decorator);
  87         }
  88         return this;
  89     }
  90
  91     @Override
  92     public void collectEffects(THashSet<Type> effects) {
  93         for(Equation equation : equations)
  94             equation.collectEffects(effects);
  95     }
  96
  97     @Override
  98     public void accept(ExpressionVisitor visitor) {
  99         visitor.visit(this);
 100     }
 101
 102     @Override
 103     public Expression inferType(TypingContext context) {
 104         context.declareEffect(this.location, effect);
 105         for(Equation equation : equations)
 106             equation.checkType(context);
 107         return this;
 108     }
 109
 110     @Override
 111     public Expression checkIgnoredType(TypingContext context) {
 112         return inferType(context);
 113     }
 114
 115     @Override
 116     public Expression simplify(SimplificationContext context) {
 117         context.getErrorLog().log(location, "Equations should be transformed into other expressions before simplification phase.");
 118         return Expressions.tuple();
 119     }
 120
 121     @Override
 122     public Expression replace(ReplaceContext context) {
 123         Equation[] newEquations = new Equation[equations.length];
 124         for(int i=0;i<equations.length;++i)
 125             newEquations[i] = equations[i].replace(context);
 126         return new EEquations(location, newEquations);
 127     }
 128
 129 }