1 files changed, 98 insertions, 0 deletions
diff --git a/tools/proxyclient/m1n1/hw/spmi.py b/tools/proxyclient/m1n1/hw/spmi.py
new file mode 100644
index 0000000..a312f61
--- /dev/null
+++ b/tools/proxyclient/m1n1/hw/spmi.py
@@ -0,0 +1,98 @@
+# SPDX-License-Identifier: MIT
+import struct
+
+from ..utils import *
+
+__all__ = ["SPMI"]
+
+CMD_EXT_WRITE   = 0x00
+CMD_EXT_READ    = 0x20
+CMD_EXT_WRITEL  = 0x30
+CMD_EXT_READL   = 0x38
+CMD_WRITE       = 0x40
+CMD_READ        = 0x60
+CMD_ZERO_WRITE  = 0x80
+
+class R_CMD(Register32):
+    REG         = 31, 16
+    ACTIVE      = 15
+    SLAVE_ID    = 14, 8
+    CMD         = 7, 0
+
+class R_STATUS(Register32):
+    RX_EMPTY    = 24
+    RX_COUNT    = 23, 16
+    TX_EMPTY    = 8
+    TX_COUNT    = 7, 0
+
+class SPMIRegs(RegMap):
+    STATUS      = 0x00, R_STATUS
+    CMD         = 0x04, R_CMD
+    REPLY       = 0x08, Register32
+    IRQ_FLAG    = 0x80, Register32
+
+class SPMI:
+    def __init__(self, u, adt_path):
+        self.u = u
+        self.p = u.proxy
+        self.iface = u.iface
+        self.base = u.adt[adt_path].get_reg(0)[0]
+        self.regs = SPMIRegs(u, self.base)
+
+    def read(self, slave, reg, size):
+        while not self.regs.STATUS.reg.RX_EMPTY:
+            print(">", self.regs.REPLY.val)
+
+        self.regs.CMD.reg = R_CMD(REG = reg, ACTIVE=1, SLAVE_ID = slave, CMD = CMD_EXT_READL | (size - 1))
+
+        buf = b""
+
+        left = size + 4
+        while left > 0:
+            while self.regs.STATUS.reg.RX_EMPTY:
+                pass
+            v = self.regs.REPLY.val
+            buf += struct.pack("<I", v)
+            left -= 4
+
+        return buf[4:4+size]
+
+    def write(self, slave, reg, data):
+        while not self.regs.STATUS.reg.RX_EMPTY:
+            self.regs.REPLY.val
+
+        size = len(data)
+        self.regs.CMD.reg = R_CMD(REG = reg, ACTIVE=1, SLAVE_ID = slave, CMD = CMD_EXT_WRITEL | (size - 1))
+
+        while data:
+            blk = (data[:4] + b"\0\0\0")[:4]
+            self.regs.CMD.val = struct.unpack("<I", blk)[0]
+            data = data[4:]
+
+        while self.regs.STATUS.reg.RX_EMPTY:
+            pass
+        return self.regs.REPLY.val
+
+    def read8(self, slave, reg):
+        return struct.unpack("<B", self.read(slave, reg, 1))[0]
+
+    def read16(self, slave, reg):
+        return struct.unpack("<H", self.read(slave, reg, 2))[0]
+
+    def read32(self, slave, reg):
+        return struct.unpack("<I", self.read(slave, reg, 4))[0]
+
+    def read64(self, slave, reg):
+        return struct.unpack("<Q", self.read(slave, reg, 8))[0]
+
+    def write8(self, slave, reg, val):
+        return self.write(slave, reg, struct.pack("<B", val))
+
+    def write16(self, slave, reg, val):
+        return self.write(slave, reg, struct.pack("<H", val))
+
+    def write32(self, slave, reg, val):
+        return self.write(slave, reg, struct.pack("<I", val))
+
+    def write64(self, slave, reg, val):
+        return self.write(slave, reg, struct.pack("<Q", val))