Advertisement
Guest User

Untitled

a guest
Jun 19th, 2019
92
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.66 KB | None | 0 0
  1. key = 0
  2. def gen_dot(T):
  3.     global key
  4.  
  5.  
  6.     def atnode(n, parent):
  7.         global key
  8.         key += 1
  9.         atn = anytree.Node(str(key),parent = parent, display_name = str(n))
  10.         if n.type == 1:
  11.             sub1 = atnode(n.sub1,atn)
  12.         if n.type == 2:
  13.             sub1 = atnode(n.sub1,atn)
  14.             sub2 = atnode(n.sub2,atn)
  15.         return atn
  16.  
  17.     root = atnode(T,None)
  18.  
  19.  
  20.     DotExporter(root,
  21.                 nodeattrfunc=lambda node: 'label="{}"'.format(node.display_name)).to_dotfile("graph.txt")
  22.     DotExporter(root,
  23.                 nodeattrfunc=lambda node: 'label="{}"'.format(node.display_name)).to_picture("graph.png")
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement