def mark_reachable_nodes(ea, source_color=COLOR_SOURCE, other_color=COLOR_REACHABLE):
graph = get_nx_graph(ea)
block_ea = get_block_start(ea)
for descendant in nx.descendants(graph, block_ea):
get_codeblock(descendant).color = other_color
get_codeblock(ea).color = source_color
def mark_reaching_nodes(ea, source_color=COLOR_SOURCE, other_color=COLOR_REACHING):
graph = get_nx_graph(ea)
graph = graph.reverse()
block_ea = get_block_start(ea)
for descendant in nx.descendants(graph, block_ea):
get_codeblock(descendant).color = other_color
get_codeblock(ea).color = source_color
def mark_not_reaching_nodes(ea, source_color=COLOR_SOURCE, other_color=COLOR_NOT_REACHING):
graph = get_nx_graph(ea)
graph = graph.reverse()
block_ea = get_block_start(ea)
reaching = nx.descendants(graph, block_ea)
for node_ea in graph.nodes_iter():
if node_ea not in reaching:
get_codeblock(node_ea).color = other_color
get_codeblock(ea).color = source_color
def mark_unreachable_nodes(ea, source_color=COLOR_SOURCE, other_color=COLOR_UNREACHABLE):
graph = get_nx_graph(ea)
block_ea = get_block_start(ea)
descendants = nx.descendants(graph, block_ea)
for block in get_flowchart(ea):
if block.startEA not in descendants:
block.color = other_color
get_codeblock(ea).color = source_color
