python

code/alu.py

@@ -1,36 +1,38 @@
 import binandint
-def alu(a,b,alu_ctrl):              #a,b为两个运算数，alu_ctrl为控制信号，返回运算结果result与是否为0 zero
-    if alu_ctrl=='00000':
+
+
+def alu(a, b, alu_ctrl):  # a,b为两个运算数，alu_ctrl为控制信号，返回运算结果result与是否为0 zero
+    if alu_ctrl == '00000':
         pass
-    elif alu_ctrl=='00001':
+    elif alu_ctrl == '00001':
         pass
-    elif alu_ctrl=='00010':
+    elif alu_ctrl == '00010':
         pass
-    elif alu_ctrl=='00011':
+    elif alu_ctrl == '00011':
         pass
-    elif alu_ctrl=='00100':
+    elif alu_ctrl == '00100':
         pass
-    elif alu_ctrl=='00101':
+    elif alu_ctrl == '00101':
         pass
-    elif alu_ctrl=='00110':
+    elif alu_ctrl == '00110':
         pass
-    elif alu_ctrl=='00111':
+    elif alu_ctrl == '00111':
         pass
-    elif alu_ctrl=='01000':
+    elif alu_ctrl == '01000':
         pass
-    elif alu_ctrl=='01001':
+    elif alu_ctrl == '01001':
         pass
-    elif alu_ctrl=='01010':
+    elif alu_ctrl == '01010':
         pass
-    elif alu_ctrl=='01011':
+    elif alu_ctrl == '01011':
         pass
-    elif alu_ctrl=='01100':
+    elif alu_ctrl == '01100':
         pass
-    elif alu_ctrl=='01101':
+    elif alu_ctrl == '01101':
         pass
-    elif alu_ctrl=='01110':
+    elif alu_ctrl == '01110':
         pass
-    elif alu_ctrl=='01111':
+    elif alu_ctrl == '01111':
         pass
     else:
         pass

code/alu_controller.py

@@ -1,48 +1,49 @@
-def alu_controller(alu_op,funct7,funct3):               #alu_op来自主控制器，funct7和funct3来自指令的特定部位   返回alu_ctrl
-    if alu_op=='00':
-        opreation='00000'        #lw.sw.auipc
-    elif alu_op=='01':
-        if funct3=='000':
-            opreation='00101'    #beq
-        elif funct3=='001':
-            opreation='00110'    #bne
-        elif funct3=='100':
-            opreation='00111'    #blt
-        elif funct3=='101':
-            opreation='01000'    #bge
-        elif funct3=='110':
-            opreation='01001'    #bltu
-        elif funct3=='111':
-            opreation='01010'    #bgeu
+def alu_controller(alu_op, funct7,
+                   funct3):  # alu_op来自主控制器，funct7和funct3来自指令的特定部位   返回alu_ctrl
+    if alu_op == '00':
+        opreation = '00000'  # lw.sw.auipc
+    elif alu_op == '01':
+        if funct3 == '000':
+            opreation = '00101'  # beq
+        elif funct3 == '001':
+            opreation = '00110'  # bne
+        elif funct3 == '100':
+            opreation = '00111'  # blt
+        elif funct3 == '101':
+            opreation = '01000'  # bge
+        elif funct3 == '110':
+            opreation = '01001'  # bltu
+        elif funct3 == '111':
+            opreation = '01010'  # bgeu
         else:
-            opreation='00000'
-    elif alu_op=='10':
-        if funct3=='000':
-            if funct7=='0100000':   #sub
-                opreation='00001'
-            else:                   #add,addi
-                opreation='00000'
-        elif funct3=='100':
-            opreation='01100'        #xor,xori
-        elif funct3=='110':
-            opreation='00010'        #or,ori
-        elif funct3=='111':
-            opreation='00011'        #and, andi
-        elif funct3=='010':
-            opreation='00100'        #slt,slti
-        elif funct3=='001':
-            opreation='01101'        #sll,slli   未存在
-        elif funct3=='011':
-            opreation='01110'        #sliu,sltiu 未存在
-        elif funct3=='101':
-            if funct7=='0100000':   #sra,srai
-                opreation='01111'
-            else:                   #srl,srli
-                opreation='10000'
+            opreation = '00000'
+    elif alu_op == '10':
+        if funct3 == '000':
+            if funct7 == '0100000':  # sub
+                opreation = '00001'
+            else:  # add,addi
+                opreation = '00000'
+        elif funct3 == '100':
+            opreation = '01100'  # xor,xori
+        elif funct3 == '110':
+            opreation = '00010'  # or,ori
+        elif funct3 == '111':
+            opreation = '00011'  # and, andi
+        elif funct3 == '010':
+            opreation = '00100'  # slt,slti
+        elif funct3 == '001':
+            opreation = '01101'  # sll,slli   未存在
+        elif funct3 == '011':
+            opreation = '01110'  # sliu,sltiu 未存在
+        elif funct3 == '101':
+            if funct7 == '0100000':  # sra,srai
+                opreation = '01111'
+            else:  # srl,srli
+                opreation = '10000'
         else:
-            opreation='00000'
-    elif alu_op=='11':              #jal
-        opreation='01011'
+            opreation = '00000'
+    elif alu_op == '11':  # jal
+        opreation = '01011'
     else:
