
    de8/                        d dl Z d dlZd dlZd dlmZ d dlmZmZ d dl	m
Z
 d Z ed          Z ed          Z ed          Z ed          Z ed	          Z ed
          Z ed          Z ed          Z ed          Z ed          Z ed          Z ed          Z ed          Z ed          Z ed          Z ed          Z ed          Z ed          Z ed          Z ed          Z ed          Z  ed          Z! ed          Z" ed          Z# ed          Z$ ed          Z% ed          Z& ed           Z' G d! d"          Z( G d# d$          Z) G d% d&e*          Z+ G d' d(e*          Z,d) Z-e j.        dd*         d+k    re-nej/        Z0 e+            Z1d, Z2d- Z3 G d. d/e*          Z4 G d0 d1e*          Z5d2 Z6d3 Z7 G d4 d5e*          Z8dS )6    N)_decode_string_is_shutting_down)get_library_namec                 b    t          | t          j        fi           }t          j        |          S N)typectypes	StructurePOINTER)namenewclss     4lib/python3.11/site-packages/llvmlite/binding/ffi.py_make_opaque_refr   
   s)    $)+R00F>&!!!    LLVMContext
LLVMModule	LLVMValueLLVMTypeLLVMExecutionEngineLLVMPassManagerBuilderLLVMPassManagerLLVMTargetDataLLVMTargetLibraryInfo
LLVMTargetLLVMTargetMachineLLVMMemoryBufferLLVMAttributeListIteratorLLVMElementIteratorLLVMAttributeSetIteratorLLVMGlobalsIteratorLLVMFunctionsIteratorLLVMBlocksIteratorLLVMArgumentsIteratorLLVMInstructionsIteratorLLVMOperandsIteratorLLVMIncomingBlocksIteratorLLVMTypesIteratorLLVMObjectCacheLLVMObjectFileLLVMSectionIteratorLLVMOrcLLJITRefLLVMOrcDylibTrackerRefc                   0    e Zd ZdZd Zd Zd Zd Zd ZdS )	_LLVMLocka6  A 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.
    c                 D    t          j                    | _        g | _        d S r   )	threadingRLock_lock_cblistselfs    r   __init__z_LLVMLock.__init__6   s     _&&
r   c                 >    | j                             ||f           dS )zRegister callbacks that are invoked immediately after the lock is
        acquired (``acq_fn()``) and immediately before the lock is released
        (``rel_fn()``).
        N)r3   appendr5   acq_fnrel_fns      r   registerz_LLVMLock.register<   s%    
 	VV,-----r   c                 >    | j                             ||f           dS )z)Remove the registered callbacks.
        N)r3   remover9   s      r   
unregisterz_LLVMLock.unregisterC   s%     	VV,-----r   c                 f    | j                                          | j        D ]\  }} |             d S r   )r2   acquirer3   r9   s      r   	__enter__z_LLVMLock.__enter__H   sB    
"l 	 	NFFFHHHH	 	r   c                 f    | j         D ]\  }} |             | j                                         d S r   )r3   r2   release)r5   exc_detailsr:   r;   s       r   __exit__z_LLVMLock.__exit__N   s>    "l 	 	NFFFHHHH
r   N)	__name__
__module____qualname____doc__r6   r<   r?   rB   rF    r   r   r.   r.   -   si           . . .. . .
      r   r.   c                        e Zd Zd Zd Zd ZdS )_suppress_cleanup_errorsc                     || _         d S r   )_context)r5   contexts     r   r6   z!_suppress_cleanup_errors.__init__V   s    r   c                 4    | j                                         S r   )rO   rB   r4   s    r   rB   z"_suppress_cleanup_errors.__enter__Y   s    }&&(((r   c                 ^    	 | j                             |||          S # t          $ r Y d S w xY wr   )rO   rF   PermissionError)r5   exc_type	exc_value	tracebacks       r   rF   z!_suppress_cleanup_errors.__exit__\   sC    	=))(IyIII 	 	 	DD	s    
