ó \ K]c@`sÜdZddlmZmZmZddlmZmZmZddl m Z m Z m Z ddl mZdefd„ƒYZd efd „ƒYZd efd „ƒYZd efd„ƒYZdefd„ƒYZdS(s" Support for lowering generators. i(tprint_functiontdivisiontabsolute_import(tConstanttTypetBuilderi(tcgutilsttypestconfig(tFunctionDescriptortGeneratorDescriptorcB`s2eZdZdZed„ƒZed„ƒZRS(s9 The descriptor for a generator's next function. c C`s¥t|tjƒst‚|j}dg}|f}|jd}|jd} ||j|j|| |j |j ||j ||j d|d|dt d|jƒ } | S(s 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. tgens.nexttargtypestmanglertinlinetenv_name(t isinstanceRt GeneratortAssertionErrort yield_typetqualnamet unique_nametnativetmodnametdocttypemapt calltypestkwstFalseR( tclstfunc_irtfndesctgentypeR trestypetargsR RRtself((s/lib/python2.7/site-packages/numba/generators.pytfrom_generator_fndescs      cC`s d|jS(sz The LLVM name of the generator's finalizer function (if .has_finalizer is true). t finalize_(t mangled_name(R#((s/lib/python2.7/site-packages/numba/generators.pytllvm_finalizer_name's((t__name__t __module__t__doc__t __slots__t classmethodR$tpropertyR'(((s/lib/python2.7/site-packages/numba/generators.pyR stBaseGeneratorLowercB`sheZdZd„Zd„Zd„Zd„Zd„Zd„Zd„Z d„Z d „Z d „Z RS( s5 Base support class for lowering generators. cC`s©|j|_|j|_|j|_|j|_|j|_|j|_|jƒ|_t j |j|j|j|jj ƒ|_ |jj |jjƒ|_i|_dS(N(tcontextRtlibraryt call_convRtgenerator_infotgeninfotget_generator_typeR R R$R tgendesctget_data_packerR t arg_packert resume_blocks(R#tlower((s/lib/python2.7/site-packages/numba/generators.pyt__init__5s      $cC`stj||ddƒS(Nii(Rt gep_inbounds(R#tbuildertgenptr((s/lib/python2.7/site-packages/numba/generators.pyt get_args_ptrEscC`stj||ddddƒS(Nitnamesgen.resume_index(RR;(R#R<R=((s/lib/python2.7/site-packages/numba/generators.pytget_resume_index_ptrHscC`stj||ddddƒS(NiiR?s gen.state(RR;(R#R<R=((s/lib/python2.7/site-packages/numba/generators.pyt get_state_ptrLsc C`sy|j|jƒ|j}|jj|j|j|jgƒ|jƒ|j ƒ|jj |jƒ}|jj t j dƒ}|jd}|jd}|jdƒ|jjrôxBt|jj|jƒD]%\}}|jjj|||ƒqÈWn|jj||jƒ} tj|ƒ} tj||| | g|ƒ} |j|| ƒ} |jdƒ|jj|| ƒ|jƒdS(s‰ Lower the generator's initialization function (which will fill up the passed-by-reference generator structure). iiis# low_init_func increfs# low_init_func before returnN( tsetup_functionRR<R/tinsert_generatorR R5R0textract_function_argumentst pre_lowertget_return_typet get_constantRtint32telementst debug_printt enable_nrttzipR tfnargstnrttincrefR7tas_dataRtnullRtmake_anonymous_structtbox_generator_structR1t return_valuet post_lower( R#R9R<trettyt resume_indextargstytstatestytargtytargvaltargsvalt statesvalt gen_structtretval((s/lib/python2.7/site-packages/numba/generators.pytlower_init_funcPs0        %    c C`sœ|j|jƒ|jdj|jjƒƒ|jjd|jksKt‚|j}|j }|j j |ƒ\}|j j ||j||ƒ|jƒ|j||ƒ|_|j||ƒ|_|jdƒ}|jƒ}|jdƒ}|j|ƒ|j j|ƒ|j|ƒ|j|j}|jd<|j|j|jƒ|ƒ} x-|jjƒD]\} } | j| | ƒq^W|j|ƒ|j|ƒdS(s¥ Lower the generator's next() function (which takes the passed-by-reference generator structure and returns the next yielded value). s# lower_next_func: {0}itgenerator_prologuetstop_iterationN( RBR5RJtformatRR R RR<tfunctionR1t get_argumentsR7t load_intoR>RMR@tresume_index_ptrRAt gen_state_ptrtappend_basic_blocktlower_function_bodytposition_at_endtreturn_stop_iterationtblkmaptfirstblkR8tswitchtloadtitemstadd_casetbranch( R#R9R<RdR=tprologuetentry_block_tailt stop_blockt first_blockRotindextblock((s/lib/python2.7/site-packages/numba/generators.pytlower_next_funcs0         cC`s¥tjtjƒ|jj|jƒgƒ}|jj|d|jj ƒ}|j dƒ}t |ƒ}|jj|jƒ}|j |j d|ƒ}|j||ƒdS(s2 Lower the generator's finalizer. R?tentryiN(RRdtvoidR/tget_value_typeR tmoduletget_or_insert_functionR5R'RiRtbitcastR"tlower_finalize_func_body(R#R9tfntyRdt entry_blockR<tgenptrtyR=((s/lib/python2.7/site-packages/numba/generators.pytlower_finalize_func¯s  cC`sHtj|jjjdƒ}|jj||jƒ|jj|jƒdS(sY Emit a StopIteration at generator end and mark the generator exhausted. iÿÿÿÿN( RtintRgttypetpointeeR<tstoreR1Rl(R#R9tindexval((s/lib/python2.7/site-packages/numba/generators.pytreturn_from_generator¾scC`s@d|f}|jj|ƒ}|jj|ƒ||j|R@RAR`RzR…R‹RRJ(((s/lib/python2.7/site-packages/numba/generators.pyR.0s     1 .   tGeneratorLowercB`s)eZdZd„Zd„Zd„ZRS(s9 Support class for lowering nopython generators. cC`s |jjS(N(RR!(R#((s/lib/python2.7/site-packages/numba/generators.pyR4ÕscC`s|S(N((R#R9R^((s/lib/python2.7/site-packages/numba/generators.pyRSØscC`sŽ|j|dƒ|jjrp|j||ƒ}x?|jj||ƒD]%\}}|jjj|||ƒqDWn|j|dƒ|jƒdS(sg Lower the body of the generator's finalizer: decref all live state variables. s# generator: finalizes# generator: finalize endN( RJR/RKR>R7RpRNtdecreftret_void(R#R<R=targs_ptrttytval((s/lib/python2.7/site-packages/numba/generators.pyRÛs " (R(R)R*R4RSR(((s/lib/python2.7/site-packages/numba/generators.pyRÐs  tPyGeneratorLowercB`s2eZdZd„Zd„Zd„Zd„ZRS(s< Support class for lowering object mode generators. c C`sZtjd|jjjdtjdtjf|jjdtjft|jj ƒdt ƒS(sx Compute the actual generator type (the generator function's return type is simply "pyobject"). tgen_funcRt arg_typest state_typest has_finalizer( RRRtfunc_idtfunctpyobjectt arg_counttlenR3t state_varstTrue(R#((s/lib/python2.7/site-packages/numba/generators.pyR4ñs   cC`s1tj|j|ƒ}|jj||j|jƒS(s> Box the raw *gen_struct* as a Python object. (Rtalloca_once_valueR<tpyapitfrom_native_generatorR tenvarg(R#R9R^tgen_ptr((s/lib/python2.7/site-packages/numba/generators.pyRSþscC`s,|jjtj|jjjƒ|jƒdS(so NULL-initialize all generator state variables, to avoid spurious decref's on cleanup. N(R<R‰RRQRhR‡Rˆ(R#R9((s/lib/python2.7/site-packages/numba/generators.pytinit_generator_statesc C`s|jj|ƒ}|j||ƒ}|j|ƒ}|jd|tj|jdƒƒ}tj ||ƒ‡|j ||ƒ}xmt t |j jƒƒD]S}tj||d|ƒ} |j j|} |jj|| | ƒ} |j| ƒq•WWdQX|jƒdS(sg Lower the body of the generator's finalizer: decref all live state variables. t>iN(R/tget_python_apiR@Rpt icmp_signedRR†R‡Rt if_unlikelyRAtrangeRŸR R™R;t unpack_valueR‘R’( R#R<R=R£RgRWt need_cleanupRht state_indext state_slotR”R•((s/lib/python2.7/site-packages/numba/generators.pyR s  (R(R)R*R4RSR§R(((s/lib/python2.7/site-packages/numba/generators.pyR–ìs   t LowerYieldcB`s)eZdZd„Zd„Zd„ZRS(s> Support class for lowering a particular yield point. cC`sª||_|j|_|j|_|j|_|jj|_|jj|_|jj|_||_|jj|_||_ g|D]}|j j j |ƒ^q‚|_ dS(N(R9R/R<tgenlowerR RhRgtyptinstt live_varsR2R Rxtlive_var_indices(R#R9t yield_pointRµtv((s/lib/python2.7/site-packages/numba/generators.pyR:.s      cC`s|jjdƒxªt|j|jƒD]“\}}tj|j|jd|ƒ}|j j |}|jj |ƒ}|j j r|j jj|j||ƒn|j j|j|||ƒq&Wtj|jjj|jjƒ}|jj||jƒ|jjdƒdS(Ns# generator suspendis# generator suspend end(R9RJRLR¶RµRR;R<RhR R™tloadvarR/RKRNROt pack_valueRR†RgR‡RˆR´RxR‰(R#R¯R?R°R”R•RŠ((s/lib/python2.7/site-packages/numba/generators.pytlower_yield_suspend=s"   cC`sí|jj|j|jjƒ|jjdƒxªt|j|jƒD]“\}}t j |j |j d|ƒ}|j j|}|jj|j ||ƒ}|jj||ƒ|jjrB|jjj|j ||ƒqBqBW|jjdƒdS(Ns# generator resumeis# generator resume end(R²RR9R´RxRJRLR¶RµRR;R<RhR R™R/R­tstorevarRKRNR‘(R#R¯R?R°R”R•((s/lib/python2.7/site-packages/numba/generators.pytlower_yield_resumePs"  #(R(R)R*R:R»R½(((s/lib/python2.7/site-packages/numba/generators.pyR±)s  N(R*t __future__RRRtllvmlite.llvmpy.coreRRRtRRRtfuncdescR R tobjectR.RR–R±(((s/lib/python2.7/site-packages/numba/generators.pyts$ =