RootNode.java

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

import edu.udel.cis.vsl.tass.ast.IF.AbstractSyntaxTreeIF;

import edu.udel.cis.vsl.tass.ast.IF.RootNodeIF;
import edu.udel.cis.vsl.tass.ast.IF.ASTNodeIF;
import edu.udel.cis.vsl.tass.ast.IF.SequenceNodeIF;
import edu.udel.cis.vsl.tass.ast.IF.FileNodeIF;
import edu.udel.cis.vsl.tass.ast.IF.declaration.FunctionDeclarationNodeIF;
import edu.udel.cis.vsl.tass.ast.IF.GlobalScopeNodeIF;

import java.util.NoSuchElementException;

import java.io.PrintWriter;

public class RootNode extends ASTNode implements RootNodeIF {

	private SequenceNodeIF<GlobalScopeNodeIF> globalScopeNodes;

	private FunctionDeclarationNodeIF mainFunction;

	private SequenceNodeIF<FileNodeIF> files;

	public RootNode(long id) {
		super(id);
	}
	
	public RootNode(long id,
					FunctionDeclarationNodeIF mMainFunction,
					SequenceNodeIF<GlobalScopeNodeIF> mGlobalScopeNodes) {
		super(id);
		globalScopeNodes = mGlobalScopeNodes;
		mainFunction = mMainFunction;
	}

	@Override
	public SequenceNodeIF<GlobalScopeNodeIF> globalScopeNodes() {
		return globalScopeNodes;
	}

	@Override
	public FunctionDeclarationNodeIF mainFunction() {
		return mainFunction;
	}

	@Override
	public void setGlobalScopeNodes(SequenceNodeIF<GlobalScopeNodeIF> mGlobalScopeNodes) {
		globalScopeNodes=mGlobalScopeNodes;
	}

	@Override
	public void setMainFunction(FunctionDeclarationNodeIF mainFunction) {
		this.mainFunction = mainFunction;
	}
	
	public void setFiles(SequenceNodeIF<FileNodeIF> files) {
		this.files = files;
	}

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

	@Override
	public ASTNodeIF child(int index) throws NoSuchElementException {
		switch (index) {
		case 0:
			return globalScopeNodes;
		default:
			throw new NoSuchElementException("Node " + id()
					+ " does not have a child with index " + index + ".");
		}
	}

	@Override
	@SuppressWarnings("unchecked")
	public void setChild(int i, ASTNodeIF child) throws NoSuchElementException {
		switch (i) {
		case 0:
			//FIXME: unchecked cast
			globalScopeNodes = (SequenceNodeIF<GlobalScopeNodeIF>) child;
			break;
		default:
			throw new NoSuchElementException("Node " + id()
					+ " does not have a child with index " + i + ".");
		}
	}
	
	@Override
	protected String nodeType() {
		return "Root";
	}

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

		out.println(prefix+"<rtn:globalScopeNodes>");
		globalScopeNodes.toXml(prefix+"  ",out,ast);
		out.println(prefix+"</rtn:globalScopeNodes>");

		if (mainFunction != null) {
			out.println(prefix+"<rtn:mainFunction>");
			mainFunction.toXml(prefix+"  ",out,ast);
			out.println(prefix+"</rtn:mainFunction>");
		}
	}
}