From a8e1526407ce7c612c6fe90a897e1389b17f1cb0 Mon Sep 17 00:00:00 2001
From: Florent Kermarrec <florent@enjoy-digital.fr>
Date: Fri, 19 Dec 2014 01:23:04 +0100
Subject: [PATCH] link_tb: simplify using implicit submodules collect

---
 lib/sata/test/common.py  |  2 +-
 lib/sata/test/link_tb.py | 33 ++++++++++++++++++---------------
 2 files changed, 19 insertions(+), 16 deletions(-)

diff --git a/lib/sata/test/common.py b/lib/sata/test/common.py
index e1819f98..b79442a6 100644
--- a/lib/sata/test/common.py
+++ b/lib/sata/test/common.py
@@ -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
 
diff --git a/lib/sata/test/link_tb.py b/lib/sata/test/link_tb.py
index cf97ff04..778b171c 100644
--- a/lib/sata/test/link_tb.py
+++ b/lib/sata/test/link_tb.py
@@ -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.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.link = InsertReset(SATALink(self.hdd.phy))
+
+		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):
-- 
2.30.2