o 8ήc"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 .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@s`eZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ dS)BaseGeneratorLowerz5 Base support class for lowering generators. cCsr|j|_|j|_|j|_|j|_|j|_|j|_||_t |j|j|j|jj |_ |j |jj|_i|_dSN)contextrlibrary call_convrgenerator_infogeninfoget_generator_typerrr"r gendescget_data_packerr arg_packer resume_blocksr lowerr r r!__init__5s  zBaseGeneratorLower.__init__cCst||ddS)Nrr gep_inboundsr buildergenptrr r r! get_args_ptrEszBaseGeneratorLower.get_args_ptrcCstj||ddddS)Nrzgen.resume_indexnamer>r@r r r!get_resume_index_ptrH z'BaseGeneratorLower.get_resume_index_ptrcCstj||ddddS)Nrz gen.staterDr>r@r r r! get_state_ptrLrGz 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=rHz# low_init_func increfNz# low_init_func before return)setup_functionrrAr0insert_generatorrr6r1extract_function_arguments pre_lowerget_return_type get_constantrint32elements debug_print enable_nrtzipr fnargsnrtincrefr8as_datarrmake_anonymous_structbox_generator_structr2 return_value post_lower) r r;rAretty resume_indexargstystatestyargtyargvalargsval statesval gen_structretvalr r r!lower_init_funcPs6        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)rJr6rRformatrr rrAfunctionr2 get_argumentsr8 load_intorCrUrFresume_index_ptrrI gen_state_ptrappend_basic_blocklower_function_bodyposition_at_endreturn_stop_iterationblkmapfirstblkr9switchloaditemsadd_casebranch) r r;rArkrBprologueentry_block_tail stop_block first_blockrvindexblockr 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&rprbitcastrlower_finalize_func_body)r r;fntyrk entry_blockrAgenptrtyrBr 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)rrntypepointeerAstorer2rs)r r;indexvalr r r!return_from_generatorsz(BaseGeneratorLower.return_from_generatorcCs0d|f}|j|}|j|||j|<dS)Nzgenerator_resume%d)rkrprArrr9)r r;r block_namerr r r!create_resumption_blocks   z*BaseGeneratorLower.create_resumption_blockcCs"tjr|j|d|dSdS)Nz DEBUGJIT: {0})r DEBUG_JITr0rRrj)r rAmsgr r r!rRszBaseGeneratorLower.debug_printN)r'r(r)r*r<rCrFrIrgrrrrrRr r r r!r.0s1. r.c@(eZdZdZddZddZddZdS) GeneratorLowerz9 Support class for lowering nopython generators. cCs|jjSr/)rrr%r r r!r5sz!GeneratorLower.get_generator_typecCs|Sr/r )r r;rer r r!rZsz#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) rRr0rSrCr8rwrVdecrefret_void)r rArBargs_ptrtyvalr r r!rs    z'GeneratorLower.lower_finalize_func_bodyN)r'r(r)r*r5rZrr 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_valuerApyapifrom_native_generatorrenvarg)r r;regen_ptrr r r!rZsz%PyGeneratorLower.box_generator_structcCs |jt|jjjd|jdS)zo NULL-initialize all generator state variables, to avoid spurious decref's on cleanup. N)rArrrorrr:r r r!init_generator_statesz%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_apirFrw icmp_signedrrr if_unlikelyrIrangerrrr? unpack_valuerr) r rArBrrnr^ need_cleanupro state_index state_slotrrr r r!r s$       z)PyGeneratorLower.lower_finalize_func_bodyN)r'r(r)r*r5rZrrr 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).0vr;r r! :sz'LowerYield.__init__..) r;r0rAgenlowerrrornypinst live_varslive_var_indices)r r; yield_pointrr rr!r<.s      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)r;rRrTrrrr?rArorrtypeof _alloca_varloadvarr0rSrVrW pack_valuerrnrrrrr)r rrErrfetyperrr r r!lower_yield_suspend=s"      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)rrr;rrrRrTrrrr?rArorrr0rstorevarrSrVr)r rrErrrr r r!lower_yield_resumeVs   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 $!=