Hardware
/
litex
mirror of https://github.com/enjoy-digital/litex.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

link_tb: simplify using implicit submodules collect

This commit is contained in:
Florent Kermarrec 2014-12-19 01:23:04 +01:00
parent 4f22bc807a
commit a8e1526407
2 changed files with 18 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lib/sata/test/common.py
View File

@ -98,7 +98,7 @@ class PacketLogger(Module):
if selfp.sink.stb == 1 and selfp.sink.eop == 1:
self.packet.done = True
class AckRandomizer(Module):
class Randomizer(Module):
def __init__(self, description, level=0):
self.level = level

31
lib/sata/test/link_tb.py
View File

@ -4,6 +4,8 @@ from lib.sata.link import SATALink
from lib.sata.test.common import *
from lib.sata.test.hdd import *
from migen.actorlib.structuring import *
class LinkStreamer(PacketStreamer):
def __init__(self):
PacketStreamer.__init__(self, link_description(32), LinkTXPacket)
@ -14,23 +16,24 @@ class LinkLogger(PacketLogger):
class TB(Module):
def __init__(self):
self.submodules.hdd = HDD(
self.hdd = HDD(
link_debug=False, link_random_level=50,
transport_debug=False, transport_loopback=True)
self.submodules.link = SATALink(self.hdd.phy)
self.link = InsertReset(SATALink(self.hdd.phy))
self.submodules.streamer = LinkStreamer()
streamer_ack_randomizer = AckRandomizer(link_description(32), level=50)
self.submodules += streamer_ack_randomizer
self.submodules.logger = LinkLogger()
logger_ack_randomizer = AckRandomizer(link_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.link.sink),
Record.connect(self.link.source, logger_ack_randomizer.sink),
Record.connect(logger_ack_randomizer.source, self.logger.sink)
]
self.streamer = LinkStreamer()
self.streamer_randomizer = Randomizer(link_description(32), level=50)
self.logger_randomizer = Randomizer(link_description(32), level=50)
self.logger = LinkLogger()
self.pipeline = Pipeline(
self.streamer,
self.streamer_randomizer,
self.link,
self.logger_randomizer,
self.logger
)
def gen_simulation(self, selfp):
for i in range(8):