-        opreation='00000'
+        opreation = '00000'
     return opreation

code/binandint.py

@@ -16,4 +16,4 @@ def UInttoBin(x, n):
 def InttoBin(x, n):
     if x < 0:
         x += 2**n
-    return UInttoBin(x, n)
+    return UInttoBin(x, n)

code/branch_unit.py

@@ -1,9 +1,10 @@
-def branch_unit(cur_pc,imm,jalr_sel,branch_taken,alu_result):           #输入值为当前pc（int），立即数，jalr信号，是否跳转信号，alu运算结果(全是int型)
-    pc_plus_4=cur_pc+4                                                  #输出为pc_plus_imm,pc_plus_4,branch_target,pc_sel(忘了这是啥了，需要回头再看)
-    pc_plus_imm=cur_pc+imm
-    pc_sel = jalr_sel | (branch_taken & (alu_result%2))
-    if jalr_sel==1:
-        branch_target=alu_result&(2**32-2)
+def branch_unit(cur_pc, imm, jalr_sel, branch_taken,
+                alu_result):  # 输入值为当前pc（int），立即数，jalr信号，是否跳转信号，alu运算结果(全是int型)
+    pc_plus_4 = cur_pc + 4  # 输出为pc_plus_imm,pc_plus_4,branch_target,pc_sel(忘了这是啥了，需要回头再看)
+    pc_plus_imm = cur_pc + imm
+    pc_sel = jalr_sel | (branch_taken & (alu_result % 2))
+    if jalr_sel == 1:
+        branch_target = alu_result & (2**32 - 2)
     else:
-        branch_target=cur_pc+imm*2
-    return pc_plus_imm,pc_plus_4,branch_target,pc_sel
+        branch_target = cur_pc + imm * 2
+    return pc_plus_imm, pc_plus_4, branch_target, pc_sel

code/data_mem.py

@@ -1,26 +1,28 @@
 class data_mem:
-    memory=[8*'0']*4*1024*16
-    def mem(self,write_en,read_en,address,data_in,funct3):      #写使能，读使能，(使能为int的0或1)地址，输入数据，func3              #输出为data_out
+    memory = [8 * '0'] * 4 * 1024 * 16
+
+    def mem(self, write_en, read_en, address, data_in, funct3
+            ):  # 写使能，读使能，(使能为int的0或1)地址，输入数据，func3              # 输出为data_out
         if write_en:
-            if funct3=='000':
+            if funct3 == '000':
                 pass
-            elif funct3=='001':
+            elif funct3 == '001':
                 pass
-            elif funct3=='010':
+            elif funct3 == '010':
                 pass
             else:
                 pass
 
         if read_en:
-            if funct3=='000':
+            if funct3 == '000':
                 pass
-            elif funct3=='001':
+            elif funct3 == '001':
                 pass
-            elif funct3=='010':
+            elif funct3 == '010':
                 pass
-            elif funct3=='100':
+            elif funct3 == '100':
                 pass
-            elif funct3=='101':
+            elif funct3 == '101':
                 pass
             else:
                 pass

code/forwarding_unit.py

@@ -1,14 +1,15 @@
-def fowardingunit(rs1,rs2,ex_mem_rd,mem_wb_rd,ex_mem_regwrite,mem_wb_regwrite):
-    if rs1!=0 and rs1==ex_mem_rd and ex_mem_regwrite:
-        forward_a='01'
-    elif rs1!=0 and rs1==mem_wb_rd and mem_wb_regwrite:
-        forward_a='10'
+def fowardingunit(rs1, rs2, ex_mem_rd, mem_wb_rd, ex_mem_regwrite,
+                  mem_wb_regwrite):
+    if rs1 != 0 and rs1 == ex_mem_rd and ex_mem_regwrite:
+        forward_a = '01'
+    elif rs1 != 0 and rs1 == mem_wb_rd and mem_wb_regwrite:
+        forward_a = '10'
     else:
-        forward_a='00'
-    if rs2!=0 and rs2==ex_mem_rd and ex_mem_regwrite:
-        forward_b='01'
-    elif rs2!=0 and rs2==mem_wb_rd and mem_wb_regwrite:
-        forward_b='10'
+        forward_a = '00'
+    if rs2 != 0 and rs2 == ex_mem_rd and ex_mem_regwrite:
+        forward_b = '01'
+    elif rs2 != 0 and rs2 == mem_wb_rd and mem_wb_regwrite:
+        forward_b = '10'
     else:
-        forward_b='00'
-    return forward_a,forward_b
+        forward_b = '00'
+    return forward_a, forward_b

code/hazard_detector.py

@@ -1,6 +1,6 @@
-def hazard_detector(if_id_rs1,if_id_rs2,id_ex_rd,id_ex_memread):
-    if id_ex_memread and (id_ex_rd==if_id_rs1 or id_ex_rd==if_id_rs2):
-        stall=1
+def hazard_detector(if_id_rs1, if_id_rs2, id_ex_rd, id_ex_memread):
+    if id_ex_memread and (id_ex_rd == if_id_rs1 or id_ex_rd == if_id_rs2):
+        stall = 1
     else:
-        stall=0
-    return stall
+        stall = 0
+    return stall