3
\ @ sh d Z ddlZddlZddlmZ ddlmZ ddlmZ G dd de Z
G d d
d
e Z i a d d Z
dS )
z+George Boutsioukis N) defaultdict )pytree) reduce_treec @ s e Zd Zej Zdd ZdS )BMNodec C s" i | _ g | _ttj| _d| _d S )N )transition_tablefixersnextr countidcontent)self r +/usr/lib64/python3.6/lib2to3/btm_matcher.py__init__ s zBMNode.__init__N)__name__
__module__ __qualname__ itertoolsr r r r r r r s r c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
Zd S )
BottomMatcherc C s0 t | _t | _| jg| _g | _tjd| _d S )NZRefactoringTool) setmatchr rootZnodesr loggingZ getLoggerZlogger)r r r r r s
zBottomMatcher.__init__c C sL | j j| t|j}|j }| j|| jd}x|D ]}|j j| q4W d S )N)start)r appendr Z pattern_treeZget_linear_subpatternaddr )r fixerZtreeZlinear match_nodesZ
match_noder r r add_fixer% s
zBottomMatcher.add_fixerc C s |s
|gS t |d trhg }xF|d D ]:}| j||d}x&|D ] }|j| j|dd | q>W q&W |S |d |jkrt }||j|d < n|j|d }|dd r| j|dd |d}n|g}|S d S )Nr )r r )
isinstancetupler extendr r )r patternr r alternativeZ end_nodesendZ next_noder r r r 1 s"
" zBottomMatcher.addc C s0 | j }tt}x|D ]}|}x|r&d|_x,|jD ]"}t|tjr8|jdkr8d|_P q8W |j dkrp|j}n|j }||j
kr|j
| }x|j D ]"}||krg ||< || j | qW nd| j }|j
d k r|j
jrP ||j
kr |j
| }x2|j D ](}||j kr g ||< || j | qW |j
}q$W qW |S )NT;Fr )r r listZ was_checkedZchildrenr! r ZLeafvaluetyper r r parentkeys) r ZleavesZcurrent_ac_nodeZresultsZleafZcurrent_ast_nodeZchildZ
node_tokenr r r r runS s>
zBottomMatcher.runc s* t d fdd | j t d d S )Nz
digraph g{c s^ xX| j j D ]J}| j | }td| j|jt|t|jf |dkrNt|j | q W d S )Nz%d -> %d [label=%s] //%sr )r r, printr type_reprstrr r
)ZnodeZ subnode_keyZsubnode)
print_noder r r1 s
z*BottomMatcher.print_ac..print_node})r. r )r r )r1 r print_ac s
zBottomMatcher.print_acN)r r r r r r r- r3 r r r r r s
"=r c C sH t s s