asm_copy2.py

import sys
PC_var = 0
PC_label = 0
type_a = {"add": "10000", "sub": "10001", "mul": "10110", "xor": "11010", "or": "11011", "and": "11100"}
type_b = {"ls":"11001", "rs":"11000", "mov":"10010"} #mov r1 imm
type_c = {"div": "10111", "not": "11101", "cmp": "11110", "mov": "10011"} # mov r1 r2
type_d = {"st": "10101", "ld": "10100"}
type_e = {"jmp": "11111", "jlt": "01100", "je": "01111", "jgt": "01101"}
reg = {"r0": "000", "r1": "001", "r2": "010", "r3": "011", "r4": "100", "r5": "101", "r6": "110", "r7": "111"}
opcode = {"add": "10000", "sub": "10001", "mul": "10110", "xor": "11010", "or": "11011", "and": "11100", "ls":"11001", "rs":"11000", "mov":"10010","div": "10111", "not": "11101", "cmp": "11110", "mov": "10011","st": "10100", "ld": "10101","jmp": "11111", "jlt": "01100", "je": "01111", "jgt": "01101"}
lab_dic = {} 
vars = {}
vars_line = {}
machine_code = []
vars_count = 0 
flag = True
flag_a = True
flag_b=True
flag_c=True
flag_d = True
flag_e = True
flag_m=True

def check_bin(check_str):   
    flag = True
    for i in range(len(check_str)):
        if check_str[i] not in "01":
            flag = False
            break
    return flag


def convert(PC_var):
    b = bin(PC_var)
    mem = str(b).replace("0b", "")
    mem = format(int(mem), '#08')
    return str(mem)


def parse_vars(lst, vars_dict, PC_var, vars_dict2):
    for i in lst:
        if "var" in i.lower():
            vars_dict[i[4:]] = convert(PC_var)
            PC_var += 1
    for i in range(len(lst)):
        if "var" in lst[i].lower():
            vars_dict2[lst[i][4:]] = i + 1

def DecBin(string):
    if string.isdigit():
        lst=[int(i) for i in string]
        for i in lst:
            if i>=0 and i<=9:
                Flag=True
            else:
                Flag=False
                break
    else:
        Flag=False
  
    if Flag:   
        num=int(string)
        Bnumlst=[]
        q=1
        while q!=0:
            q=num//2
            Bnum=num%2
            Bnumlst.append(Bnum)
            num=q
        zeroes=8-len(Bnumlst)
        for i in range(zeroes):
            Bnumlst.append(0)
        Bnumlst.reverse()
        Anslst=[str(i) for i in Bnumlst]
        Binary_Number="".join(Anslst)
        return(Binary_Number)

    # if not(Flag):
    #     return("Enter the correct number")

    
# def BinDec(string):
    #Checking if input number is binary.
    # if string.isdigit():
    #     lst=[int(i) for i in string]
    #     for i in lst:
    #         if i>=0 and i<=1:
    #             Flag=True
    #         else:
    #             Flag=False
    #             break
    # else:
    #     Flag=False


    #Converting binary number to decimal number.
    # if Flag:
    #     lst=[int(i) for i in string]
    #     DecNum=0
    #     highestPower=len(lst)
    #     for i in range(len(lst)):
    #         DecNum+=(lst[i]*(2**(highestPower-1)))
    #         highestPower-=1
    #     return(DecNum)

    # if not(Flag):
    #     return("Enter the correct number")

s = sys.stdin.read()
line = s.split("\n")
line_copy = line
inst_lst = []
for i in line:
    ele = i.split()
    inst_lst.append(ele)
#inst_lst=inst_lst[:256]
for i in line:
    if "var" in i.lower():
        vars_count += 1

c = line.count("")
for i in range(c):
    line.remove("")
inst_lst2 = []
for i in line:
    ele = i.split()
    inst_lst2.append(ele)

PC_var = len(inst_lst2) - vars_count

parse_vars(line_copy, vars, PC_var, vars_line)
    
