] @ 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 <gboutsioukis@gmail.com> 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 <n | j | d } | d d r | j | d d d | } n | g } | Sd S)Nr r r )
isinstancetupler extendr r )r patternr r! Z alternativeZ end_nodesendZ next_noder r r r 1 s"
+ " zBottomMatcher.addc C s | j } t t } xr| D]j} | } x[| rd | _ x< | j D]1 } t | t j rD | j d k rD d | _ PqD W| j d k r | j } n | j } | | j
k r | j
| } x | j D]- } | | k r g | | <| | j | q Wn | j } | j
d k r | j
j r P| | j
k ry| j
| } x> | 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.<locals>.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 | <q& Wt j | | S)Nr )python_symbols) _type_reprsZpygramr6 __dict__itemsr+ int
setdefault)Ztype_numr6 namevalr r r r1 s r1 )
__author__r r collectionsr r r Z btm_utilsr objectr r r7 r1 r r r r <module> s