CodeGenerator.java

import java.util.ArrayList;
import java.util.Arrays;

public class CodeGenerator {

	private static String lineNumber = "0";
	private static String mechCode = "";
	private static String staticCode = "00 ";
	private static String heapCode = "";
	
	private static ArrayList<String> varTable = new ArrayList<String>();
	//private static ArrayList<String[]> strTable = new ArrayList<String[]>();
	private static ArrayList<String> jumpTable = new ArrayList<String>();
	
	public static void generateCode(ArrayList<Object> ast) {
		//need more than just a "T0"...
		varTable.add("T0");
		for(Object branch : ast) {
			if(branch instanceof Object[]) {
				generateBranch((Object[])branch);
			}
		}
		mechCode += "00 ";
		replaceTemps();
		replaceJumps();
		String result = mechCode.toUpperCase() + staticCode;
		int filler = 255 - (result.length() + heapCode.length()) / 3;
		for(int i = 0; i < filler; i++) {
			result += "00 ";
		}
		System.out.println(result + " " + heapCode);
	}
	
	private static void generateBranch(Object[] branch) {
		String temp = "";
		if(branch[0].equals("LINE")) {
			lineNumber = (String) branch[1];
		}
		else if(branch[0].equals("BLOCK")) {
			for(int i = 1; i < branch.length; i++) {
				generateBranch((Object[])branch[i]);
			}
		}
		else if(branch[0].equals("DECLARE")) {
			if(branch[1].equals("INT") || branch[1].equals("BOOL")) {
				varTable.add("T" + branch[2]);
				staticCode += "00 ";
			}
			else if(branch[1].equals("STR")) {
				varTable.add("T" + branch[2]);
				staticCode += "FF ";
				//strTable.add(new String[]{("S" + branch[2]), Integer.toHexString(255 - Math.round(heapCode.length()/3)*4)});
			}
		}
		else if(branch[0].equals("SET")) {
			if(branch[2] instanceof Object[]) {
				temp = generateExpr((Object[])branch[2], branch[1]);
			}
			else if(branch[2] instanceof String) {
				temp = generateExpr((String)branch[2]);
			}
			else {
				MainDisplay.errorReport = "[Line: " + lineNumber + "]You should never get this message.\n";
			}
			mechCode += temp + " 8D T" + branch[1] + " 00 ";
		}
		else if(branch[0].equals("PRINT")) {
			if(branch[1] instanceof Object[]) {
				temp = generateExpr((Object[])branch[1], "0");
				if(((Object[])branch[1])[0].equals("QUOTE")) {
					mechCode += temp + " 8D T0 00 AC T0 00 A2 02 FF ";
					return;
				}
			}
			else if(branch[1] instanceof String) {
				temp = generateExpr((String)branch[1]);
			}
			else {
				MainDisplay.errorReport = "[Line: " + lineNumber + "]You should never get this message.\n";
			}
			mechCode += temp + " 8D T0 00 AC T0 00 A2 01 FF ";
		}
		else if(branch[0].equals("BRANCH") || branch[0].equals("LOOP")) {
			boolean isLoop = branch[0].equals("LOOP");
			int jStart, jEnd, jPos;
			String jump = "00";
			jPos = jumpTable.size();
			jumpTable.add(jump);
			temp = generateExpr((Object[])branch[1], "0");
			mechCode += temp + " 8D T0 00 A2 00 EC T0 00 D0 J" + jPos + " ";
			jStart = mechCode.replaceAll("J\\d+", "J0").length()/3;
			generateBranch((Object[])branch[2]);
			jEnd = mechCode.replaceAll("J\\d+", "J0").length()/3;
			if(isLoop) {
				jEnd += 7;
			}
			jump = Integer.toHexString(jEnd - jStart);
			if(jump.length() == 1)
				jump = "0" + jump;
			jumpTable.set(jPos, jump);
			
			if(isLoop) {
				jump = Integer.toHexString(jEnd - jStart);
				if(jump.length() == 1)
					jump = "0" + jump;
				jumpTable.set(jPos, jump);
				
				jPos = temp.replaceAll("J\\d+", "J0").length()/3 + 17;
				jump = Integer.toHexString(256 - jPos);
				if(jump.length() == 1)
					jump = "0" + jump;
				mechCode += "A9 00 EC FF 00 D0 " + jump + " ";
			}
		}
		else if(branch[0].equals("")) {
			
		}
	}
	
