�
!A?h� � � � d Z ddlZddlZddlmZ ddlmZ ddlmZ G d� de � � Z
G d � d
e � � Z i a d � Z
dS ) z+George Boutsioukis � N)� defaultdict� )�pytree)� reduce_treec �4 � e Zd Z ej � � Zd� ZdS )�BMNodec �l � i | _ g | _ t t j � � | _ d| _ d S )N� )�transition_table�fixers�nextr �count�id�content��selfs �@/opt/alt/python-internal/lib64/python3.11/lib2to3/btm_matcher.py�__init__zBMNode.__init__ s- � � "�� ��� ��v�|�$�$����� � � � N)�__name__�
__module__� __qualname__� itertoolsr r � r r r r s5 � � � � � �I� �I�O� � �E�� � � � r r c �. � e Zd Z d� Zd� Zd� Zd� Zd� ZdS )�
BottomMatcherc � � t � � | _ t � � | _ | j g| _ g | _ t
j d� � | _ d S )N�RefactoringTool) �set�matchr �root�nodesr �logging� getLogger�loggerr s r r zBottomMatcher.__init__ sA � ��U�U��
��H�H�� ��i�[��
��� � �'�(9�:�:�� � � r c � � | j � |� � t |j � � }|� � � }| � || j �� � }|D ] }|j � |� � � d S )N��start)r �appendr � pattern_tree�get_linear_subpattern�addr! )r �fixer�tree�linear� match_nodes�
match_nodes r � add_fixerzBottomMatcher.add_fixer% s� � � �
� ���5�!�!�!��5�-�.�.���+�+�-�-���h�h�v�T�Y�h�7�7� �%� ,� ,�J� � � $� $�U� +� +� +� +� ,� ,r c �� � |s|gS t |d t � � r\g }|d D ]O}| � ||�� � }|D ]3}|� | � |dd � |� � � � �4�P|S |d |j vr t � � }||j |d < n|j |d }|dd � r | � |dd � |�� � }n|g}|S )Nr r' r )�
isinstance�tupler, �extendr r )r �patternr( r0 � alternative� end_nodes�end� next_nodes r r, zBottomMatcher.add1 s' � �?�� ��7�N� �g�a�j�%� (� (� � �K� &�q�z�
C�
C� � !�H�H�[��H� >� >� �$� C� C�C� �&�&�t�x�x����� �S�'A�'A�B�B�B�B�C� � � �q�z��!7�7�7� "�H�H� �5>��&�w�q�z�2�2� "� 2�7�1�:� >� ��q�r�r�{�
(� �H�H�W�Q�R�R�[� �H� B� B� � � &�K� � � r c �8 � | j }t t � � }|D ]�}|}|r�d|_ |j D ]0}t |t j � � r|j dk r d|_ n�1|j dk r|j }n|j }||j
v r3|j
| }|j D ] }|| � |� � � nV| j }|j
�
|j
j rnD||j
v r2|j
| }|j D ] }|| � |� � � |j
}|���|S )NT�;Fr )r! r �list� was_checked�childrenr4 r �Leaf�value�typer r r) �parent) r �leaves�current_ac_node�results�leaf�current_ast_node�child�
node_tokenr- s r �runzBottomMatcher.runS s� � � � �)�� �d�#�#��� # ;� # ;�D� #� �"� !
;�/3� �,� -� 6� � �E�!�%���5�5� �%�+��:L�:L�7<�(�4� ���#�(�A�-�-�!1�!7�J�J�!1�!6�J� � �!A�A�A�&5�&F�z�&R�O�!0�!7� @� @�� ���-�-�.>�?�?�?�?�@� '+�i�O�(�/�;� ,� 3� ?� <� � "�_�%E�E�E�*9�*J�:�*V��%4�%;� D� D�E� #�E�N� 1� 1�2B� C� C� C� C�#3�#:� �C #� !
;��D �r c �p �� t d� � �fd�� �| j � � t d� � d S )Nz
digraph g{c
�" �� | j � � � D ]s}| j | }t d| j |j t |� � t |j � � fz � � |dk rt |j � � �|� � �td S )Nz%d -> %d [label=%s] //%sr )r �keys�printr � type_repr�strr r )�node� subnode_key�subnode�
print_nodes �r rV z*BottomMatcher.print_ac..print_node� s� �� � #� 4� 9� 9� ;� ;�
$�
$� � �/� �<���0��w��
�I�k�,B�,B�C���DW�DW�X�Y� Z� Z� Z� �!�#�#��'�/�*�*�*��
�7�#�#�#�#�
$�
$r �})rP r! )r rV s @r �print_aczBottomMatcher.print_ac� sP �� �F�
�l���� $� $� $� $� $� �
�4�9� � � �
�c�
�
�
�
�
r N)r r r r r2 r, rL rX r r r r r sf � � � � � �+�;� ;� ;�
,�
,�
,� � � �D6� 6� 6�p � � � � r r c �� � t sGddlm} |j � � � D ]'\ }}t |� � t k r
|t |< �(t � | | � � S )Nr )�python_symbols)� _type_reprs�pygramrZ �__dict__�itemsrC �int�
setdefault)�type_numrZ �name�vals r rQ rQ � sq � � � 9�*�*�*�*�*�*� (�0�6�6�8�8� 9� 9�I�D�#��C�y�y�C� � �D��S�!1�� � !� !�(�H� 5� 5�5r )�
__author__r# r � collectionsr r
r � btm_utilsr �objectr r r[ rQ r r r �rh s� ��G� ;�
� ���� � � � � #� #� #� #� #� #� � � � � � � "� "� "� "� "� "�� � � � �V� � � �}� }� }� }� }�F� }� }� }�@ � �6� 6� 6� 6� 6r