\ K]c@sdZddlZddlZddlmZddlmZddlmZddlm Z ddl m Z m Z m Z mZmZmZmZddlmZdd lmZdd lmZmZmZmZdd lmZmZdd lmZdd lmZddlm Z m!Z!m"Z"m#Z#m$Z$m%Z%ej&Z'ej&Z(ej&Z)ej*Z+ej,Z-ej&Z.ej/ej0Z1ej2Z3eedej4fdYZ5eeeedej4fdYZ6defdYZ7dZ8edZ9edZ:edZ;eej<ejejdZ=dZ>dZ?dZ@edZAedZBe e@dZCe eDd ZEed!ZFed"ZGe ejd#d$ZHe d%ZIe d&ZJed'ZKed(ZLd)ZMe ejNd*ZOed+ZPe ejQd,ZRe ejd-dd.ZSed/ZTe ejUd0ZVe ejWd1ZXe ejd2d3ZYe ejd4d5ZZe ejd6d7Z[e ejd8d9Z\e ejd:d;Z]e ejd<d=Z^e ejd>d?Z_e ejd@dddAZae dBZbdCZce ejddDZee ejfdEZge ejhdFZie ejjdGZke ejldHZme ejndIZoedJejdKZpedLejeejqdMZrdS(Ns8 Compiler-side implementation of the Numba typed-list. iN(tIntEnum(tir(tcgutils(t _helperlib(toverloadtoverload_methodtregister_jitablet intrinsictregister_modeltmodelst lower_builtin(t iternext_impl(ttypes(tListTypetListTypeIterableTypetListTypeIteratorTypetType(timpl_ret_borrowedtRefType(t TypingError(ttyping(t _as_bytest_castt _nonoptionalt_get_incref_decreft_container_get_datat_container_get_meminfot ListModelcBseZdZRS(cCs;dtfdtjfg}tt|j|||dS(Ntmeminfotdata(t_meminfo_listptrR tvoidptrtsuperRt__init__(tselftdmmtfe_typetmembers((s/lib/python2.7/site-packages/numba/listobject.pyR!9s (t__name__t __module__R!(((s/lib/python2.7/site-packages/numba/listobject.pyR7st ListIterModelcBseZdZRS(cCs>d|jfdtjfg}tt|j|||dS(Ntparenttstate(R)R RR R(R!(R"R#R$R%((s/lib/python2.7/site-packages/numba/listobject.pyR!Ds (R&R'R!(((s/lib/python2.7/site-packages/numba/listobject.pyR(Ast ListStatuscBs,eZdZdZdZdZdZdZRS(s+Status code for other list operations. iiiii(i(R&R't__doc__tLIST_OKtLIST_ERR_INDEXtLIST_ERR_NO_MEMORYtLIST_ERR_MUTATEDtLIST_ERR_ITER_EXHAUSTED(((s/lib/python2.7/site-packages/numba/listobject.pyR+Ls cCs`|jttj}|j|jd||dt|jj|t |fWdQXdS(s?Raise an internal error depending on the value of *status* s!=tlikelyN( ttypetintR+R-tif_thent icmp_signedtTruet call_convtreturn_user_exct RuntimeError(tcontexttbuildertstatustmsgt ok_status((s/lib/python2.7/site-packages/numba/listobject.pyt_raise_if_errorVs cCs@t|tjs!tdnd}t|}||fS(s*Returns the MemInfoPointer of a list. s"expected *lstobj* to be a ListTypecSsV|j\}|\}|jj|||tj|}|||d|}|jS(Ntvalue(targstnrttincrefRtcreate_struct_proxyR(R;R<tsigRBttltltctortlstruct((s/lib/python2.7/site-packages/numba/listobject.pytcodegenfs   (t isinstanceR R RR(t typingctxtlstobjRKRF((s/lib/python2.7/site-packages/numba/listobject.pyt _as_meminfo_s  css|tkrtdn|jttsKtdjtnfd}||}||fS(s*Recreate a list from a MemInfoPointer s#expected a MemInfoPointer for list.s expected a {}c s|j\}}|j}|\}}tj|} | ||} |jj||} |j| tj} |j | | _ || _ t ||| j S(N(RBt instance_typeRRERCt meminfo_datatbitcastt ll_list_typet as_pointertloadRRRt _getvalue( R;R<RFRBttmittdrefttdtmit_RItdstructt data_pointer(tlisttype(s/lib/python2.7/site-packages/numba/listobject.pyRK~s   (RRRPRLR tformat(RMRZt listtyperefRKRF((R^s/lib/python2.7/site-packages/numba/listobject.pyt _from_meminfots  cs1tj}||}fd}||fS(s%Wrap numba_list_set_method_table csAtjttg}tjtjt|jg}tj|j|dd}|d}t j ||dt }t j ||dd} t j ||dd} |j j} | jr't||j| d\} } |j|j| | jj| |j|j| | jj| n|j|||gdS(Ntnametnumba_list_set_method_tableitzfillitlist(RtLiteralStructTypetll_voidptr_typet FunctionTypetVoidTypeRSRTtFunctiontmoduleRt alloca_onceR7t gep_inboundstdata_model_managerRPtcontains_nrt_meminfoRtstoreRRR3tpointeetcall(R;R<RFRBtvtabletytsetmethod_fntyt setmethod_fntdptvtabletitem_incref_ptrtitem_decref_ptrtdm_itemt item_increft item_decref(titemty(s/lib/python2.7/site-packages/numba/listobject.pyRKs2      (R tvoid(RMtlpR}trestyRFRK((R}s/lib/python2.7/site-packages/numba/listobject.pyt_list_set_method_tables %cCs}t|||jd|d}t|||jd|d}|j|tj}|j|tj}|jd||S(Niis==(RRBtptrtointRtintp_tR6(R;R<RFRBt a_meminfot b_meminfotmatmb((s/lib/python2.7/site-packages/numba/listobject.pytlist_iss   cCsJtjtjtg}|jj|dd}|j||gdS(sCall numba_list_free(ptr) Rbtnumba_list_freeN(RRhRiRSRktget_or_insert_functionRr(R;R<tptrtfntytfree((s/lib/python2.7/site-packages/numba/listobject.pyt_call_list_frees   c Cs|jtj}|jtj}tjtj|||g}d}|j|d|}|jrd|_ tj |j }|j |j dtj}|j|} t||| |jn|S(sDefine the dtor for list t_numba_list_dtorRbt linkonce_odri(tget_value_typeR RtuintpRRhRiRtis_declarationtlinkaget IRBuildertappend_basic_blockRRRBRSRTRURtret_void( R;Rkt llvoidptrtllsizeRtfnametfnR<RRH((s/lib/python2.7/site-packages/numba/listobject.pyt _imp_dtors    cCs tdS(sConstruct a new list. (Not implemented in the interpreter yet) Parameters ---------- item: TypeRef Item type of the new list. N(tNotImplementedError(titem((s/lib/python2.7/site-packages/numba/listobject.pytnew_listscs:tj|jfd}||}||fS(sMake a list struct with the given *ptr* Parameters ---------- itemty: Type Type of the item. ptr : llvm pointer value Points to the list object. c s|\}}tj}|||}||_|j|jtj}t||j} |j j ||j tj || } |j j || } |j| tj} |j|| | |_|jS(N(RRERtget_abi_sizeofRR RRRkRCtmeminfo_alloc_dtort get_constantRRQRRRSRTRpRRV( R;R<t signatureRBR[RRIRJt alloc_sizetdtorRR](tlist_ty(s/lib/python2.7/site-packages/numba/listobject.pyRK s      (R R RP(RMR}RRKRF((Rs/lib/python2.7/site-packages/numba/listobject.pyt _make_lists cs.tj}|}fd}||fS(sWrap numba_list_new. Allocate a new list object with zero capacity. Parameters ---------- itemty: Type Type of the items c stjttjttg}|jj|dd}|jj }|j |}t j |tdt }|j||t|tdg} t||| dd|j|} | S(NRbtnumba_list_newRdiR>sFailed to allocate list(RRht ll_statusRSRTt ll_ssize_tRkRt get_data_typeRPRRRlR7RrR@RU( R;R<RFRBRRtll_itemtsz_itemtreflpR=R(R}(s/lib/python2.7/site-packages/numba/listobject.pyRK7s (R R(RMR}RRFRK((R}s/lib/python2.7/site-packages/numba/listobject.pyt _list_new(s  cs7t|tstdn|fd}|S(sECreates a new list with *item* as the type of the list item. s#expecting *item* to be a numba Typecs,t}t|t|}|S(N(RRR(RRRH(R}(s/lib/python2.7/site-packages/numba/listobject.pytimpYs  (RLRt TypeError(RR((R}s/lib/python2.7/site-packages/numba/listobject.pyt impl_new_listOs cCs#t|tjrd}|SdS(slen(list) cSs t|S(N(t _list_length(RH((s/lib/python2.7/site-packages/numba/listobject.pytimplgsN(RLR R (RHR((s/lib/python2.7/site-packages/numba/listobject.pytimpl_lenbs cCs(tj}||}d}||fS(s@Wrap numba_list_length Returns the length of the list. c Ssptjttg}|jj|dd}|\}|j\}t||||}|j||g} | S(NRbtnumba_list_length( RRhRRSRkRRBRRr( R;R<RFRBRRRHRGRtn((s/lib/python2.7/site-packages/numba/listobject.pyRKvs   (R tintp(RMRHRRFRK((s/lib/python2.7/site-packages/numba/listobject.pyRms   cCs.tj}|||j}d}||fS(sWrap numba_list_append cSstjtttg}|\}}|j\}}|jj|dd} |j|} | j ||} t j || } t ||||} |j | | t|| g}|S(NRbtnumba_list_append(RRhRRStll_bytesRBRkRRntas_dataRtalloca_once_valueRRrR(R;R<RFRBRRHRRGttitemRRzt data_itemtptr_itemRR=((s/lib/python2.7/site-packages/numba/listobject.pyRKs  (R tint32t item_type(RMRHRRRFRK((s/lib/python2.7/site-packages/numba/listobject.pyt _list_appends  tappendcs|t|tjsdS|jfd}|jr>|S|j|}|jtjtj|}||fSdS(Ncs_t|}t||}|tjkr1dS|tjkrOtdn tddS(Ns(Unable to allocate memory to append itemslist.append failed unexpectedly(RRR+R-R/t MemoryErrorR:(RHRt casteditemR=(R}(s/lib/python2.7/site-packages/numba/listobject.pyRs( RLR R Rt is_precisetrefineRRR~(RHRRRF((R}s/lib/python2.7/site-packages/numba/listobject.pyt impl_appends   cCsT|dkrt||}nd|ko<t|knsPtdn|S(stHandle index. If the index is negative, convert it. If the index is out of range, raise an IndexError. islist index out of range(tlent IndexError(RHtindex((s/lib/python2.7/site-packages/numba/listobject.pyt handle_indexs  "cCsHt|dkrtdSt||j|j|jf\}}}}|dkr|jdkrwt||dn t||}|dkrt||dn t||}n~|dkr)|jdkrt||dnt|d|}|dkrt||dn t||}n tdt|||jS(sHandle slice. Convert a slice object for a given list into a range object that can be used to index the list. Many subtle caveats here, especially if the step is negative. iiisslice step cannot be zero(Rtrangetstarttstoptsteptmaxtmint ValueError(RHtstlltsatsotseRR((s/lib/python2.7/site-packages/numba/listobject.pyt handle_slices - 11 51 cCst|||dS(Ntgetitem(t_list_getitem_pop_helper(RMRHR((s/lib/python2.7/site-packages/numba/listobject.pyt _list_getitemscCst|||dS(Ntpop(R(RMRHR((s/lib/python2.7/site-packages/numba/listobject.pyt _list_popscsadksttjtjtj|jg||}fd}||fS(sWrap numba_list_getitem and numba_list_pop Returns 2-tuple of (intp, ?item_type) This is a helper that is parametrized on the type of operation, which can be either 'pop' or 'getitem'. This is because, signature wise, getitem and pop and are the same. RRcstjttttg}|j\}}|\}}|jj|ddj } |j |j } |j |j } t j|| } t||||} |j| | |t|| g}|jd||jttj}|j||j }t j||}|j|X| j|| }|jj||j ||j||j |}|j||WdQX|j|}|j |||gS(NRbs numba_list_{}s>=(!RRhRRSRRRBRkRR_RnRRRRlRRrRR6R3R4R+R-tmake_optional_noneRR5tload_from_data_pointerRCRDtmake_optional_valueRpRUt make_tuple(R;R<RFRBRRGttindexRHRRRzRRRR=tfoundtouttpoutRtloaded(topR(s/lib/python2.7/site-packages/numba/listobject.pyRKs4 '(RR(tAssertionErrorR tTupleRtOptionalR(RMRHRRRFRK((RRs/lib/python2.7/site-packages/numba/listobject.pyRs $&cs|t|tjsdSt|j|tjkrGfd}|St|tjrlfd}|StddS(Ncs\t||}t|}t||\}}|tjkrLt|StddS(Ns"internal list error during getitem(RRRR+R-RR(RHRt castedindexR=R(tindexty(s/lib/python2.7/site-packages/numba/listobject.pyt integer_impl6s  cs;t}x(t||D]}|j||qW|S(N(RRR(RHRtnewlti(R}(s/lib/python2.7/site-packages/numba/listobject.pyt slice_implBs s.list indices must be signed integers or slices(RLR R tINDEXTYRt signed_domaint SliceTypeR(RHRRR((RR}s/lib/python2.7/site-packages/numba/listobject.pyt impl_getitem-s  cCs.tj}||||}d}||fS(sWrap numba_list_setitem cSstjttttg}|\}}}|j\}} } |jj|dd} |j | } | j ||} t j || }t ||||}|j| ||t||g}|S(NRbtnumba_list_setitem(RRhRRSRRRBRkRRnRRRRRrR(R;R<RFRBRRHRRRGRRRRzRRRR=((s/lib/python2.7/site-packages/numba/listobject.pyRKUs"  (R R(RMRHRRRRFRK((s/lib/python2.7/site-packages/numba/listobject.pyt _list_setitemNs  cst|tjsdSt|j|tjkrJfd}|St|tjrt|tjs}tdnd}|StddS(Ncsbt||}t|}t|}t|||}|tjkrRdStddS(Ns#internal list error during settitem(RRRR+R-R(RHRRRRR=(RR}(s/lib/python2.7/site-packages/numba/listobject.pyt impl_integerzssFcan only assign an iterable when using a slice with assignment/setitemc Ss||kr|j}nt||}|jdkrt|t|kr~x*t||D]\}}||||S|j|}|j}tjtj |t |}||fSn t ddS(NcSs|t|ks$t|dkr4|j|n|dkr\tt||d}n|j|dt|d}x,||kr||d||<|d8}qW||| other return 1 iiiN(RR(tthistotherRt this_itemt other_item((s/lib/python2.7/site-packages/numba/listobject.pytcompares  csJt|tjsdSt|tjs7tdnfd}|S(Ns!list can only be compared to listcst||kS(N(R.(R*R+(taccepted(s/lib/python2.7/site-packages/numba/listobject.pyRs(RLR R R(R*R+R/R((R/s/lib/python2.7/site-packages/numba/listobject.pytcompare_helpers cCst||dS(Ni(i(R0(R*R+((s/lib/python2.7/site-packages/numba/listobject.pyt impl_equalscCst||dS(Nii(ii(R0(R*R+((s/lib/python2.7/site-packages/numba/listobject.pytimpl_not_equalscCst||dS(Ni(i(R0(R*R+((s/lib/python2.7/site-packages/numba/listobject.pytimpl_less_thanscCst||dS(Nii(ii(R0(R*R+((s/lib/python2.7/site-packages/numba/listobject.pytimpl_less_than_or_equalscCst||dS(Ni(i(R0(R*R+((s/lib/python2.7/site-packages/numba/listobject.pytimpl_greater_thanscCst||dS(Nii(ii(R0(R*R+((s/lib/python2.7/site-packages/numba/listobject.pytimpl_greater_than_or_equal stgetitercCs6|j\}|\}tj|}|j||j}tjtjtt g}|j j |dd} t j t j} | tjd} tjtjd| } tj|| dt} t|| |_||_t|||j|d}|j| |j|gt|||j|jS(sImplement iter(List). Rbtnumba_list_itertlist_iter_sizeofiRdi(RBR Rt make_helpert iterator_typeRRhRitll_listiter_typeRSRkRtctypest CFUNCTYPEtc_size_tRt c_helperst ArrayTypetIntTypeRRlR7RR*R)RRrRt return_typeRV(R;R<RFRBRGRHt iterabletytitRRtprototlistiter_sizeoft state_typetpstateRv((s/lib/python2.7/site-packages/numba/listobject.pytimpl_list_getiters*    titernextcCs|jd}|j|||d}tjtttjg}|jj |dd}t j |t} |j ||j | f} | jttj} |jd| | } |j| dt|jj|tdWdQX| jttj} |jd| | }|j||j|dt|jj}|j|}|j|j| |j j}|j!||}t"|j#t$r|j%|nt&dj'|j#WdQXdS( NiRbtnumba_list_iter_nexts==R2s!list was mutated during iterationsunknown type: {}(s!list was mutated during iteration((RBR:RRhRR<RRTRkRRRlRrR*R3R4R+R0R6R5R R8R9R:R-t set_validR7R)RRnRRRURRRLRRtyield_RR_(R;R<RFRBtresultt iter_typeREt iternext_fntyRKt item_raw_ptrR=tmutated_statust is_mutatedR?tis_validtitem_tyRztitem_ptrR((s/lib/python2.7/site-packages/numba/listobject.pytimpl_iterator_iternext4s8        (sR,R=toperatortenumRtllvmliteRtnumbaRRtnumba.extendingRRRRRR R tnumba.targets.imputilsR R t numba.typesR RRRRRt numba.errorsRRtnumba.typedobjectutilsRRRRRRt voidptr_tRSR<Rgtint32_tRRRRtMemInfoPointerRRRRt StructModelRR(R+R@RORaRtis_RRRRRRRRRRRRRRRRRRRRtsetitemRRR tdelitemR tcontainsR RRRRRR!R"R#R)R.R0teqR1tneR2tltR3tleR4tgtR5tgeR6RJtBORROWEDRX(((s/lib/python2.7/site-packages/numba/listobject.pyts  4 ".            #/$  )' # 6!#I +*   "