141LProcessor/assembler1.py at main · 21KennethTran/141LProcessor · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
import sys

opcode = {"lwr":"0000", "str":"0001", "mov":"0010", "mof":"0011", "and":"0100", "or":"0101", "xor":"0110", "add":"0111",
          "sub":"1000", "lsl":"1001", "rsl":"1010", "cmp":"1011", "beq":"1100", "bne":"1101", "j":"1110", "don":"1111"}

operand = {"$t0":"0000", "$t1":"0001", "$t2":"0010", "$t3":"0011", "$t4":"0100", "$t5":"0101", "$t6":"0110", "$t7":"0111",
          "$t8":"1000", "$t9":"1001", "$t10":"1010", "$t11":"1011", "$t12":"1100", "$t13":"1101", "$t14":"1110", "$t15":"1111"}

Lookup = {"$t0":"0000", "$t1":"0001", "$t2":"0010", "$t3":"0011", "$t4":"0100", "$t5":"0101", "$t6":"0110", "$t7":"0111",
          "$t8":"1000", "$t9":"1001", "$t10":"1010", "$t11":"1011", "$t12":"1100", "$t13":"1101", "$t14":"1110", "$t15":"1111"}

def assembleLine(part):
    print(part)
    final = ""

    opc = part[0].lower()

    if opc.endswith('*'):
        opc = opc.strip('*')
        if opc not in opcode:
            return None
        final += opcode[opc] + "1"
    else:
        if opc not in opcode:
            return None
        final += opcode[opc] + "0"

    ope = part[1]
    if ope in operand:
        final += operand[ope]
    else:
        final += ope # deal with jump mapping later

    return final

def main():
    if len(sys.argv) != 2 or sys.argv[1] not in ['1', '2', '3']:
        print("Usage: python script.py [1|2|3]")
        sys.exit(1)

    input_file = f"code{sys.argv[1]}.txt"
    output_file = f"machcode.txt"

    with open(input_file, "r") as file:
        lines = file.readlines()
        full = []
        for line in lines:
            line = line.split('#')[0].strip()
            if line:
                part = line.split()
                instr = assembleLine(part)
                if instr is not None:
                    full.append(instr)

    with open(output_file, "w") as out:
        for instruction in full:
            out.write(instruction + '\n')

if __name__ == "__main__":
    main()