litex/migen/flow/plumbing.py

44 lines
1.3 KiB
Python

from migen.fhdl.structure import *
from migen.flow.actor import *
from migen.corelogic.record import *
from migen.corelogic.misc import optree
class Buffer(Actor):
def __init__(self, layout):
self.d = Record(layout)
self.q = Record(layout)
Actor.__init__(self,
SchedulingModel(SchedulingModel.PIPELINE, 1),
self.d, self.q)
def get_process_fragment(self):
sigs_d = self.d.flatten()
sigs_q = self.q.flatten()
sync = [If(self.pipe_ce, Cat(*sigs_q).eq(Cat(*sigs_d)))]
return Fragment(sync=sync)
class Combinator(Actor):
def __init__(self, layout, *subrecords):
self.destination = Record(layout)
self.ins = [self.destination.subrecord(*subr) for subr in subrecords]
Actor.__init__(self,
SchedulingModel(SchedulingModel.COMBINATORIAL),
self.ins, self.destination)
def get_fragment(self):
source = self.sources()[0]
sinks = self.sinks()
comb = [source.stb.eq(optree('&', [sink.stb for sink in sinks]))]
comb += [sink.ack.eq(source.ack & source.stb) for sink in sinks]
return Fragment(comb)
class Splitter(Actor):
def __init__(self, layout, *subrecords):
self.source = Record(layout)
self.outs = [self.source.subrecord(*subr) for subr in subrecords]
Actor.__init__(self,
SchedulingModel(SchedulingModel.COMBINATORIAL),
self.source, self.outs)
# TODO def get_fragment(self):