104 lines
3.4 KiB
Python
104 lines
3.4 KiB
Python
from lib.sata.common import *
|
|
from lib.sata.link import SATALink
|
|
from lib.sata.transport import SATATransport
|
|
from lib.sata.command import SATACommand
|
|
|
|
from lib.sata.test.hdd import *
|
|
from lib.sata.test.common import *
|
|
|
|
class CommandTXPacket(list):
|
|
def __init__(self, write=0, read=0, identify=0, sector=0, count=0, data=[]):
|
|
self.ongoing = False
|
|
self.done = False
|
|
self.write = write
|
|
self.read = read
|
|
self.identify = identify
|
|
self.sector = sector
|
|
self.count = count
|
|
for d in data:
|
|
self.append(d)
|
|
|
|
class CommandStreamer(PacketStreamer):
|
|
def __init__(self):
|
|
PacketStreamer.__init__(self, command_tx_description(32), CommandTXPacket)
|
|
|
|
def do_simulation(self, selfp):
|
|
PacketStreamer.do_simulation(self, selfp)
|
|
selfp.source.write = self.packet.write
|
|
selfp.source.read = self.packet.read
|
|
selfp.source.identify = self.packet.identify
|
|
selfp.source.sector = self.packet.sector
|
|
selfp.source.count = self.packet.count
|
|
|
|
class CommandRXPacket(list):
|
|
def __init__(self):
|
|
self.ongoing = False
|
|
self.done = False
|
|
self.write = 0
|
|
self.read = 0
|
|
self.identify = 0
|
|
self.success = 0
|
|
self.failed = 0
|
|
|
|
class CommandLogger(PacketLogger):
|
|
def __init__(self):
|
|
PacketLogger.__init__(self, command_rx_description(32), CommandRXPacket)
|
|
|
|
def do_simulation(self, selfp):
|
|
selfp.sink.ack = 1
|
|
if selfp.sink.stb == 1 and selfp.sink.sop == 1:
|
|
self.packet = CommandRXPacket()
|
|
self.packet.write = selfp.sink.write
|
|
self.packet.read = selfp.sink.read
|
|
self.packet.identify = selfp.sink.identify
|
|
self.packet.sucess = selfp.sink.success
|
|
self.packet.failed = selfp.sink.failed
|
|
self.packet.append(selfp.sink.data)
|
|
elif selfp.sink.stb:
|
|
self.packet.append(selfp.sink.data)
|
|
if selfp.sink.stb == 1 and selfp.sink.eop == 1:
|
|
self.packet.done = True
|
|
|
|
class TB(Module):
|
|
def __init__(self):
|
|
self.submodules.hdd = HDD(
|
|
link_debug=False, link_random_level=50,
|
|
transport_debug=False, transport_loopback=False,
|
|
hdd_debug=True)
|
|
self.submodules.link = SATALink(self.hdd.phy)
|
|
self.submodules.transport = SATATransport(self.link)
|
|
self.submodules.command = SATACommand(self.transport)
|
|
|
|
self.submodules.streamer = CommandStreamer()
|
|
streamer_ack_randomizer = AckRandomizer(command_tx_description(32), level=50)
|
|
self.submodules += streamer_ack_randomizer
|
|
self.submodules.logger = CommandLogger()
|
|
logger_ack_randomizer = AckRandomizer(command_rx_description(32), level=50)
|
|
self.submodules += logger_ack_randomizer
|
|
self.comb += [
|
|
Record.connect(self.streamer.source, streamer_ack_randomizer.sink),
|
|
Record.connect(streamer_ack_randomizer.source, self.command.sink),
|
|
Record.connect(self.command.source, logger_ack_randomizer.sink),
|
|
Record.connect(logger_ack_randomizer.source, self.logger.sink)
|
|
]
|
|
|
|
def gen_simulation(self, selfp):
|
|
hdd = self.hdd
|
|
hdd.malloc(0, 64)
|
|
write_data = [i for i in range(hdd.sectors2dwords(2))]
|
|
write_len = hdd.dwords2sectors(len(write_data))
|
|
write_packet = CommandTXPacket(write=1, sector=2, count=write_len, data=write_data)
|
|
yield from self.streamer.send(write_packet)
|
|
yield from self.logger.receive()
|
|
read_packet = CommandTXPacket(read=1, sector=2, count=write_len)
|
|
yield from self.streamer.send(read_packet)
|
|
yield from self.logger.receive()
|
|
read_data = self.logger.packet
|
|
|
|
# check results
|
|
s, l, e = check(write_data, read_data)
|
|
print("shift "+ str(s) + " / length " + str(l) + " / errors " + str(e))
|
|
|
|
if __name__ == "__main__":
|
|
run_simulation(TB(), ncycles=2048, vcd_name="my.vcd", keep_files=True)
|