ž zyMa±c@s–dZdZddlZddlZddlmZddlmZddlm Z Gdd „d e ƒZ Gd d „d e ƒZ ia d d „ZdS(uŒA bottom-up tree matching algorithm implementation meant to speed up 2to3's matching process. After the tree patterns are reduced to their rarest linear path, a linear Aho-Corasick automaton is created. The linear automaton traverses the linear paths from the leaves to the root of the AST and returns a set of nodes for further matching. This reduces significantly the number of candidate nodes.u+George Boutsioukis iN(u defaultdicti(upytree(u reduce_treecBs2|EeZdZdZejƒZdd„ZdS(uBMNodeu?Class for a node of the Aho-Corasick automaton used in matchingcCs1i|_g|_ttjƒ|_d|_dS(Nu(utransition_tableufixersunextuBMNodeucountuiducontent(uself((u8/opt/alt/python33/lib64/python3.3/lib2to3/btm_matcher.pyu__init__s  uBMNode.__init__N(u__name__u __module__u __qualname__u__doc__u itertoolsucountu__init__(u __locals__((u8/opt/alt/python33/lib64/python3.3/lib2to3/btm_matcher.pyuBMNodes uBMNodecBsV|EeZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd S( u BottomMatcherugThe main matcher class. After instantiating the patterns should be added using the add_fixer methodcCsFtƒ|_tƒ|_|jg|_g|_tjdƒ|_dS(NuRefactoringTool( usetumatchuBMNodeurootunodesufixersuloggingu getLoggerulogger(uself((u8/opt/alt/python33/lib64/python3.3/lib2to3/btm_matcher.pyu__init__s    uBottomMatcher.__init__cCsh|jj|ƒt|jƒ}|jƒ}|j|d|jƒ}x|D]}|jj|ƒqJWdS(uÒReduces a fixer's pattern tree to a linear path and adds it to the matcher(a common Aho-Corasick automaton). The fixer is appended on the matching states and called when they are reachedustartN(ufixersuappendu reduce_treeu pattern_treeuget_linear_subpatternuadduroot(uselfufixerutreeulinearu match_nodesu match_node((u8/opt/alt/python33/lib64/python3.3/lib2to3/btm_matcher.pyu add_fixer%s   uBottomMatcher.add_fixercCs |s |gSt|dtƒrˆg}x[|dD]O}|j|d|ƒ}x1|D])}|j|j|dd…|ƒƒqSWq1W|S|d|jkr¸tƒ}||j|d     !          uBottomMatcher.runcs7tdƒ‡fdd†‰ˆ|jƒtdƒdS(u<Prints a graphviz diagram of the BM automaton(for debugging)u digraph g{cs€xy|jjƒD]h}|j|}td|j|jt|ƒt|jƒfƒ|dkrnt|jƒnˆ|ƒqWdS(Nu%d -> %d [label=%s] //%si(utransition_tableukeysuprintuidu type_reprustrufixersucontent(unodeu subnode_keyusubnode(u print_node(u8/opt/alt/python33/lib64/python3.3/lib2to3/btm_matcher.pyu print_node“s ) u*BottomMatcher.print_ac..print_nodeu}N(uprinturoot(uself((u print_nodeu8/opt/alt/python33/lib64/python3.3/lib2to3/btm_matcher.pyuprint_acs  uBottomMatcher.print_acN( u__name__u __module__u __qualname__u__doc__u__init__u add_fixeruaddurunuprint_ac(u __locals__((u8/opt/alt/python33/lib64/python3.3/lib2to3/btm_matcher.pyu BottomMatchers   " =u BottomMatchercCshtsXddlm}x?|jjƒD]+\}}t|ƒtkr&|t|s   …