o @ad@sVdZddlmZmZddlmZddlZddlZGdddeZ Gdddej Z dS) z! Inlining inline functions body. ) InlinableAliases)TransformationNcsXeZdZdZfddZddZeZeZeZeZ eZ eZ eZ eZ eZeZddZZS)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))) cs*d|_t|_d|_tt|ttdS)z6 fun : Function {name :body} for inlinable functions. FrN) updatelistdefs call_countsuperr__init__rr)self __class__@/usr/lib/python3/dist-packages/pythran/optimizations/inlining.pyr %szInlining.__init__cCs8|jptt}|_|||j|}|_||gS)z3 Add new variable definition before the Statement. )rrZ generic_visit)r nodeZ save_defsZnew_defsrrr visit_Stmt,s  zInlining.visit_Stmtc Cs$|j|j}t|dkrtt|}t|tjr|j|j vrd|_ t |j |j}t }|j}||jjt|jt|jjd7}t|jj|D]4\}}d|j|j|j} tj| tddd} |jtj| g|ddtj| tddd||j<qJ|jd7_t||jdS|S)z~ Replace function call by inlined function's body. We can inline if it aliases on only one function. TNz__pythran_inline{}{}{})idctx annotation type_comment)targetsvaluerr)aliasesfunclennextiter isinstanceastZ FunctionDefnameZ inlinablercopyZdeepcopydictargsdefaultszipformatrr NameZStorerappendZAssignZLoadInlinervisitbody) r rZ func_aliasesZ function_defZ to_inlineZ arg_to_valuevaluesZarg_funZarg_callZv_nameZnew_varrrr visit_Call>sD       zInlining.visit_Call)__name__ __module__ __qualname____doc__r r visit_ReturnZ visit_AssignZvisit_AugAssignZ visit_PrintZ visit_ForZ visit_WhileZvisit_IfZ visit_WithZ visit_AssertZ visit_Exprr. __classcell__rrr rr s rcs0eZdZdZfddZddZddZZS)r*z> Helper transform that performed inlined body transformation. cs||_tt|dS)z; match : {original_variable_name : Arguments use on call}. N)matchr r*r )r r5r rrr fszInliner.__init__cCs|j|j|S)z0 Transform name from match values if available. )r5getrr rrrr visit_NamekszInliner.visit_NamecCs ||jS)z% Remove return keyword after inline. )r+rr7rrrr3os zInliner.visit_Return)r/r0r1r2r r8r3r4rrr rr*bs  r*) r2Zpythran.analysesrrZpythran.passmanagerrZgastr r"rZNodeTransformerr*rrrrs X