o JAf@sj dZddlZddlZddlmZddlmZddlmZGddde Z Gd d d e Z ia d d Z dS) z+George Boutsioukis N) defaultdict)pytree) reduce_treec@seZdZ eZddZdS)BMNodecCs"i|_g|_ttj|_d|_dS)N)transition_tablefixersnextrcountidcontentselfr*/usr/lib/python3.10/lib2to3/btm_matcher.py__init__s  zBMNode.__init__N)__name__ __module__ __qualname__ itertoolsr rrrrrrs rc@s6eZdZ ddZddZddZddZd d Zd S) BottomMatchercCs0t|_t|_|jg|_g|_td|_dS)NZRefactoringTool) setmatchrrootZnodesr logging getLoggerloggerrrrrrs  zBottomMatcher.__init__cCsJ |j|t|j}|}|j||jd}|D]}|j|qdS)Nstart)r appendrZ pattern_treeZget_linear_subpatternaddr)rfixerZtreeZlinear match_nodesZ match_noderrr add_fixer%s  zBottomMatcher.add_fixerc Cs |s|gSt|dtr1g}|dD]}|j||d}|D]}|||dd|qq|S|d|jvrCt}||j|d<n|j|d}|ddr]|j|dd|d}|S|g}|S)Nrrr) isinstancetupler!extendrr)rpatternrr#Z alternativeZ end_nodesendZ next_noderrrr!1s(  zBottomMatcher.addc Cs |j}tt}|D]l}|}|rvd|_|jD]}t|tjr(|jdkr(d|_nq|j dkr2|j}n|j }||j vrM|j |}|j D] }|| |qBn$|j}|j durZ|j jrZn||j vrq|j |}|j D] }|| |qg|j }|sq |S)NT;Fr)rrlistZ was_checkedchildrenr%rZLeafvaluetyperr r parent) rZleavesZcurrent_ac_noderesultsZleafZcurrent_ast_nodechildZ node_tokenr"rrrrunSsB         "zBottomMatcher.runcs, tdfdd|jtddS)Nz digraph g{csZ|jD]%}|j|}td|j|jt|t|jf|dkr&t|j|qdS)Nz%d -> %d [label=%s] //%sr)rkeysprintr type_reprstrr r )ZnodeZ subnode_keyZsubnode print_noderrr8s   z*BottomMatcher.print_ac..print_node})r4rrrr7rprint_acs    zBottomMatcher.print_acN)rrrrr$r!r2r:rrrrrs " 8rcCsDtsddlm}|jD]\}}t|tkr|t|<q t||S)Nr)python_symbols) _type_reprsZpygramr;__dict__itemsr.int setdefault)Ztype_numr;namevalrrrr5s   r5) __author__rr collectionsrrrZ btm_utilsrobjectrrr<r5rrrrs