--- /dev/null
+package org.simantics.scl.compiler.elaboration.expressions;\r
+\r
+import gnu.trove.map.hash.TObjectIntHashMap;\r
+import gnu.trove.set.hash.THashSet;\r
+import gnu.trove.set.hash.TIntHashSet;\r
+\r
+import org.simantics.scl.compiler.elaboration.contexts.ReplaceContext;\r
+import org.simantics.scl.compiler.elaboration.contexts.SimplificationContext;\r
+import org.simantics.scl.compiler.elaboration.contexts.TranslationContext;\r
+import org.simantics.scl.compiler.elaboration.contexts.TypingContext;\r
+import org.simantics.scl.compiler.elaboration.java.Builtins;\r
+import org.simantics.scl.compiler.elaboration.java.ListConstructor;\r
+import org.simantics.scl.compiler.errors.Locations;\r
+import org.simantics.scl.compiler.internal.codegen.utils.Constants;\r
+import org.simantics.scl.compiler.internal.elaboration.utils.ExpressionDecorator;\r
+import org.simantics.scl.compiler.internal.interpreted.IExpression;\r
+import org.simantics.scl.compiler.internal.interpreted.IListLiteral;\r
+import org.simantics.scl.compiler.top.ExpressionInterpretationContext;\r
+import org.simantics.scl.compiler.types.Type;\r
+import org.simantics.scl.compiler.types.Types;\r
+import org.simantics.scl.compiler.types.exceptions.MatchException;\r
+\r
+public class EListLiteral extends SimplifiableExpression {\r
+\r
+ Expression[] components;\r
+ Type componentType;\r
+\r
+ public EListLiteral(Expression[] components) {\r
+ this.components = components;\r
+ }\r
+\r
+ private EListLiteral(Expression[] components, Type componentType) {\r
+ this.components = components;\r
+ this.componentType = componentType;\r
+ }\r
+\r
+ public Expression[] getComponents() {\r
+ return components;\r
+ }\r
+ \r
+ @Override\r
+ public void collectRefs(TObjectIntHashMap<Object> allRefs,\r
+ TIntHashSet refs) {\r
+ for(Expression component : components)\r
+ component.collectRefs(allRefs, refs);\r
+ }\r
+\r
+ @Override\r
+ public void collectVars(TObjectIntHashMap<Variable> allVars,\r
+ TIntHashSet vars) {\r
+ for(Expression component : components)\r
+ component.collectVars(allVars, vars);\r
+ }\r
+ \r
+ @Override\r
+ public void collectFreeVariables(THashSet<Variable> vars) {\r
+ for(Expression component : components)\r
+ component.collectFreeVariables(vars);\r
+ }\r
+ \r
+ @Override\r
+ public Expression simplify(SimplificationContext context) {\r
+ context.pushLocation(location);\r
+ try {\r
+ for(int i=0;i<components.length;++i)\r
+ components[i] = components[i].simplify(context);\r
+ \r
+ if(components.length <= Constants.MAX_LIST_LITERAL_LENGTH) {\r
+ Expression result = new EConstant(location, Builtins.LIST_CONSTRUCTORS[components.length], \r
+ componentType);\r
+ if(components.length > 0)\r
+ result = new EApply(location, result, components);\r
+ return result;\r
+ }\r
+ else {\r
+ Expression result = new EApplyType(new ELiteral(location, new ListConstructor(components.length)), componentType);\r
+ result = new EApply(location, result, components);\r
+ return result;\r
+ }\r
+ } finally {\r
+ context.popLocation();\r
+ }\r
+ }\r
+\r
+ @Override\r
+ public Expression resolve(TranslationContext context) {\r
+ for(int i=0;i<components.length;++i)\r
+ components[i] = components[i].resolve(context);\r
+ return this;\r
+ }\r
+ \r
+ @Override\r
+ public Expression resolveAsPattern(TranslationContext context) {\r
+ for(int i=0;i<components.length;++i)\r
+ components[i] = components[i].resolveAsPattern(context);\r
+ return this;\r
+ }\r
+ \r
+ @Override\r
+ protected void updateType() throws MatchException {\r
+ setType(Types.list(componentType));\r
+ }\r
+ \r
+ @Override\r
+ public Expression checkBasicType(TypingContext context, Type requiredType) {\r
+ try {\r
+ componentType = Types.unifyApply(Types.LIST, requiredType);\r
+ } catch (MatchException e) {\r
+ context.getErrorLog().log(location, "Expected a value with type " + requiredType + " but got a list.");\r
+ return new EError(location);\r
+ }\r
+ for(int i=0;i<components.length;++i)\r
+ components[i] = components[i].checkType(context, componentType);\r
+ return this;\r
+ }\r
+\r
+ @Override\r
+ public Expression decorate(ExpressionDecorator decorator) {\r
+ for(int i=0;i<components.length;++i)\r
+ components[i] = components[i].decorate(decorator);\r
+ return decorator.decorate(this);\r
+ }\r
+\r
+ @Override\r
+ public void collectEffects(THashSet<Type> effects) {\r
+ for(Expression component : components)\r
+ component.collectEffects(effects);\r
+ }\r
+\r
+ @Override\r
+ public void setLocationDeep(long loc) {\r
+ if(location == Locations.NO_LOCATION) {\r
+ location = loc;\r
+ for(Expression component : components)\r
+ component.setLocationDeep(loc);\r
+ }\r
+ }\r
+ \r
+ @Override\r
+ public void accept(ExpressionVisitor visitor) {\r
+ visitor.visit(this);\r
+ }\r
+ \r
+ @Override\r
+ public IExpression toIExpression(ExpressionInterpretationContext target) {\r
+ IExpression[] componentExpressions = new IExpression[components.length];\r
+ for(int i=0;i<components.length;++i)\r
+ componentExpressions[i] = components[i].toIExpression(target);\r
+ return new IListLiteral(componentExpressions);\r
+ }\r
+\r
+ @Override\r
+ public void forVariables(VariableProcedure procedure) {\r
+ for(Expression component : components)\r
+ component.forVariables(procedure);\r
+ }\r
+ \r
+ @Override\r
+ public Expression replace(ReplaceContext context) {\r
+ Expression[] newComponents = new Expression[components.length];\r
+ for(int i=0;i<components.length;++i)\r
+ newComponents[i] = components[i].replace(context);\r
+ return new EListLiteral(newComponents, componentType.replace(context.tvarMap));\r
+ }\r
+ \r
+ @Override\r
+ public boolean isPattern(int arity) {\r
+ if(arity != 0)\r
+ return false;\r
+ for(Expression component : components)\r
+ if(!component.isPattern(0))\r
+ return false;\r
+ return true;\r
+ }\r
+ \r
+ @Override\r
+ public Expression accept(ExpressionTransformer transformer) {\r
+ return transformer.transform(this);\r
+ }\r
+\r
+}\r