,,N)rG   rH   rI   r6   rB   rF   rK   r   r   rM   rM   U   sA             ) ) )    r   rM   c                   n    e Zd ZdZg dZd Zd Zed             Zd Z	ed             Z
e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_handle_fntabr2   c                 H    d | _         i | _        t                      | _        d S r   )rY   rZ   r.   r2   r4   s    r   r6   z_lib_wrapper.__init__k   s     [[


r   c                    	 t          t          t                              d          d         t	                                          5 }t          j        t          |                    | _        | j        	                                }d d d            d S # 1 swxY w Y   d S # t          t          f$ r}t          d          |d }~ww xY w)N.r   z&Could not find/load shared object file)rM   _importlib_resources_pathrG   
rpartitionr   r	   CDLLstrrY   LLVMPY_GetVersionInfoOSErrorAttributeError)r5   lib_path_es       r   	_load_libz_lib_wrapper._load_libp   s   	K)*C'',,Q/$&&+( +( ) ) =,4#);s8}}#=#= $::<<= = = = = = = = = = = = = = = = = = ( 	K 	K 	K
 BCCJ	Ks=   AB# 	A B	B# BB# BB# #C	4CC	c                 F    | j         s|                                  | j         S r   )rY   rh   r4   s    r   _libz_lib_wrapper._lib   s(      	NNr   c                     	 | j         |         S # t          $ r9 t          | j        |          }t	          | j        |          }|| j         |<   |cY S w xY wr   )rZ   KeyErrorgetattrrj   _lib_fn_wrapperr2   )r5   r   cfnwrappeds       r   __getattr__z_lib_wrapper.__getattr__   sf    	;t$$ 	 	 	$)T**C%dj#66G 'DKNNN	s    A AAc                     | j         j        S )zgThe name of the library passed in the CDLL constructor.

        For duck-typing a ctypes.CDLL
        )rj   _namer4   s    r   rs   z_lib_wrapper._name   s     yr   c                     | j         j        S )z]The system handle used to access the library.

        For duck-typing a ctypes.CDLL
        )rj   _handler4   s    r   ru   z_lib_wrapper._handle   s     y  r   N)rG   rH   rI   rJ   	__slots__r6   rh   propertyrj   rq   rs   ru   rK   r   r   rX   rX   c   s         
 322I! ! !
K K K     X      X ! ! X! ! !r   rX   c                       e Zd ZdZddgZd Zed             Zej        d             Zed             Z	e	j        d             Z	d	 Z
d
S )rn   zWraps 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.
    r2   _cfnc                 "    || _         || _        d S r   r2   ry   )r5   lockro   s      r   r6   z_lib_fn_wrapper.__init__   s    
			r   c                     | j         j        S r   ry   argtypesr4   s    r   r   z_lib_fn_wrapper.argtypes   s    y!!r   c                     || j         _        d S r   r~   )r5   r   s     r   r   z_lib_fn_wrapper.argtypes   s    %	r   c                     | j         j        S r   ry   restyper4   s    r   r   z_lib_fn_wrapper.restype   s    y  r   c                     || j         _        d S r   r   )r5   r   s     r   r   z_lib_fn_wrapper.restype   s    #	r   c                 ^    | j         5   | j        |i |cd d d            S # 1 swxY w Y   d S r   r{   )r5   argskwargss      r   __call__z_lib_fn_wrapper.__call__   s    Z 	. 	.49d-f--	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	.s   "&&N)rG   rH   rI   rJ   rv   r6   rw   r   setterr   r   rK   r   r   rn   rn      s          &!I   " " X" _& & _& ! ! X! ^$ $ ^$. . . . .r   rn   c                 T    t          j        t          j        |           |z            S )a  Replacement 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.
    )_impresas_filefiles)packageresources     r   _importlib_resources_path_replr      s#     ?7=11H<===r      )   	   c                 F    t           j                            | |           dS )z~Register callback functions for lock acquire and release.
    *acq_fn* and *rel_fn* are callables that take no arguments.
    N)libr2   r<   r:   r;   s     r   register_lock_callbackr      s"     Ivv&&&&&r   c                 F    t           j                            | |           dS )zRemove the registered callback functions for lock acquire and release.
    The arguments are the same as used in `register_lock_callback()`.
    N)r   r2   r?   r   s     r   unregister_lock_callbackr      s"     I(((((r   c                       e Zd ZdZdS )_DeadPointerz:
    Dummy class to make error messages more helpful.
    N)rG   rH   rI   rJ   rK   r   r   r   r      s           r   r   c                       e Zd ZdZ e            Zed             ZddZd Z	d Z
