47 lines
1.6 KiB
Python
47 lines
1.6 KiB
Python
|
import networkx as nx
|
||
|
|
|
||
|
|
from migen.flow.actor import *
|
||
|
|
|
||
|
|
class CompositeActor(Actor):
|
||
|
|
def __init__(self, dfg):
|
||
|
|
self.dfg = dfg
|
||
|
|
# Internal unconnected endpoints become our endpoints. Determine them.
|
||
|
|
our_endpoints = []
|
||
|
|
for node in self.dfg:
|
||
|
|
endpoints = set(node.endpoints)
|
||
|
|
for u, v, d in self.dfg.in_edges([node], data=True):
|
||
|
|
endpoints.remove(d['sink'])
|
||
|
|
for u, v, d in self.dfg.out_edges([node], data=True):
|
||
|
|
endpoints.remove(d['source'])
|
||
|
|
our_endpoints += list(endpoints)
|
||
|
|
Actor.__init__(self,
|
||
|
|
SchedulingModel(SchedulingModel.DYNAMIC),
|
||
|
|
endpoints=our_endpoints)
|
||
|
|
|
||
|
|
def get_control_fragment(self):
|
||
|
|
this = sum([get_conn_control_fragment(x[2]['source'], x[2]['sink'])
|
||
|
|
for x in self.dfg.edges(data=True)], Fragment())
|
||
|
|
others = sum([node.get_control_fragment() for node in self.dfg], Fragment())
|
||
|
|
return this + others
|
||
|
|
|
||
|
|
def get_process_fragment(self):
|
||
|
|
this = sum([get_conn_process_fragment(x[2]['source'], x[2]['sink'])
|
||
|
|
for x in self.dfg.edges(data=True)], Fragment())
|
||
|
|
others = sum([node.get_process_fragment() for node in self.dfg], Fragment())
|
||
|
|
return this + others
|
||
|
|
|
||
|
|
def add_connection(dfg, 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]
|
||
|
|
assert(isinstance(source_ep, Source))
|
||
|
|
assert(isinstance(sink_ep, Sink))
|
||
|
|
assert(source_ep in source_node.endpoints)
|
||
|
|
assert(sink_ep in sink_node.endpoints)
|
||
|
|
dfg.add_edge(source_node, sink_node, source=source_ep, sink=sink_ep)
|