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

flow/network: require ActorNode be passed to add_connection

Browse source
This commit is contained in:
Sebastien Bourdeauducq 2012-06-16 22:40:26 +02:00
parent b0b0380ea7
commit 9af87367eb
1 changed files with 2 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
migen/flow/network.py
View file

@ -53,10 +53,8 @@ class DataFlowGraph(MultiDiGraph):
def add_connection(self, source_node, sink_node,
source_ep=None, sink_ep=None, # default: assume nodes have 1 source/sink and use that one
source_subr=None, sink_subr=None): # default: use whole record
if not isinstance(source_node, ActorNode):
source_node = ActorNode(source_node)
if not isinstance(sink_node, ActorNode):
sink_node = ActorNode(sink_node)
assert(isinstance(source_node, ActorNode))
assert(isinstance(sink_node, ActorNode))
self.add_edge(source_node, sink_node,
source=source_ep, sink=sink_ep,
source_subr=source_subr, sink_subr=sink_subr)