← Back
Editing: normalize_ifelse.cpython-311.pyc
� cR�a� � �F � d Z ddlmZ ddlmZ ddlZ G d� de� � ZdS )z: NormalizeIfElse transform early exit in if into if-else. � )� Ancestors)�TransformationNc �. � � e Zd ZdZ� fd�Zd� Zd� Z� xZS )�NormalizeIfElsea� >>> import gast as ast >>> from pythran import passmanager, backend >>> node = ast.parse(""" ... def foo(y): ... if y: return 1 ... return 2""") >>> pm = passmanager.PassManager("test") >>> _, node = pm.apply(NormalizeIfElse, node) >>> print(pm.dump(backend.Python, node)) def foo(y): if y: return 1 else: return 2 >>> node = ast.parse(""" ... def foo(y): ... if y: ... z = y + 1 ... if z: ... return 1 ... else: ... return 3 ... return 2""") >>> pm = passmanager.PassManager("test") >>> _, node = pm.apply(NormalizeIfElse, node) >>> print(pm.dump(backend.Python, node)) def foo(y): if y: z = (y + 1) if z: return 1 else: return 3 else: return 2 c �b �� t t | � � � t � � d S )N)�superr �__init__r )�self� __class__s ��J/usr/lib/python3/dist-packages/pythran/transformations/normalize_ifelse.pyr zNormalizeIfElse.__init__2 s' �� � �o�t�$�$�-�-�i�8�8�8�8�8� c �2 � t |t j t j t j f� � rdS t |t j � � rI| � |j d � � sdS |j o | � |j d � � S d S )NT���F)� isinstance�ast�Return�Break�If� check_lasts�body�orelse)r �nodes r r zNormalizeIfElse.check_lasts5 s� � � �d�S�Z���C�J� ?� @� @� ��4� �d�C�F� #� #� E��#�#�D�I�b�M�2�2� � �u��;�D�4�#3�#3�D�K��O�#D�#D� D� E� Er c �� � | � |� � | � |j d � � s|S | j | d }dD ]�} t ||� � }|� |� � }|t |� � dz k r|c S |j sg |_ |j � ||dz d � � � g ||dz d �< d| _ |c S # t $ r Y ��t $ r Y ��w xY w|S )Nr )r r � finalbody� T) � generic_visitr r � ancestors�getattr�index�lenr �extend�update� ValueError�AttributeError)r r �parent�attrr r s r �visit_IfzNormalizeIfElse.visit_If= s, � � ���4� � � �� � �� �"� �.�.� ��K�� ��%�b�)��3� � �D� � �v�t�,�,�� � � �4�(�(���C��I�I��M�)�)� �K�K�K��{� %�"$�D�K�� �"�"�4��� � � �#3�4�4�4�#%��U�Q�Y�Z�Z� � "�� �� � � �� � � � ���!� � � ��� ����� s �<C�AC� C/�# C/�.C/)�__name__� __module__� __qualname__�__doc__r r r' � __classcell__)r s @r r r sd �� � � � � �&� &�P9� 9� 9� 9� 9�E� E� E�� � � � � � r r )r+ �pythran.analysesr �pythran.passmanagerr �gastr r � r r �<module>r1 sy �� @� @� &� &� &� &� &� &� .� .� .� .� .� .� � � � �I� I� I� I� I�n� I� I� I� I� Ir
Save File
Cancel