package slangc.model.statements;

import slangc.api.BytecodeInstructionWriter;
import slangc.api.Reporter;
import slangc.bytecode.TypeSignature;
import slangc.model.ClauseModel;
import slangc.model.ExpressionModel;
import slangc.model.ExpressionOwner;
import slangc.model.InnerTypeScope;
import slangc.model.MemberModel;
import slangc.model.StatementModel;
import slangc.model.StatementOwner;
import slangc.model.TypeModel;
import slangc.model.clauses.ElseClause;
import slangc.parser.Branch;
import slangc.parser.Node;

public class IfStatement extends StatementModel implements StatementOwner, ExpressionOwner {
	private ExpressionModel condition;
	private StatementModel innerStatement;
	private ElseClause elseClause;

	public IfStatement(StatementOwner owner, Branch source) {
		super(owner, source);
		condition = ExpressionModel.construct(this, source.getSubnode(1));
		innerStatement = StatementModel.construct(this, source.getSubnode(2));
		elseClause = (ElseClause)ClauseModel.construct(this, source.getSubnode(3));
	}

	public MethodModel[] lookupSimpleMethod(String name) {
		// TODO: Better interface support
		return super.lookupSimpleMethod(name);
	}

	public slangc.model.Named lookupSimpleName(String name) {
		// TODO: Better interface support
		return super.lookupSimpleName(name);
	}

	@Override
	public void dump(Reporter reporter, String indent, String incr) {
		reporter.note("DUMP", indent + "> If:");
		condition.dump(reporter, indent + incr, incr);
		innerStatement.dump(reporter, indent + incr, incr);
		if (elseClause != null) {
			elseClause.dump(reporter, indent + incr, incr);
		}
	}

	//@Override
	public TypeModel resolveType(Node subnode) {
		return getOwner().resolveType(subnode);
	}

	//@Override
	public MemberModel getMethodOrField() {
		return getOwner().getMethodOrField();
	}


	//@Override
	public InnerTypeScope getTypeScope() {
		return getOwner().getTypeScope();
	}
	
	//@Override
	public TypeModel getExpectedResult(ExpressionModel e) {
		// TODO Probably should be boolean
		return null;
	}
	
	@Override
	public int resolveExpressions() {
		return condition.resolveExpressions() + innerStatement.resolveExpressions() + (elseClause == null ? 0 : elseClause.resolveExpressions());
	}
	
	@Override
	public void innerGenerate(BytecodeInstructionWriter w) {
		BytecodeInstructionWriter.Label elseLabel = w.newLabel("if_else");
		BytecodeInstructionWriter.Label endLabel = w.newLabel("if_end");
		
		TypeSignature xt = getCondition().generate(w);
		w.genJump(BytecodeInstructionWriter.JumpOp.IFFALSE, elseLabel);
		getInnerStatement().generate(w);
		w.genJump(BytecodeInstructionWriter.JumpOp.GOTO, endLabel);
		w.labelHere(elseLabel);
		if (getElseClause() != null) {
			getElseClause().getInnerStatement().generate(w);
		}
		w.labelHere(endLabel);
	}
	
	public ExpressionModel getCondition() {
		return condition;
	}
	
	public StatementModel getInnerStatement() {
		return innerStatement;
	}
	
	public ElseClause getElseClause() {
		return elseClause;
	}
}