o ۾i`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|tjsJ|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 Generator yield_typequalname unique_namenativemodnamedoctypemap calltypeskwsr) clsfunc_irfndescgentyper restypeargsr rrselfr r I/home/ubuntu/.local/lib/python3.10/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). finalize_) mangled_namer r r r!llvm_finalizer_name's z'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 gendescget_data_packerr arg_packer resume_blocksr r2r r r!__init__5s  zBaseGeneratorLower.__init__cC|jjSr/)r2 call_convr%r r r!r=EszBaseGeneratorLower.call_convcCst||ddS)Nrr gep_inboundsr buildergenptrr r r! get_args_ptrIszBaseGeneratorLower.get_args_ptrcCstj||ddddS)Nrzgen.resume_indexnamer?rAr r r!get_resume_index_ptrL z'BaseGeneratorLower.get_resume_index_ptrcCstj||ddddS)Nrz gen.staterEr?rAr r r! get_state_ptrPrHz BaseGeneratorLower.get_state_ptrc Cs||j|j}|j|j|j|jg|| |j |j}|j t j d}|jd}|jd}|d|jjrUt|jj|jD] \}}|jj|||qG|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>rIz# low_init_func increfNz# low_init_func before return)setup_functionrrBr0insert_generatorrr6r1extract_function_arguments pre_lowerget_return_type get_constantrint32elements debug_print enable_nrtzipr fnargsnrtincrefr8as_datarrmake_anonymous_structbox_generator_structr= return_value post_lower) r r2rBretty resume_indexargstystatestyargtyargvalargsval statesval gen_structretvalr r r!lower_init_funcTs6        z"BaseGeneratorLower.lower_init_funcc Cs$||j|d|jj|jjd|jksJ|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}rgenerator_prologuestop_iterationN)rKr6rSformatrr rrBfunctionr= get_argumentsr8 load_intorDrVrGresume_index_ptrrJ gen_state_ptrappend_basic_blocklower_function_bodyposition_at_endreturn_stop_iterationblkmapfirstblkr9switchloaditemsadd_casebranch) r r2rBrlrCprologueentry_block_tail stop_block first_blockrwindexblockr 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)llvmliteir FunctionTypeVoidTyper0get_value_typerrget_or_insert_functionmoduler6r&rqrbitcastrlower_finalize_func_body)r r2fntyrl entry_blockrBgenptrtyrCr 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)rrotypepointeerBstorer=rt)r r2indexvalr r r!return_from_generatorsz(BaseGeneratorLower.return_from_generatorcCs0d|f}|j|}|j|||j|<dS)Nzgenerator_resume%d)rlrqrBrsr9)r r2r block_namerr r r!create_resumption_blocks   z*BaseGeneratorLower.create_resumption_blockcCs"tjr|j|d|dSdS)Nz DEBUGJIT: {0})r DEBUG_JITr0rSrk)r rBmsgr r r!rSszBaseGeneratorLower.debug_printN)r'r(r)r*r;r-r=rDrGrJrhrrrrrSr r r r!r.0s 1. r.c@(eZdZdZddZddZddZdS) GeneratorLowerz9 Support class for lowering nopython generators. cCr<r/)rrr%r r r!r5sz!GeneratorLower.get_generator_typecCs|Sr/r )r r2rfr r r!r[sz#GeneratorLower.box_generator_structcCsb||d|jjr%|||}|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) rSr0rTrDr8rxrWdecrefret_void)r rBrCargs_ptrtyvalr r r!rs    z'GeneratorLower.lower_finalize_func_bodyN)r'r(r)r*r5r[rr r r r!rs  rc@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)gen_funcr arg_types state_types has_finalizer) rrrfunc_idfuncpyobject arg_countlenr4 state_varsr%r r r!r5sz#PyGeneratorLower.get_generator_typecCs"t|j|}|j||j|jS)z> Box the raw *gen_struct* as a Python object. )ralloca_once_valuerBpyapifrom_native_generatorrenvarg)r r2rfgen_ptrr r r!r[sz%PyGeneratorLower.box_generator_structcCs |jt|jjjd|jdS)zo NULL-initialize all generator state variables, to avoid spurious decref's on cleanup. N)rBrrrprrr:r r r!init_generator_state sz%PyGeneratorLower.init_generator_statec Cs|j|}|||}||}|d|t|jd}t||4| ||}t t |j j D]}t||d|} |j j |} |j|| | } || q1Wdn1sYwY|dS)r>rN)r0get_python_apirGrx icmp_signedrrr if_unlikelyrJrangerrrr@ unpack_valuerr) r rBrCrror_ need_cleanuprp state_index state_slotrrr r r!rs$       z)PyGeneratorLower.lower_finalize_func_bodyN)r'r(r)r*r5r[rrr r r r!rs   rc@r) LowerYieldz> Support class for lowering a particular yield point. csj|_j|_j|_j|_|jj|_|jj|_|jj|_||_|jj|_||_ fdd|D|_ dS)Ncsg|] }jj|qSr )r3rr).0vr2r r! >sz'LowerYield.__init__..) r2r0rBgenlowerrrproypinst live_varslive_var_indices)r r2 yield_pointrr rr!r;2s      zLowerYield.__init__cCs|jdt|j|jD]?\}}t|j|jd|}|j j |}|j |}|j |||j |}|jjrB|jj|j|||j|j|||q t|jjj|jj}|j||j|jddS)Nz# generator suspendrz# generator suspend end)r2rSrUrrrr@rBrprrtypeof _alloca_varloadvarr0rTrWrX pack_valuerrorrrrr)r rrFrrfetyperrr r r!lower_yield_suspendAs"      zLowerYield.lower_yield_suspendcCs|j|j|jj|jdt|j|jD]2\}}t |j |j d|}|j j|}|j|j ||}|j|||jjrI|jj|j ||q|jddS)Nz# generator resumerz# generator resume end)rrr2rrrSrUrrrr@rBrprrr0rstorevarrTrWr)r rrFrrrr r r!lower_yield_resumeZs   zLowerYield.lower_yield_resumeN)r'r(r)r*r;rrr r r r!r-s  r)r* llvmlite.irrrr numba.corerrrnumba.core.funcdescrrobjectr.rrrr r r r!s $%=