bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/elaboration/expressions/list/ListGenerator.java

   1 package org.simantics.scl.compiler.elaboration.expressions.list;
   2
   3 import org.simantics.scl.compiler.elaboration.contexts.SimplificationContext;
   4 import org.simantics.scl.compiler.elaboration.contexts.TranslationContext;
   5 import org.simantics.scl.compiler.elaboration.contexts.TypingContext;
   6 import org.simantics.scl.compiler.elaboration.expressions.Case;
   7 import org.simantics.scl.compiler.elaboration.expressions.EVariable;
   8 import org.simantics.scl.compiler.elaboration.expressions.Expression;
   9 import org.simantics.scl.compiler.elaboration.expressions.Variable;
  10 import org.simantics.scl.compiler.errors.Locations;
  11 import org.simantics.scl.compiler.types.TMetaVar;
  12 import org.simantics.scl.compiler.types.Types;
  13 import org.simantics.scl.compiler.types.kinds.Kinds;
  14
  15 import gnu.trove.map.hash.TObjectIntHashMap;
  16 import gnu.trove.set.hash.THashSet;
  17 import gnu.trove.set.hash.TIntHashSet;
  18
  19 public class ListGenerator extends ListQualifier {
  20     public Expression pattern;
  21     public Expression value;
  22
  23     public ListGenerator(Expression pattern, Expression value) {
  24         this.pattern = pattern;
  25         this.value = value;
  26     }
  27
  28     @Override
  29     public void checkType(TypingContext context) {
  30         TMetaVar componentType = Types.metaVar(Kinds.STAR);
  31         value.checkType(context, Types.apply(Types.LIST, componentType));
  32         pattern.checkTypeAsPattern(context, componentType);
  33     }
  34
  35     @Override
  36     public void collectVars(TObjectIntHashMap<Variable> allVars,
  37             TIntHashSet vars) {
  38         value.collectVars(allVars, vars);
  39     }
  40
  41     @Override
  42     public void collectFreeVariables(THashSet<Variable> vars) {
  43         value.collectFreeVariables(vars);
  44         pattern.collectFreeVariables(vars);
  45     }
  46
  47     @Override
  48     public CompiledQualifier compile(SimplificationContext context) {
  49         if(pattern instanceof EVariable)
  50             return new CompiledQualifier(value, pattern);
  51         else {
  52             THashSet<Variable> variables = pattern.getFreeVariables();
  53             Variable[] variableArray = variables.toArray(new Variable[variables.size()]);
  54             Expression[] variableExps = new Expression[variableArray.length];
  55             for(int i=0;i<variableArray.length;++i)
  56                 variableExps[i] = new EVariable(variableArray[i]);
  57             Expression newPattern = context.tuple(variableExps);
  58
  59             EVariable blank = context.blank();
  60             blank.getVariable().setType(pattern.getType());
  61
  62             return new CompiledQualifier(
  63                     context.concatMap(
  64                             context.lambda(
  65                                     new Case(pattern, context.singletonList(newPattern.copy())),
  66                                     new Case(blank, context.emptyList(newPattern.getType()))
  67                                     ),
  68                             value
  69                             ),
  70                     newPattern);
  71         }
  72     }
  73
  74     @Override
  75     public void resolve(TranslationContext context) {
  76         value = value.resolve(context);
  77         pattern = pattern.resolveAsPattern(context);
  78     }
  79
  80     @Override
  81     public void setLocationDeep(long loc) {
  82         if(location == Locations.NO_LOCATION) {
  83             location = loc;
  84             pattern.setLocationDeep(loc);
  85             value.setLocationDeep(loc);
  86         }
  87     }
  88
  89     @Override
  90     public void accept(ListQualifierVisitor visitor) {
  91         visitor.visit(this);
  92     }
  93
  94     @Override
  95     public ListQualifier accept(ListQualifierTransformer transformer) {
  96         return transformer.transform(this);
  97     }
  98 }