ForLoopNode.java

package edu.udel.cis.vsl.tass.ast.impl.statement;

import java.io.PrintWriter;
import java.util.Collection;
import java.util.NoSuchElementException;

import edu.udel.cis.vsl.tass.ast.IF.ASTNodeIF;
import edu.udel.cis.vsl.tass.ast.IF.AbstractSyntaxTreeIF;
import edu.udel.cis.vsl.tass.ast.IF.SequenceNodeIF;
import edu.udel.cis.vsl.tass.ast.IF.expression.ExpressionNodeIF;
import edu.udel.cis.vsl.tass.ast.IF.expression.PureExpressionNodeIF;
import edu.udel.cis.vsl.tass.ast.IF.statement.ForLoopNodeIF;
import edu.udel.cis.vsl.tass.ast.IF.statement.StatementNodeIF;

public class ForLoopNode extends StatementNode implements ForLoopNodeIF {

	private StatementNodeIF body;
	private SequenceNodeIF<PureExpressionNodeIF> invariants;
	private StatementNodeIF initializer;
	private ExpressionNodeIF condition;
	private StatementNodeIF incrementer;

	public ForLoopNode(long id) {
		super(id);
	}
	
	public ForLoopNode(long id,
					   StatementNodeIF body,
					   StatementNodeIF initializer,
					   ExpressionNodeIF condition,
					   StatementNodeIF incrementer) {
		super(id);
		this.body = body;
		this.initializer = initializer;
		this.condition = condition;
		this.incrementer = incrementer;
	}

	@Override
	public void setBody(StatementNodeIF body) {
		this.body = body;
	}

	@Override
	public StatementNodeIF body() {
		return body;
	}

	@Override
	public void setInvariants(SequenceNodeIF<PureExpressionNodeIF> invariants) {
		this.invariants = invariants;
	}

	@Override
	public SequenceNodeIF<PureExpressionNodeIF> invariants() {
		return invariants;
	}

	@Override
	public void setInitializer(StatementNodeIF initializer) {
		this.initializer = initializer;
	}

	@Override
	public StatementNodeIF initializer() {
		return initializer;
	}
	
	@Override
	public void setCondition(ExpressionNodeIF condition) {
		this.condition = condition;
	}

	@Override
	public ExpressionNodeIF condition() {
		return condition;
	}

	@Override
	public void setIncrementer(StatementNodeIF incrementer) {
		this.incrementer = incrementer;
	}
	
	@Override
	public StatementNodeIF incrementer() {
		return incrementer;
	}

	@Override
	public int numChildren() {
		return 5;
	}

	@Override
	public ASTNodeIF child(int index) throws NoSuchElementException {
		switch (index) {
		case 0:
			return super.child(index);
		case 1:
			return body;
		case 2:
			return initializer;
		case 3:
			return condition;
		case 4:
			return incrementer;
		default:
			throw new NoSuchElementException("Node " + id()
					+ " does not have a child with index " + index + ".");
		}
	}

	@Override
	public void setChild(int i, ASTNodeIF child) throws NoSuchElementException {
		switch (i) {
		case 0:
			super.setChild(i, child);
			break;
		case 1:
			body = (StatementNodeIF)child;
			break;
		case 2:
			initializer = (StatementNodeIF)child;
			break;
		case 3:
			condition = (ExpressionNodeIF)child;
			break;
		case 4:
			incrementer = (StatementNodeIF)child;
			break;
		default:
			throw new NoSuchElementException("Node " + id()
					+ " does not have a child with index " + i + ".");
		}
	}

	@Override
	protected String nodeType() {
		return "ForLoop";
	}

	@Override
	public String toString() {
		return super.toString();
	}

	@Override
	public void toXml(String prefix, PrintWriter out,AbstractSyntaxTreeIF ast) {
		//TODO: Implement
	}
}