ó \ K]c@@sdZddlmZmZddlmZmZddlmZm Z ddl m Z m Z m Z ddlmZd„Zd„Zed d ƒZd „Zd „Zd „Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„ZdS(s& Implement transformation on Numba IR i(tabsolute_importtprint_function(t namedtuplet defaultdict(tcompute_cfg_from_blockstfind_top_level_loops(tirterrorstir_utils(tcompute_use_defsc@s‘‡fd†}d„}‡fd†}g}x]tˆƒD]O}||ƒr:||ƒr:||ƒr:ˆjƒ|jkr:|j|ƒq:q:W|S(sE Returns a list of loops that are candidate for loop lifting c@s`tƒ}xD|jD]9}td„ˆj|ƒDƒƒ}|sBtS||O}qWt|ƒdkS(s)all exits must point to the same locationcs@s|]\}}|VqdS(N((t.0txt_((s/lib/python2.7/site-packages/numba/transforms.pys si(tsettexitst successorstFalsetlen(tlooptoutedgestktsuccs(tcfg(s/lib/python2.7/site-packages/numba/transforms.pytsame_exit_points cS@st|jƒdkS(sthere is one entryi(Rtentries(R((s/lib/python2.7/site-packages/numba/transforms.pyt one_entry sc@sŒt|jƒt|jƒBt|jƒB}x\tˆj|ƒD]H}x?|jD]4}t|tjƒrLt|j tj ƒr€t SqLqLWq<Wt S(s!cannot have yield inside the loop( R tbodyRRtmapt __getitem__t isinstanceRtAssigntvaluetYieldRtTrue(Rtinsiderstblktinst(tblocks(s/lib/python2.7/site-packages/numba/transforms.pyt cannot_yield$s)(Rt entry_pointRtappend(RR%RRR&t candidatesR((R%Rs/lib/python2.7/site-packages/numba/transforms.pyt _extract_loop_lifting_candidatess  $cC@sã||}||}i}x|D]}||||§sR%t arg_namest arg_counttforce_non_generator(tnumba.dispatcherRWRR RRRtdictRVtderivettupleR3RR!ROR0R4R1(tfunc_irRSR%t typingctxt targetctxtflagstlocalsRWRt loopblockkeysR5t lifted_irRMt callblockR((R%s/lib/python2.7/site-packages/numba/transforms.pyt_loop_lift_modify_blocksœs$ )       c C@s‘|jjƒ}t|ƒ}t|||jjƒ}g}x9|D]1} t|| |||||ƒ} |j| ƒq@W|jd|ƒ} | |fS(sš Loop lifting transformation. Given a interpreter `func_ir` returns a 2 tuple of `(toplevel_interp, [loop0_interp, loop1_interp, ....])` R%( R%RXRREtvariable_lifetimeR/RhR(R^( R`RaRbRcRdR%RRAR@RStliftedtmain((s/lib/python2.7/site-packages/numba/transforms.pyt loop_lifting¿s    c@sˆtˆƒ‰ˆjƒ‰‡fd†‰‡fd†‰‡‡fd†}d„‰‡‡‡fd†}x|ƒD]}||ƒqpWˆS(s5 Rewrite loops that have multiple backedges. c@stˆjƒƒdS(Ni(tmaxtkeys((t newblocks(s/lib/python2.7/site-packages/numba/transforms.pyt new_block_idÝsc@scd}xV|jD]K}ˆ|}|jjƒ}|j|kr|d7}|dkr[tSqqWtS(Nii(Rt terminatort get_targetstheaderR!R(RtcountRR#tedges(R%(s/lib/python2.7/site-packages/numba/transforms.pythas_multiple_backedgesàs    c3@s5x.ˆjƒjƒD]}ˆ|ƒr|VqqWdS(N(R@R,(tlp(RRv(s/lib/python2.7/site-packages/numba/transforms.pyt#yield_loops_with_multiple_backedgesís c @s­‡‡fd†}t|tjƒr^tjd|jd||jƒd||jƒd|jƒSt|tjƒr’tjd||jƒd|jƒS|j ƒ s¥t ‚|SdS(Nc@s|ˆkrˆS|S(N((ttarget(tdsttsrc(s/lib/python2.7/site-packages/numba/transforms.pytreplaceóstcondttruebrtfalsebrRGRy( RRtBranchR}R~RRGtJumpRyRrtAssertionError(ttermR{RzR|((RzR{s/lib/python2.7/site-packages/numba/transforms.pytreplace_targetòs "c@sÒ|j}ˆƒ}xe|jD]Z}ˆ|}||jjƒkr|jƒ}ˆ|j||ƒ|jd<|ˆ||j}||ƒ\}}t|dƒsŸtjddˆˆjƒ‚n||fSq>Wtjddˆˆjƒ‚dS( s7Get the global object used for the context manager sIllegal use of context-manager.c@s ˆj|ƒS(s#Get the definition given a variable(tget_definition(tvar(R`(s/lib/python2.7/site-packages/numba/transforms.pyt get_var_dfnbsc@sˆj|ƒ}t|tjƒr‹|jdkr‹g|jD]}ˆ|ƒ^q:}‡fd†|jDƒ}i|d6|d6}|j}nd}t j t j ˆ|ƒ}|tj kr×t jddˆˆjƒ‚n|dkrþt jˆd|jƒ‚n||fS(s–Return the context-manager object and extra info. The extra contains the arguments if the context-manager is used as a call. tcallc@s%i|]\}}ˆ|ƒ|“qS(((R Rtv(R°(s/lib/python2.7/site-packages/numba/transforms.pys ps targsR•s*Undefined variable used as context managerRGN(R®RRtExprtopR³tkwstfunctNoneRtguardtfind_global_valuet UNDEFINEDRt CompilerErrorRG(tvar_reftdfnR R³R¶Rªtctxobj(t_illegal_cm_msgR¥R%R`R°(s/lib/python2.7/site-packages/numba/transforms.pytget_ctxmgr_objfs!"  R¡s"Unsupported context manager in useRGsmalformed with-context usageN( RRRt EnterWithtcontextmanagerthasattrRR¼RG(R`R%R¥RÁtstmtR½R¿Rª((RÀR¥R%R`R°s/lib/python2.7/site-packages/numba/transforms.pyR ]s cC@sÔttƒ}x'|jD]}|t|ƒcd7s, +      #  F  9 9