o 4ήc' @sddlZddlZddlZddlmZmZddlmZddZ e dZ e dZ e dZ e d Z e d Ze d Ze d Ze d Ze dZe dZe dZe dZe dZe dZe dZe dZe dZe dZe dZe dZe dZe dZe dZ e dZ!GdddZ"Gd d!d!e#Z$Gd"d#d#e#Z%eZ&d$'e()d$dd%Z*zej+,e*e&Z-e.e/e-0Z1Wne2yZ3z d&e&d'e3Z4e2e4dZ3[3wwe$e1Z1d(d)Z5d*d+Z6Gd,d-d-e#Z7Gd.d/d/e#Z8d0d1Z9d2d3Z:Gd4d5d5e#Z;dS)6N)_decode_string_is_shutting_down)get_library_namecCst|tjfi}t|SN)typectypes StructurePOINTER)namenewclsr ;/tmp/pip-target-vg8gfxp4/lib/python/llvmlite/binding/ffi.py_make_opaque_ref s r LLVMContext LLVMModule LLVMValueLLVMTypeLLVMExecutionEngineLLVMPassManagerBuilderLLVMPassManagerLLVMTargetDataLLVMTargetLibraryInfo LLVMTargetLLVMTargetMachineLLVMMemoryBufferLLVMAttributeListIteratorLLVMAttributeSetIteratorLLVMGlobalsIteratorLLVMFunctionsIteratorLLVMBlocksIteratorLLVMArgumentsIteratorLLVMInstructionsIteratorLLVMOperandsIteratorLLVMTypesIteratorLLVMObjectCacheLLVMObjectFileLLVMSectionIteratorc@s8eZdZdZddZddZddZdd Zd d Zd S) _LLVMLocka6A Lock to guarantee thread-safety for the LLVM C-API. This class implements __enter__ and __exit__ for acquiring and releasing the lock as a context manager. Also, callbacks can be attached so that every time the lock is acquired and released the corresponding callbacks will be invoked. cCst|_g|_dSr) threadingRLock_lock_cblistselfr r r __init__1s  z_LLVMLock.__init__cC|j||fdS)zRegister callbacks that are invoked immediately after the lock is acquired (``acq_fn()``) and immediately before the lock is released (``rel_fn()``). N)r+appendr-acq_fnrel_fnr r r register7sz_LLVMLock.registercCr/)z)Remove the registered callbacks. N)r+remover1r r r unregister>sz_LLVMLock.unregistercCs$|j|jD]\}}|qdSr)r*acquirer+r1r r r __enter__Cs z_LLVMLock.__enter__cGs$|jD]\}}|q|jdSr)r+r*release)r- exc_detailsr2r3r r r __exit__Isz_LLVMLock.__exit__N) __name__ __module__ __qualname____doc__r.r4r6r8r;r r r r r'(s r'c@s@eZdZdZgdZddZddZeddZed d Z d S) _lib_wrapperz{Wrap libllvmlite with a lock such that only one thread may access it at a time. This class duck-types a CDLL. )_lib_fntabr*cCs||_i|_t|_dSr)rArBr'r*)r-libr r r r.Xs z_lib_wrapper.__init__cCsFz|j|WSty"t|j|}t|j|}||j|<|YSwr)rBKeyErrorgetattrrA_lib_fn_wrapperr*)r-r cfnwrappedr r r __getattr__]s     z_lib_wrapper.__getattr__cC|jjS)zgThe name of the library passed in the CDLL constructor. For duck-typing a ctypes.CDLL )rA_namer,r r r rKgz_lib_wrapper._namecCrJ)z]The system handle used to access the library. For duck-typing a ctypes.CDLL )rA_handler,r r r rMorLz_lib_wrapper._handleN) r<r=r>r? __slots__r.rIpropertyrKrMr r r r r@Ps r@c@s\eZdZdZddgZddZeddZejddZed d Z e jd d Z d d Z dS)rFzWraps and duck-types a ctypes.CFUNCTYPE to provide automatic locking when the wrapped function is called. TODO: we can add methods to mark the function as threadsafe and remove the locking-step on call when marked. r*_cfncCs||_||_dSrr*rP)r-lockrGr r r r.s z_lib_fn_wrapper.__init__cCrJrrPargtypesr,r r r rTz_lib_fn_wrapper.argtypescC ||j_dSrrS)r-rTr r r rT cCrJrrPrestyper,r r r rYrUz_lib_fn_wrapper.restypecCrVrrX)r-rYr r r rYrWcOs:|j|j|i|WdS1swYdSrrQ)r-argskwargsr r r __call__s$z_lib_fn_wrapper.__call__N) r<r=r>r?rNr.rOrTsetterrYr\r r r r rFxs     rF.z(Could not find/load shared object file: z Error was: cCtj||dS)z~Register callback functions for lock acquire and release. *acq_fn* and *rel_fn* are callables that take no arguments. N)rCr*r4r2r3r r r register_lock_callbackrbcCr`)zRemove the registered callback functions for lock acquire and release. The arguments are the same as used in `register_lock_callback()`. N)rCr*r6rar r r unregister_lock_callbackrcrdc@seZdZdZdS) _DeadPointerz: Dummy class to make error messages more helpful. N)r<r=r>r?r r r r resrec@speZdZdZeZeddZdddZdd Z d d Z d d Z e fddZ ddZddZeZeddZdS) OutputStringz< Object for managing the char* output of LLVM APIs. cCs|t|tjdS)a%Constructing from a pointer returned from the C-API. The pointer must be allocated with LLVMPY_CreateString. Note ---- Because ctypes auto-converts *restype* of *c_char_p* into a python string, we must use *c_void_p* to obtain the raw pointer. )init)rcastc_char_p)clsptrr r r from_returns zOutputString.from_returnTNcCs0|dur|ntd|_t|j|_||_dSr)rri_ptrbyref_as_parameter__owned)r-ownedrgr r r r.s zOutputString.__init__cCs.|jdur|jrt|jd|_|`dSdSr)rmrprCLLVMPY_DisposeStringror,r r r closes  zOutputString.closecCs|Srr r,r r r r8szOutputString.__enter__cC |dSrrsr-exc_typeexc_valexc_tbr r r r; zOutputString.__exit__cC$|s|jdur|dSdSdSrrur-rr r r __del__s   zOutputString.__del__cCs*|jdurdS|jj}|dusJt|S)Nz)rmvaluer)r-sr r r __str__s  zOutputString.__str__cC t|jSrboolrmr,r r r __bool__ zOutputString.__bool__cCrJ)z:Get the raw bytes of content of the char pointer. )rmr~r,r r r bytesszOutputString.bytes)TN)r<r=r>r?rero classmethodrlr.rsr8r;rr}rr __nonzero__rOrr r r r rfs   rfcCs|dur tt|SdS)z,To wrap string return-value from C-API. N)strrfrlrkr r r ret_stringsrcCs|dur t|jSdS)z+To wrap bytes return-value from C-API. N)rfrlrrr r r ret_bytess rc@seZdZdZdZeZdZddZddZ ddZ d d Z e d d Z d dZddZefddZddZddZeZddZdS) ObjectRefzJ A wrapper around a ctypes pointer to a LLVM object ("resource"). FcCs&|durtd||_||_t|_dS)Nz NULL pointer) ValueErrorrmrorC_capi)r-rkr r r r.s  zObjectRef.__init__cCsJz|js|js|W|dSW|dSW|dS|w)zI Close this object and do any required clean-up actions. N)_closedrp_disposedetachr,r r r rss    zObjectRef.closecCs|js |`d|_d|_dSdS)zN Detach the underlying LLVM resource without disposing of it. TN)rrormr,r r r r's  zObjectRef.detachcCsdS)z Dispose of the underlying LLVM resource. Should be overriden by subclasses. Automatically called by close(), __del__() and __exit__() (unless the resource has been detached). Nr r,r r r r0szObjectRef._disposecCs|jS)zf Whether this object has been closed. A closed object can't be used anymore. )rr,r r r closed7szObjectRef.closedcCs(t|dsJ|jrtd|jf|S)Nrsz%s instance already closed)hasattrr RuntimeError __class__r,r r r r8?szObjectRef.__enter__cCrtrrurvr r r r;ErzzObjectRef.__exit__cCr{rrur|r r r r}Hs   zObjectRef.__del__cCrrrr,r r r rMrzObjectRef.__bool__cCs.t|dsdSt|jdt|jdkS)NrmFr)rr addressofrm)r-otherr r r __eq__Ps zObjectRef.__eq__cCstt|jtjjSr)hashrrhrmc_void_pr~r,r r r __hash__YszObjectRef.__hash__N)r<r=r>r?rrerorpr.rsrrrOrr8r;rr}rrrrr r r r r s$     r)sj ((">