Ñò V}òOc@sTdZddklZddkZdefd„ƒYZdeieid„Z dS(sŸ @package antlr3.dottreegenerator @brief ANTLR3 runtime package, tree module This module contains all support classes for AST construction and tree parsers. iÿÿÿÿ(tCommonTreeAdaptorNtDOTTreeGeneratorcBs¦eZdZeiddddddddd d ƒZeidd ƒZeidd ƒZd „Zdeed„Z dd„Z d„Z d„Z d„ZRS(s¾ A utility class to generate DOT diagrams (graphviz) from arbitrary trees. You can pass in your own templates and can pass in any kind of tree or use Tree interface method. ttemplates digraph { s ordering=out; s ranksep=.4; sJ node [shape=plaintext, fixedsize=true, fontsize=11, fontname="Courier", s width=.25, height=.25]; s edge [arrowsize=.5] s $nodes$ s $edges$ s} s$name$ [label="$text$"]; s7$parent$ -> $child$ // "$parentText$" -> "$childText$" cCsh|_d|_dS(Ni(tnodeToNumberMapt nodeNumber(tself((sx/afs/athena.mit.edu/user/x/a/xavid/lib/python2.7/site-packages/antlr_python_runtime-3.1.3-py2.6.egg/antlr3/dottreegen.pyt__init__Hs cCse|djo tƒ}n|iƒ}d|_|i|||ƒd|_|i||||ƒ|S(Ni(tNoneRt getInstanceOfRttoDOTDefineNodesttoDOTDefineEdges(RttreetadaptorttreeSTtedgeST((sx/afs/athena.mit.edu/user/x/a/xavid/lib/python2.7/site-packages/antlr_python_runtime-3.1.3-py2.6.egg/antlr3/dottreegen.pyttoDOTPs     c Cs.|djo tƒ}n|djodS|i|ƒ}|djodS|i|ƒ}||jo3|i||ƒ}|id|ƒ|i|ƒnx‹t|ƒD]}}|i||ƒ} |i| ƒ}||jo3|i|| ƒ} |id| ƒ|i|ƒn|i | |||ƒq©WdS(Nitnodes( Rtsett getChildCountt getNodeNumbert getNodeSTt setAttributetaddtrangetgetChildR ( RR R R t knownNodestntnumbert parentNodeSTtitchildtnodeST((sx/afs/athena.mit.edu/user/x/a/xavid/lib/python2.7/site-packages/antlr_python_runtime-3.1.3-py2.6.egg/antlr3/dottreegen.pyR ^s*       c Cs|djodS|i|ƒ}|djodSd|i|ƒ}|i|ƒ}xºt|ƒD]¬}|i||ƒ} |i| ƒ} d|i| ƒ} |iƒ}|id|ƒ|id| ƒ|id|ƒ|id| ƒ|id|ƒ|i| |||ƒqbWdS(Nisn%dtparentRt parentTextt childTexttedges( RRRtgetTextRRRRR ( RR R R RRt parentNameR!RRR"t childName((sx/afs/athena.mit.edu/user/x/a/xavid/lib/python2.7/site-packages/antlr_python_runtime-3.1.3-py2.6.egg/antlr3/dottreegen.pyR s&    cCsx|i|ƒ}|iiƒ}d|i|ƒ}|id|ƒ|dj o|iddƒ}n|id|ƒ|S(Nsn%dtnamet"s\\"ttext(R$t_nodeSTRRRRtreplace(RR ttR)Rt uniqueName((sx/afs/athena.mit.edu/user/x/a/xavid/lib/python2.7/site-packages/antlr_python_runtime-3.1.3-py2.6.egg/antlr3/dottreegen.pyRšs cCsPy|i|SWn:tj o.|i|i|<|id7_|idSXdS(Ni(RtKeyErrorR(RR,((sx/afs/athena.mit.edu/user/x/a/xavid/lib/python2.7/site-packages/antlr_python_runtime-3.1.3-py2.6.egg/antlr3/dottreegen.pyR¥s N(t__name__t __module__t__doc__tstringtemplate3tStringTemplatet_treeSTR*t_edgeSTRRRR R RR(((sx/afs/athena.mit.edu/user/x/a/xavid/lib/python2.7/site-packages/antlr_python_runtime-3.1.3-py2.6.egg/antlr3/dottreegen.pyR,s        !  cCstƒ}|i||||ƒS(s Generate DOT (graphviz) for a whole tree not just a node. For example, 3+4*5 should generate: digraph { node [shape=plaintext, fixedsize=true, fontsize=11, fontname="Courier", width=.4, height=.2]; edge [arrowsize=.7] "+"->3 "+"->"*" "*"->4 "*"->5 } Return the ST not a string in case people want to alter. Takes a Tree interface object. Example of invokation: import antlr3 import antlr3.extras input = antlr3.ANTLRInputStream(sys.stdin) lex = TLexer(input) tokens = antlr3.CommonTokenStream(lex) parser = TParser(tokens) tree = parser.e().tree print tree.toStringTree() st = antlr3.extras.toDOT(t) print st (RR(R R R Rtgen((sx/afs/athena.mit.edu/user/x/a/xavid/lib/python2.7/site-packages/antlr_python_runtime-3.1.3-py2.6.egg/antlr3/dottreegen.pyR®s# ( R1t antlr3.treeRR2tobjectRRR4R5R(((sx/afs/athena.mit.edu/user/x/a/xavid/lib/python2.7/site-packages/antlr_python_runtime-3.1.3-py2.6.egg/antlr3/dottreegen.pyts# ‚