← Back
Editing: list_to_tuple.cpython-311.pyc
� cR�a` � � � d Z ddlmZmZ ddlmZ ddlmZ ddlm Z ddl Z ed d ed d ed d ed d ed d ed d fZ d� Z d� Z G d� de� � ZdS )zH ListToTuple transforms some List node into more Efficient Tuple nodes. � )�Aliases� FixedSizeList)�MODULES)�Transformation)� path_to_attrN�numpy�full�ones�zeros�empty� concatenate�builtins�tuplec �6 � t | t j � � S �N)� isinstance�ast�List��nodes �E/usr/lib/python3/dist-packages/pythran/optimizations/list_to_tuple.py�islistr s � � �d�C�H� %� %�%� c �@ � t j | j | j � � S r )r �Tuple�elts�ctxr s r �totupler s � � �9�T�Y��� )� )�)r c �@ � � e Zd ZdZ� fd�Zd� Zd� Zd� Zd� Zd� Z � xZ S )� ListToTuplea� Replace list nodes by tuple nodes when possible >>> import gast as ast >>> from pythran import passmanager, backend >>> node = ast.parse("def foo(n): import numpy; return numpy.ones([n,n])") >>> pm = passmanager.PassManager("test") >>> _, node = pm.apply(ListToTuple, node) >>> print(pm.dump(backend.Python, node)) def foo(n): import numpy return numpy.ones((n, n)) c �| �� d| _ t t | � � � t t � � d S )NF)�update�superr �__init__r r )�self� __class__s �r r$ zListToTuple.__init__* s0 �� ��� � �k�4� � �)�)�'�=�A�A�A�A�Ar c � � t |j � � s| � |� � S t |j � � |_ d| _ | � |� � S )NT)r �value� generic_visitr r" �r% r s r �visit_AugAssignzListToTuple.visit_AugAssign. sT � ��d�j�!�!� ,��%�%�d�+�+� +� �T�Z�(�(�� ��� ��!�!�$�'�'�'r c �> � | j � |j t � � � � }|� t � � rCt |j d � � r)d| _ t |j d � � |j d< | � |� � S )Nr T) �aliases�get�func�set�issubset�patternsr �argsr" r r) )r% r � func_aliasess r � visit_CallzListToTuple.visit_Call5 s � ��|�'�'�� �3�5�5�9�9� � � � �� *� *� 5��d�i��l�#�#� 5� "�� � &�t�y��|� 4� 4�� �!� ��!�!�$�'�'�'r c �l � | � |� � || j v r| � |� � S |S r )r) �fixed_size_list�convertr* s r � visit_ListzListToTuple.visit_List= s; � � ���4� � � � �4�'� '� '��<�<��%�%� %��Kr c � � | � |� � |j | j vr|S | � |j � � |_ |S )a6 Replace list calls by static_list calls when possible >>> import gast as ast >>> from pythran import passmanager, backend >>> node = ast.parse("def foo(n):\n" ... " x = builtins.list(n)\n" ... " x[0] = 0\n" ... " return builtins.tuple(x)") >>> pm = passmanager.PassManager("test") >>> _, node = pm.apply(ListToTuple, node) >>> print(pm.dump(backend.Python, node)) def foo(n): x = builtins.pythran.static_list(n) x[0] = 0 return builtins.tuple(x) >>> node = ast.parse("def foo(n):\n" ... " x = builtins.list(n)\n" ... " x[0] = 0\n" ... " return x") >>> pm = passmanager.PassManager("test") >>> _, node = pm.apply(ListToTuple, node) >>> print(pm.dump(backend.Python, node)) def foo(n): x = builtins.list(n) x[0] = 0 return x )r) r( r7 r8 r* s r � visit_AssignzListToTuple.visit_AssignD sG � �< ���4� � � � �:�T� 1� 1� 1��K��\�\�$�*�-�-�� �� r c �j � d| _ t |t j � � r*|j st j g � � }nS|j d }nEt |t j � � r+t j |j t j � � � � }t j t d� � |gg � � S )NTr )r �pythran� static_list) r" r r �Callr3 r r r �Loadr r* s r r8 zListToTuple.converti s� � ��� � �d�C�H� %� %� 4��9� $��y��}�}����y��|��� ��c�h� '� '� 4��9�T�Y��� � �3�3�D��x� �%K�L�L� ���$� $� $r ) �__name__� __module__� __qualname__�__doc__r$ r+ r5 r9 r; r8 � __classcell__)r& s @r r r s� �� � � � � � � �B� B� B� B� B�(� (� (�(� (� (�� � �#� #� #�J $� $� $� $� $� $� $r r )rD �pythran.analysesr r �pythran.tablesr �pythran.passmanagerr � pythran.utilsr �gastr r2 r r r � r r �<module>rL s �� N� N� 3� 3� 3� 3� 3� 3� 3� 3� "� "� "� "� "� "� .� .� .� .� .� .� &� &� &� &� &� &� � � � � �G� �V� $� �G� �V� $� �G� �W� %� �G� �W� %� �G� �]� +� �J� �� (� ��&� &� &�*� *� *�Z$� Z$� Z$� Z$� Z$�.� Z$� Z$� Z$� Z$� Z$r
Save File
Cancel