try:
    for i in range(len(inst_lst2)):
        if "var" in inst_lst2[i]:
            continue
        elif len(inst_lst2[i]) == 0:
            continue
        else:
            if inst_lst2[i][0][-1] == ":" and len(inst_lst2[i]) > 1 and inst_lst2[i][1] in opcode:
                lab_dic[inst_lst2[i][0][:-1].lower()] = DecBin(str(PC_label))
            PC_label += 1

    if flag == True:
        for i in range(len(inst_lst2)):
            
            if len(inst_lst2[i]) == 0:
                continue
            if inst_lst2[i][0][-1]==":":
                label=inst_lst2[i][0]
                inst_lst2.pop(0)
                flag_l=True
            if inst_lst2[i][0].lower() in type_a:
                flag_a = True
                if len(inst_lst2[i]) == 4:
                    if inst_lst2[i][1].lower() in reg and inst_lst2[i][2].lower() in reg and inst_lst2[i][3].lower() in reg:
                        op = type_a[inst_lst2[i][0]]
                        r1 = reg[inst_lst2[i][1].lower()]
                        r2 = reg[inst_lst2[i][2].lower()]
                        r3 = reg[inst_lst2[i][3].lower()]
                        machine_code.append(op+"00"+r1+r2+r3)
                    else:
                        print(f'Error in line {i+1}: Undefined Register name')
                        flag_a = False
                        break
                else:
                    print(f'Error in line {i+1}: Wrong Instruction syntax for {inst_lst[i][0].lower()}')
                    flag_a = False
                    break
                       
                        
            elif inst_lst2[i][0].lower() in type_b and inst_lst2[i][2][0]=="$":
                flag_b = True
                if len(inst_lst2[i])!=3:
                    flag_b=False
                    print(f'Error in line {i+1}: Number of operands exceed requirement')
                if int(inst_lst2[i][1][1])<0 or int(inst_lst2[i][1][1])>6:
                    flag_b=False
                    print(f'Error in line {i+1}: Undefined Register name')
                if inst_lst2[i][2].lower()=="flag":
                    flag_b=False
                    print(f'Error in line {i+1}: Illegal use of flags register')
                Imm=inst_lst2[i][2][1:]
                if int(Imm)<0 or int(Imm)>255:
                    flag_b=False
                    print(f'Error in line {i+1}: Illegal immediate values')
                if flag_b:
                    op=type_b[inst_lst2[i][0].lower()]
                    r1=reg[(inst_lst2[i][1]).lower()]
                    Im=DecBin(Imm)
                    machine_code.append(op+r1+Im)
                
            elif inst_lst[i][0] in type_c:
                flag_c = True
                if len(inst_lst2[i]) != 3:
                    flag_c=False
                    print(f'Error in line {i+1}: Number of operand exceed requirement')
                if not(inst_lst2[i][1][1].isnumeric()) or int(inst_lst2[i][1][1])<0 or int(inst_lst2[i][1][1])>6:
                    flag_c=False
                    print(f'Error in line {i+1}: Undefined Register name')
                if inst_lst2[i][2].lower()=="flag":
                    if inst_lst2[i][0]!="mov":
                        flag_c=False
                        print(f'Error in line {i+1}: Illegal use of flag register')
                if inst_lst2[i][1].lower()=="flag":
                    flag_c=False
                    print(f'Error in line {i+1}: Illegal use of flag register')
                if flag_c:
                    op=type_c[inst_lst2[i][0]]
                    r1=reg[(inst_lst2[i][1]).lower()]
                    if inst_lst2[i][2].lower()=="flag":
                        r2="111"
                    else:
                        r2=reg[(inst_lst2[i][2]).lower()]
                    machine_code.append(op+"00000"+r1+r2)  
                    
            elif inst_lst2[i][0].lower() in type_d:  
                flag_d = True
                if len(inst_lst2[i]) == 3:
                    if inst_lst2[i][1].lower() in reg:
                        if inst_lst2[i][2] in vars:
                            op = type_d[inst_lst2[i][0]]
                            r1 = reg[inst_lst2[i][1].lower()]
                            mem = vars[inst_lst2[i][2].lower()]
                            machine_code.append(op + r1 + mem)
                        elif inst_lst2[i][2] in lab_dic:
                            print(f'Error in line {i+1}: Use of labels as variables')
                            flag_d = False
                            break
                        else:
                            print(f'Error in line {i+1}: Use of undefined variable')
                            flag_d = False
                            break
                    else:
                        print(f'Error in line {i+1}: Undefined Register name')
                        flag_d = False
                        break
                else:
                    print(f'Error in line {i+1}: Wrong Instruction syntax for {inst_lst[i][0].lower()}')
                    flag_d = False
                    break
            if flag_l:
                inst_lst2.insert(0,label)
            elif inst_lst2[i][0].lower() in type_e:
                flag_e = True
                if len(inst_lst2[i]) == 2:
                    if inst_lst2[i][1].lower() in lab_dic:
                        op = type_e[inst_lst2[i][0]]
                        mem = lab_dic[inst_lst2[i][1].lower()]
                        machine_code.append(op+"000"+mem)
                    elif inst_lst2[i][1].lower() in vars:
                        print(f'Error in line {i+1}: Use of variables as labels')
                        flag_d = False
                        break
                    else:
                        print(f'Error in line {i+1}: Memory address is not a label')
                        flag_e = False
                        break
                else:
                    print(f'Error in line {i+1}: Wrong Instruction syntax for {inst_lst2[i][0].lower()}')
                    flag_e = False
                    break
            
            elif inst_lst2[i][0][-1]=":" and inst_lst2[i][1].lower() in opcode:
                if inst_lst2[i][1].lower in type_a:
                    

            elif inst_lst2[i][0].lower() == "hlt":
                machine_code.append("0101000000000000")

            elif inst_lst[i][0].lower() not in opcode or inst_lst[i][0].lower() != "var":
                print(f'Error in line {i+1}: Not an instruction syntax')
                flag = False
                break

    if flag and (len(inst_lst2[-1]) != 1 or inst_lst2[-1][0]) != "hlt":
        flag = False
        print(f'Error in line {len(inst_lst2)}: Invalid/Absent hlt declaration') 
        
    for i in range(vars_count):
        if inst_lst2[0][0].lower() == "var":
            inst_lst2.remove(inst_lst2[0])

    for i in range(len(inst_lst2)):
        if flag and inst_lst2[i][0] == "var":
            flag = False
            print(f'Error in line {vars_line[inst_lst2[i][1]]}: Variable not declared at the begining')
         
    if len(machine_code)>256:
        flag_m=False
        print(f'Number of instructions exceed limit')
    if flag and flag_a and flag_b and flag_c and flag_d and flag_e:
        for i in machine_code:
            sys.stdout.write(i)
            sys.stdout.write("\n")

except:
    print("General Syntax Error")