U md+ @s4ddlZddlZddlZddlmZddlmZmZddl 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#GdddZ$Gd d!d!e%Z&Gd"d#d#e%Z'd$d%Z(ej)dd&d'krVe(nej*Z+e Z,d(-e./d(dd)Z0z e+e0e,Z1e2e3e14Z5Wn<e6k rZ7zd*e,d+e7Z8e6e8W5dZ7[7XYnXe&e5Z5d,d-Z9d.d/Z:Gd0d1d1e%Z;Gd2d3d3e%ZGd8d9d9e%Z?dS):N)_decode_string_is_shutting_down)get_library_namecCst|tjfi}t|SN)typectypesZ StructureZPOINTER)nameZnewclsr M/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/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__2s 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 register8sz_LLVMLock.registercCs|j||fdS)z)Remove the registered callbacks. N)rremoverr r r unregister?sz_LLVMLock.unregistercCs$|j|jD]\}}|qdSr)racquirerrr r r __enter__Ds z_LLVMLock.__enter__cGs$|jD]\}}|q|jdSr)rrrelease)r exc_detailsrr r r r __exit__Jsz_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)r.r/rr)rlibr r r rYsz_lib_wrapper.__init__cCsLz |j|WStk rFt|j|}t|j|}||j|<|YSXdSr)r/KeyErrorgetattrr._lib_fn_wrapperr)rrcfnwrappedr 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 )r._namerr r r r7hsz_lib_wrapper._namecCs|jjS)z]The system handle used to access the library. For duck-typing a ctypes.CDLL )r._handlerr r r r8psz_lib_wrapper._handleN) r)r*r+r, __slots__rr6propertyr7r8r r r r r-Qs   r-c@s\eZdZdZddgZddZeddZejddZed d Z e jd d Z d d Z dS)r3zWraps 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;)rlockr4r r r rsz_lib_fn_wrapper.__init__cCs|jjSrr;argtypesrr r r r?sz_lib_fn_wrapper.argtypescCs ||j_dSrr>)rr?r r r r?scCs|jjSrr;restyperr r r rAsz_lib_fn_wrapper.restypecCs ||j_dSrr@)rrAr r r rAsc Os*|j|j||W5QRSQRXdSrr<)rargskwargsr r r __call__sz_lib_fn_wrapper.__call__N) r)r*r+r,r9rr:r?setterrArDr r r r r3ys    r3cCstt||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 rJ) .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)r0rr!rr r r r register_lock_callbacksrQcCstj||dS)zRemove the registered callback functions for lock acquire and release. The arguments are the same as used in `register_lock_callback()`. N)r0rr#rPr r r unregister_lock_callbacksrRc@seZdZdZdS) _DeadPointerz: Dummy class to make error messages more helpful. N)r)r*r+r,r r r r rSsrSc@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)rrW_ptrZbyref_as_parameter__owned)rZownedrUr r r rszOutputString.__init__cCs*|jdk r&|jrt|jd|_|`dSr)r[r]r0ZLLVMPY_DisposeStringr\rr r r closes   zOutputString.closecCs|Srr rr r r r%szOutputString.__enter__cCs |dSrr^rexc_typeexc_valexc_tbr r r r(szOutputString.__exit__cCs|s|jdk r|dSrr_rrr r r __del__s zOutputString.__del__cCs*|jdkrdS|jj}|dk s"tt|S)Nz)r[valueAssertionErrorr)rsr r r __str__s   zOutputString.__str__cCs t|jSrboolr[rr r r __bool__szOutputString.__bool__cCs|jjS)z:Get the raw bytes of content of the char pointer. )r[rfrr r r bytes szOutputString.bytes)TN)r)r*r+r,rSr\ classmethodrZrr^r%r(rrerirl __nonzero__r:rmr r r r rTs  rTcCs|dk rtt|SdS)z,To wrap string return-value from C-API. N)strrTrZrYr r r ret_stringsrrcCs|dk rt|jSdS)z+To wrap bytes return-value from C-API. N)rTrZrmrqr r r ret_bytessrsc@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) ValueErrorr[r\r0Z_capi)rrYr 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_closedr]_disposerr r r r^1s  zObjectRef.closecCs|js|`d|_d|_dS)zN Detach the underlying LLVM resource without disposing of it. TN)rwr\r[rr r r rv;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 rxDszObjectRef._disposecCs|jS)zf Whether this object has been closed. A closed object can't be used anymore. )rwrr r r closedKszObjectRef.closedcCs(t|dst|jr$td|jf|S)Nr^z%s instance already closed)hasattrrgrw RuntimeError __class__rr r r r%SszObjectRef.__enter__cCs |dSrr_r`r r r r(YszObjectRef.__exit__cCs|s|jdk r|dSrr_rdr r r re\s zObjectRef.__del__cCs t|jSrrjrr r r rlaszObjectRef.__bool__cCs.t|dsdSt|jdt|jdkS)Nr[Fr)rzrZ addressofr[)rotherr r r __eq__ds  zObjectRef.__eq__cCstt|jtjjSr)hashrrVr[Zc_void_prfrr r r __hash__mszObjectRef.__hash__N)r)r*r+r,rwrSr\r]rr^rvrxr:ryr%r(rrerlr~rorr r r r rt!s"    rt)@sysrrimportlib.resources resourcesrFZllvmlite.binding.commonrrZllvmlite.utilsrr ZLLVMContextRefZ LLVMModuleRefZ LLVMValueRefZ LLVMTypeRefZLLVMExecutionEngineRefZLLVMPassManagerBuilderRefZLLVMPassManagerRefZLLVMTargetDataRefZLLVMTargetLibraryInfoRefZ LLVMTargetRefZLLVMTargetMachineRefZLLVMMemoryBufferRefr r rrrrrrrZLLVMObjectCacheRefZLLVMObjectFileRefZLLVMSectionIteratorRefrobjectr-r3rJ version_infopathZ_importlib_resources_pathZ _lib_namejoinr)splitpkgnameZ _lib_handleCDLLrpr%r0OSErroremsgrQrRrSrTrrrsrtr r r r sp  ((" >