-package org.simantics.scl.compiler.elaboration.expressions;\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.types.Type;\r
-import org.simantics.scl.compiler.types.exceptions.MatchException;\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 EAsPattern extends Expression {\r
-\r
- Variable var;\r
- EVar eVar;\r
- Expression pattern;\r
-\r
- public EAsPattern(EVar eVar, Expression pattern) {\r
- this.eVar = eVar;\r
- this.pattern = pattern;\r
- }\r
- \r
- public EAsPattern(Variable var, Expression pattern) {\r
- this.var = var;\r
- this.pattern = pattern;\r
- }\r
- \r
- public Variable getVariable() {\r
- return var;\r
- }\r
- \r
- public Expression getPattern() {\r
- return pattern;\r
- }\r
-\r
- @Override\r
- public void collectRefs(TObjectIntHashMap<Object> allRefs,\r
- TIntHashSet refs) {\r
- pattern.collectRefs(allRefs, refs);\r
- }\r
-\r
- @Override\r
- public void collectVars(TObjectIntHashMap<Variable> allVars,\r
- TIntHashSet vars) {\r
- int id = allVars.get(var);\r
- if(id >= 0)\r
- vars.add(id);\r
- pattern.collectVars(allVars, vars);\r
- }\r
-\r
- @Override\r
- public void collectFreeVariables(THashSet<Variable> vars) {\r
- vars.add(var);\r
- pattern.collectFreeVariables(vars);\r
- }\r
- \r
- @Override\r
- public Expression simplify(SimplificationContext context) {\r
- pattern = pattern.simplify(context);\r
- return this;\r
- }\r
-\r
- @Override\r
- public Expression resolve(TranslationContext context) {\r
- context.getErrorLog().log(location, "As binding (@) can be used only in patterns.");\r
- return this;\r
- }\r
- \r
- @Override\r
- public Expression resolveAsPattern(TranslationContext context) {\r
- var = context.newVariable(eVar.name);\r
- pattern = pattern.resolveAsPattern(context);\r
- return this;\r
- }\r
- \r
- @Override\r
- protected void updateType() throws MatchException {\r
- setType(pattern.getType());\r
- }\r
-\r
- @Override\r
- public IVal toVal(Environment env, CodeWriter w) {\r
- throw new InternalCompilerError("EAsPattern allowed only in patterns.");\r
- }\r
- \r
- @Override\r
- public Expression inferType(TypingContext context) {\r
- pattern = pattern.inferType(context);\r
- var.setType(pattern.getType());\r
- return this;\r
- }\r
- \r
- @Override\r
- public Expression checkBasicType(TypingContext context, Type requiredType) {\r
- pattern = pattern.checkBasicType(context, requiredType);\r
- var.setType(pattern.getType());\r
- return this;\r
- }\r
-\r
- @Override\r
- public Expression decorate(ExpressionDecorator decorator) {\r
- pattern = pattern.decorate(decorator);\r
- return decorator.decorate(this);\r
- }\r
- \r
- @Override\r
- public Expression replace(ReplaceContext context) {\r
- if(!context.inPattern) \r
- throw new IllegalStateException();\r
- \r
- Variable newVariable = new Variable(var.name);\r
- newVariable.setType(var.getType().replace(context.tvarMap));\r
- context.varMap.put(var, new EVariable(newVariable));\r
- \r
- EAsPattern result = new EAsPattern(eVar, pattern.replace(context));\r
- result.var = newVariable;\r
- \r
- return result;\r
- }\r
-\r
- @Override\r
- public void collectEffects(THashSet<Type> effects) {\r
- pattern.collectEffects(effects);\r
- }\r
- \r
- @Override\r
- public void setLocationDeep(long loc) {\r
- if(location == Locations.NO_LOCATION) {\r
- location = loc;\r
- pattern.setLocationDeep(loc);\r
- if(eVar != null)\r
- eVar.setLocationDeep(loc);\r
- }\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
- procedure.execute(eVar.location, var);\r
- pattern.forVariables(procedure);\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 org.simantics.scl.compiler.common.exceptions.InternalCompilerError;
+import org.simantics.scl.compiler.compilation.CompilationContext;
+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.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.types.Type;
+import org.simantics.scl.compiler.types.exceptions.MatchException;
+
+import gnu.trove.map.hash.TObjectIntHashMap;
+import gnu.trove.set.hash.THashSet;
+import gnu.trove.set.hash.TIntHashSet;
+
+public class EAsPattern extends Expression {
+
+ Variable var;
+ EVar eVar;
+ Expression pattern;
+
+ public EAsPattern(EVar eVar, Expression pattern) {
+ this.eVar = eVar;
+ this.pattern = pattern;
+ }
+
+ public EAsPattern(Variable var, Expression pattern) {
+ this.var = var;
+ this.pattern = pattern;
+ }
+
+ public Variable getVariable() {
+ return var;
+ }
+
+ public Expression getPattern() {
+ return pattern;
+ }
+
+ @Override
+ public void collectRefs(TObjectIntHashMap<Object> allRefs,
+ TIntHashSet refs) {
+ pattern.collectRefs(allRefs, refs);
+ }
+
+ @Override
+ public void collectVars(TObjectIntHashMap<Variable> allVars,
+ TIntHashSet vars) {
+ int id = allVars.get(var);
+ if(id >= 0)
+ vars.add(id);
+ pattern.collectVars(allVars, vars);
+ }
+
+ @Override
+ public void collectFreeVariables(THashSet<Variable> vars) {
+ throw new InternalCompilerError(location, "Cannot collect free variables for a pattern.");
+ }
+
+ @Override
+ public void removeFreeVariables(THashSet<Variable> vars) {
+ vars.remove(var);
+ pattern.removeFreeVariables(vars);
+ }
+
+ @Override
+ public Expression simplify(SimplificationContext context) {
+ pattern = pattern.simplify(context);
+ return this;
+ }
+
+ @Override
+ public Expression resolve(TranslationContext context) {
+ context.getErrorLog().log(location, "As binding (@) can be used only in patterns.");
+ return this;
+ }
+
+ @Override
+ public Expression resolveAsPattern(TranslationContext context) {
+ var = context.newVariable(eVar.name);
+ pattern = pattern.resolveAsPattern(context);
+ return this;
+ }
+
+ @Override
+ protected void updateType() throws MatchException {
+ setType(pattern.getType());
+ }
+
+ @Override
+ public IVal toVal(CompilationContext context, CodeWriter w) {
+ throw new InternalCompilerError("EAsPattern allowed only in patterns.");
+ }
+
+ @Override
+ public Expression inferType(TypingContext context) {
+ pattern = pattern.inferType(context);
+ var.setType(pattern.getType());
+ return this;
+ }
+
+ @Override
+ public Expression checkBasicType(TypingContext context, Type requiredType) {
+ pattern = pattern.checkBasicType(context, requiredType);
+ var.setType(pattern.getType());
+ return this;
+ }
+
+ @Override
+ public Expression decorate(ExpressionDecorator decorator) {
+ pattern = pattern.decorate(decorator);
+ return decorator.decorate(this);
+ }
+
+ @Override
+ public Expression replace(ReplaceContext context) {
+ if(!context.inPattern)
+ throw new IllegalStateException();
+
+ Variable newVariable = new Variable(var.name);
+ newVariable.setType(var.getType().replace(context.tvarMap));
+ context.varMap.put(var, new EVariable(newVariable));
+
+ EAsPattern result = new EAsPattern(eVar, pattern.replace(context));
+ result.var = newVariable;
+
+ return result;
+ }
+
+ @Override
+ public void collectEffects(THashSet<Type> effects) {
+ pattern.collectEffects(effects);
+ }
+
+ @Override
+ public void setLocationDeep(long loc) {
+ if(location == Locations.NO_LOCATION) {
+ location = loc;
+ pattern.setLocationDeep(loc);
+ if(eVar != null)
+ eVar.setLocationDeep(loc);
+ }
+ }
+
+ @Override
+ public void accept(ExpressionVisitor visitor) {
+ visitor.visit(this);
+ }
+
+ @Override
+ public void forVariables(VariableProcedure procedure) {
+ procedure.execute(eVar.location, var);
+ pattern.forVariables(procedure);
+ }
+
+ @Override
+ public Expression accept(ExpressionTransformer transformer) {
+ return transformer.transform(this);
+ }
+
+}