freet15
/
Linux_Kernel_map4vehicle
forked from OpenSDV/Linux_Kernel_map4vehicle
1
0
Fork 0
Code Pull Requests Activity

fix digraph_tree

This commit is contained in:
Costa Shulyupin 2018-07-29 16:03:45 +03:00
parent f6f4513543
commit 75da8dd37a
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
srcxray.py
View File

@ -416,14 +416,16 @@ def starts(dg): # roots
return {n: dg.out_degree(n) for (n, d) in dg.in_degree if not d}
def digraph_tree(dg, starts=None):
def digraph_tree(dg, starts=None, black_list=black_list):
tree = nx.DiGraph()
def sub(node):
tree.add_node(node)
for o in dg.successors(node):
if o in black_list or tree.has_node(o):
if o in black_list or tree.has_edge(node, o) or o in starts:
# print(o)
continue
tree.add_edge(node,o)
tree.add_edge(node, o)
sub(o)
printed = set()
@ -438,7 +440,6 @@ def digraph_tree(dg, starts=None):
for o in starts:
if o in black_list:
continue
tree.add_edge(default_root,o)
sub(o)
return tree