Pass1 Assembler Implementation

Design suitable data structures and implement pass-I of a two-pass assembler for a pseudo machine in Java/C++ using object oriented features. Implementation should consist of a few instructions from each category and few assembler directives.

 
import java.util.*;

public class pass1 {
    
    // MOT (Machine Opcode Table) for IS, AD, DL
    static Map IS = new HashMap();
    static Map AD = new HashMap();
    static Map DL = new HashMap();
    static Map REG = new HashMap();

    // SYMTAB and LITTAB
    static List symtab = new ArrayList();
    static Map symtabAddr = new HashMap();

    static List littab = new ArrayList();
    static Map littabAddr = new HashMap();

    // IC
    static List intermediate = new ArrayList();

    public static void main(String[] args) {

        // Initialize MOT
        IS.put("ADD", "01");
        IS.put("SUB", "02");
        IS.put("MULT", "03");
        IS.put("MOVER", "04");
        IS.put("MOVEM", "05");
        IS.put("COMP", "06");
        IS.put("BC", "07");
        IS.put("DIV", "08");
        IS.put("READ", "09");
        IS.put("PRINT", "10");

        AD.put("START", "01");
        AD.put("END", "02");

        DL.put("DS", "01");
        DL.put("DC", "02");

        REG.put("AREG", "1");
        REG.put("BREG", "2");
        REG.put("CREG", "3");
        REG.put("DREG", "4");

        // Example input program
        String program[] = {
            "START 200",
            "MOVER AREG, ONE",
            "ADD BREG, TWO",
            "ADD BREG, ='5'",
            "MOVEM CREG, RESULT",
            "ADD BREG, ='3'",
            "ONE DS 4",
            "TWO DC 2",
            "RESULT DS 1",
            "END"
        };

        pass1(program);

        // Print tables
        System.out.println("\nSYMTAB:");
        for (int i = 0; i < symtab.size(); i++) {
            String sym = symtab.get(i);
            System.out.println("" + (i) + "  " + sym + "  " + symtabAddr.get(sym));
        }

        System.out.println("\nLITTAB:");
        for (int i = 0; i < littab.size(); i++) {
            String lit = littab.get(i);
            System.out.println("" + (i) + "  " + lit + "  " + littabAddr.get(lit));
        }

        System.out.println("\nIntermediate Code:");
        for (String line : intermediate) {
            System.out.println(line);
        }
    }

    static void pass1(String[] program) {
    	String value=null;
        int lc = 0; // location counter

        for (String line : program) {
            String parts[] = line.split("[ ,]+"); // split by space or comma

            if (AD.containsKey(parts[0])) {
                // START / END
                if (parts[0].equals("START")) {
                    lc = Integer.parseInt(parts[1]);
                    intermediate.add("AD " + AD.get("START") + " C " + lc);
                } else if (parts[0].equals("END")) {
                    intermediate.add("AD " + AD.get("END"));

                    // Assign addresses to literals
                    for (int i = 0; i < littab.size(); i++) {
                        String lit = littab.get(i);
                        littabAddr.put(lit, lc);
                        value = lit.substring(2, lit.length() - 1);
                        intermediate.add("DL " + DL.get("DC") + " C " + value);

                        lc++;
                    }

                }
            }
            else if (IS.containsKey(parts[0])) {
                // Instruction without label
                String ic = "IS " + IS.get(parts[0]);
                if (parts.length > 1 && REG.containsKey(parts[1])) {
                    ic += " RG " + REG.get(parts[1]);
                }
                if (parts.length > 2) {
                    if (parts[2].startsWith("=")) {
                        // literal
                        if (!littab.contains(parts[2])) littab.add(parts[2]);
                        int idx = littab.indexOf(parts[2]);
                        ic += " L " + idx;
                    } else {
                        // symbol
                        if (!symtab.contains(parts[2])) symtab.add(parts[2]);
                        int idx = symtab.indexOf(parts[2]);
                        ic += " S " + idx;
                    }
                }
                intermediate.add(ic);
                lc++;
            }
            else {
                // Label + DL (DS/DC)
                String label = parts[0];
                if (!symtab.contains(label))
                	symtab.add(label);
                int idx = symtab.indexOf(label);
                symtabAddr.put(label, lc);

                if (DL.containsKey(parts[1])) {
                    if (parts[1].equals("DS")) {
                        intermediate.add("DL " + DL.get("DS") + " C " + parts[2]);
                        lc += Integer.parseInt(parts[2]);
                    } else if (parts[1].equals("DC")) {
                        intermediate.add("DL " + DL.get("DC") + " C " + parts[2]);
                        lc++;
                    }
                }
            }
        }
    }
}

Output:

    SYMTAB:
0  ONE  205
1  TWO  209
2  RESULT  210

LITTAB:
0  ='5'  211
1  ='3'  212

Intermediate Code:
AD 01 C 200
IS 04 RG 1 S 0
IS 01 RG 2 S 1
IS 01 RG 2 L 0
IS 05 RG 3 S 2
IS 01 RG 2 L 1
DL 01 C 4
DL 02 C 2
DL 01 C 1
AD 02
DL 02 C 5
DL 02 C 3