31 lines
1.0 KiB
Python
31 lines
1.0 KiB
Python
from networkx import MultiDiGraph
|
|
|
|
from migen.fhdl.structure import *
|
|
from migen.flow.actor import *
|
|
from migen.corelogic.misc import optree
|
|
|
|
class DataFlowGraph(MultiDiGraph):
|
|
def add_connection(self, source_node, sink_node, source_ep=None, sink_ep=None):
|
|
if source_ep is None:
|
|
source_eps = source_node.sources()
|
|
assert(len(source_eps) == 1)
|
|
source_ep = source_eps[0]
|
|
if sink_ep is None:
|
|
sink_eps = sink_node.sinks()
|
|
assert(len(sink_eps) == 1)
|
|
sink_ep = sink_eps[0]
|
|
self.add_edge(source_node, sink_node, source=source_ep, sink=sink_ep)
|
|
|
|
class CompositeActor(Actor):
|
|
def __init__(self, dfg): # TODO: endpoints
|
|
self.dfg = dfg
|
|
super().__init__()
|
|
|
|
def get_fragment(self):
|
|
this_fragments = [get_conn_fragment(x[0].endpoints[x[2]["source"]], x[1].endpoints[x[2]["sink"]])
|
|
for x in self.dfg.edges(data=True)]
|
|
this = sum(this_fragments, Fragment())
|
|
others = sum([node.get_fragment() for node in self.dfg], Fragment())
|
|
busy = Fragment([self.busy.eq(optree("|", [node.busy for node in self.dfg]))])
|
|
return this + others + busy
|