1 package org.simantics.scl.compiler.elaboration.expressions.list;
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;
15 import gnu.trove.map.hash.TObjectIntHashMap;
16 import gnu.trove.set.hash.THashSet;
17 import gnu.trove.set.hash.TIntHashSet;
19 public class ListGenerator extends ListQualifier {
20 public Expression pattern;
21 public Expression value;
23 public ListGenerator(Expression pattern, Expression value) {
24 this.pattern = pattern;
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);
36 public void collectVars(TObjectIntHashMap<Variable> allVars,
38 value.collectVars(allVars, vars);
42 public void collectFreeVariables(THashSet<Variable> vars) {
43 value.collectFreeVariables(vars);
44 pattern.collectFreeVariables(vars);
48 public CompiledQualifier compile(SimplificationContext context) {
49 if(pattern instanceof EVariable)
50 return new CompiledQualifier(value, pattern);
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);
59 EVariable blank = context.blank();
60 blank.getVariable().setType(pattern.getType());
62 return new CompiledQualifier(
65 new Case(pattern, context.singletonList(newPattern.copy())),
66 new Case(blank, context.emptyList(newPattern.getType()))
75 public void resolve(TranslationContext context) {
76 value = value.resolve(context);
77 pattern = pattern.resolveAsPattern(context);
81 public void setLocationDeep(long loc) {
82 if(location == Locations.NO_LOCATION) {
84 pattern.setLocationDeep(loc);
85 value.setLocationDeep(loc);
90 public void accept(ListQualifierVisitor visitor) {
95 public ListQualifier accept(ListQualifierTransformer transformer) {
96 return transformer.transform(this);