import java.util.*;
import java.io.*;

public class Z80compiler {
    
    static byte[] mc = new byte[32768]; //memory map
    static int loc = 0; //location in memory map
    static Hashtable labels = new Hashtable(); //hashtable storing references "name" / loc
    static String[] fixref = new String[32768]; //reference of labels to plug back in
    static Hashtable vars = new Hashtable(); //table storing var names/loc
    static String[] fixvar = new String[32768]; //reference of vars to plug back in
    static Stack ifstack = new Stack(); //pop ifcount and increment ifcount when going if
    static int ifcount = 0;
    
    public static void main(String[] args) {
    	
    	System.out.println("STRANGE BEHAVIOR MAY OCCUR IF FILE IS NOT ENDED WITH A NEWLINE");
    	
    	//make sure we've got two args
        if (args.length != 2) {
        	System.out.println("Usage: java Z80compiler <input> <output>");
        	System.exit(0);
        }
        
        //open the source file
        EasyReader source = new EasyReader(args[0]); //source file
        
        //clear out the fixcharts
        for (int i=0; i<fixref.length; i++) {
        	fixref[i] = null;
        	fixvar[i] = null;
        }
        
        //1. initialize the stack to 0x70FF
        writeByte(49);
        writeByte(255);
        writeByte(112);
        
        //2. go through code
        String command = source.readWord();
        while (!source.eof()) {
        	
        	if (command.equals("end")) //end   HALT
        		writeByte(118);
        	
        	if (command.charAt(command.length()-1)==':' && command.charAt(0)!='#') {//label:  write to the labels table
        		labels.put(command.substring(0,command.length()-1), new Integer(loc));
        		System.out.println("CREATED LABEL: " + command.substring(0,command.length()-1) + " at " + loc + "d");
        	}
        	
        	if (command.equals("var")) { //var name   write to the vars table
        		String variable = source.readWord();
        		vars.put(variable, new Integer(0));
        	}
        	
        	if (command.equals("set")) {//set name value/var   
      			String varname = source.readWord();
      			String value = source.readWord();
      			if (isNumber(value)) {
      				//it's a number!
      				writeByte(62); //LD A, n
      				writeByte(Integer.parseInt(value));
      			}
      			else {
      				//it's a variable!
      				writeByte(58); //LD A, (nn)
      				fixvar[loc] = value;
      				writeByte(0);
      				writeByte(0);
      			}
      			//something is loaded into A, so load it to variable
      			writeByte(50); //LD (nn), A
      			fixvar[loc] = varname;
      			writeByte(0);
      			writeByte(0);
      		}
      		
      		if (command.equals("out")) { //out port value/var
      			String port = source.readWord();
      			String value = source.readWord();
      			if (isNumber(value)) {
      				//number
      				writeByte(62); //LD A, n
      				writeByte(Integer.parseInt(value));
      			}
      			else {
      				//variable
      				writeByte(58); //LD A, (nn)
      				fixvar[loc] = value;
      				writeByte(0);
      				writeByte(0);
      			}
      			//value loaded into A, so out it to port
      			writeByte(211); //OUT n, A
      			writeByte(Integer.parseInt(port));
      		}
      		
      		if (command.equals("in")) { //in port variable
      			String port = source.readWord();
      			String var = source.readWord();
      			writeByte(219); //IN A, n
      			writeByte(Integer.parseInt(port));
      			writeByte(50); //LD (nn), A
      			fixvar[loc] = var;
      			writeByte(0);
      			writeByte(0);
      		}
      		
      		if (command.equals("goto")) { //goto label
      			writeByte(195); //JP label
      			fixref[loc] = source.readWord();
      			writeByte(0);
      			writeByte(0);
      		}
      		
      		if (command.equals("inc")) { //inc var
      			writeByte(33); //LD HL, var
      			fixvar[loc] = source.readWord();
      			writeByte(0);
      			writeByte(0);
      			writeByte(52); //INC (HL)
      		}
      		
      		if (command.equals("dec")) { //dec var
      			writeByte(33); //LD HL, var
      			fixvar[loc] = source.readWord();
      			writeByte(0);
      			writeByte(0);
      			writeByte(53); //DEC (HL)
      		}
        	
        	if (command.charAt(0)=='#') { //# blah blah blah comment
        		String blah = source.readLine();
        	}
        	
        	if (command.equals("if")) { //if var1 * value/var2
        		/*
        			LD A, (var1)
        		if (value)
        			SUB value        			
        		if (var2)
        			LD HL, var2	
        			SUB (HL)
        		end      		
        		*/
        		String variable = source.readWord();
        		String operand = source.readWord();
        		String value = source.readWord();
				
				writeByte(58); //LD A (var1)
				fixvar[loc] = variable;
				writeByte(0);
				writeByte(0);
				if (isNumber(value)) {
					//number
					writeByte(214); //SUB value
					writeByte(Integer.parseInt(value));				
				}
				else {
					//variable
					writeByte(33); //LD HL, var
      				fixvar[loc] = value;
      				writeByte(0);
      				writeByte(0);
      				writeByte(150); //SUB (HL)
				}
				//IT'S SUBTRACTED!!!!!
				
				if (operand.equals("=")) {
					writeByte(194); //JP NZ endiflabel
					fixref[loc] = "_RESERVED_ENDIF_LABEL_" + ifcount;
					writeByte(0);
					writeByte(0);
					ifstack.push(new Integer(ifcount));
					ifcount++;
				}
				
				if (operand.equals("!=")) {
					writeByte(202); //JP Z endiflabel
					fixref[loc] = "_RESERVED_ENDIF_LABEL_" + ifcount;
					writeByte(0);
					writeByte(0);
					ifstack.push(new Integer(ifcount));
					ifcount++;
				}
				
				if (operand.equals("<")) { //NEITHER POSITIVE OR ZERO AFTER SUBTRACTION
					writeByte(242); //JP P endiflabel
					fixref[loc] = "_RESERVED_ENDIF_LABEL_" + ifcount;
					writeByte(0);
					writeByte(0);
					writeByte(202); //JP Z endiflabel
					fixref[loc] = "_RESERVED_ENDIF_LABEL_" + ifcount;
					writeByte(0);
					writeByte(0);
					ifstack.push(new Integer(ifcount));
					ifcount++;
				}
				
				if (operand.equals(">")) { //NEITHER NEGATIVE OR ZERO AFTER SUBTRACTION
					writeByte(250); //JP M endiflabel
					fixref[loc] = "_RESERVED_ENDIF_LABEL_" + ifcount;
					writeByte(0);
					writeByte(0);
					writeByte(202); //JP Z endiflabel
					fixref[loc] = "_RESERVED_ENDIF_LABEL_" + ifcount;
					writeByte(0);
					writeByte(0);
					ifstack.push(new Integer(ifcount));
					ifcount++;
				}
				
        	}
        	
        	if (command.equals("endif")) { //endif - push a label
        		labels.put("_RESERVED_ENDIF_LABEL_" + ifstack.peek(), new Integer(loc));
        		//System.out.println("_RESERVED_ENDIF_LABEL_" + ifstack.pop() + " at " + loc + "d");
        	}
        	
        	if (command.equals("add")) { // add var1 value/var2
        		String variable = source.readWord();
        		String value = source.readWord();
        		writeByte(58); //LD A, (var1)
        		fixvar[loc] = variable;
        		writeByte(0);
        		writeByte(0);
        		if (isNumber(value)) {
        			//number
        			writeByte(198); //ADD a, value
        			writeByte(Integer.parseInt(value));
        		}
        		else {
        			//variable
        			writeByte(33); //LD HL, value
      				fixvar[loc] = value;
      				writeByte(0);
      				writeByte(0);
      				writeByte(134); //ADD A, (HL)
        		}
        		writeByte(50); //LD (var1), A
        		fixvar[loc] = variable;
        		writeByte(0);
        		writeByte(0);
        	}
        	       	
        	if (command.equals("return")) { //ret
        		writeByte(201); //RET
        	}
        	
        	if (command.equals("call")) { //cal label
        		writeByte(205); //CALL label
        		fixref[loc] = source.readWord();
        		writeByte(0);
        		writeByte(0);
        	}
        	
        	if (command.equals("sub")) { // sub var1 value/var2
        		String variable = source.readWord();
        		String value = source.readWord();
        		writeByte(58); //LD A, (var1)
        		fixvar[loc] = variable;
        		writeByte(0);
        		writeByte(0);
        		if (isNumber(value)) {
        			//number
        			writeByte(214); //SUB a, value
        			writeByte(Integer.parseInt(value));
        		}
        		else {
        			//variable
        			writeByte(33); //LD HL, value
      				fixvar[loc] = value;
      				writeByte(0);
      				writeByte(0);
      				writeByte(150); //SUB A, (HL)
        		}
        		writeByte(50); //LD (var1), A
        		fixvar[loc] = variable;
        		writeByte(0);
        		writeByte(0);
        	}
        	
        	command = source.readWord();
        }
        
        //3. write in variable space, record the address
        writeByte(118); //HALT, just incase they forgot
        Iterator iter = vars.keySet().iterator();
        while (iter.hasNext()) {
        	Object key = iter.next();
        	int value = ((Integer)vars.get(key)).intValue();
        	writeByte(value);
        	vars.put(key, new Integer(loc-1));
        	System.out.println("CREATED VARIBLE: " + key + "=" + value + " at " + (loc-1) + "d");
        }
        int END_OF_MEMORY = loc;
        
        //4. go through fixvars and plug in variable addresses
        for (int i=0; i<fixvar.length; i++) {
        	if (fixvar[i]!=null) {
        		int address = ((Integer)vars.get(fixvar[i])).intValue();
        		mc[i] = byteLow(address);
        		mc[i+1] = byteHigh(address);
        	}
        }
        
        //5. go through fixrefs and plug in label addresses
        for (int i=0; i<fixref.length; i++) {
        	if (fixref[i]!=null) {
        		int address = ((Integer)labels.get(fixref[i])).intValue();
        		mc[i] = byteLow(address);
        		mc[i+1] = byteHigh(address);
        	}
        }
        
        //6. save the bizatch
        try {
        File f = new File(args[1]);
        f.delete();
        f = new File(args[1]);
        RandomAccessFile raf = new RandomAccessFile(f, "rw");
        raf.seek(f.length());
        for (int i=0; i<END_OF_MEMORY; i++)
        	raf.writeByte(mc[i]);
        raf.close();
	    } catch (IOException e) {
    	}

        
    }
    
    static byte byteLow(int addr)
    {
    	double temp = addr/256.0;
    	temp = temp - ((int)addr/256);
    	temp = temp * 256.0;
    	return ((byte)((int)temp));
  	}
  	
  	static byte byteHigh(int addr)
  	{
  		return ((byte)(addr/256));
  	}
    
    static boolean isNumber(String n) {
    try  {
      double d = Double.valueOf(n).doubleValue();
      return true;
      }
    catch (NumberFormatException e) {
      //e.printStackTrace();
      return false;
      }
    }
    
    static void writeByte(int data)
    {
    	mc[loc] = (byte)data;
    	loc++;
    }
}