421 lines
12 KiB
Python
421 lines
12 KiB
Python
"""
|
|
|
|
Copyright (c) 2020 Alex Forencich
|
|
|
|
Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
of this software and associated documentation files (the "Software"), to deal
|
|
in the Software without restriction, including without limitation the rights
|
|
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
copies of the Software, and to permit persons to whom the Software is
|
|
furnished to do so, subject to the following conditions:
|
|
|
|
The above copyright notice and this permission notice shall be included in
|
|
all copies or substantial portions of the Software.
|
|
|
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY
|
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
THE SOFTWARE.
|
|
|
|
"""
|
|
|
|
import logging
|
|
from collections import deque
|
|
|
|
import cocotb
|
|
from cocotb.triggers import RisingEdge, Event, First, Timer
|
|
from cocotb.bus import Bus
|
|
|
|
|
|
class StreamTransaction(object):
|
|
|
|
_signals = ["data"]
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
for sig in self._signals:
|
|
if sig in kwargs:
|
|
setattr(self, sig, kwargs[sig])
|
|
del kwargs[sig]
|
|
else:
|
|
setattr(self, sig, 0)
|
|
|
|
super().__init__(*args, **kwargs)
|
|
|
|
def __repr__(self):
|
|
return f"{type(self).__name__}({', '.join(f'{s}={int(getattr(self, s))}' for s in self._signals)})"
|
|
|
|
|
|
class StreamBase(object):
|
|
|
|
_signals = ["data", "valid", "ready"]
|
|
_optional_signals = []
|
|
|
|
_signal_widths = {"valid": 1, "ready": 1}
|
|
|
|
_init_x = False
|
|
|
|
_valid_signal = "valid"
|
|
_valid_init = None
|
|
_ready_signal = "ready"
|
|
_ready_init = None
|
|
|
|
_transaction_obj = StreamTransaction
|
|
|
|
def __init__(self, entity, name, clock, reset=None, *args, **kwargs):
|
|
self.log = logging.getLogger(f"cocotb.{entity._name}.{name}")
|
|
self.entity = entity
|
|
self.clock = clock
|
|
self.reset = reset
|
|
self.bus = Bus(self.entity, name, self._signals, optional_signals=self._optional_signals, **kwargs)
|
|
|
|
super().__init__(*args, **kwargs)
|
|
|
|
self.queue = deque()
|
|
self.queue_sync = Event()
|
|
|
|
self.ready = None
|
|
self.valid = None
|
|
|
|
if self._ready_signal is not None and hasattr(self.bus, self._ready_signal):
|
|
self.ready = getattr(self.bus, self._ready_signal)
|
|
if self._ready_init is not None:
|
|
self.ready.setimmediatevalue(self._ready_init)
|
|
|
|
if self._valid_signal is not None and hasattr(self.bus, self._valid_signal):
|
|
self.valid = getattr(self.bus, self._valid_signal)
|
|
if self._valid_init is not None:
|
|
self.valid.setimmediatevalue(self._valid_init)
|
|
|
|
for sig in self._signals+self._optional_signals:
|
|
if hasattr(self.bus, sig):
|
|
if sig in self._signal_widths:
|
|
assert len(getattr(self.bus, sig)) == self._signal_widths[sig]
|
|
if self._init_x and sig not in (self._valid_signal, self._ready_signal):
|
|
v = getattr(self.bus, sig).value
|
|
v.binstr = 'x'*len(v)
|
|
getattr(self.bus, sig).setimmediatevalue(v)
|
|
|
|
def count(self):
|
|
return len(self.queue)
|
|
|
|
def empty(self):
|
|
return not self.queue
|
|
|
|
def clear(self):
|
|
self.queue.clear()
|
|
|
|
|
|
class StreamPause(object):
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
|
|
self.pause = False
|
|
self._pause_generator = None
|
|
self._pause_cr = None
|
|
|
|
def set_pause_generator(self, generator=None):
|
|
if self._pause_cr is not None:
|
|
self._pause_cr.kill()
|
|
self._pause_cr = None
|
|
|
|
self._pause_generator = generator
|
|
|
|
if self._pause_generator is not None:
|
|
self._pause_cr = cocotb.fork(self._run_pause())
|
|
|
|
def clear_pause_generator(self):
|
|
self.set_pause_generator(None)
|
|
|
|
async def _run_pause(self):
|
|
for val in self._pause_generator:
|
|
self.pause = val
|
|
await RisingEdge(self.clock)
|
|
|
|
|
|
class StreamSource(StreamBase, StreamPause):
|
|
|
|
_signals = ["data", "valid", "ready"]
|
|
_optional_signals = []
|
|
|
|
_signal_widths = {"valid": 1, "ready": 1}
|
|
|
|
_init_x = True
|
|
|
|
_valid_signal = "valid"
|
|
_valid_init = 0
|
|
_ready_signal = "ready"
|
|
_ready_init = None
|
|
|
|
_transaction_obj = StreamTransaction
|
|
|
|
def __init__(self, entity, name, clock, reset=None, *args, **kwargs):
|
|
super().__init__(entity, name, clock, reset, *args, **kwargs)
|
|
|
|
self.drive_obj = None
|
|
self.drive_sync = Event()
|
|
|
|
self.active = False
|
|
|
|
cocotb.fork(self._run_source())
|
|
cocotb.fork(self._run())
|
|
|
|
async def drive(self, obj):
|
|
if self.drive_obj is not None:
|
|
self.drive_sync.clear()
|
|
await self.drive_sync.wait()
|
|
|
|
self.drive_obj = obj
|
|
|
|
async def send(self, obj):
|
|
self.send_nowait(obj)
|
|
|
|
def send_nowait(self, obj):
|
|
self.queue.append(obj)
|
|
self.queue_sync.set()
|
|
|
|
def idle(self):
|
|
return self.empty() and not self.active
|
|
|
|
async def wait(self):
|
|
while not self.idle():
|
|
await RisingEdge(self.clock)
|
|
|
|
def clear(self):
|
|
self.queue.clear()
|
|
self.drive_obj = None
|
|
self.drive_sync.set()
|
|
|
|
async def _run_source(self):
|
|
while True:
|
|
await RisingEdge(self.clock)
|
|
|
|
# read handshake signals
|
|
ready_sample = self.ready is None or self.ready.value
|
|
valid_sample = self.valid is None or self.valid.value
|
|
|
|
if self.reset is not None and self.reset.value:
|
|
self.clear()
|
|
if self.valid is not None:
|
|
self.valid <= 0
|
|
self.active = False
|
|
continue
|
|
|
|
if (ready_sample and valid_sample) or (not valid_sample):
|
|
if self.drive_obj and not self.pause:
|
|
self.bus.drive(self.drive_obj)
|
|
self.drive_obj = None
|
|
self.drive_sync.set()
|
|
if self.valid is not None:
|
|
self.valid <= 1
|
|
self.active = True
|
|
else:
|
|
if self.valid is not None:
|
|
self.valid <= 0
|
|
self.active = bool(self.drive_obj)
|
|
|
|
async def _run(self):
|
|
while True:
|
|
while not self.queue:
|
|
self.queue_sync.clear()
|
|
await self.queue_sync.wait()
|
|
|
|
await self.drive(self.queue.popleft())
|
|
|
|
|
|
class StreamSink(StreamBase, StreamPause):
|
|
|
|
_signals = ["data", "valid", "ready"]
|
|
_optional_signals = []
|
|
|
|
_signal_widths = {"valid": 1, "ready": 1}
|
|
|
|
_init_x = False
|
|
|
|
_valid_signal = "valid"
|
|
_valid_init = None
|
|
_ready_signal = "ready"
|
|
_ready_init = 0
|
|
|
|
_transaction_obj = StreamTransaction
|
|
|
|
def __init__(self, entity, name, clock, reset=None, *args, **kwargs):
|
|
super().__init__(entity, name, clock, reset, *args, **kwargs)
|
|
|
|
cocotb.fork(self._run_sink())
|
|
|
|
async def recv(self):
|
|
while self.empty():
|
|
self.queue_sync.clear()
|
|
await self.queue_sync.wait()
|
|
return self.recv_nowait()
|
|
|
|
def recv_nowait(self):
|
|
if self.queue:
|
|
return self.queue.popleft()
|
|
return None
|
|
|
|
async def wait(self, timeout=0, timeout_unit=None):
|
|
if not self.empty():
|
|
return
|
|
self.queue_sync.clear()
|
|
if timeout:
|
|
await First(self.queue_sync.wait(), Timer(timeout, timeout_unit))
|
|
else:
|
|
await self.queue_sync.wait()
|
|
|
|
def callback(self, obj):
|
|
self.queue.append(obj)
|
|
self.queue_sync.set()
|
|
|
|
async def _run_sink(self):
|
|
while True:
|
|
await RisingEdge(self.clock)
|
|
|
|
# read handshake signals
|
|
ready_sample = self.ready is None or self.ready.value
|
|
valid_sample = self.valid is None or self.valid.value
|
|
|
|
if self.reset is not None and self.reset.value:
|
|
self.clear()
|
|
if self.ready is not None:
|
|
self.ready <= 0
|
|
continue
|
|
|
|
if ready_sample and valid_sample:
|
|
obj = self._transaction_obj()
|
|
self.bus.sample(obj)
|
|
self.callback(obj)
|
|
|
|
if self.ready is not None:
|
|
self.ready <= (not self.pause)
|
|
|
|
|
|
class StreamMonitor(StreamBase):
|
|
|
|
_signals = ["data", "valid", "ready"]
|
|
_optional_signals = []
|
|
|
|
_signal_widths = {"valid": 1, "ready": 1}
|
|
|
|
_init_x = False
|
|
|
|
_valid_signal = "valid"
|
|
_valid_init = None
|
|
_ready_signal = "ready"
|
|
_ready_init = None
|
|
|
|
_transaction_obj = StreamTransaction
|
|
|
|
def __init__(self, entity, name, clock, reset=None, *args, **kwargs):
|
|
super().__init__(entity, name, clock, reset, *args, **kwargs)
|
|
|
|
cocotb.fork(self._run_monitor())
|
|
|
|
async def recv(self):
|
|
while self.empty():
|
|
self.queue_sync.clear()
|
|
await self.queue_sync.wait()
|
|
return self.recv_nowait()
|
|
|
|
def recv_nowait(self):
|
|
if self.queue:
|
|
return self.queue.popleft()
|
|
return None
|
|
|
|
def count(self):
|
|
return len(self.queue)
|
|
|
|
def empty(self):
|
|
return not self.queue
|
|
|
|
async def wait(self, timeout=0, timeout_unit=None):
|
|
if not self.empty():
|
|
return
|
|
self.queue_sync.clear()
|
|
if timeout:
|
|
await First(self.queue_sync.wait(), Timer(timeout, timeout_unit))
|
|
else:
|
|
await self.queue_sync.wait()
|
|
|
|
def callback(self, obj):
|
|
self.queue.append(obj)
|
|
self.queue_sync.set()
|
|
|
|
async def _run_monitor(self):
|
|
while True:
|
|
await RisingEdge(self.clock)
|
|
|
|
# read handshake signals
|
|
ready_sample = self.ready is None or self.ready.value
|
|
valid_sample = self.valid is None or self.valid.value
|
|
|
|
if self.reset is not None and self.reset.value:
|
|
self.clear()
|
|
continue
|
|
|
|
if ready_sample and valid_sample:
|
|
obj = self._transaction_obj()
|
|
self.bus.sample(obj)
|
|
self.callback(obj)
|
|
|
|
|
|
def define_stream(name, signals, optional_signals=None, valid_signal=None, ready_signal=None, signal_widths=None):
|
|
all_signals = signals.copy()
|
|
|
|
if optional_signals is None:
|
|
optional_signals = []
|
|
else:
|
|
all_signals += optional_signals
|
|
|
|
if valid_signal is None:
|
|
for s in all_signals:
|
|
if s.lower().endswith('valid'):
|
|
valid_signal = s
|
|
if valid_signal not in all_signals:
|
|
signals += valid_signal
|
|
|
|
if ready_signal is None:
|
|
for s in all_signals:
|
|
if s.lower().endswith('ready'):
|
|
ready_signal = s
|
|
else:
|
|
if ready_signal not in all_signals:
|
|
signals += ready_signal
|
|
|
|
if signal_widths is None:
|
|
signal_widths = {}
|
|
|
|
if valid_signal not in signal_widths:
|
|
signal_widths[valid_signal] = 1
|
|
|
|
if ready_signal not in signal_widths:
|
|
signal_widths[ready_signal] = 1
|
|
|
|
filtered_signals = []
|
|
|
|
for s in all_signals:
|
|
if s not in (ready_signal, valid_signal):
|
|
filtered_signals.append(s)
|
|
|
|
attrib = {s: 0 for s in filtered_signals}
|
|
attrib['_signals'] = filtered_signals
|
|
|
|
transaction = type(name+"Transaction", (StreamTransaction,), attrib)
|
|
|
|
attrib = {}
|
|
attrib['_signals'] = signals
|
|
attrib['_optional_signals'] = optional_signals
|
|
attrib['_signal_widths'] = signal_widths
|
|
attrib['_ready_signal'] = ready_signal
|
|
attrib['_valid_signal'] = valid_signal
|
|
attrib['_transaction_obj'] = transaction
|
|
|
|
source = type(name+"Source", (StreamSource,), attrib)
|
|
sink = type(name+"Sink", (StreamSink,), attrib)
|
|
monitor = type(name+"Monitor", (StreamMonitor,), attrib)
|
|
|
|
return transaction, source, sink, monitor
|