B   ›\º5ã@s dZddlmZmZmZddlmZmZmZddl m Z m Z m Z ddl mZGdd„deƒZGd d „d eƒZGd d „d eƒZGd d„deƒZGdd„deƒZdS)z" Support for lowering generators. é)Úprint_functionÚdivisionÚabsolute_import)ÚConstantÚTypeÚBuilderé)ÚcgutilsÚtypesÚconfig)ÚFunctionDescriptorc@s,eZdZdZdZedd„ƒZedd„ƒZdS)ÚGeneratorDescriptorz9 The descriptor for a generator's next function. ©c Cslt|tjƒst‚|j}dg}|f}|jd}|jd} ||j|j|| |j |j ||j ||j ||d|j d} | S)a Build a GeneratorDescriptor for the generator returned by the function described by *fndesc*, with type *gentype*. The generator inherits the env_name from the *fndesc*. All emitted functions for the generator shares the same Env. Úgenz.nextT)ÚargtypesÚmanglerZinlineÚenv_name)Ú isinstancer Ú GeneratorÚAssertionErrorÚ yield_typeÚqualnameÚ unique_nameZnativeÚmodnameÚdocZtypemapZ calltypesZkwsr) ÚclsÚfunc_irÚfndescÚgentyperÚrestypeÚargsrrrÚselfrrú/lib/python3.7/site-packages/numba/generators.pyÚfrom_generator_fndescs     z)GeneratorDescriptor.from_generator_fndesccCs d|jS)zz The LLVM name of the generator's finalizer function (if .has_finalizer is true). Z finalize_)Z mangled_name)r!rrr"Úllvm_finalizer_name'sz'GeneratorDescriptor.llvm_finalizer_nameN) Ú__name__Ú __module__Ú __qualname__Ú__doc__Ú __slots__Ú classmethodr#Úpropertyr$rrrr"r s r c@s`eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„Z dd„Z dd„Z dd„Z dS)ÚBaseGeneratorLowerz5 Base support class for lowering generators. cCsr|j|_|j|_|j|_|j|_|j|_|j|_| ¡|_t   |j|j|j|jj ¡|_ |j  |jj¡|_i|_dS)N)ÚcontextrÚlibraryÚ call_convrÚgenerator_infoÚgeninfoÚget_generator_typerr r#rÚgendescZget_data_packerrÚ arg_packerÚ resume_blocks)r!Úlowerrrr"Ú__init__5s zBaseGeneratorLower.__init__cCst ||dd¡S)Nrr)r Ú gep_inbounds)r!ÚbuilderÚgenptrrrr"Ú get_args_ptrEszBaseGeneratorLower.get_args_ptrcCstj||ddddS)Nrzgen.resume_index)Úname)r r8)r!r9r:rrr"Úget_resume_index_ptrHs z'BaseGeneratorLower.get_resume_index_ptrcCstj||ddddS)Nréz gen.state)r<)r r8)r!r9r:rrr"Ú get_state_ptrLs z BaseGeneratorLower.get_state_ptrc Cs | |j¡|j}|j |j|j|jg¡| ¡|  ¡|j  |j¡}|j  t j d¡}|jd}|jd}| d¡|jjr®x.t|jj|jƒD]\}}|jj |||¡qW|j ||j¡} t |¡} t ||| | g|¡} | || ¡} | d¡|j || ¡| ¡dS)z‰ Lower the generator's initialization function (which will fill up the passed-by-reference generator structure). rrr>z# low_init_func increfz# low_init_func before returnN) Úsetup_functionrr9r-Zinsert_generatorrr3r.Zextract_function_argumentsZ pre_lowerZget_return_typeZ get_constantr Zint32ÚelementsÚ debug_printÚ enable_nrtÚziprÚfnargsÚnrtÚincrefr4Zas_datarÚnullr Zmake_anonymous_structÚbox_generator_structr/Z return_valueZ post_lower) r!r6r9ZrettyÚ resume_indexZargstyZstatestyZargtyZargvalZargsvalZ statesvalÚ gen_structZretvalrrr"Úlower_init_funcPs0        z"BaseGeneratorLower.lower_init_funcc Cs(| |j¡| d |jj¡¡|jjd|jks6t‚|j}|j }|j   |¡\}|j   || ||¡|j¡| ||¡|_| ||¡|_| d¡}| ¡}| d¡}| |¡|j  |¡| |¡|j|j}|jd<| | |j¡|¡} x"|j ¡D]\} } |  | | ¡qøW| |¡| |¡dS)z¥ Lower the generator's next() function (which takes the passed-by-reference generator structure and returns the next yielded value). z# lower_next_func: {0}rZgenerator_prologueZstop_iterationN) r@r3rBÚformatrrrrr9Úfunctionr/Z get_argumentsr4Z load_intor;rEr=Úresume_index_ptrr?Ú gen_state_ptrÚappend_basic_blockZlower_function_bodyÚposition_at_endÚreturn_stop_iterationZblkmapZfirstblkr5ÚswitchÚloadÚitemsZadd_caseÚbranch) r!r6r9rNr:ZprologueZentry_block_tailZ stop_blockZ first_blockrTÚindexÚblockrrr"Úlower_next_funcs0        z"BaseGeneratorLower.lower_next_funccCsrt t ¡|j |j¡g¡}|jj||jj d}|  d¡}t |ƒ}|j |j¡}|  |j d|¡}| ||¡dS)z2 Lower the generator's finalizer. )r<ÚentryrN)rrNZvoidr-Zget_value_typerÚmoduleZget_or_insert_functionr3r$rQrZbitcastr Úlower_finalize_func_body)r!r6ZfntyrNZ entry_blockr9Zgenptrtyr:rrr"Úlower_finalize_func¯s  z&BaseGeneratorLower.lower_finalize_funccCs4t |jjjd¡}|j ||j¡|j |j¡dS)zY Emit a StopIteration at generator end and mark the generator exhausted. éÿÿÿÿN) rÚintrOÚtypeÚpointeer9Ústorer/rS)r!r6Úindexvalrrr"Úreturn_from_generator¾sz(BaseGeneratorLower.return_from_generatorcCs0d|f}|j |¡}|j |¡||j|<dS)Nzgenerator_resume%d)rNrQr9rRr5)r!r6rXZ block_namerYrrr"Úcreate_resumption_blockÆs   z*BaseGeneratorLower.create_resumption_blockcCstjr|j |d |¡¡dS)Nz DEBUGJIT: {0})r Z DEBUG_JITr-rBrM)r!r9Úmsgrrr"rBÌszBaseGeneratorLower.debug_printN)r%r&r'r(r7r;r=r?rLrZr^rerfrBrrrr"r,0s1.r,c@s(eZdZdZdd„Zdd„Zdd„ZdS) ÚGeneratorLowerz9 Support class for lowering nopython generators. cCs|jjS)N)rr)r!rrr"r2Õsz!GeneratorLower.get_generator_typecCs|S)Nr)r!r6rKrrr"rIØsz#GeneratorLower.box_generator_structcCsf| |d¡|jjrN| ||¡}x,|j ||¡D]\}}|jj |||¡q0W| |d¡| ¡dS)zg Lower the body of the generator's finalizer: decref all live state variables. z# generator: finalizez# generator: finalize endN) rBr-rCr;r4rUrFÚdecrefÚret_void)r!r9r:Zargs_ptrÚtyÚvalrrr"r]Ûs   z'GeneratorLower.lower_finalize_func_bodyN)r%r&r'r(r2rIr]rrrr"rhÐsrhc@s0eZdZdZdd„Zdd„Zdd„Zdd „Zd S) ÚPyGeneratorLowerz< Support class for lowering object mode generators. cCs8tj|jjjtjtjf|jjtjft|jj ƒddS)zx Compute the actual generator type (the generator function's return type is simply "pyobject"). T)Zgen_funcrZ arg_typesÚ state_typesZ has_finalizer) r rrZfunc_idÚfuncZpyobjectZ arg_countÚlenr1Ú state_vars)r!rrr"r2ñs z#PyGeneratorLower.get_generator_typecCs"t |j|¡}|j ||j|j¡S)z> Box the raw *gen_struct* as a Python object. )r Zalloca_once_valuer9ÚpyapiZfrom_native_generatorrZenvarg)r!r6rKZgen_ptrrrr"rIþsz%PyGeneratorLower.box_generator_structcCs |j t |jjj¡|j¡dS)zo NULL-initialize all generator state variables, to avoid spurious decref's on cleanup. N)r9rcrrHrPrarb)r!r6rrr"Úinit_generator_statesz%PyGeneratorLower.init_generator_statec Csº|j |¡}| ||¡}| |¡}| d|t |jd¡¡}t  ||¡b|  ||¡}xNt t |j jƒƒD]:}t ||d|¡} |j j|} |j || | ¡} | | ¡qfWWdQRX| ¡dS)zg Lower the body of the generator's finalizer: decref all live state variables. ú>rN)r-Zget_python_apir=rUZ icmp_signedrr`rar Z if_unlikelyr?Úrangerprrnr8Ú unpack_valuerirj) r!r9r:rrrOrJZ need_cleanuprPÚ state_indexÚ state_slotrkrlrrr"r] s     z)PyGeneratorLower.lower_finalize_func_bodyN)r%r&r'r(r2rIrsr]rrrr"rmìs  rmc@s(eZdZdZdd„Zdd„Zdd„ZdS) Ú LowerYieldz> Support class for lowering a particular yield point. csjˆ|_ˆj|_ˆj|_ˆj|_|jj|_|jj|_|jj|_||_|jj|_||_ ‡fdd„|Dƒ|_ dS)Ncsg|]}ˆjj |¡‘qSr)r0rqrX)Ú.0Úv)r6rr"ú :sz'LowerYield.__init__..) r6r-r9ÚgenlowerrrPrOZypÚinstÚ live_varsÚlive_var_indices)r!r6Z yield_pointrr)r6r"r7.s     zLowerYield.__init__cCsº|j d¡xvt|j|jƒD]d\}}t |j|jd|¡}|j j |}|j  |¡}|j j rl|j j |j||¡|j  |j|||¡qWt |jjj|jj¡}|j ||j¡|j d¡dS)Nz# generator suspendrz# generator suspend end)r6rBrDr€rr r8r9rPrrnZloadvarr-rCrFrGZ pack_valuerr`rOrarbr~rXrc)r!rwr<rxrkrlrdrrr"Úlower_yield_suspend=s      zLowerYield.lower_yield_suspendcCs¨|j |j|jj¡|j d¡xvt|j|jƒD]d\}}t   |j |j d|¡}|j j|}|j |j ||¡}|j ||¡|jjr0|jj |j ||¡q0W|j d¡dS)Nz# generator resumerz# generator resume end)r}rfr6r~rXrBrDr€rr r8r9rPrrnr-rvZstorevarrCrFri)r!rwr<rxrkrlrrr"Úlower_yield_resumePs   zLowerYield.lower_yield_resumeN)r%r&r'r(r7rr‚rrrr"ry)sryN)r(Z __future__rrrZllvmlite.llvmpy.corerrrÚr r r Zfuncdescr r Úobjectr,rhrmryrrrr"Ús $!=