Hardware/litex
1
0
Fork 0
mirror of https://github.com/enjoy-digital/litex.git synced 2025-01-04 09:52:26 -05:00
Code Issues Projects Releases Packages Wiki Activity

soc/interconnect/stream_sim: use passive generators and some cleanup

Browse source
This commit is contained in:
Florent Kermarrec 2016-03-23 01:04:33 +01:00
parent 5e16516706
commit 9517b9b870
1 changed files with 5 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
litex/soc/interconnect/stream_sim.py
View file

@ -74,18 +74,6 @@ def randn(max_n):
return random.randint(0, max_n-1) return random.randint(0, max_n-1)
class RandRun:
def __init__(self, level=0):
self.run = True
self.level = level
def do_simulation(self, selfp):
self.run = True
n = randn(100)
if n < self.level:
self.run = False
class Packet(list): class Packet(list):
def __init__(self, init=[]): def __init__(self, init=[]):
self.ongoing = False self.ongoing = False
@ -115,6 +103,7 @@ class PacketStreamer(Module):
while not packet.done: while not packet.done:
yield yield
@passive
def generator(self): def generator(self):
while True: while True:
if len(self.packets) and self.packet.done: if len(self.packets) and self.packet.done:
@ -155,6 +144,7 @@ class PacketLogger(Module):
while not self.packet.done: while not self.packet.done:
yield yield
@passive
def generator(self): def generator(self):
while True: while True:
yield self.sink.ready.eq(1) yield self.sink.ready.eq(1)
@ -171,7 +161,7 @@ class PacketLogger(Module):
yield yield
class AckRandomizer(Module): class Randomizer(Module):
def __init__(self, description, level=0): def __init__(self, description, level=0):
self.level = level self.level = level
@ -182,12 +172,13 @@ class AckRandomizer(Module):
self.comb += \ self.comb += \
If(self.ce, If(self.ce,
Record.connect(self.sink, self.source) self.sink.connect(self.source)
).Else( ).Else(
self.source.valid.eq(0), self.source.valid.eq(0),
self.sink.ready.eq(0), self.sink.ready.eq(0),
) )
@passive
def generator(self): def generator(self):
while True: while True:
n = randn(100) n = randn(100)