U ?h`7@sdZddlZddlmZmZddlmZmZmZddl m Z Gddde Z Gdd d e Z Gd d d e ZGd d d e ZGddde ZdS)z" Support for lowering generators. N)Constant IRBuilder)typesconfigcgutils)FunctionDescriptorc@s,eZdZdZdZeddZeddZdS)GeneratorDescriptorz9 The descriptor for a generator's next function. c Cslt|tjst|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.nextF)argtypesmanglerinlineenv_name) isinstancer GeneratorAssertionError yield_typequalname unique_namenativemodnamedoctypemapZ calltypeskwsr) clsfunc_irfndescgentyper restypeargsr rrselfr r G/var/www/html/venv/lib/python3.8/site-packages/numba/core/generators.pyfrom_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_namer r r r!llvm_finalizer_name'sz'GeneratorDescriptor.llvm_finalizer_nameN) __name__ __module__ __qualname____doc__ __slots__ classmethodr"propertyr$r r r r!r s  rc@sleZdZdZddZeddZddZdd Zd d Z d d Z ddZ ddZ ddZ ddZddZdS)BaseGeneratorLowerz5 Base support class for lowering generators. cCsp|j|_|j|_|j|_|j|_||_|j|_||_t |j|j|j|jj |_ |j |jj|_i|_dSN)contextrlibraryrlowergenerator_infogeninfoget_generator_typerrr"r gendescZget_data_packerr arg_packer resume_blocksr r0r r r!__init__5s zBaseGeneratorLower.__init__cCs|jjSr-)r0 call_convr#r r r!r9EszBaseGeneratorLower.call_convcCst||ddS)Nrr gep_inboundsr buildergenptrr r r! get_args_ptrIszBaseGeneratorLower.get_args_ptrcCstj||ddddS)Nrzgen.resume_indexnamer;r=r r r!get_resume_index_ptrLs z'BaseGeneratorLower.get_resume_index_ptrcCstj||ddddS)Nrz gen.staterAr;r=r r r! get_state_ptrPs z BaseGeneratorLower.get_state_ptrc Cs||j|j}|j|j|j|jg|| |j |j}|j t j d}|jd}|jd}|d|jjrt|jj|jD]\}}|jj|||q|j||j} t|d} t||| | g|} ||| } |d|j|| |dS)z Lower the generator's initialization function (which will fill up the passed-by-reference generator structure). rr:rDz# low_init_func increfNz# low_init_func before return)setup_functionrr>r.Zinsert_generatorrr4r/Zextract_function_argumentsZ pre_lowerZget_return_typeZ get_constantrZint32elements debug_print enable_nrtzipr fnargsnrtincrefr5Zas_datarrZmake_anonymous_structbox_generator_structr9Z return_valueZ post_lower) r r0r>Zretty resume_indexZargstyZstatestyZargtyargvalZargsvalZ statesval gen_structretvalr r r!lower_init_funcTs6       z"BaseGeneratorLower.lower_init_funcc Cs$||j|d|jj|jjd|jks6t|j}|j }|j |\}|j |||||j||||_||||_|d}|}|d}|||j ||||j|j}|jd<|||j|} |jD]\} } | | | q||||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) rFr4rHformatrr rrr>functionr9Z get_argumentsr5Z load_intor@rKrCresume_index_ptrrE gen_state_ptrappend_basic_blockZlower_function_bodyposition_at_endreturn_stop_iterationZblkmapZfirstblkr6switchloaditemsZadd_casebranch) r r0r>rUr?ZprologueZentry_block_tailZ stop_blockZ first_blockr[indexblockr r r!lower_next_funcs4        z"BaseGeneratorLower.lower_next_funccCsvtjtj|j|jg}t|j ||j j }| d}t |}|j|j}||jd|}|||dS)z2 Lower the generator's finalizer. entryrN)llvmliteZir FunctionTypeZVoidTyper.Zget_value_typerrZget_or_insert_functionmoduler4r$rXrZbitcastrlower_finalize_func_body)r r0ZfntyrUZ entry_blockr>Zgenptrtyr?r r r!lower_finalize_funcs z&BaseGeneratorLower.lower_finalize_funccCs2t|jjjd}|j||j|j|jdS)zY Emit a StopIteration at generator end and mark the generator exhausted. N)rrVtypepointeer>storer9rZ)r r0indexvalr r r!return_from_generatorsz(BaseGeneratorLower.return_from_generatorcCs0d|f}|j|}|j|||j|<dS)Nzgenerator_resume%d)rUrXr>rYr6)r r0r_Z block_namer`r r r!create_resumption_blocks   z*BaseGeneratorLower.create_resumption_blockcCstjr|j|d|dS)Nz DEBUGJIT: {0})rZ DEBUG_JITr.rHrT)r r>msgr r r!rHszBaseGeneratorLower.debug_printN)r%r&r'r(r8r+r9r@rCrErSrargrmrnrHr r r r!r,0s 1.r,c@s(eZdZdZddZddZddZdS) GeneratorLowerz9 Support class for lowering nopython generators. cCs|jjSr-)rrr#r r r!r3sz!GeneratorLower.get_generator_typecCs|Sr-r )r r0rQr r r!rNsz#GeneratorLower.box_generator_structcCsb||d|jjrJ|||}|j||D]\}}|jj|||q.||d|dS)g Lower the body of the generator's finalizer: decref all live state variables. z# generator: finalizez# generator: finalize endN) rHr.rIr@r5r\rLdecrefret_void)r r>r?Zargs_ptrtyvalr r r!rfs   z'GeneratorLower.lower_finalize_func_bodyN)r%r&r'r(r3rNrfr r r r!rpsrpc@s0eZdZdZddZddZddZdd 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) rrrZfunc_idfuncZpyobject arg_countlenr2 state_varsr#r r r!r3sz#PyGeneratorLower.get_generator_typecCs"t|j|}|j||j|jS)z> Box the raw *gen_struct* as a Python object. )rZalloca_once_valuer>pyapiZfrom_native_generatorrZenvarg)r r0rQZgen_ptrr r r!rNsz%PyGeneratorLower.box_generator_structcCs |jt|jjjd|jdS)zo NULL-initialize all generator state variables, to avoid spurious decref's on cleanup. N)r>rkrrWrirjr7r r r!init_generator_state sz%PyGeneratorLower.init_generator_statec Cs|j|}|||}||}|d|t|jd}t||^| ||}t t |j j D]:}t||d|} |j j |} |j|| | } || qbW5QRX|dS)rq>rN)r.Zget_python_apirCr\Z icmp_signedrrirZ if_unlikelyrErangerzrrwr< unpack_valuerrrs) r r>r?r|rVrOZ need_cleanuprW state_index state_slotrtrur r r!rfs&      z)PyGeneratorLower.lower_finalize_func_bodyN)r%r&r'r(r3rNr}rfr r r r!rvs  rvc@s(eZdZdZddZddZddZdS) LowerYieldz> Support class for lowering a particular yield point. csj|_j|_j|_j|_|jj|_|jj|_|jj|_||_|jj|_||_ fdd|D|_ dS)Ncsg|]}jj|qSr )r1r{r_).0vr0r r! >sz'LowerYield.__init__..) r0r.r>genlowerrrWrVZypinst live_varslive_var_indices)r r0Z yield_pointrr rr!r82s     zLowerYield.__init__cCs|jdt|j|jD]~\}}t|j|jd|}|j j |}|j |}|j |||j |}|jjr|jj|j|||j|j|||qt|jjj|jj}|j||j|jddS)Nz# generator suspendrz# generator suspend end)r0rHrJrrrr<r>rWrrwZtypeofZ _alloca_varZloadvarr.rIrLrMZ pack_valuerrVrirjrr_rk)r rrBrrtZfetyperurlr r r!lower_yield_suspendAs$      zLowerYield.lower_yield_suspendcCs|j|j|jj|jdt|j|jD]d\}}t |j |j d|}|j j|}|j|j ||}|j|||jjr.|jj|j ||q.|jddS)Nz# generator resumerz# generator resume end)rrnr0rr_rHrJrrrr<r>rWrrwr.rZstorevarrIrLrr)r rrBrrtrur r r!lower_yield_resumeZs   zLowerYield.lower_yield_resumeN)r%r&r'r(r8rrr r r r!r-sr)r(Z llvmlite.irrcrrZ numba.corerrrZnumba.core.funcdescrrobjectr,rprvrr r r r!s $%=