AssumeStatementNode.java

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

import java.util.NoSuchElementException;

import edu.udel.cis.vsl.tass.ast.IF.ASTNodeIF;
import edu.udel.cis.vsl.tass.ast.IF.expression.ExpressionNodeIF;
import edu.udel.cis.vsl.tass.ast.IF.statement.AssumeStatementNodeIF;

public class AssumeStatementNode extends StatementNode
	implements AssumeStatementNodeIF {

	private ExpressionNodeIF predicate;

	public AssumeStatementNode(long id) {
		super(id);
	}
	
	public AssumeStatementNode(long id, ExpressionNodeIF predicate) {
		super(id);
		this.predicate = predicate;
	}
	
	@Override
	public ExpressionNodeIF predicate() {
		return predicate;
	}

	@Override
	public void setPredicate(ExpressionNodeIF predicate) {
		this.predicate = predicate;
	}

	@Override
	public ASTNodeIF child(int index) throws NoSuchElementException {
		switch (index) {
		case 0:
			return super.child(index);
		case 1:
			return predicate;
		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:
			predicate = (ExpressionNodeIF) child;
			break;
		default:
			throw new NoSuchElementException("Node " + id()
					+ " does not have a child with index " + i + ".");
		}
	}

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

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

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