U jg.@sddlZddlZddlZddlmZddlmZmZddl m Z ddZ e dZ e dZ e dZe d Ze d Ze d Ze d Ze d Ze dZe dZe dZe dZe dZe dZe dZe dZe dZe dZe dZe dZe dZ e dZ!e dZ"e dZ#e dZ$e dZ%Gd d!d!Z&Gd"d#d#Z'Gd$d%d%e(Z)Gd&d'd'e(Z*d(d)Z+ej,dd*d+krte+nej-Z.e)Z/d,d-Z0d.d/Z1Gd0d1d1e(Z2Gd2d3d3e(Z3d4d5Z4d6d7Z5Gd8d9d9e(Z6dS):N)_decode_string_is_shutting_down)get_library_namecCst|tjfi}t|SN)typectypes StructurePOINTER)nameZnewclsr 8/tmp/pip-unpacked-wheel-lnc_7cdn/llvmlite/binding/ffi.py_make_opaque_ref sr Z LLVMContextZ LLVMModuleZ LLVMValueZLLVMTypeZLLVMExecutionEngineZLLVMPassManagerBuilderZLLVMPassManagerZLLVMTargetDataZLLVMTargetLibraryInfoZ LLVMTargetZLLVMTargetMachineZLLVMMemoryBufferLLVMAttributeListIteratorLLVMAttributeSetIteratorLLVMGlobalsIteratorLLVMFunctionsIteratorLLVMBlocksIteratorLLVMArgumentsIteratorLLVMInstructionsIteratorLLVMOperandsIteratorLLVMTypesIteratorZLLVMObjectCacheZLLVMObjectFileZLLVMSectionIteratorLLVMOrcLLJITRefLLVMOrcDylibTrackerRefc@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__4s 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 register:sz_LLVMLock.registercCs|j||fdS)z)Remove the registered callbacks. N)rremover"r r r unregisterAsz_LLVMLock.unregistercCs$|j|jD]\}}|qdSr)racquirerr"r r r __enter__Fs z_LLVMLock.__enter__cGs$|jD]\}}|q|jdSr)rrrelease)r exc_detailsr#r$r r r __exit__Lsz_LLVMLock.__exit__N) __name__ __module__ __qualname____doc__r r%r'r)r,r r r r r+s rc@s$eZdZddZddZddZdS)_suppress_cleanup_errorscCs ||_dSr)_context)rcontextr r r r Tsz!_suppress_cleanup_errors.__init__cCs |jSr)r2r)rr r r r)Wsz"_suppress_cleanup_errors.__enter__cCs,z|j|||WStk r&YnXdSr)r2r,PermissionError)rexc_type exc_value tracebackr r r r,Zsz!_suppress_cleanup_errors.__exit__N)r-r.r/r r)r,r r r r r1Ssr1c@sVeZdZdZdddgZddZddZed d Zd d Z ed dZ eddZ dS) _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_handle_fntabrcCsd|_i|_t|_dSr)r9r:rrrr r r r isz_lib_wrapper.__init__c CszzDtttddt }tt||_|j }W5QRXWn0t t fk rt}zt d|W5d}~XYnXdS)N.rz&Could not find/load shared object file) r1_importlib_resources_pathr- rpartitionrrCDLLstrr9ZLLVMPY_GetVersionInfoOSErrorAttributeError)rZlib_path_er r r _load_libns z_lib_wrapper._load_libcCs|js||jSr)r9rDrr r r _lib}sz_lib_wrapper._libcCsLz |j|WStk rFt|j|}t|j|}||j|<|YSXdSr)r:KeyErrorgetattrrE_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 )rE_namerr r r rLsz_lib_wrapper._namecCs|jjS)z]The system handle used to access the library. For duck-typing a ctypes.CDLL )rE_handlerr r r rMsz_lib_wrapper._handleN) r-r.r/r0 __slots__r rDpropertyrErKrLrMr r r r r8as    r8c@s\eZdZdZddgZddZeddZejddZed d Z e jd d Z d d Z dS)rHzWraps 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||_||_dSrrrP)rlockrIr r r r sz_lib_fn_wrapper.__init__cCs|jjSrrPargtypesrr r r rTsz_lib_fn_wrapper.argtypescCs ||j_dSrrS)rrTr r r rTscCs|jjSrrPrestyperr r r rVsz_lib_fn_wrapper.restypecCs ||j_dSrrU)rrVr r r rVsc Os*|j|j||W5QRSQRXdSrrQ)rargskwargsr r r __call__sz_lib_fn_wrapper.__call__N) r-r.r/r0rNr rOrTsetterrVrYr r r r rHs    rHcCstt||S)aReplacement implementation of `import.resources.path` to avoid deprecation warning following code at importlib_resources/_legacy.py as suggested by https://importlib-resources.readthedocs.io/en/latest/using.html#migrating-from-legacy Notes on differences from importlib.resources implementation: The `_common.normalize_path(resource)` call is skipped because it is an internal API and it is unnecessary for the use here. What it does is ensuring `resource` is a str and that it does not contain path separators. )_impresZas_filefiles)packageresourcer r r _importlib_resources_path_repls r_) cCstj||dS)z~Register callback functions for lock acquire and release. *acq_fn* and *rel_fn* are callables that take no arguments. N)librr%r#r$r r r register_lock_callbacksrecCstj||dS)zRemove the registered callback functions for lock acquire and release. The arguments are the same as used in `register_lock_callback()`. N)rcrr'rdr r r unregister_lock_callbacksrfc@seZdZdZdS) _DeadPointerz: Dummy class to make error messages more helpful. N)r-r.r/r0r r r r rgsrgc@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)rrk_ptrbyref_as_parameter__owned)rZownedrir r r r szOutputString.__init__cCs*|jdk r&|jrt|jd|_|`dSr)rorrrcZLLVMPY_DisposeStringrqrr r r closes   zOutputString.closecCs|Srr rr r r r) szOutputString.__enter__cCs |dSrrsrr5exc_valexc_tbr r r r, szOutputString.__exit__cCs|s|jdk r|dSrrtrrr r r __del__s zOutputString.__del__cCs*|jdkrdS|jj}|dk s"tt|S)Nz)rovalueAssertionErrorr)rsr r r __str__s   zOutputString.__str__cCs t|jSrboolrorr r r __bool__szOutputString.__bool__cCs|jjS)z:Get the raw bytes of content of the char pointer. )rorzrr r r bytes"szOutputString.bytes)TN)r-r.r/r0rgrq classmethodrnr rsr)r,rryr}r __nonzero__rOrr r r r rhs  rhcCs|dk rtt|SdS)z,To wrap string return-value from C-API. N)r?rhrnrmr r r ret_string)srcCs|dk rt|jSdS)z+To wrap bytes return-value from C-API. N)rhrnrrr r r ret_bytes0src@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) ValueErrorrorqrcZ_capi)rrmr r r r @s zObjectRef.__init__cCs(z|js|js|W5|XdS)zI Close this object and do any required clean-up actions. N)detach_closedrr_disposerr r r rsGs  zObjectRef.closecCs|js|`d|_d|_dS)zN Detach the underlying LLVM resource without disposing of it. TN)rrqrorr r r rQszObjectRef.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 rZszObjectRef._disposecCs|jS)zf Whether this object has been closed. A closed object can't be used anymore. )rrr r r closedaszObjectRef.closedcCs(t|dst|jr$td|jf|S)Nrsz%s instance already closed)hasattrr{r RuntimeError __class__rr r r r)iszObjectRef.__enter__cCs |dSrrtrur r r r,oszObjectRef.__exit__cCs|s|jdk r|dSrrtrxr r r ryrs zObjectRef.__del__cCs t|jSrr~rr r r rwszObjectRef.__bool__cCs.t|dsdSt|jdt|jdkS)NroFr)rr addressofro)rotherr r r __eq__zs  zObjectRef.__eq__cCstt|jtjjSr)hashrrjroc_void_przrr r r __hash__szObjectRef.__hash__N)r-r.r/r0rrgrqrrr rsrrrOrr)r,rryrrrrr r r r r7s"    r)7sysrrimportlib.resources resourcesr[Zllvmlite.binding.commonrrZllvmlite.utilsrr ZLLVMContextRefZ LLVMModuleRefZ LLVMValueRefZ LLVMTypeRefZLLVMExecutionEngineRefZLLVMPassManagerBuilderRefZLLVMPassManagerRefZLLVMTargetDataRefZLLVMTargetLibraryInfoRefZ LLVMTargetRefZLLVMTargetMachineRefZLLVMMemoryBufferRefrrrrrrrrrZLLVMObjectCacheRefZLLVMObjectFileRefZLLVMSectionIteratorRefrrrr1objectr8rHr_ version_infopathr<rcrerfrgrhrrrr r r r sb  (>">