← Back
Editing: modindex.cpython-311.pyc
� cR�az � �v � d Z ddlmZmZmZmZ ddlmZ ddlmZ ddl m Z ddl Z ddl mZ G d� d e� � ZdS ) z+ Simplify modulo computation based on index� )� UseDefChains� Ancestors�Aliases� RangeValues)� Identifiers)�Transformation)�MODULESN)�deepcopyc �* � e Zd ZdZd� Zd� Zd� Zd� ZdS )�ModIndexa] Simplify modulo on loop index >>> import gast as ast >>> from pythran import passmanager, backend >>> pm = passmanager.PassManager("test") >>> code = """ ... def foo(x): ... y = builtins.len(x) ... for i in builtins.range(8): ... z = i % y""" >>> node = ast.parse(code) >>> _, node = pm.apply(ModIndex, node) >>> print(pm.dump(backend.Python, node)) def foo(x): y = builtins.len(x) i_m = ((0 - 1) % y) for i in builtins.range(8): i_m = (0 if ((i_m + 1) == y) else (i_m + 1)) z = i_m c � � t j | t t t t t � � t � � | _ d S )N) r �__init__r r r r r �dict� loops_mod)�selfs �@/usr/lib/python3/dist-packages/pythran/optimizations/modindex.pyr zModIndex.__init__# s3 � �� ��l�I�w� +�[� :� :� :� ������� c �\ � | j | }t |� � dk o |d j S )N� r )�use_def_chains�len�node)r r �chains r � single_defzModIndex.single_def( s* � ��#�D�)���5�z�z�Q��0�5��8�=�0r c �~ � t |j t j � � s| � |� � S t |j t j � � s| � |� � S | � |j � � }|s| � |� � S | j |j j j dk r| � |� � S t |j t j � � s| � |� � S | � |j � � }|s| � |� � S | j | d }t |t j � � s| � |� � S t |j t j � � s| � |� � S || j | v r| � |� � S d }| j |j j D ]}|t$ d d u r|}� |�| � |� � S |j j dz }d}|| j v r.d� |j j |� � }|dz }|| j v �.|j j } t- | � � dk r| d nt j dd � � } t j t j |t j � � d d � � gt j t j t7 | � � t j � � t j dd � � � � t j � � t7 |j � � � � d � � } t j t j |t j � � d d � � t j � � t j dd � � � � } t j t j |t j � � d d � � gt j t j | t j! � � gt7 |j � � g� � t j dd � � t7 | � � � � d � � } | j" �# |g � � �$ | | f� � d| _% t j |t j � � d d � � S ) Nr ����builtins�range�_mz{}_m{}r T)&� isinstance�op�ast�Mod� generic_visit�right�Namer � range_values�id�low�left� ancestors�For�iter�Call�aliases�funcr � identifiers�format�argsr �Constant�Assign�Store�BinOpr �Sub�Load�Add�IfExp�Compare�Eqr � setdefault�append�update)r r � right_def�head�loop�range_�alias�new_id�i�rargs�lower�header�incr�steps r � visit_BinOpzModIndex.visit_BinOp, s � ��$�'�3�7�+�+� ,��%�%�d�+�+� +� �$�*�c�h�/�/� ,��%�%�d�+�+� +��O�O�D�J�/�/� �� ,��%�%�d�+�+� +� � �T�Z�]� +� /�!� 3� 3��%�%�d�+�+� +� �$�)�S�X�.�.� ,��%�%�d�+�+� +����t�y�)�)��� ,��%�%�d�+�+� +� �~�d�#�B�'���$���(�(� ,��%�%�d�+�+� +��$�)�S�X�.�.� ,��%�%�d�+�+� +� �4�>�)�,� ,� ,��%�%�d�+�+� +� ���\�$�)�.�1� � �E��� �+�G�4�4�4� ���� �>��%�%�d�+�+� +� �� ��$�� ����(�(�(� �_�_�T�Y�\�1�5�5�F� ��F�A� ��(�(�(� � � �� ��J�J��N�N��a���� �Q��0E�0E����S�X�f�c�i�k�k�4�� F� F� G� �I� #� �(�5�/�/�*-�'�)�)�*-�,�q�$�*?�*?�!A� !A� !$�� � � (��� 4� 4� 6� 6� !�"� "�� �y���&�#�(�*�*�d�D�A�A� ���� � �a��.�.�0� 0�� �z�3�8�F�C�I�K�K��t�D�D�E� �)� !�k�$�+.�6�8�8�*�x�� �7K�7K�6L� N� N� !�l�1�d� 3� 3� &�t�n�n� .� .� � � �� ��!�!�$��+�+�2�2�F�D�>�B�B�B��� ��x���� � �D�$�7�7�7r c �� � | � |� � || j vr|S d� | j | D � � }d� | j | D � � }||j z |_ ||gz S )Nc � � g | ]\ }}|��S � rP )�.0�h�_s r � <listcomp>z&ModIndex.visit_For.<locals>.<listcomp>� s � �6�6�6���A�1�6�6�6r c � � g | ]\ }}|��S rP rP )rQ rS �ss r rT z&ModIndex.visit_For.<locals>.<listcomp>� s � �4�4�4�t�q�!��4�4�4r )r$ r �body)r r �headers�stepss r � visit_ForzModIndex.visit_For| sw � � ���4� � � � �t�~� %� %��K�6�6�� ��!5�6�6�6��4�4�t�~�d� 3�4�4�4���D�I�%�� ��$�� � r N)�__name__� __module__� __qualname__�__doc__r r rM rZ rP r r r r s^ � � � � � �� �, � � � 1� 1� 1�N8� N8� N8�` � � � � r r )r^ �pythran.analysesr r r r r �pythran.passmanagerr �pythran.tablesr �gastr" �copyr r rP r r �<module>rd s� �� 1� 1� J� J� J� J� J� J� J� J� J� J� J� J� (� (� (� (� (� (� .� .� .� .� .� .� "� "� "� "� "� "� � � � � � � � � � �x � x � x � x � x �~� x � x � x � x � x r
Save File
Cancel