← Back
Editing: square.cpython-311.pyc
� 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
Save File
Cancel