← Back
Editing: scope.cpython-311.pyc
� cR�a` � �^ � d Z ddlmZ ddlmZ ddlmZ ddlmZ ddl Z G d� de� � Z dS ) z" Scope computes scope information � )�AncestorsWithBody)� DefUseChains)�FunctionAnalysis)� defaultdictNc �. � � e Zd ZdZ� fd�Zd� Zd� Z� xZS )�ScopeaN Associate each variable declaration with the node that defines it Whenever possible, associate the variable declaration to an assignment, otherwise to a node that defines a bloc (e.g. a For) This takes OpenMP information into accounts! The result is a dictionary with nodes as key and set of names as values c �( �� t t � � | _ t j t j t j t j t j t f| _ t t | � � � t t � � d S �N)r �set�result�ast� FunctionDef�For� excepthandler�While�If�tuple� decl_holders�superr �__init__r r )�self� __class__s ��8/usr/lib/python3/dist-packages/pythran/analyses/scope.pyr zScope.__init__ s_ �� �!�#�&�&�� � �_�c�g� � .� �Y���� 7��� �e�T���#�#�$5�|�D�D�D�D�D� c �� � |j D ]Y}||j v r� t |t j � � r3| j � |j g � � � |� � �Zd S r ) �deps� private_deps� isinstancer �Name� openmp_deps� setdefault�id�append)r �node�deps r �visit_OMPDirectivezScope.visit_OMPDirective sq � ��9� D� D�C��d�'�'�'���#�s�x�(�(� D�� �+�+�C�F�B�7�7�>�>�s�C�C�C�� D� Dr c � � � � j � _ t � � � _ � � |� � t � � }� j j | D ]=}|� |� � � g � � � |� � �>|� � � D �]�\ }}d� |D � � d� |D � � z }|� � fd�� j � |g � � D � � � � � fd�|D � � }d� t |� D � � }|d d } t | � j � � r�d� |D � � }|� � j � |g � � � � t! � � } |D ]V} | } � j | d | ur(� j | d } � j | d | u�(| � | � � �W| | vrEt | t$ � � r| n| j } | D ]$}|| v r t |t( j � � r|} n�%� j | � |� � ���d S ) Nc � � g | ] }|j �� S � �r$ ��.0�ds r � <listcomp>z+Scope.visit_FunctionDef.<locals>.<listcomp>4 s � �)�)�)�q�A�F�)�)�)r c �J � g | ] }|� � � D ] }|j �� �!S r) )�usersr$ )r, r- �us r r. z+Scope.visit_FunctionDef.<locals>.<listcomp>4 sO � � -O� -O� -O�12�A�G�G�I�I�-O� -O�?@� ./�V� -O� -O� -O� -Or c 3 �> �K � | ]}�j | d V � �dS )���N�� ancestors)r, r- r s �r � <genexpr>z*Scope.visit_FunctionDef.<locals>.<genexpr>7 sK �� � � � A� A� � ��q�)�"�-� A� A� A� A� A� Ar c �* �� g | ]}�j | ��S r) r4 )r, �refr s �r r. z+Scope.visit_FunctionDef.<locals>.<listcomp>: s �� �=�=�=��� ��,�=�=�=r c �R � g | ]$}t t |� � � � d k �"|��%S )� )�lenr )r, �ps r r. z+Scope.visit_FunctionDef.<locals>.<listcomp>< s. � �G�G�G�a�c�#�a�&�&�k�k�Q�6F�6F��6F�6F�6Fr ���r c � � g | ] }|j �� S r) r* r+ s r r. z+Scope.visit_FunctionDef.<locals>.<listcomp>C s � �-�-�-�1���-�-�-r )�ancestors_with_bodyr5 �dictr � generic_visit�def_use_chains�localsr! �namer# �items�extend�get�zipr r r �addr �bodyr �Assignr )r r$ � name_to_defs�def_rD �defs�refsr5 �prefixes�common�prefs�r�ancestorrJ �cs ` r �visit_FunctionDefzScope.visit_FunctionDef% s� �� � �1��� �6�6��� ���4� � � ��v�v� ��'�.�t�4� B� B�D� � #� #�D�I�I�K�K�� 4� 4� ;� ;�D� A� A� A� A� '�,�,�.�.� ( *� ( *�J�D�$�)�)�D�)�)�)� -O� -O�6:�-O� -O� -O� O�D� �K�K� A� A� A� A�!%�!1�!5�!5�d�B�!?�!?�A� A� A� A� A� A� >�=�=�=��=�=�=�I�G�G�3� �?�G�G�G�H� �b�\�!�_�F� �&�$�"3�4�4� "�-�-��-�-�-��� � �D� ,� 0� 0��r� :� :�;�;�;� ���� � (� (�A� �H� �.��2�2�6�f�D�D�#'�>�(�#;�B�#?�� �.��2�2�6�f�D�D��I�I�h�'�'�'�'� ��&�&�%/���%>�%>�O�6�6�F�K�D� !� "� "�� ��:�:� )�!�S�Z� 8� 8� +�)*�� !�E� &� �K�� � #� #�D� )� )� )� )�Q( *� ( *r )�__name__� __module__� __qualname__�__doc__r r& rV � __classcell__)r s @r r r sh �� � � � � �� �E� E� E� E� E�D� D� D�5*� 5*� 5*� 5*� 5*� 5*� 5*r r ) rZ �pythran.analyses.ancestorsr � pythran.analyses.use_def_chainr �pythran.passmanagerr � collectionsr �gastr r r) r r �<module>ra s� ��� � 9� 8� 8� 8� 8� 8� 7� 7� 7� 7� 7� 7� 0� 0� 0� 0� 0� 0� #� #� #� #� #� #� � � � �M*� M*� M*� M*� M*� � M*� M*� M*� M*� M*r
Save File
Cancel