blackteay/ffmpeg-python
1
0
Fork 0
mirror of https://github.com/kkroening/ffmpeg-python.git synced 2025-04-06 04:15:44 +08:00
Code Issues Projects Releases Wiki Activity

#18: improve edge labelling

Browse Source
This commit is contained in:
Karl Kroening 2017-07-06 02:32:30 -06:00
parent 1955547202
commit 543cd1b4e3
1 changed files with 12 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
ffmpeg/_view.py
View File

@ -62,10 +62,18 @@ def view(*streams, **kwargs):
for edge in get_outgoing_edges(node, outgoing_edge_map): for edge in get_outgoing_edges(node, outgoing_edge_map):
kwargs = {} kwargs = {}
if show_labels and (edge.upstream_label is not None or edge.downstream_label is not None): up_label = edge.upstream_label
upstream_label = edge.upstream_label if edge.upstream_label is not None else '' down_label = edge.downstream_label
downstream_label = edge.downstream_label if edge.downstream_label is not None else '' if show_labels and (up_label is not None or down_label is not None):
kwargs['label'] = '{} {}'.format(upstream_label, downstream_label) if up_label is None:
up_label = ''
if down_label is None:
down_label = ''
if up_label != '' and down_label != '':
middle = ' -> '
else:
middle = ''
kwargs['label'] = '{} {} {}'.format(up_label, middle, down_label)
upstream_node_id = str(hash(edge.upstream_node)) upstream_node_id = str(hash(edge.upstream_node))
downstream_node_id = str(hash(edge.downstream_node)) downstream_node_id = str(hash(edge.downstream_node))
graph.edge(upstream_node_id, downstream_node_id, **kwargs) graph.edge(upstream_node_id, downstream_node_id, **kwargs)