1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
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))
|
