memory_unit.py

# The main memory
class MainMemory:
    def __init__(self):
        self.list = ["0"] * 1024
    
    def print_memory(self, range_n):
        print(f"************Memory {range_n} from start************")
        for i in range(range_n):
            print(f"Memory[{i}] = {self.list[i]}")
    
    # address and value are integers
    def write_memory(self, address, value):
        self.list[int(address)] = str(value)

    # address is integer
    def read_memory(self, address):
        return self.list[int(address)]
    

# A single Register
class Register:
    def __init__(self, num):
        self.num = num
        self.data = "0"
        self.busy = False
        self.tag = 0
    

# A set of registers
class Registers:
    def __init__(self):
        self.list = [Register(i) for i in range(16)]

    def print_registers(self):
        print("---------------Registers----------------------")
        for i in self.list:
            print("Register",i.num,i.data)
        print("______________________________________________")

    def write_register(self, idx, value):
        self.list[idx] = str(value)

    # address is integer
    def read_register(self, idx):
        return self.list[idx]