Program.java

package de.fuberlin.projecta.analysis.ast;

import de.fuberlin.projecta.analysis.SemanticException;
import de.fuberlin.projecta.analysis.SymbolTableStack;


public class Program extends AbstractSyntaxTree {
	
	@Override
	public void buildSymbolTable(SymbolTableStack stack){
		stack.push();
		for(int i = 0; i < this.getChildrenCount(); i++){
			((AbstractSyntaxTree)this.getChild(i)).buildSymbolTable(stack);
		}
		table = stack.pop();			
	}

	@Override
	public void checkSemantics() {
		int mains = 0;
		for (int i = 0; i < this.getChildrenCount(); i++) {
			AbstractSyntaxTree child = (AbstractSyntaxTree) this.getChild(i);
			child.checkSemantics();

			if (child instanceof FuncDef) {
				String name = ((Id) child.getChild(1)).getValue();
				if (name.equals("main")) {
					mains++;
				}
			}
		}

		if (mains != 1) {
			throw new SemanticException(
					"Program needs exactly one main method! Program contains "
							+ mains + " main methods.", this);
		}
	}

	/*
	 * GenCode already implemented by AbstractSyntaxTree
	 * - i don't care, we need some declarations
	 */
	public String genCode(){
		//we use puts to print to screen
		String out = "declare i32 @puts(i8*) nounwind\n";
		out += "declare i32 @printf(i8*, ...) nounwind\n";
		out += super.genStruct();
		out += super.genCode();
		return out;
	}
}