flow/network/DataFlowGraph: add add_pipeline

This commit is contained in:
Sebastien Bourdeauducq 2013-04-30 15:49:51 +02:00
parent fb83794ef4
commit 4f13c5b74d
1 changed files with 4 additions and 0 deletions

View File

@ -38,6 +38,10 @@ class DataFlowGraph(MultiDiGraph):
source=source_ep, sink=sink_ep, source=source_ep, sink=sink_ep,
source_subr=source_subr, sink_subr=sink_subr) source_subr=source_subr, sink_subr=sink_subr)
def add_pipeline(self, *nodes):
for n1, n2 in zip(nodes, nodes[1:]):
self.add_connection(n1, n2)
def del_connections(self, source_node, sink_node, data_requirements): def del_connections(self, source_node, sink_node, data_requirements):
edges_to_delete = [] edges_to_delete = []
edge_data = self.get_edge_data(source_node, sink_node) edge_data = self.get_edge_data(source_node, sink_node)