-package org.simantics.scl.compiler.elaboration.expressions;\r
-\r
-import java.util.ArrayList;\r
-\r
-import org.simantics.scl.compiler.common.exceptions.InternalCompilerError;\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.environment.Environment;\r
-import org.simantics.scl.compiler.errors.Locations;\r
-import org.simantics.scl.compiler.internal.codegen.references.IVal;\r
-import org.simantics.scl.compiler.internal.codegen.writer.CodeWriter;\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.IVariable;\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
-import org.simantics.scl.compiler.types.kinds.Kinds;\r
-import org.simantics.scl.compiler.types.util.TypeUnparsingContext;\r
-\r
-import gnu.trove.map.hash.TObjectIntHashMap;\r
-import gnu.trove.set.hash.THashSet;\r
-import gnu.trove.set.hash.TIntHashSet;\r
-\r
-public class EVariable extends Expression {\r
- public static final EVariable[] EMPTY_ARRAY = new EVariable[0];\r
- \r
- Variable variable;\r
- \r
- public EVariable(Variable variable) {\r
- this.variable = variable;\r
- }\r
-\r
- public EVariable(long loc, Variable variable) {\r
- super(loc);\r
- this.variable = variable;\r
- }\r
-\r
- public Variable getVariable() {\r
- return variable;\r
- }\r
- \r
- public void setVariable(Variable variable) {\r
- this.variable = variable;\r
- }\r
-\r
- public void collectRefs(TObjectIntHashMap<Object> allRefs, TIntHashSet refs) {\r
- }\r
- \r
- @Override\r
- public void collectVars(TObjectIntHashMap<Variable> allVars,\r
- TIntHashSet vars) {\r
- int id = allVars.get(variable);\r
- if(id >= 0)\r
- vars.add(id);\r
- }\r
-\r
- public void toString(StringBuilder b, TypeUnparsingContext tuc) {\r
- b.append(variable == null ? "???" : variable.toString());\r
- }\r
-\r
- @Override\r
- protected void updateType() throws MatchException {\r
- setType(variable.getType());\r
- }\r
- \r
- @Override\r
- public IVal toVal(Environment env, CodeWriter w) {\r
- return variable.getVal();\r
- }\r
-\r
- @Override\r
- public void collectFreeVariables(THashSet<Variable> vars) {\r
- vars.add(variable);\r
- }\r
-\r
- @Override\r
- public Expression simplify(SimplificationContext context) {\r
- Expression expression = context.getInlinedValue(variable);\r
- if(expression != null)\r
- return expression.copy();\r
- else\r
- return this;\r
- }\r
-\r
- @Override\r
- public Expression resolve(TranslationContext context) {\r
- return this;\r
- }\r
- \r
- @Override\r
- public void getParameters(TranslationContext translationContext,\r
- ArrayList<Expression> parameters) {\r
- }\r
- \r
- @Override\r
- public void removeFreeVariables(THashSet<Variable> vars) {\r
- vars.remove(variable);\r
- }\r
- \r
- @Override\r
- public Expression resolveAsPattern(TranslationContext context) {\r
- return this;\r
- }\r
-\r
- @Override\r
- public Expression replace(ReplaceContext context) {\r
- if(context.inPattern) {\r
- Type type = variable.getType().replace(context.tvarMap);\r
- \r
- Variable newVariable = new Variable(variable.name);\r
- newVariable.setType(type);\r
- EVariable result = new EVariable(newVariable);\r
- context.varMap.put(variable, result);\r
- return result;\r
- }\r
- else {\r
- if(variable == null) {\r
- EVariable newVariable = new EVariable(location, null);\r
- newVariable.setType(getType().replace(context.tvarMap));\r
- if(context.typingContext == null)\r
- throw new InternalCompilerError(location, "Encountered unresolved variable but not in type checking phase.");\r
- context.typingContext.addConstraintDemand(newVariable);\r
- return newVariable;\r
- }\r
- else {\r
- Expression expression = context.varMap.get(variable);\r
- if(expression != null)\r
- return expression.copy(context.typingContext);\r
- else {\r
- return new EVariable(variable);\r
- }\r
- }\r
- }\r
- }\r
- \r
- @Override\r
- public IExpression toIExpression(ExpressionInterpretationContext target) {\r
- return new IVariable(target.getVariableId(variable));\r
- }\r
- \r
- @Override\r
- public Expression inferType(TypingContext context) {\r
- if(context.isInPattern()) {\r
- variable.setType(Types.metaVar(Kinds.STAR));\r
- return this;\r
- }\r
- else\r
- return applyPUnit(context);\r
- }\r
- \r
- @Override\r
- public Expression checkBasicType(TypingContext context, Type requiredType) {\r
- if(context.isInPattern()) {\r
- variable.setType(requiredType);\r
- return this;\r
- }\r
- else\r
- return context.subsume(this, requiredType);\r
- }\r
- \r
- @Override\r
- public Expression decorate(ExpressionDecorator decorator) { \r
- return decorator.decorate(this);\r
- }\r
- \r
- @Override\r
- public boolean isEffectful() {\r
- return false;\r
- }\r
- \r
- @Override\r
- public void collectEffects(THashSet<Type> effects) {\r
- }\r
- \r
- @Override\r
- public void setLocationDeep(long loc) {\r
- if(location == Locations.NO_LOCATION)\r
- location = loc;\r
- }\r
- \r
- @Override\r
- public void accept(ExpressionVisitor visitor) {\r
- visitor.visit(this);\r
- }\r
-\r
- @Override\r
- public void forVariables(VariableProcedure procedure) {\r
- if(variable != null)\r
- procedure.execute(location, variable);\r
- }\r
- \r
- @Override\r
- public boolean isPattern(int arity) {\r
- return arity == 0;\r
- }\r
- \r
- @Override\r
- public Expression accept(ExpressionTransformer transformer) {\r
- return transformer.transform(this);\r
- }\r
-\r
-}\r
+package org.simantics.scl.compiler.elaboration.expressions;
+
+import java.util.ArrayList;
+
+import org.simantics.scl.compiler.common.exceptions.InternalCompilerError;
+import org.simantics.scl.compiler.elaboration.contexts.ReplaceContext;
+import org.simantics.scl.compiler.elaboration.contexts.SimplificationContext;
+import org.simantics.scl.compiler.elaboration.contexts.TranslationContext;
+import org.simantics.scl.compiler.elaboration.contexts.TypingContext;
+import org.simantics.scl.compiler.environment.Environment;
+import org.simantics.scl.compiler.errors.Locations;
+import org.simantics.scl.compiler.internal.codegen.references.IVal;
+import org.simantics.scl.compiler.internal.codegen.writer.CodeWriter;
+import org.simantics.scl.compiler.internal.elaboration.utils.ExpressionDecorator;
+import org.simantics.scl.compiler.internal.interpreted.IExpression;
+import org.simantics.scl.compiler.internal.interpreted.IVariable;
+import org.simantics.scl.compiler.top.ExpressionInterpretationContext;
+import org.simantics.scl.compiler.types.Type;
+import org.simantics.scl.compiler.types.Types;
+import org.simantics.scl.compiler.types.exceptions.MatchException;
+import org.simantics.scl.compiler.types.kinds.Kinds;
+import org.simantics.scl.compiler.types.util.TypeUnparsingContext;
+
+import gnu.trove.map.hash.TObjectIntHashMap;
+import gnu.trove.set.hash.THashSet;
+import gnu.trove.set.hash.TIntHashSet;
+
+public class EVariable extends Expression {
+ public static final EVariable[] EMPTY_ARRAY = new EVariable[0];
+
+ Variable variable;
+
+ public EVariable(Variable variable) {
+ this.variable = variable;
+ }
+
+ public EVariable(long loc, Variable variable) {
+ super(loc);
+ this.variable = variable;
+ }
+
+ public Variable getVariable() {
+ return variable;
+ }
+
+ public void setVariable(Variable variable) {
+ this.variable = variable;
+ }
+
+ public void collectRefs(TObjectIntHashMap<Object> allRefs, TIntHashSet refs) {
+ }
+
+ @Override
+ public void collectVars(TObjectIntHashMap<Variable> allVars,
+ TIntHashSet vars) {
+ int id = allVars.get(variable);
+ if(id >= 0)
+ vars.add(id);
+ }
+
+ public void toString(StringBuilder b, TypeUnparsingContext tuc) {
+ b.append(variable == null ? "???" : variable.toString());
+ }
+
+ @Override
+ protected void updateType() throws MatchException {
+ setType(variable.getType());
+ }
+
+ @Override
+ public IVal toVal(Environment env, CodeWriter w) {
+ return variable.getVal();
+ }
+
+ @Override
+ public void collectFreeVariables(THashSet<Variable> vars) {
+ vars.add(variable);
+ }
+
+ @Override
+ public Expression simplify(SimplificationContext context) {
+ Expression expression = context.getInlinedValue(variable);
+ if(expression != null)
+ return expression.copy();
+ else
+ return this;
+ }
+
+ @Override
+ public Expression resolve(TranslationContext context) {
+ return this;
+ }
+
+ @Override
+ public void getParameters(TranslationContext translationContext,
+ ArrayList<Expression> parameters) {
+ }
+
+ @Override
+ public void removeFreeVariables(THashSet<Variable> vars) {
+ vars.remove(variable);
+ }
+
+ @Override
+ public Expression resolveAsPattern(TranslationContext context) {
+ return this;
+ }
+
+ @Override
+ public Expression replace(ReplaceContext context) {
+ if(context.inPattern) {
+ Type type = variable.getType().replace(context.tvarMap);
+
+ Variable newVariable = new Variable(variable.name);
+ newVariable.setType(type);
+ EVariable result = new EVariable(newVariable);
+ context.varMap.put(variable, result);
+ return result;
+ }
+ else {
+ if(variable == null) {
+ EVariable newVariable = new EVariable(location, null);
+ newVariable.setType(getType().replace(context.tvarMap));
+ if(context.typingContext == null)
+ throw new InternalCompilerError(location, "Encountered unresolved variable but not in type checking phase.");
+ context.typingContext.addConstraintDemand(newVariable);
+ return newVariable;
+ }
+ else {
+ Expression expression = context.varMap.get(variable);
+ if(expression != null)
+ return expression.copy(context.typingContext);
+ else {
+ return new EVariable(variable);
+ }
+ }
+ }
+ }
+
+ @Override
+ public IExpression toIExpression(ExpressionInterpretationContext target) {
+ return new IVariable(target.getVariableId(variable));
+ }
+
+ @Override
+ public Expression inferType(TypingContext context) {
+ if(context.isInPattern()) {
+ variable.setType(Types.metaVar(Kinds.STAR));
+ return this;
+ }
+ else
+ return applyPUnit(context);
+ }
+
+ @Override
+ public Expression checkBasicType(TypingContext context, Type requiredType) {
+ if(context.isInPattern()) {
+ variable.setType(requiredType);
+ return this;
+ }
+ else
+ return context.subsume(this, requiredType);
+ }
+
+ @Override
+ public Expression decorate(ExpressionDecorator decorator) {
+ return decorator.decorate(this);
+ }
+
+ @Override
+ public boolean isEffectful() {
+ return false;
+ }
+
+ @Override
+ public void collectEffects(THashSet<Type> effects) {
+ }
+
+ @Override
+ public void setLocationDeep(long loc) {
+ if(location == Locations.NO_LOCATION)
+ location = loc;
+ }
+
+ @Override
+ public void accept(ExpressionVisitor visitor) {
+ visitor.visit(this);
+ }
+
+ @Override
+ public void forVariables(VariableProcedure procedure) {
+ if(variable != null)
+ procedure.execute(location, variable);
+ }
+
+ @Override
+ public boolean isPattern(int arity) {
+ return arity == 0;
+ }
+
+ @Override
+ public Expression accept(ExpressionTransformer transformer) {
+ return transformer.transform(this);
+ }
+
+ @Override
+ public boolean equalsExpression(Expression expression) {
+ if(expression.getClass() != getClass())
+ return false;
+ EVariable other = (EVariable)expression;
+ return variable == other.variable;
+ }
+
+}