|
|
from binandint import *
|
|
|
|
|
|
|
|
|
class data_mem:
|
|
|
memory = [8 * '0'] * 4 * 1024 * 16
|
|
|
|
|
|
def __init__(self) -> None:
|
|
|
pass
|
|
|
|
|
|
def mem(self, write_en, read_en, address, data_in, funct3
|
|
|
): # 写使能，读使能，(使能为int的0或1)地址，输入数据，func3 #输出为data_out
|
|
|
if write_en:
|
|
|
if funct3 == '000': # sb
|
|
|
self.memory[address] = InttoBin(data_in % 2**8, 8)
|
|
|
elif funct3 == '001': # sh
|
|
|
temp = InttoBin(data_in % 2**16, 16)
|
|
|
self.memory[address + 1] = temp[0:8]
|
|
|
self.memory[address] = temp[8:16]
|
|
|
elif funct3 == '010': # sw
|
|
|
temp = InttoBin(data_in, 32)
|
|
|
self.memory[address + 3] = temp[0:8]
|
|
|
self.memory[address + 2] = temp[8:16]
|
|
|
self.memory[address + 1] = temp[16:24]
|
|
|
self.memory[address] = temp[24:32]
|
|
|
else:
|
|
|
pass
|
|
|
|
|
|
if read_en:
|
|
|
if funct3 == '000': # lb
|
|
|
data_out = BintoInt(self.memory[address])
|
|
|
elif funct3 == '001': # lh
|
|
|
data_out = BintoInt(self.memory[address + 1] +
|
|
|
self.memory[address])
|
|
|
elif funct3 == '010': # lw
|
|
|
data_out = BintoInt(self.memory[address + 3] +
|
|
|
self.memory[address + 2] +
|
|
|
self.memory[address + 1] +
|
|
|
self.memory[address])
|
|
|
elif funct3 == '100': # lbu
|
|
|
data_out = BintoUInt(self.memory[address + 1] +
|
|
|
self.memory[address])
|
|
|
elif funct3 == '101': # lhu
|
|
|
data_out = BintoUInt(self.memory[address + 3] +
|
|
|
self.memory[address + 2] +
|
|
|
self.memory[address + 1] +
|
|
|
self.memory[address])
|
|
|
else:
|
|
|
data_out = '0' * 32
|
|
|
return data_out
|
