ó \ K]c@`s\ddlmZmZmZddlmZddlmZmZde fd„ƒYZ dS(i(tprint_functiontabsolute_importtdivision(tir(tcgutilsttypest NRTContextcB`s°eZdZd„Zd„Zd„Zd„Zd„Zd„Zd„Z d„Z d „Z d „Z d „Z d „Zd „Zd„Zd„Zd„Zd„Zd„ZRS(sF An object providing access to NRT APIs in the lowering pass. cC`s||_||_dS(N(t_contextt_enabled(tselftcontexttenabled((s4lib/python2.7/site-packages/numba/runtime/context.pyt__init__ s cC`s|jstdƒ‚ndS(NsNRT required but not enabled(Rt RuntimeError(R ((s4lib/python2.7/site-packages/numba/runtime/context.pyt _require_nrts cC`sf|jƒ|j}tjtjtjgƒ}|j|ddƒ}|jj dƒ|j ||gƒS(sG Low-level allocate a new memory area of `size` bytes. tnamet NRT_Allocatetnoalias( RtmoduleRt FunctionTypeRt voidptr_ttintp_ttget_or_insert_functiont return_valuet add_attributetcall(R tbuildertsizetmodtfntytfn((s4lib/python2.7/site-packages/numba/runtime/context.pytallocates   cC`sY|jƒ|j}tjtjƒtjgƒ}|j|ddƒ}|j||gƒS(sI Low-level free a memory area allocated with allocate(). RtNRT_Free( RRRRtVoidTypeRRRR(R RtptrRRR((s4lib/python2.7/site-packages/numba/runtime/context.pytfree!s   cC`sf|jƒ|j}tjtjtjgƒ}|j|ddƒ}|jj dƒ|j ||gƒS(s| Allocate a new MemInfo with a data payload of `size` bytes. A pointer to the MemInfo is returned. RtNRT_MemInfo_alloc_safeR( RRRRRRRRRRR(R RRRRR((s4lib/python2.7/site-packages/numba/runtime/context.pyt meminfo_alloc,s   cC`s~|jƒ|j}tjtjtjtjgƒ}|j|ddƒ}|jj dƒ|j |||j |tjƒgƒS(NRtNRT_MemInfo_alloc_dtor_safeR( RRRRRRRRRRRtbitcast(R RRtdtorRRR((s4lib/python2.7/site-packages/numba/runtime/context.pytmeminfo_alloc_dtor:s      cC`sÀ|jƒ|j}tjdƒ}tjtjtj|gƒ}|j|ddƒ}|j j dƒt |t ƒr|j jtj|ƒ}n|j|ksªtdƒ‚|j|||gƒS(sÿ Allocate a new MemInfo with an aligned data payload of `size` bytes. The data pointer is aligned to `align` bytes. `align` can be either a Python int or a LLVM uint32 value. A pointer to the MemInfo is returned. i RtNRT_MemInfo_alloc_safe_alignedRsalign must be a uint32(RRRtIntTypeRRRRRRRt isinstancetintRt get_constantRtuint32ttypetAssertionErrorR(R RRtalignRtu32RR((s4lib/python2.7/site-packages/numba/runtime/context.pytmeminfo_alloc_alignedFs    cC`sf|jƒ|j}tjtjtjgƒ}|j|ddƒ}|jj dƒ|j ||gƒS(s  Allocate a MemInfo pointing to a variable-sized data area. The area is separately allocated (i.e. two allocations are made) so that re-allocating it doesn't change the MemInfo's address. A pointer to the MemInfo is returned. RtNRT_MemInfo_new_varsizeR( RRRRRRRRRRR(R RRRRR((s4lib/python2.7/site-packages/numba/runtime/context.pytmeminfo_new_varsize\s   cC`s_|jƒ|j}tjtjtjtjgƒ}|j|ddƒ}|j|||gƒS(s Like meminfo_new_varsize() but also set the destructor for cleaning up references to objects inside the allocation. RtNRT_MemInfo_new_varsize_dtor( RRRRRRRRR(R RRR(RRR((s4lib/python2.7/site-packages/numba/runtime/context.pytmeminfo_new_varsize_dtorls   cC`s|j|||dƒS(s¾ Allocate a new data area for a MemInfo created by meminfo_new_varsize(). The new data pointer is returned, for convenience. Contrary to realloc(), this always allocates a new area and doesn't copy the old data. This is useful if resizing a container needs more than simply copying the data area (e.g. for hash tables). The old pointer will have to be freed with meminfo_varsize_free(). tNRT_MemInfo_varsize_alloc(t_call_varsize_alloc(R RtmeminfoR((s4lib/python2.7/site-packages/numba/runtime/context.pytmeminfo_varsize_alloczs cC`s|j|||dƒS(s‡ Reallocate a data area allocated by meminfo_new_varsize(). The new data pointer is returned, for convenience. tNRT_MemInfo_varsize_realloc(R:(R RR;R((s4lib/python2.7/site-packages/numba/runtime/context.pytmeminfo_varsize_reallocˆscC`sb|jƒ|j}tjtjƒtjtjgƒ}|j|ddƒ}|j|||fƒS(s Free a memory area allocated for a NRT varsize object. Note this does *not* free the NRT object itself! RtNRT_MemInfo_varsize_free( RRRRR!RRRR(R RR;R"RRR((s4lib/python2.7/site-packages/numba/runtime/context.pytmeminfo_varsize_frees   cC`so|jƒ|j}tjtjtjtjgƒ}|j|d|ƒ}|jj dƒ|j |||gƒS(NRR( RRRRRRRRRRR(R RR;RtfuncnameRRR((s4lib/python2.7/site-packages/numba/runtime/context.pyR:s   cC`sK|jƒddlm}|j}|j|ddƒ}|j||gƒS(s¯ Given a MemInfo pointer, return a pointer to the allocated data managed by it. This works for MemInfos allocated with all the above methods. i(tmeminfo_data_tyRtNRT_MemInfo_data_fast(Rtnumba.runtime.nrtdynmodRBRRR(R RR;RBRR((s4lib/python2.7/site-packages/numba/runtime/context.pyt meminfo_data§s     c C`s¢|jj|}|j|ƒ}g}|jƒrY|j||ƒ}|j||fƒnxB|D]:\}} | |ƒ} |j||| ƒ} |j| ƒq`W|S(sCReturn a list of *(type, meminfo)* inside the given value. (Rtdata_model_managerttraversethas_nrt_meminfotget_nrt_meminfotappendt get_meminfostextend( R Rttytvalt datamodeltmemberstmeminfostmitmtyptgettertfieldtinner_meminfos((s4lib/python2.7/site-packages/numba/runtime/context.pyRK¶s  c C`s£|jƒddlm}|j|||ƒ}xm|D]e\}}|j} | j|d|ƒ} | jdjdƒ| jdjdƒ|j| |gƒq6WdS(sGCall function of *funcname* on every meminfo found in *value*. i(tincref_decref_tyRRt nocaptureN( RRDRWRKRRtargsRR( R RttyptvalueRARWRQt_RRRR((s4lib/python2.7/site-packages/numba/runtime/context.pyt_call_incref_decrefÇs  cC`s|j|||dƒdS(sG Recursively incref the given *value* and its members. t NRT_increfN(R](R RRZR[((s4lib/python2.7/site-packages/numba/runtime/context.pytincrefØscC`s|j|||dƒdS(sG Recursively decref the given *value* and its members. t NRT_decrefN(R](R RRZR[((s4lib/python2.7/site-packages/numba/runtime/context.pytdecrefÞs(t__name__t __module__t__doc__R RRR#R%R)R4R6R8R<R>R@R:RERKR]R_Ra(((s4lib/python2.7/site-packages/numba/runtime/context.pyRs&            N( t __future__RRRtllvmliteRtnumbaRRtobjectR(((s4lib/python2.7/site-packages/numba/runtime/context.pyts