U +d' @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*z$ej+,e*e&Z-e.e/e-0Z1Wn<e2k rZ3zd&e&d'e3Z4e2e4W5dZ3[3XYnXe$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)nameZnewclsr 8/tmp/pip-unpacked-wheel-stw2luzp/llvmlite/binding/ffi.py_make_opaque_ref sr Z LLVMContextZ LLVMModuleZ LLVMValueZLLVMTypeZLLVMExecutionEngineZLLVMPassManagerBuilderZLLVMPassManagerZLLVMTargetDataZLLVMTargetLibraryInfoZ LLVMTargetZLLVMTargetMachineZLLVMMemoryBufferLLVMAttributeListIteratorLLVMAttributeSetIteratorLLVMGlobalsIteratorLLVMFunctionsIteratorLLVMBlocksIteratorLLVMArgumentsIteratorLLVMInstructionsIteratorLLVMOperandsIteratorLLVMTypesIteratorZLLVMObjectCacheZLLVMObjectFileZLLVMSectionIteratorc@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__cCs|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)rappendracq_fnrel_fnr r r register7sz_LLVMLock.registercCs|j||fdS)z)Remove the registered callbacks. N)rremover r r r unregister>sz_LLVMLock.unregistercCs$|j|jD]\}}|qdSr)racquirerr r r r __enter__Cs z_LLVMLock.__enter__cGs$|jD]\}}|q|jdSr)rrrelease)r exc_detailsr!r"r r r __exit__Isz_LLVMLock.__exit__N) __name__ __module__ __qualname____doc__rr#r%r'r*r r r r r(s rc@sBeZdZdZdddgZddZddZed d Zed 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_fntabrcCs||_i|_t|_dSr)r0r1rr)rlibr r r rXsz_lib_wrapper.__init__cCsLz |j|WStk rFt|j|}t|j|}||j|<|YSXdSr)r1KeyErrorgetattrr0_lib_fn_wrapperr)rr cfnwrappedr r r __getattr__]s    z_lib_wrapper.__getattr__cCs|jjS)zgThe name of the library passed in the CDLL constructor. For duck-typing a ctypes.CDLL )r0_namerr r r r9gsz_lib_wrapper._namecCs|jjS)z]The system handle used to access the library. For duck-typing a ctypes.CDLL )r0_handlerr r r r:osz_lib_wrapper._handleN) r+r,r-r. __slots__rr8propertyr9r:r r r r r/Ps   r/c@s\eZdZdZddgZddZeddZejddZed d Z e jd d Z d d Z dS)r5zWraps 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||_||_dSrrr=)rlockr6r r r rsz_lib_fn_wrapper.__init__cCs|jjSrr=argtypesrr r r rAsz_lib_fn_wrapper.argtypescCs ||j_dSrr@)rrAr r r rAscCs|jjSrr=restyperr r r rCsz_lib_fn_wrapper.restypecCs ||j_dSrrB)rrCr r r rCsc Os*|j|j||W5QRSQRXdSrr>)rargskwargsr r r __call__sz_lib_fn_wrapper.__call__N) r+r,r-r.r;rr<rAsetterrCrFr r r r r5xs    r5.z(Could not find/load shared object file: z Error was: cCstj||dS)z~Register callback functions for lock acquire and release. *acq_fn* and *rel_fn* are callables that take no arguments. N)r2rr#r!r"r r r register_lock_callbacksrKcCstj||dS)zRemove the registered callback functions for lock acquire and release. The arguments are the same as used in `register_lock_callback()`. N)r2rr%rJr r r unregister_lock_callbacksrLc@seZdZdZdS) _DeadPointerz: Dummy class to make error messages more helpful. N)r+r,r-r.r r r r rMsrMc@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|dk r |ntd|_t|j|_||_dSr)rrQ_ptrbyref_as_parameter__owned)rZownedrOr r r rszOutputString.__init__cCs*|jdk r&|jrt|jd|_|`dSr)rUrXr2ZLLVMPY_DisposeStringrWrr r r closes   zOutputString.closecCs|Srr rr r r r'szOutputString.__enter__cCs |dSrrYrexc_typeexc_valexc_tbr r r r*szOutputString.__exit__cCs|s|jdk r|dSrrZrrr r r __del__s zOutputString.__del__cCs*|jdkrdS|jj}|dk s"tt|S)Nz)rUvalueAssertionErrorr)rsr r r __str__s   zOutputString.__str__cCs t|jSrboolrUrr r r __bool__szOutputString.__bool__cCs|jjS)z:Get the raw bytes of content of the char pointer. )rUrarr r r bytesszOutputString.bytes)TN)r+r,r-r.rMrW classmethodrTrrYr'r*rr`rdrg __nonzero__r<rhr r r r rNs  rNcCs|dk rtt|SdS)z,To wrap string return-value from C-API. N)strrNrTrSr r r ret_stringsrmcCs|dk rt|jSdS)z+To wrap bytes return-value from C-API. N)rNrTrhrlr r r ret_bytessrnc@seZdZdZdZeZdZddZddZ ddZ d d Z e d d Z d dZddZefddZddZddZeZddZdS) ObjectRefzJ A wrapper around a ctypes pointer to a LLVM object ("resource"). FcCs&|dkrtd||_||_t|_dS)Nz NULL pointer) ValueErrorrUrWr2Z_capi)rrSr r r rs zObjectRef.__init__cCs(z|js|js|W5|XdS)zI Close this object and do any required clean-up actions. N)detach_closedrX_disposerr r r rYs  zObjectRef.closecCs|js|`d|_d|_dS)zN Detach the underlying LLVM resource without disposing of it. TN)rrrWrUrr r r rq'szObjectRef.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 rr r r rs0szObjectRef._disposecCs|jS)zf Whether this object has been closed. A closed object can't be used anymore. )rrrr r r closed7szObjectRef.closedcCs(t|dst|jr$td|jf|S)NrYz%s instance already closed)hasattrrbrr RuntimeError __class__rr r r r'?szObjectRef.__enter__cCs |dSrrZr[r r r r*EszObjectRef.__exit__cCs|s|jdk r|dSrrZr_r r r r`Hs zObjectRef.__del__cCs t|jSrrerr r r rgMszObjectRef.__bool__cCs.t|dsdSt|jdt|jdkS)NrUFr)rur addressofrU)rotherr r r __eq__Ps  zObjectRef.__eq__cCstt|jtjjSr)hashrrPrUc_void_prarr r r __hash__YszObjectRef.__hash__N)r+r,r-r.rrrMrWrXrrYrqrsr<rtr'r*rr`rgrzrjr}r r r r ro s"    ro)sd ((">