��]��@s�dZddlZddlZddlmZddlmZddlmZGdd�de �Z Gd d �d e �Ziadd�Z dS) z+George Boutsioukis �N)�defaultdict�)�pytree)�reduce_treec@s(eZdZej�Zdd�ZdS)�BMNodecCs1i|_g|_ttj�|_d|_dS)N�)�transition_table�fixers�nextr�count�id�content)�self�r�0/opt/alt/python35/lib64/python3.5/btm_matcher.py�__init__s zBMNode.__init__N)�__name__� __module__�__qualname__� itertoolsrrrrrrrsrc@sLeZdZdd�Zdd�Zdd�Zdd�Zd d �ZdS)� BottomMatchercCsFt�|_t�|_|jg|_g|_tjd�|_dS)NZRefactoringTool) �set�matchr�rootZnodesr �logging� getLogger�logger)rrrrrs zBottomMatcher.__init__cCsh|jj|�t|j�}|j�}|j|d|j�}x|D]}|jj|�qJWdS)N�start)r �appendrZpattern_treeZget_linear_subpattern�addr)r�fixerZtreeZlinear�match_nodesZ match_noderrr� add_fixer%s zBottomMatcher.add_fixercCs|s |gSt|dt�r�g}x[|dD]O}|j|d|�}x1|D])}|j|j|dd�|��qSWq1W|S|d|jkr�t�}||j|d|jD]3}||j�krdg||<||j|�qBW|j }q+WqW|S)NT�;Fr)rr�listZwas_checkedZchildrenr#rZLeaf�value�typerr r�parent�keys) rZleavesZcurrent_ac_node�resultsZleafZcurrent_ast_nodeZchildZ node_tokenr rrr�runSs> ! zBottomMatcher.runcs7td��fdd��|j�td�dS)Nz digraph g{cs}xv|jj�D]e}|j|}td|j|jt|�t|j�f�|dkrkt|j��|�qWdS)Nz%d -> %d [label=%s] //%sr)rr-�printr� type_repr�strr r )ZnodeZsubnode_keyZsubnode)� print_noderrr3�s ) z*BottomMatcher.print_ac..print_node�})r0r)rr)r3r�print_ac�s zBottomMatcher.print_acN)rrrrr"rr/r5rrrrrs "=rcCsbtsRddlm}x9|jj�D](\}}t|�tkr&|t|s �