	private static String generateExpr(Object[] expr, Object id) {
		String temp = "";
		if(expr[0].equals("PLUS")) {
			if(expr[1] instanceof Object[]) {
				temp = generateExpr((Object[])expr[1], id);
			}
			else if (expr[1] instanceof String) {
				temp = generateExpr((String)expr[1]);
			}
			else {
				MainDisplay.errorReport = "[Line: " + lineNumber + "] You should never get this message.\n";
			}
			temp += " 8D T" + id + " 00 ";
			
			if(expr[2] instanceof Object[]) {
				temp += generateExpr((Object[])expr[2], id);
			}
			else if (expr[2] instanceof String) {
				temp += generateExpr((String)expr[2]);
			}
			else {
				MainDisplay.errorReport = "[Line: " + lineNumber + "] You should never get this message.\n";
			}
			temp += " 6D T" + id + " 00";
		}
		else if(expr[0].equals("EQTO") || expr[0].equals("NOTEQ")) {
			if(expr[1] instanceof Object[]) {
				temp = generateExpr((Object[])expr[1], id);
			}
			else if (expr[1] instanceof String) {
				temp = generateExpr((String)expr[1]);
			}
			else {
				MainDisplay.errorReport = "[Line: " + lineNumber + "] You should never get this message.\n";
			}
			temp += " 8D T" + id + " 00 AE T" + id + " 00 ";
			
			if(expr[2] instanceof Object[]) {
				temp += generateExpr((Object[])expr[2], id);
			}
			else if (expr[2] instanceof String) {
				temp += generateExpr((String)expr[2]);
			}
			else {
				MainDisplay.errorReport = "[Line: " + lineNumber + "] You should never get this message.\n";
			}
			temp += " 8D T" + id + " 00 EC T" + id + " 00";
			
			if(expr[0].equals("EQTO")) {
				temp += " A9 01 D0 02 A9 00";
			}
			else {
				temp += " A9 00 D0 02 A9 01";
			}
		}
		else if(expr[0].equals("QUOTE")) {
			String quote = "";
			for(int i = 1; i < expr.length; i++) {
				temp = (String) expr[i];
				if(temp.equals("SPACE")) {
					quote += "20 ";
				}
				else if (!temp.equals("EMPTY")){
					char[] test = temp.toCharArray();
					for(char tst : test) {
						quote += Integer.toHexString(tst) + " ";
					}
				}
			}
			quote += "00 ";
			heapCode = quote + heapCode;
			return "A9 " + Integer.toHexString(255 - (heapCode.length()/3));
		}
		else {
			MainDisplay.errorReport = "[Line: " + lineNumber + "] You should never get this message.\n";
		}
		return temp;
	}
	private static String generateExpr(String expr) {
		if(expr.matches("\\d")) {
			expr = Integer.toHexString(Integer.parseInt(expr));
			if(expr.length() < 2) {
				expr = "0" + expr;
			}
		}
		else if(expr.equals("TRUE")) {
			expr = "01";
		}
		else if(expr.equals("FALSE")) {
			expr = "00";
		}
		else if(expr.matches("[a-z]")) {
			return "AD T" + expr + " 00";
		}
		else {
			MainDisplay.errorReport = "[Line: " + lineNumber + "] You should never get this message.\n";
		}
		
		return "A9 " + expr;
	}
	
	private static void replaceTemps() {
		String temp = "T0",
			   sPos = "00";
		int sStart = mechCode.replaceAll("J\\d+", "J0").length()/3;
		System.out.println("Start Static at " + sStart);
		for(int i = 0; i < varTable.size(); i++) {
			temp = varTable.get(i);
			sPos = Integer.toHexString(sStart + i);
			if(sPos.length() < 2) {
				sPos = "0" + sPos;
			}
			System.out.println("Next Static at " + sPos);
			mechCode = mechCode.replaceAll(temp, sPos);
		}
	}
	
	private static void replaceJumps() {
		String temp = "J0";
		for(int i = 0; i < jumpTable.size(); i++) {
			temp = "J" + i;
			mechCode = mechCode.replaceAll(temp, jumpTable.get(i));
		}
	}

}
/*{intxintyintzbooleanabooleanbbooleanc
if((x==y)==((z != 3 + 4)==(a == (b != (c == true))))) {
print(2)}}$*/