] @ s d Z d d l Z d d l Z d d l m Z d d l m Z d d l m Z Gd d d e Z
Gd d
d
e Z i a d d Z
d S)
z+George Boutsioukis N) defaultdict )pytree) reduce_treec @ s( e Z d Z e j Z d d Z d S)BMNodec C s1 i | _ g | _ t t j | _ d | _ d S)N )transition_tablefixersnextr countidcontent)self r 0/opt/alt/python35/lib64/python3.5/btm_matcher.py__init__ s zBMNode.__init__N)__name__
__module__ __qualname__ itertoolsr r r r r r r s r c @ sL e Z d Z d d Z d d Z d d Z d d Z d d
Z d S)
BottomMatcherc C sF t | _ t | _ | j g | _ g | _ t j d | _ d S)NZRefactoringTool) setmatchr rootZnodesr logging getLoggerlogger)r r r r r s
zBottomMatcher.__init__c C sh | j j | t | j } | j } | j | d | j } x | D] } | j j | qJ Wd S)Nstart)r appendr Z pattern_treeZget_linear_subpatternaddr )r fixerZtreeZlinear match_nodesZ
match_noder r r add_fixer% s
zBottomMatcher.add_fixerc C s | s
| g St | d t r g } x[ | d D]O } | j | d | } x1 | D]) } | j | j | d d | qS Wq1 W| S| d | j k r t } | | j | d | j D]3 } | | j k rdg | | <| | j | qBW| j
} q+ Wq W| S)NT;Fr )r r listZ was_checkedZchildrenr# r ZLeafvaluetyper r r parentkeys) r ZleavesZcurrent_ac_noderesultsZleafZcurrent_ast_nodeZchildZ
node_tokenr r r r runS s>
!
zBottomMatcher.runc s7 t d f d d | j t d d S)Nz
digraph g{c s} xv | j j D]e } | j | } t d | j | j t | t | j f | d k rk t | j | q Wd S)Nz%d -> %d [label=%s] //%sr )r r- printr type_reprstrr r
)ZnodeZ subnode_keyZsubnode)
print_noder r r3 s
)
z*BottomMatcher.print_ac..print_node})r0 r )r r )r3 r print_ac s
zBottomMatcher.print_acN)r r r r r" r r/ r5 r r r r r s
" =r c C sb t sR d d l m } x9 | j j D]( \ } } t | t k r& | t | s