package slangc.model.statements;

import slangc.api.BytecodeInstructionWriter;
import slangc.api.Reporter;
import slangc.bytecode.TypeSignature;
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.parser.Branch;
import slangc.parser.Node;

public class WhileStatement extends StatementModel implements StatementOwner, ExpressionOwner {
	private ExpressionModel condition;
	private StatementModel innerStatement;

	public WhileStatement(StatementOwner owner, Branch source) {
		super(owner, source);
		condition = ExpressionModel.construct(this, source.getSubnode(1));
		innerStatement = StatementModel.construct(this, source.getSubnode(2));
	}

	@Override
	public void dump(Reporter reporter, String indent, String incr) {
		reporter.note("DUMP", indent + "> While:");
		condition.dump(reporter, indent + incr, incr);
		innerStatement.dump(reporter, indent + incr, incr);
	}
	
	@Override
	public int resolveExpressions() {
		return condition.resolveExpressions() + innerStatement.resolveExpressions();
	}

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

	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 MemberModel getMethodOrField() {
		return getOwner().getMethodOrField();
	}

	//@Override
	public InnerTypeScope getTypeScope() {
		return getOwner().getTypeScope();
	}
	
	//@Override
	public TypeModel getExpectedResult(ExpressionModel e) {
		// TODO Should probably be boolean
		return null;
	}
	
	public ExpressionModel getCondition() {
		return condition;
	}
	
	public StatementModel getInnerStatement() {
		return innerStatement;
	}
	
	@Override
	public void innerGenerate(BytecodeInstructionWriter w) {
		BytecodeInstructionWriter.Label beginLabel = w.newLabel("while_begin");
		BytecodeInstructionWriter.Label endLabel = w.newLabel("while_end");
		
		w.pushContinue(beginLabel);
		w.pushBreak(endLabel);
		
		w.labelHere(beginLabel);
		TypeSignature xt = getCondition().generate(w);
		w.genJump(BytecodeInstructionWriter.JumpOp.IFFALSE, endLabel);
		getInnerStatement().generate(w);
		w.genJump(BytecodeInstructionWriter.JumpOp.GOTO, beginLabel);
		w.labelHere(endLabel);
		
		w.popBreak(endLabel);
		w.popContinue(beginLabel);
	}
}