-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.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.IIf;\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 EIf extends Expression {\r
- public Expression condition;\r
- public Expression then_;\r
- public Expression else_;\r
- \r
- public EIf(Expression condition, Expression then_, Expression else_) {\r
- this.condition = condition;\r
- this.then_ = then_;\r
- this.else_ = else_;\r
- }\r
-\r
- public EIf(long loc, Expression condition, Expression then_, Expression else_) {\r
- super(loc);\r
- this.condition = condition;\r
- this.then_ = then_;\r
- this.else_ = else_;\r
- }\r
-\r
- public void collectRefs(TObjectIntHashMap<Object> allRefs, TIntHashSet refs) {\r
- condition.collectRefs(allRefs, refs);\r
- then_.collectRefs(allRefs, refs);\r
- else_.collectRefs(allRefs, refs);\r
- }\r
-\r
- @Override\r
- public void collectVars(TObjectIntHashMap<Variable> allVars,\r
- TIntHashSet vars) {\r
- condition.collectVars(allVars, vars);\r
- then_.collectVars(allVars, vars);\r
- else_.collectVars(allVars, vars);\r
- }\r
-\r
- @Override\r
- protected void updateType() throws MatchException {\r
- setType(then_.getType());\r
- }\r
- \r
- @Override\r
- public IVal toVal(Environment env, CodeWriter w) {\r
- IVal conditionVal = condition.toVal(env, w); \r
- \r
- CodeWriter thenBlock = w.createBlock();\r
- CodeWriter elseBlock = w.createBlock();\r
- \r
- CodeWriter joinPoint = w.createBlock(getType());\r
- \r
- w.if_(conditionVal, thenBlock.getContinuation(), elseBlock.getContinuation());\r
- \r
- IVal thenVal = then_.toVal(env, thenBlock);\r
- thenBlock.jump(joinPoint.getContinuation(), thenVal);\r
- \r
- IVal elseVal = else_.toVal(env, elseBlock);\r
- elseBlock.jump(joinPoint.getContinuation(), elseVal);\r
- \r
- w.continueAs(joinPoint);\r
- \r
- return w.getParameters()[0];\r
- }\r
-\r
- @Override\r
- public void collectFreeVariables(THashSet<Variable> vars) {\r
- condition.collectFreeVariables(vars);\r
- then_.collectFreeVariables(vars);\r
- else_.collectFreeVariables(vars);\r
- }\r
-\r
- @Override\r
- public Expression simplify(SimplificationContext context) {\r
- condition = condition.simplify(context);\r
- then_ = then_.simplify(context);\r
- else_ = else_.simplify(context);\r
- return this;\r
- }\r
-\r
- @Override\r
- public Expression resolve(TranslationContext context) {\r
- condition = condition.resolve(context);\r
- then_ = then_.resolve(context);\r
- else_ = else_.resolve(context);\r
- return this;\r
- }\r
-\r
- @Override\r
- public Expression replace(ReplaceContext context) {\r
- return new EIf(condition.replace(context), \r
- then_.replace(context), \r
- else_.replace(context));\r
- }\r
- \r
- @Override\r
- public Expression checkBasicType(TypingContext context, Type requiredType) {\r
- condition = condition.checkType(context, Types.BOOLEAN);\r
- then_ = then_.checkType(context, requiredType);\r
- else_ = else_.checkType(context, requiredType);\r
- return this;\r
- }\r
- \r
- @Override\r
- public Expression decorate(ExpressionDecorator decorator) {\r
- condition = condition.decorate(decorator);\r
- then_ = then_.decorate(decorator);\r
- else_ = else_.decorate(decorator); \r
- return decorator.decorate(this);\r
- }\r
- \r
- @Override\r
- public boolean isEffectful() {\r
- return condition.isEffectful() || then_.isEffectful() || else_.isEffectful();\r
- }\r
-\r
- @Override\r
- public void collectEffects(THashSet<Type> effects) {\r
- condition.collectEffects(effects);\r
- then_.collectEffects(effects);\r
- else_.collectEffects(effects);\r
- }\r
- \r
- @Override\r
- public void setLocationDeep(long loc) {\r
- if(location == Locations.NO_LOCATION) {\r
- location = loc;\r
- condition.setLocationDeep(loc);\r
- then_.setLocationDeep(loc);\r
- else_.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
- return new IIf(condition.toIExpression(target), then_.toIExpression(target), else_.toIExpression(target));\r
- }\r
- \r
- public Expression getCondition() {\r
- return condition;\r
- }\r
- \r
- public Expression getThen() {\r
- return then_;\r
- }\r
- \r
- public Expression getElse() {\r
- return else_;\r
- }\r
-\r
- @Override\r
- public void forVariables(VariableProcedure procedure) {\r
- condition.forVariables(procedure);\r
- then_.forVariables(procedure);\r
- else_.forVariables(procedure);\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.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.interpreted.IConstant;
+import org.simantics.scl.compiler.internal.interpreted.IExpression;
+import org.simantics.scl.compiler.internal.interpreted.IIf;
+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.runtime.tuple.Tuple0;
+
+public class EIf extends Expression {
+ public Expression condition;
+ public Expression then_;
+ public Expression else_; // may be null
+
+ public EIf(Expression condition, Expression then_, Expression else_) {
+ this.condition = condition;
+ this.then_ = then_;
+ this.else_ = else_;
+ }
+
+ public EIf(long loc, Expression condition, Expression then_, Expression else_) {
+ super(loc);
+ this.condition = condition;
+ this.then_ = then_;
+ this.else_ = else_;
+ }
+
+ @Override
+ protected void updateType() throws MatchException {
+ setType(then_.getType());
+ }
+
+ @Override
+ public IVal toVal(CompilationContext context, CodeWriter w) {
+ IVal conditionVal = condition.toVal(context, w);
+ CodeWriter joinPoint = w.createBlock(getType());
+ CodeWriter thenBlock = w.createBlock();
+ if(else_ != null) {
+ CodeWriter elseBlock = w.createBlock();
+ w.if_(conditionVal, thenBlock.getContinuation(), elseBlock.getContinuation());
+
+ IVal elseVal = else_.toVal(context, elseBlock);
+ elseBlock.jump(joinPoint.getContinuation(), elseVal);
+ }
+ else {
+ w.if_(conditionVal, thenBlock.getContinuation(), joinPoint.getContinuation());
+ }
+ IVal thenVal = then_.toVal(context, thenBlock);
+ thenBlock.jump(joinPoint.getContinuation(), thenVal);
+ w.continueAs(joinPoint);
+
+ return w.getParameters()[0];
+ }
+
+ @Override
+ public Expression simplify(SimplificationContext context) {
+ condition = condition.simplify(context);
+ then_ = then_.simplify(context);
+ if(else_ != null)
+ else_ = else_.simplify(context);
+ return this;
+ }
+
+ @Override
+ public Expression resolve(TranslationContext context) {
+ condition = condition.resolve(context);
+ then_ = then_.resolve(context);
+ if(else_ != null)
+ else_ = else_.resolve(context);
+ return this;
+ }
+
+ @Override
+ public Expression replace(ReplaceContext context) {
+ return new EIf(condition.replace(context),
+ then_.replace(context),
+ else_ == null ? null : else_.replace(context));
+ }
+
+ @Override
+ public Expression checkBasicType(TypingContext context, Type requiredType) {
+ condition = condition.checkType(context, Types.BOOLEAN);
+ then_ = then_.checkType(context, requiredType);
+ if(else_ != null)
+ else_ = else_.checkType(context, requiredType);
+ else
+ context.getErrorLog().log(location, "Else branch is required because the return value of the if expression is used.");
+ return this;
+ }
+
+ @Override
+ public Expression checkIgnoredType(TypingContext context) {
+ condition = condition.checkType(context, Types.BOOLEAN);
+ then_ = then_.checkIgnoredType(context);
+ if(else_ != null)
+ else_ = else_.checkIgnoredType(context);
+ return this;
+ }
+
+ @Override
+ public boolean isEffectful() {
+ return condition.isEffectful() || then_.isEffectful() || (else_ != null && else_.isEffectful());
+ }
+
+ @Override
+ public void setLocationDeep(long loc) {
+ if(location == Locations.NO_LOCATION) {
+ location = loc;
+ condition.setLocationDeep(loc);
+ then_.setLocationDeep(loc);
+ if(else_ != null)
+ else_.setLocationDeep(loc);
+ }
+ }
+
+ @Override
+ public void accept(ExpressionVisitor visitor) {
+ visitor.visit(this);
+ }
+
+ @Override
+ public IExpression toIExpression(ExpressionInterpretationContext target) {
+ return new IIf(condition.toIExpression(target), then_.toIExpression(target),
+ else_ != null ? else_.toIExpression(target) : new IConstant(Tuple0.INSTANCE));
+ }
+
+ @Override
+ public Expression accept(ExpressionTransformer transformer) {
+ return transformer.transform(this);
+ }
+
+ @Override
+ public int getSyntacticFunctionArity() {
+ return Math.max(then_.getSyntacticFunctionArity(), else_.getSyntacticFunctionArity());
+ }
+}