dot Namespace Reference

Functions
def	dot (G, fn, nodesToHighlight=[])

Function Documentation

dot()

def dot.dot	(		G,
			fn,
			nodesToHighlight = []
	)

dot() is a function to output a graph which suppords adj() that writes
out a .dot file for graph visualisation

Definition at line 5 of file dot.py.

def dot(G, fn, nodesToHighlight=[]):
    """dot() is a function to output a graph which suppords adj() that writes
    out a .dot file for graph visualisation"""
    
    lines = [
        "digraph G{\n",
        "layout=twopi ranksep=3 ratio=auto\n"]
 
    for v in range(G.V):
 
        for w in G.adj(v): 
            line = '%d -> %d;' % (v, w)
            lines.append(line)
 
    for v in nodesToHighlight:
        assert v < G.V
        lines.append('%d  [color=red, style=filled];' % v)
 
    lines.append('}\n')
 
    text = '\n'.join(lines)
 
    with open(fn, 'w') as fh:
        fh.write(text)