AssignmentNode.java

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

import java.io.PrintWriter;
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.expression.AssignmentNodeIF;
import edu.udel.cis.vsl.tass.ast.IF.expression.ExpressionNodeIF;
import edu.udel.cis.vsl.tass.ast.IF.expression.LHSExpressionNodeIF;

public class AssignmentNode extends ModificationNode implements
		AssignmentNodeIF {

	ExpressionNodeIF rhs;

	AssignmentNodeIF.AST_ASSIGNMENT_TYPE assignmentType =
		AssignmentNodeIF.AST_ASSIGNMENT_TYPE.SIMPLE;
	
	public AssignmentNode(long id) {
		super(id);
	}
	
	public AssignmentNode(long id, LHSExpressionNodeIF lhs, ExpressionNodeIF rhs) {
		super(id, lhs);
		this.rhs = rhs;
	}

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

	@Override
	public void setRhs(ExpressionNodeIF rhs) {
		this.rhs = rhs;
	}

	@Override
	public void setAssignmentType(AssignmentNodeIF.AST_ASSIGNMENT_TYPE assignmentType) {
		this.assignmentType = assignmentType;
	}

	@Override
	public AssignmentNodeIF.AST_ASSIGNMENT_TYPE assignmentType() {
		return assignmentType;
	}
	
	@Override
	public ASTNodeIF child(int index) throws NoSuchElementException {
		switch (index) {
		case 0:
			//Fallthrough
		case 1:
			return super.child(index);
		case 2:
			return rhs;
		default:
			throw new NoSuchElementException("Node " + id()
					+ " does not have a child with index " + index + ".");
		}
	}

	@Override
	@SuppressWarnings("fallthrough")
	public void setChild(int i, ASTNodeIF child) throws NoSuchElementException {
		switch (i) {
		case 0:
			//Fallthrough
		case 1:
			super.setChild(i,child);
			break;
		case 2:
			rhs = (ExpressionNodeIF) child;
			break;
		default:
			throw new NoSuchElementException("Node " + id()
					+ " does not have a child with index " + i + ".");
		}
	}

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

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

	@Override
	public void toXml(String prefix, PrintWriter out,AbstractSyntaxTreeIF ast) {
		super.toXml(prefix,out,ast);

		out.println(prefix+"<asn:assignmentType>"
                            +assignmentType +
                            "</asn:assignmentType>");

		out.println(prefix+"<asn:rhs>");
		rhs.toXml(prefix+"  ",out,ast);
		out.println(prefix+"</asn:rhs>");
	}
}