| Server IP : 217.160.0.135 / Your IP : 216.73.217.85 Web Server : Apache System : Linux www 6.18.52-i1-ampere #1203 SMP Mon Sep 14 18:29:59 CEST 2026 aarch64 User : sws1074145052 ( 1074145052) PHP Version : 8.3.32 Disable Function : NONE MySQL : OFF | cURL : ON | WGET : ON | Perl : ON | Python : OFF | Sudo : OFF | Pkexec : OFF Directory : /lib/python3/dist-packages/pythran/optimizations/__pycache__/ |
Upload File : |
�
cR�a� � �j � d Z ddlmZ ddlmZmZ ddlmZ ddlm Z ddl
Z ddl Z G d� de� � Z
dS ) z) Replaces **2 by a call to numpy.square. � )�Transformation)�
ASTMatcher�AST_any)�mangle)�isnumNc
� � e Zd ZdZ ej e� � ej � � ej dd� � � � Z ej
ej ej e
d� � ej � � dd� � d ej � � � � e� � ej dd� � gg � � Zd� Zd� Zd� Zd � Zd
� Zd � ZdS ) �Squarea�
Replaces **2 by a call to numpy.square.
>>> import gast as ast
>>> from pythran import passmanager, backend
>>> node = ast.parse('a**2')
>>> pm = passmanager.PassManager("test")
>>> _, node = pm.apply(Square, node)
>>> print(pm.dump(backend.Python, node))
import numpy as __pythran_import_numpy
__pythran_import_numpy.square(a)
>>> node = ast.parse('__pythran_import_numpy.power(a,2)')
>>> pm = passmanager.PassManager("test")
>>> _, node = pm.apply(Square, node)
>>> print(pm.dump(backend.Python, node))
import numpy as __pythran_import_numpy
__pythran_import_numpy.square(a)
� N�numpy�powerc �. � t j | � � d S )N)r �__init__)�selfs �>/usr/lib/python3/dist-packages/pythran/optimizations/square.pyr zSquare.__init__) s � �� ��%�%�%�%�%� c � � dx| _ | _ t j t d� � t j � � d d � � }t j t j |dt j � � � � |gg � � S )NTr �square)�update� need_import�ast�Namer �Load�Call� Attribute)r �value� module_names r �replacezSquare.replace, sc � �)-�-�� �d�&��h�v�g�����
�
�D�$�G�G� ��x��
�k�8�S�X�Z�Z�H�H� ���%� %� %r c �� � d| _ | � |� � | j rTt j dt d� � �� � }t j |g�� � }|j � d|� � |S )NFr )�name�asname)�namesr )r �
generic_visitr �aliasr �Import�body�insert)r �node� import_alias�importIts r � visit_ModulezSquare.visit_Module2 st � � ��� ���4� � � � � � *� �9�'�&��/�/�J�J�J�L��z� ��7�7�7�H� �I� � �Q�� )� )� )�� r c �$ � |dk rt j dd � � S |dk r|S | � |� � }| � ||dz � � }|dz r9t j |t j � � t
j |� � � � S |S )Nr � )r �Constantr �
expand_pow�BinOp�Mult�copy�deepcopy)r r' �n� node_square�node_pows r r. zSquare.expand_pow; s� � � ��6�6��<��4�(�(� (�
�!�V�V��K� �,�,�t�,�,�K����{�A��F�;�;�H��1�u�
��y��3�8�:�:�t�}�T�7J�7J�K�K�K� �r c � � | � |� � t t j � � � |� � r| � |j � � S t |j t j
� � rVt |j � � rB|j j
}t |� � |k r!|dk r| � |j |� � S |S |S �Nr )r" r r � POW_PATTERN�searchr �left�
isinstance�opr �Powr �rightr �intr. )r r' r3 s r � visit_BinOpzSquare.visit_BinOpH s� � � ���4� � � � �f�(� )� )� 0� 0�� 6� 6� ��<�<�� �*�*� *�
�����
)�
)� �e�D�J�.?�.?� ��
� �A��1�v�v��{�{�q�1�u�u����t�y�!�4�4�4�� ��Kr c �� � | � |� � t t j � � � |� � r | � |j d � � S |S r7 )r" r r �
POWER_PATTERNr9 r �args)r r' s r �
visit_CallzSquare.visit_CallU sT � � ���4� � � � �f�*� +� +� 2� 2�4� 8� 8� ��<�<�� �!� �-�-� -��Kr )�__name__�
__module__� __qualname__�__doc__r r/ r r= r- r8 r r r r r rB r r r* r. r@ rD � r r r r s" � � � � � �� �& �#�)�G�G�I�I�w�s�w�y�y�,�#�,�q�$�2G�2G�H�H�K� �C�H���
�h�c�h�v�v�g������
�
�D�$�G�G� � �c�h�j�j� "� "�
����L�C�L��D�)�)�*�
� � �M�&� &� &�%� %� %� � � � � � � � � �� � � � r r )rH �pythran.passmanagerr � pythran.analyses.ast_matcherr r �pythran.conversionr �
pythran.utilsr �gastr r1 r rI r r �<module>rO s� �� /� /� .� .� .� .� .� .� <� <� <� <� <� <� <� <� %� %� %� %� %� %� � � � � � � � � � � � � � �N� N� N� N� N�^� N� N� N� N� Nr