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.Named;
import slangc.model.StatementModel;
import slangc.model.StatementOwner;
import slangc.model.TypeModel;
import slangc.parser.Branch;
import slangc.parser.Node;

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

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

	@Override
	public void dump(Reporter reporter, String indent, String incr) {
		reporter.note("DUMP", indent + "> Do While:");
		condition.dump(reporter, indent + incr, incr);
		innerStatement.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 Should probably be boolean
		return null;
	}
	
	@Override
	public Named lookupSimpleName(String name) {
		return getOwner().lookupSimpleName(name);
	}

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

	
	@Override
	public int resolveExpressions() {
		return innerStatement.resolveExpressions() + condition.resolveExpressions();
	}
	
	public ExpressionModel getCondition() {
		return condition;
	}
	
	public StatementModel getInnerStatement() {
		return innerStatement;
	}
	
	@Override
	public void innerGenerate(BytecodeInstructionWriter w) {
		BytecodeInstructionWriter.Label beginLabel = w.newLabel("dowhile_begin");
		BytecodeInstructionWriter.Label condLabel = w.newLabel("dowhile_cond");
		BytecodeInstructionWriter.Label endLabel = w.newLabel("dowhile_end");
		
		w.pushContinue(condLabel);
		w.pushBreak(endLabel);
		
		w.labelHere(beginLabel);
		getInnerStatement().generate(w);
		w.labelHere(condLabel);
		TypeSignature xt = getCondition().generate(w);
		w.genJump(BytecodeInstructionWriter.JumpOp.IFTRUE, beginLabel);
		w.labelHere(endLabel);
		
		w.popBreak(endLabel);
		w.popContinue(condLabel);
	}
	
}