getType.py

# This function looks at the opCode and returns the type of the function.

from printColor import printColor
from writeToFile import writeToFile


def getType(inst):
    rType = [
        "AND",
        "SUBU",
        "MFLO",
        "MFHI",
        "SLTU",
        "ADDU",
        "XOR",
        "SUB",
        "ADD",
        "NOR",
        "SRLV",
        "SRAV",
        "DIVU",
        "MULT",
        "SRA",
        "SLT",
        "SLLV",
        "SRL",
        "SLL",
        "MTLO",
        "MTHI",
        "DIV",
        "OR",
        "MULTU",
        "MOVE",
        "NEGU",
        "NOT",
        "MOVN",
        "MOVZ",
    ]
    iType = [
        "ADDI",
        "LW",
        "SW",
        "SUBI",
        "SLTI",
        "BLEZ",
        "MULTI",
        "BNE",
        "BEQ",
        "BGTZ",
        "BLT",
        "LI",
        "ORI",
        "ANDI",
        "XORI",
    ]

    opcode = {
        # R - TYPE
        # ALL THE R TYPE HAVE 000000 OPCODE.
        # I - TYPE
        "ADDI": "001000",
        "SLTI": "001010",
        "LW": "100011",
        "SW": "101011",
        "BEQ": "000100",
        "BGTZ": "000111",
        "BNE": "000101",
        "BLEZ": "000110",
        "BLT": "100011",
        "BLTZ": "000001",
        "LI": "001001",
        "ORI": "001101",
        "ANDI": "001100",
        "XORI": "001110",
        "SUBI": "001000",
        "CLO": "",
        # J - TYPE
        "J": "000010",
    }

    if inst.upper() in rType:
        printColor("000000", "yellow")
        writeToFile("000000")
        return "R"

    if inst.upper() in iType:
        printColor(opcode[inst.upper()], "yellow")
        writeToFile(opcode[inst.upper()])
        return "I"

    if inst.upper() == "J":
        printColor(opcode["J"], "yellow")
        writeToFile(opcode["J"])
        return "J"

    raise Exception("Invalid OpCode : {0}".format(inst))