d Zefd	Zd
 Zd ZeZed             ZdS )OutputStringz<
    Object for managing the char* output of LLVM APIs.
    c                 T     | t          j        |t           j                            S )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)r	   castc_char_p)clsptrs     r   from_returnzOutputString.from_return   s%     sC99::::r   TNc                     ||nt          j        d           | _        t          j        | j                  | _        || _        d S r   )r	   r   _ptrbyref_as_parameter__owned)r5   ownedr   s      r   r6   zOutputString.__init__   s;     ,DD&/$2G2G	$l4955r   c                 v    | j         1| j        rt                              | j                    d | _         | `d S d S r   )r   r   r   LLVMPY_DisposeStringr   r4   s    r   closezOutputString.close  sE    9 { 4((333DI###	 ! r   c                     | S r   rK   r4   s    r   rB   zOutputString.__enter__  s    r   c                 .    |                                   d S r   r   r5   rT   exc_valexc_tbs       r   rF   zOutputString.__exit__      

r   c                 X     |            s| j         |                                   d S d S d S r   r   r5   r   s     r   __del__zOutputString.__del__  s>     ! "" 	z%

	 	%%r   c                 R    | j         dS | j         j        }|J t          |          S )Nz<dead OutputString>)r   valuer   )r5   ss     r   __str__zOutputString.__str__  s0    9((IO}}}a   r   c                 *    t          | j                  S r   boolr   r4   s    r   __bool__zOutputString.__bool__      DIr   c                     | j         j        S )z:Get the raw bytes of content of the char pointer.
        )r   r   r4   s    r   byteszOutputString.bytes$  s     yr   )TN)rG   rH   rI   rJ   r   r   classmethodr   r6   r   rB   rF   r   r   r   r   __nonzero__rw   r   rK   r   r   r   r      s          "\^^N	; 	; [	;   
$ $ $     ):    ! ! !   K  X  r   r   c                 X    | 't          t                              |                     S dS )z,To wrap string return-value from C-API.
    N)ra   r   r   r   s    r   
ret_stringr   +  s,     <++C00111 r   c                 H    | t                               |           j        S dS )z+To wrap bytes return-value from C-API.
    N)r   r   r   r   s    r   	ret_bytesr   2  s'     '',,22 r   c                       e Zd ZdZdZ e            ZdZd Zd Z	d Z
d Zed             Zd Zd	 Zefd
Zd Zd ZeZd ZdS )	ObjectRefzJ
    A wrapper around a ctypes pointer to a LLVM object ("resource").
    Fc                 \    |t          d          || _        || _        t          | _        d S )NzNULL pointer)
ValueErrorr   r   r   _capi)r5   r   s     r   r6   zObjectRef.__init__B  s/    ;^,,,	!


r   c                     	 | j         s| j        s|                                  |                                  dS # |                                  w xY w)zI
        Close this object and do any required clean-up actions.
        N)_closedr   _disposedetachr4   s    r   r   zObjectRef.closeI  sJ    	<    KKMMMMMDKKMMMMs	   ": Ac                 8    | j         s| `d| _         d| _        dS dS )zN
        Detach the underlying LLVM resource without disposing of it.
        TN)r   r   r   r4   s    r   r   zObjectRef.detachS  s0     | 	#DLDIII	 	r   c                     dS )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).
        NrK   r4   s    r   r   zObjectRef._dispose\  s      r   c                     | j         S )zf
        Whether this object has been closed.  A closed object can't
        be used anymore.
        )r   r4   s    r   closedzObjectRef.closedc  s     |r   c                 f    t          | d          sJ | j        rt          | j        d          | S )Nr   z instance already closed)hasattrr   RuntimeError	__class__r4   s    r   rB   zObjectRef.__enter__k  s?    tW%%%%%< 	Qt~~~OPPPr   c                 .    |                                   d S r   r   r   s       r   rF   zObjectRef.__exit__q  r   r   c                 X     |            s| j         |                                   d S d S d S r   r   r   s     r   r   zObjectRef.__del__t  s<      "" 	z%

	 	%%r   c                 *    t          | j                  S r   r   r4   s    r   r   zObjectRef.__bool__y  r   r   c                     t          |d          sdS t          j        | j        d                   t          j        |j        d                   k    S )Nr   Fr   )r   r	   	addressofr   )r5   others     r   __eq__zObjectRef.__eq__|  sK    uf%% 	5	!--UZ]++, 	,r   c                 n    t          t          j        | j        t          j                  j                  S r   )hashr	   r   r   c_void_pr   r4   s    r   __hash__zObjectRef.__hash__  s#    FK	6?;;ABBBr   N)rG   rH   rI   rJ   r   r   r   r   r6   r   r   r   rw   r   rB   rF   r   r   r   r   r   r   rK   r   r   r   r   9  s          G!\^^NF           X     ):    
  , , , KC C C C Cr   r   )9sysr	   r0   importlib.resources	resourcesr   llvmlite.binding.commonr   r   llvmlite.utilsr   r   LLVMContextRefLLVMModuleRefLLVMValueRefLLVMTypeRefLLVMExecutionEngineRefLLVMPassManagerBuilderRefLLVMPassManagerRefLLVMTargetDataRefLLVMTargetLibraryInfoRefLLVMTargetRefLLVMTargetMachineRefLLVMMemoryBufferRefr   r   r   r    r!   r"   r#   r$   r%   r&   r'   LLVMObjectCacheRefLLVMObjectFileRefLLVMSectionIteratorRefr+   r,   r.   rM   objectrX   rn   r   version_infopathr^   r   r   r   r   r   r   r   r   rK   r   r   <module>r      s$   



      % % % % % % E E E E E E E E + + + + + +" " "
 "!-00  ..,,z**))*?@@ ,,-EFF %%&788 $$%566 ++,CDD   ..''(;<< &&'9:: ,,-HII &&'<== ++,FGG &&'<== (()@AA %%&:;; (()@AA ++,FGG ''(>?? --.JKK $$%899 %%&788 $$%566 ))*?@@ ""#455))*BCC % % % % % % % %P       ;! ;! ;! ;! ;!6 ;! ;! ;!|. . . . .f . . .D> > > !$ 0! 4 > > <;"), 
 lnn' ' ') ) )    6   ; ; ; ; ;6 ; ; ;|2 2 23 3 3MC MC MC MC MC MC MC MC MC MCr   