convert from public static functions/properties for regspecs

[soc.git] / src / soc / fu / trap / pipe_data.py

diff --git a/src/soc/fu/trap/pipe_data.py b/src/soc/fu/trap/pipe_data.py
index 581040c42c0828a41269738a25a3bbd2758f0a77..b9c829bccc1811a1e7e334aba22fc3b09e7a907d 100644 (file)
--- a/src/soc/fu/trap/pipe_data.py
+++ b/src/soc/fu/trap/pipe_data.py
@@ -1,48 +1,40 @@
-from nmigen import Signal, Const
-from ieee754.fpcommon.getop import FPPipeContext
-from soc.fu.alu.pipe_data import IntegerData
+from soc.fu.pipe_data import FUBaseData, CommonPipeSpec
+from soc.fu.trap.trap_input_record import CompTrapOpSubset
 
 
-class TrapInputData(IntegerData):
+class TrapInputData(FUBaseData):
+    regspec = [('INT', 'ra', '0:63'),  # RA
+               ('INT', 'rb', '0:63'),  # RB/immediate
+               ('FAST', 'fast1', '0:63'), # SRR0
+               ('FAST', 'fast2', '0:63'), # SRR1
+               ('FAST', 'fast3', '0:63'), # SVSRR0
+                # note here that MSR CIA and SVSTATE are *not* read as regs:
+                # they are passed in as incoming "State", via the
+                # CompTrapOpSubset
+               ] 
     def __init__(self, pspec):
-        super().__init__(pspec)
-        self.a = Signal(64, reset_less=True)  # RA
-        self.b = Signal(64, reset_less=True)  # RB/immediate
-        self.cia = Signal(64, reset_less=True)  # Program counter
-        self.msr = Signal(64, reset_less=True)  # MSR
+        super().__init__(pspec, False)
+        # convenience
+        self.srr0, self.srr1, self.svsrr0 = self.fast1, self.fast2, self.fast3
+        self.a, self.b = self.ra, self.rb
 
-    def __iter__(self):
-        yield from super().__iter__()
-        yield self.a
-        yield self.b
-        yield self.cia
-        yield self.msr
 
-    def eq(self, i):
-        lst = super().eq(i)
-        return lst + [self.a.eq(i.a), self.b.eq(i.b),
-                      self.cia.eq(i.nia), self.msr.eq(i.msr)]
-
-class TrapOutputData(IntegerData):
+class TrapOutputData(FUBaseData):
+    regspec = [('INT', 'o', '0:63'),     # RA
+               ('FAST', 'fast1', '0:63'), # SRR0 SPR
+               ('FAST', 'fast2', '0:63'), # SRR1 SPR
+               ('FAST', 'fast3', '0:63'), # SRR2 SPR
+               # ... however we *do* need to *write* MSR, NIA, SVSTATE (RFID)
+               ('STATE', 'nia', '0:63'),  # NIA (Next PC)
+               ('STATE', 'msr', '0:63'),  # MSR
+               ('STATE', 'svstate', '0:63')]  # SVSTATE
     def __init__(self, pspec):
-        super().__init__(pspec)
-        self.nia = Signal(64, reset_less=True) # NIA (Next PC)
-        self.msr = Signal(64, reset_less=True) # MSR
-        self.srr0 = Signal(64, reset_less=True) # SRR0 SPR
-        self.srr1 = Signal(64, reset_less=True) # SRR1 SPR
-        self.should_trap = Signal(reset_less=True)
+        super().__init__(pspec, True)
+        # convenience
+        self.srr0, self.srr1, self.svsrr0 = self.fast1, self.fast2, self.fast3
+
 
-    def __iter__(self):
-        yield from super().__iter__()
-        yield self.nia
-        yield self.msr
-        yield self.srr0
-        yield self.srr1
-        yield self.should_trap
 
-    def eq(self, i):
-        lst = super().eq(i)
-        return lst + [
-            self.nia.eq(i.nia), self.msr.eq(i.msr),
-            self.srr0.eq(i.srr0), self.srr1.eq(i.srr1),
-            self.should_trap.eq(i.should_trap)]
+class TrapPipeSpec(CommonPipeSpec):
+    regspecklses = (TrapInputData, TrapOutputData)
+    opsubsetkls = CompTrapOpSubset