X-Git-Url: https://git.libre-soc.org/?p=rv32.git;a=blobdiff_plain;f=cpu_fetch_stage.py;h=e3a3467284fffa3882a06ec4c6fb2e4dfb6fad3e;hp=f3269fb75a9fe890c5e11c9be55c2f7bdcce3e39;hb=HEAD;hpb=2ab5b48210e064bc87131bd6e1451ff89c064931

diff --git a/cpu_fetch_stage.py b/cpu_fetch_stage.py
index f3269fb..e3a3467 100644
--- a/cpu_fetch_stage.py
+++ b/cpu_fetch_stage.py
@@ -34,6 +34,8 @@ class CPUFetchStage(Module):
     def __init__(self):
         self.clk = ClockSignal()
         self.reset = ResetSignal()
+        self.reset_vector = Signal(32) #32'hXXXXXXXX; - parameter
+        self.mtvec = Signal(32) # 32'hXXXXXXXX; - parameter
         #output [31:2] memory_interface_fetch_address,
         self.memory_interface_fetch_address = Signal(32)
         #input [31:0] memory_interface_fetch_data,
@@ -41,27 +43,25 @@ class CPUFetchStage(Module):
         self.memory_interface_fetch_valid = Signal()
         self.fetch_action = Signal(fetch_action)
         self.target_pc = Signal(32)
-        self.output_pc = Signal(32, reset=reset_vector)
+        self.output_pc = Signal(32, reset=self.reset_vector)
         self.output_instruction = Signal(32)
         self.output_state = Signal(fetch_output_state,
-                                   reset=fetch_output_state_empty)
-        self.reset_vector = Signal(32) #32'hXXXXXXXX; - parameter
-        self.mtvec = Signal(32) # 32'hXXXXXXXX; - parameter
+                                   reset=FOS.empty)
 
         #self.comb += [
         #    self.cd_sys.clk.eq(self.clk),
         #    self.cd_sys.rst.eq(self.reset)
         #]
 
-        fetch_pc = Signal(32, reset=reset_vector)
+        fetch_pc = Signal(32, reset=self.reset_vector)
 
-        self.sync += If(self.fetch_action != fetch_action_wait,
+        self.sync += If(self.fetch_action != FA.wait,
                         self.output_pc.eq(fetch_pc))
 
         self.comb += self.memory_interface_fetch_address.eq(fetch_pc[2:])
 
-        #initial output_pc <= reset_vector;
-        #initial output_state <= `fetch_output_state_empty;
+        #initial output_pc <= self.reset_vector;
+        #initial output_state <= `FOS.empty;
 
         delayed_instruction = Signal(32, reset=0)
         delayed_instruction_valid = Signal(reset=0)
@@ -75,38 +75,38 @@ class CPUFetchStage(Module):
                 )
 
         self.sync += delayed_instruction_valid.eq(self.fetch_action ==
-                                                  fetch_action_wait)
+                                                  FA.wait)
 
         fc = {
-            fetch_action_ack_trap:
+            FA.ack_trap:
                 If(self.memory_interface_fetch_valid,
                    [fetch_pc.eq(fetch_pc + 4),
-                    self.output_state.eq(fetch_output_state_valid)]
+                    self.output_state.eq(FOS.valid)]
                 ).Else(
-                   [fetch_pc.eq(mtvec),
-                    self.output_state.eq(fetch_output_state_trap)]
+                   [fetch_pc.eq(self.mtvec),
+                    self.output_state.eq(FOS.trap)]
                 ),
-            fetch_action_fence:
+            FA.fence:
                 [ fetch_pc.eq(self.output_pc + 4),
-                  self.output_state.eq(fetch_output_state_empty)
+                  self.output_state.eq(FOS.empty)
                 ],
-            fetch_action_jump:
+            FA.jump:
                 [ fetch_pc.eq(self.target_pc),
-                  self.output_state.eq(fetch_output_state_empty)
+                  self.output_state.eq(FOS.empty)
                 ],
-            fetch_action_error_trap:
-                   [fetch_pc.eq(mtvec),
-                    self.output_state.eq(fetch_output_state_empty)
+            FA.error_trap:
+                   [fetch_pc.eq(self.mtvec),
+                    self.output_state.eq(FOS.empty)
                 ],
-            fetch_action_wait:
+            FA.wait:
                    [fetch_pc.eq(fetch_pc),
-                    self.output_state.eq(fetch_output_state_valid)
+                    self.output_state.eq(FOS.valid)
                 ]
         }
-        fc[fetch_action_default] = fc[fetch_action_ack_trap]
-        fc[fetch_action_noerror_trap] = fc[fetch_action_error_trap]
+        fc[FA.default] = fc[FA.ack_trap]
+        fc[FA.noerror_trap] = fc[FA.error_trap]
         self.sync += Case(self.fetch_action,
-                          fc).makedefault(fetch_action_default)
+                          fc).makedefault(FA.default)
 
 if __name__ == "__main__":
     example = CPUFetchStage()
@@ -121,7 +121,7 @@ if __name__ == "__main__":
            example.target_pc,
            example.output_pc,
            example.output_instruction,
-           example.output_state
+           example.output_state,
            example.reset_vector,
            example.mtvec
             }))