← Back
Editing: inlining.cpython-311.pyc
� cR�ad � �x � d Z ddlmZmZ ddlmZ ddlZddlZ G d� de� � Z G d� dej � � Z dS ) z! Inlining inline functions body. � )� Inlinable�Aliases)�TransformationNc �V � � e Zd ZdZ� fd�Zd� ZeZeZeZeZ eZ eZ eZ eZ eZeZd� Z� xZS )�Inlininga Inline one line functions. >>> import gast as ast >>> from pythran import passmanager, backend >>> pm = passmanager.PassManager("test") >>> node = ast.parse(''' ... def foo(a, b): ... return b + b * a ... def bar(b): ... return foo(2 * b, b) * foo(b, b)''') >>> _, node = pm.apply(Inlining, node) >>> print(pm.dump(backend.Python, node)) def foo(a, b): return (b + (b * a)) def bar(b): __pythran_inlinefooa0 = (2 * b) __pythran_inlinefoob0 = b __pythran_inlinefooa1 = b __pythran_inlinefoob1 = b return ((__pythran_inlinefoob0 + (__pythran_inlinefoob0 * __pythran_inlinefooa0)) * (__pythran_inlinefoob1 + (__pythran_inlinefoob1 * __pythran_inlinefooa1))) c � �� d| _ t � � | _ d| _ t t | � � � t t � � dS )z6 fun : Function {name :body} for inlinable functions. Fr N) �update�list�defs� call_count�superr �__init__r r )�self� __class__s ��@/usr/lib/python3/dist-packages/pythran/optimizations/inlining.pyr zInlining.__init__% sB �� ��� ��F�F�� ���� �h�� � �&�&�y�'�:�:�:�:�:� c � � | j p t � � t � � c}| _ | � |� � | j |c}| _ ||gz S )z3 Add new variable definition before the Statement. )r r � generic_visit)r �node� save_defs�new_defss r � visit_StmtzInlining.visit_Stmt, sP � � #�y� 2�D�F�F�D�F�F� � �4�9� ���4� � � � "�i����$�)��4�&� � r c � � | j |j }t |� � dk �r�t t |� � � � }t |t j � � �r�|j | j v �r�d| _ t j | j |j � � }t � � }|j }||j j t |j � � t |j j � � z d� z }t! |j j |� � D ]�\ }}d� |j |j | j � � } t j | t j � � dd�� � } | j � t j | g|d�� � � � t j | t j � � dd�� � ||j < ��| xj dz c_ t5 |� � � |j d � � S |S )z~ Replace function call by inlined function's body. We can inline if it aliases on only one function. � TNz__pythran_inline{}{}{})�id�ctx� annotation� type_comment)�targets�valuer r ) �aliases�func�len�next�iter� isinstance�ast� FunctionDef�name� inlinabler �copy�deepcopy�dict�args�defaults�zip�formatr r �Name�Storer �append�Assign�Load�Inliner�visit�body) r r � func_aliases� function_def� to_inline� arg_to_value�values�arg_fun�arg_call�v_name�new_vars r � visit_CallzInlining.visit_Call> s� � � �|�D�I�.� � �|� � �� !� !� ��\� 2� 2�3�3�L��<�� �9�9� F� �%�� �7�7� "�� � �M�$�.� �9J�*K� L� L� � #�v�v� � ����)�.�1�#�d�i�.�.�25�i�n�6I�2J�2J�3K� 3L� 3L� M� M��),�Y�^�-@�&�)I�)I� K� K�%�G�X� 5� <� <�\�=N�=D�Z�=A�_� N� N�F� "�h�&�+.�9�;�;�26�T� K� K� K�G� �I�$�$�S�Z�� �6>�<@�&B� &B� &B� C� C� C� 03�x�6�<?�H�J�J�CG�EI�0K� 0K� 0K�L���,�,� ���1�$��� �|�,�,�2�2�9�>�!�3D�E�E�E�� r )�__name__� __module__� __qualname__�__doc__r r � visit_Return� visit_Assign�visit_AugAssign� visit_Print� visit_For� visit_While�visit_If� visit_With� visit_Assert� visit_ExprrC � __classcell__�r s @r r r s� �� � � � � �� �2;� ;� ;� ;� ;�!� !� !� �L� �L� �O� �K��I� �K��H��J� �L��J�!� !� !� !� !� !� !r r c �. � � e Zd ZdZ� fd�Zd� Zd� Z� xZS )r7 z> Helper transform that performed inlined body transformation. c �d �� || _ t t | � � � � � dS )z; match : {original_variable_name : Arguments use on call}. N)�matchr r7 r )r rV r s �r r zInliner.__init__f s, �� ��� � �g�t� � �%�%�'�'�'�'�'r c �B � | j � |j |� � S )z0 Transform name from match values if available. )rV �getr �r r s r � visit_NamezInliner.visit_Namek s � ��z�~�~�d�g�t�,�,�,r c �6 � | � |j � � S )z% Remove return keyword after inline. )r8 r rY s r rH zInliner.visit_Returno s � ��z�z�$�*�%�%�%r )rD rE rF rG r rZ rH rR rS s @r r7 r7 b s\ �� � � � � �H�H�(� (� (� (� (� -� -� -�&� &� &� &� &� &� &r r7 ) rG �pythran.analysesr r �pythran.passmanagerr �gastr' r+ r �NodeTransformerr7 � r r �<module>ra s� �� '� '� /� /� /� /� /� /� /� /� .� .� .� .� .� .� � � � � � � � �U� U� U� U� U�~� U� U� U�p&� &� &� &� &�c�!� &� &� &� &� &r
Save File
Cancel