import random from migen.fhdl.std import * from migen.genlib.record import * from migen.sim.generic import run_simulation from lib.sata.common import * from lib.sata.link import SATALink from lib.sata.test.bfm import * from lib.sata.test.common import * class LinkStreamer(Module): def __init__(self): self.source = Source(link_description(32)) ### self.packets = [] self.packet = LinkTXPacket() self.packet.done = 1 def send(self, packet, blocking=True): self.packets.append(packet) if blocking: while packet.done == 0: yield def do_simulation(self, selfp): if len(self.packets) and self.packet.done: self.packet = self.packets.pop(0) if not self.packet.ongoing and not self.packet.done: selfp.source.stb = 1 selfp.source.sop = 1 selfp.source.d = self.packet.pop(0) self.packet.ongoing = True elif selfp.source.stb == 1 and selfp.source.ack == 1: selfp.source.sop = 0 selfp.source.eop = (len(self.packet) == 1) if len(self.packet) > 0: selfp.source.stb = 1 selfp.source.d = self.packet.pop(0) else: self.packet.done = 1 selfp.source.stb = 0 class LinkLogger(Module): def __init__(self): self.sink = Sink(link_description(32)) ### self.packet = LinkRXPacket() def receive(self): self.packet.done = 0 while self.packet.done == 0: yield def do_simulation(self, selfp): selfp.sink.ack = 1 if selfp.sink.stb == 1 and selfp.sink.sop == 1: self.packet = LinkRXPacket() self.packet.append(selfp.sink.d) elif selfp.sink.stb: self.packet.append(selfp.sink.d) if (selfp.sink.stb ==1 and selfp.sink.eop ==1): self.packet.done = True class TB(Module): def __init__(self): self.submodules.bfm = BFM(phy_debug=False, link_random_level=50, transport_debug=False, transport_loopback=True) self.submodules.link = SATALink(self.bfm.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) ] def gen_simulation(self, selfp): for i in range(24): yield for i in range(8): streamer_packet = LinkTXPacket([i for i in range(64)]) yield from self.streamer.send(LinkTXPacket([i for i in range(64)])) yield from self.logger.receive() # check results s, l, e = check(streamer_packet, self.logger.packet) 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)