B   ›\Ô ã@sDddlmZmZmZddlmZddlmZmZGdd„de ƒZ dS)é)Úprint_functionÚabsolute_importÚdivision)Úir)ÚcgutilsÚtypesc@s eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„Z dd„Z dd„Z dd„Z dd„Zdd„Zdd„Zdd„Zdd„Zd d!„Zd"d#„Zd$d%„Zd&S)'Ú NRTContextzF An object providing access to NRT APIs in the lowering pass. cCs||_||_dS)N)Ú_contextÚ_enabled)ÚselfÚcontextZenabled©r ú4lib/python3.7/site-packages/numba/runtime/context.pyÚ__init__ szNRTContext.__init__cCs|jstdƒ‚dS)NzNRT required but not enabled)r Ú RuntimeError)r r r rÚ _require_nrtszNRTContext._require_nrtcCsH| ¡|j}t tjtjg¡}|j|dd}|j  d¡|  ||g¡S)zG Low-level allocate a new memory area of `size` bytes. Z NRT_Allocate)ÚnameÚnoalias) rÚmodulerÚ FunctionTyperÚ voidptr_tÚintp_tÚget_or_insert_functionÚ return_valueÚ add_attributeÚcall)r ÚbuilderÚsizeÚmodÚfntyÚfnr r rÚallocates  zNRTContext.allocatecCs>| ¡|j}t t ¡tjg¡}|j|dd}| ||g¡S)zI Low-level free a memory area allocated with allocate(). ZNRT_Free)r) rrrrÚVoidTyperrrr)r rÚptrrrr r r rÚfree!s zNRTContext.freecCsH| ¡|j}t tjtjg¡}|j|dd}|j  d¡|  ||g¡S)z| Allocate a new MemInfo with a data payload of `size` bytes. A pointer to the MemInfo is returned. ZNRT_MemInfo_alloc_safe)rr) rrrrrrrrrrr)r rrrrr r r rÚ meminfo_alloc,s  zNRTContext.meminfo_alloccCsX| ¡|j}t tjtjtjg¡}|j|dd}|j  d¡|  |||  |tj¡g¡S)NZNRT_MemInfo_alloc_dtor_safe)rr) rrrrrrrrrrrZbitcast)r rrÚdtorrrr r r rÚmeminfo_alloc_dtor:s zNRTContext.meminfo_alloc_dtorcCs„| ¡|j}t d¡}t tjtj|g¡}|j|dd}|j   d¡t |t ƒrb|j  tj|¡}n|j|ksttdƒ‚| |||g¡S)zÿ 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. é ZNRT_MemInfo_alloc_safe_aligned)rrzalign must be a uint32)rrrZIntTyperrrrrrrÚ isinstanceÚintr Z get_constantrZuint32ÚtypeÚAssertionErrorr)r rrZalignrÚu32rr r r rÚmeminfo_alloc_alignedFs   z NRTContext.meminfo_alloc_alignedcCsH| ¡|j}t tjtjg¡}|j|dd}|j  d¡|  ||g¡S)a  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. ZNRT_MemInfo_new_varsize)rr) rrrrrrrrrrr)r rrrrr r r rÚmeminfo_new_varsize\s  zNRTContext.meminfo_new_varsizecCsB| ¡|j}t tjtjtjg¡}|j|dd}| |||g¡S)z Like meminfo_new_varsize() but also set the destructor for cleaning up references to objects inside the allocation. ZNRT_MemInfo_new_varsize_dtor)r) rrrrrrrrr)r rrr&rrr r r rÚmeminfo_new_varsize_dtorls z#NRTContext.meminfo_new_varsize_dtorcCs| |||d¡S)a¾ 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(). ZNRT_MemInfo_varsize_alloc)Ú_call_varsize_alloc)r rÚmeminforr r rÚmeminfo_varsize_alloczs z NRTContext.meminfo_varsize_alloccCs| |||d¡S)z‡ Reallocate a data area allocated by meminfo_new_varsize(). The new data pointer is returned, for convenience. ZNRT_MemInfo_varsize_realloc)r1)r rr2rr r rÚmeminfo_varsize_reallocˆs z"NRTContext.meminfo_varsize_realloccCsD| ¡|j}t t ¡tjtjg¡}|j|dd}| |||f¡S)z Free a memory area allocated for a NRT varsize object. Note this does *not* free the NRT object itself! ZNRT_MemInfo_varsize_free)r) rrrrr"rrrr)r rr2r#rrr r r rÚmeminfo_varsize_frees  zNRTContext.meminfo_varsize_freecCsN| ¡|j}t tjtjtjg¡}|j||d}|j  d¡|  |||g¡S)N)rr) rrrrrrrrrrr)r rr2rÚfuncnamerrr r r rr1s zNRTContext._call_varsize_alloccCs6| ¡ddlm}|j}|j|dd}| ||g¡S)z¯ Given a MemInfo pointer, return a pointer to the allocated data managed by it. This works for MemInfos allocated with all the above methods. r)Úmeminfo_data_tyZNRT_MemInfo_data_fast)r)rÚnumba.runtime.nrtdynmodr7rrr)r rr2r7rr r r rÚ meminfo_data§s  zNRTContext.meminfo_datac Csr|jj|}| |¡}g}| ¡r<| ||¡}| ||f¡x0|D](\}} | |ƒ} | ||| ¡} | | ¡qBW|S)zCReturn a list of *(type, meminfo)* inside the given value. )r Zdata_model_managerZtraverseZhas_nrt_meminfoZget_nrt_meminfoÚappendÚ get_meminfosÚextend) r rZtyÚvalZ datamodelÚmembersÚmeminfosÚmiZmtypÚgetterZfieldZinner_meminfosr r rr;¶s   zNRTContext.get_meminfosc Csz| ¡ddlm}| |||¡}xR|D]J\}}|j} | j||d} | jd d¡| jd d¡| | |g¡q(WdS)zGCall function of *funcname* on every meminfo found in *value*. r)Úincref_decref_ty)rrZ nocaptureN) rr8rBr;rrÚargsrr) r rÚtypÚvaluer6rBr?Ú_r@rr r r rÚ_call_incref_decrefÇs zNRTContext._call_incref_decrefcCs| |||d¡dS)zG Recursively incref the given *value* and its members. Z NRT_increfN)rG)r rrDrEr r rÚincrefØszNRTContext.increfcCs| |||d¡dS)zG Recursively decref the given *value* and its members. Z NRT_decrefN)rG)r rrDrEr r rÚdecrefÞszNRTContext.decrefN)Ú__name__Ú __module__Ú __qualname__Ú__doc__rrr!r$r%r'r.r/r0r3r4r5r1r9r;rGrHrIr r r rrs&     rN) Z __future__rrrZllvmliterZnumbarrÚobjectrr r r rÚs