← Back
Editing: expand_globals.cpython-311.pyc
� cR�aC � � � d Z ddlmZ ddlmZ ddlmZ ddlmZ ddl m Z ddl Z G d� d e j � � Z G d � d e� � ZdS ) zu ExpandGlobals replaces globals variables by function call. It also turn globals assignment in function definition. � )�LocalNameDeclarations)�Transformation)�PythranSyntaxError)� path_to_attr)�metadataNc � � e Zd ZdZd� Zd� ZdS )�GlobalTransformerz? Use assumptions on globals to improve code generation c � � |S �N� ��self�nodes �H/usr/lib/python3/dist-packages/pythran/transformations/expand_globals.py� visit_Callz GlobalTransformer.visit_Call s � �� � c � � � t j t d� � t j � fd�|j D � � t j � � � � gg � � S )N)�builtins�pythran� static_listc �: �� g | ]}�� |� � ��S r )�visit)�.0�eltr s �r � <listcomp>z0GlobalTransformer.visit_List.<locals>.<listcomp> s# �� �#I�#I�#I��D�J�J�s�O�O�#I�#I�#Ir )�ast�Callr �Tuple�elts�Loadr s ` r � visit_Listz GlobalTransformer.visit_List s[ �� ��x� �%K�L�L� ��#I�#I�#I�#I�t�y�#I�#I�#I�"%�(�*�*�.� .� /�� � � r N)�__name__� __module__� __qualname__�__doc__r r! r r r r r s< � � � � � �� �� � � � � � � r r c �. � � e Zd ZdZ� fd�Zd� Zd� Z� xZS )� ExpandGlobalsa� Expands all builtins into full paths. >>> import gast as ast >>> from pythran import passmanager, backend >>> node = ast.parse(''' ... a = 1 ... def foo(): ... return a''') >>> pm = passmanager.PassManager("test") >>> _, node = pm.apply(ExpandGlobals, node) >>> print(pm.dump(backend.Python, node)) def a(): return 1 def foo(): return a() c � �� t � � | _ t � � | _ t t | � � � � � dS )z; Initialize local declaration and constant name to expand. N)�set� local_decl� to_expand�superr' �__init__)r � __class__s �r r- zExpandGlobals.__init__4 s; �� � �%�%��� ����� �m�T�"�"�+�+�-�-�-�-�-r c �� � � t � � }t � � }|j D �]n}t |t j t j f� � r.|j D ]%}|j p|j }|� |� � �&nUt |t j � � r;|j |v rt d|j z |� � �|� |j � � t |t j � � s��|j D ]�}t |t j � � st d|� � �|j � j v rt d|j z |� � �t |j t j � � r|j j |v r��� j � |j � � ����p|j D �]�}t |t j � � �r>t' � fd�|j D � � � � r|� |� � �Tt � � � _ t- � � � � � |j � � � � }|j D ]�}t |t j � � sJ �|� t j |j t j g g dg g dg � � t j |�� � gg dd� � � � t5 j |d j d t5 j � � � � ����\� � t: |� � � _ |� � � |� � � � ���� xj t? � j � � z c_ ||_ |S )z@Turn globals assignment to functionDef and visit function defs. z$Multiple top-level definition of %s.z&Top-level assignment to an expression.c 3 �h �K � | ],}t |t j � � o |j �j vV � �-d S r )� isinstancer �Name�idr+ )r �tr s �r � <genexpr>z-ExpandGlobals.visit_Module.<locals>.<genexpr>` sQ �� � � � .� .� � "�!�S�X�.�.�M�1�4�t�~�3M� .� .� .� .� .� .r N)�value���r ) �listr) �bodyr1 r �Import� ImportFrom�names�asname�name�add� FunctionDefr �Assign�targetsr2 r3 r+ r6 �all�appendr* r r � arguments�Returnr � StaticReturn�gatherr �update�bool) r r � module_body�symbols�stmt�aliasr> �target� cst_values ` r � visit_ModulezExpandGlobals.visit_Module: sw �� ��f�f� ��%�%���I� .� .�D��$���S�^� <�=�=� +� !�Z� &� &�E� �<�5�5�:�D��K�K��%�%�%�%�&� �D�#�/�2�2� +��9��'�'�,�>���J� � � � � �K�K�� �*�*�*� �d�C�J�/�/� �� �,� .� .��!�&�#�(�3�3� �,�@� � � � � �9���.�.�,�>���J� � � � � �d�j�#�(�3�3� !��z�}��/�/� ���"�"�6�9�-�-�-�-� .� �I� 5� 5�D��$�� �+�+� 5�� .� .� .� .� $� �.� .� .� .� .� � �&�&�t�,�,�,� �"%�%�%��� -� /� /� 5� 5�d�j�j���6L�6L� M� M� � "�l� :� :�F�%�f�c�h�7�7�7�7�7� �&�&���� �(+� �b�"�d�B�68�$��)D� )D�),��)�)D�)D�)D�(E�(*�D�$� 8� 8�9� 9� 9� �L��R� �!5�a�!8�!)�!6�!8�!8�:� :� :� :� :� #'�+�+�)�4�#1� #1����"�"�4�:�:�d�#3�#3�4�4�4�4� � � �t�D�N�+�+�+� � � �� �� r c � � t |j t j � � r:|j | j vr,|j | j v r d| _ t j |g g �� � S |S )z� Turn global variable used not shadows to function call. We check it is a name from an assignment as import or functions use should not be turn into call. T)�func�args�keywords) r1 �ctxr r r3 r* r+ rI r r s r � visit_NamezExpandGlobals.visit_Name{ sd � � �t�x��� *� *� 2���t��.�.���4�>�)�)� �D�K��8��!#�b�2� 2� 2� 2�� r )r" r# r$ r% r- rQ rW � __classcell__)r. s @r r' r' sa �� � � � � �� �$.� .� .� .� .� ?� ?� ?�B � � � � � � r r' )r% �pythran.analysesr �pythran.passmanagerr �pythran.syntaxr � pythran.utilsr r r �gastr �NodeTransformerr r' r r r �<module>r_ s� ��� � 3� 2� 2� 2� 2� 2� .� .� .� .� .� .� -� -� -� -� -� -� &� &� &� &� &� &� � � � � � � � � � � � � � � ��+� � � �"h� h� h� h� h�N� h� h� h� h� hr
Save File
Cancel