U
    ¤½|e+  ã                
   @   s4  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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„ dƒZ$G d d!„ d!e%ƒZ&G d"d#„ d#e%ƒZ'd$d%„ Z(e j)dd&… d'k�rVe(nej*Z+e
ƒ Z,d( -e. /d(¡d d)… ¡Z0z e+e0e,ƒZ1e 2e3e1 4¡ ƒ¡Z5W n< e6k
�rÖ Z7 zd*e,› d+e7› �Z8e6e8ƒ‚W 5 dZ7[7X Y nX e&e5ƒZ5d,d-„ Z9d.d/„ Z:G d0d1„ d1e%ƒZ;G d2d3„ d3e%ƒZ<d4d5„ Z=d6d7„ Z>G d8d9„ d9e%ƒZ?dS ):é    N)Ú_decode_stringÚ_is_shutting_down)Úget_library_namec                 C   s   t | tjfi ƒ}t |¡S ©N)ÚtypeÚctypesÚ	StructureÚPOINTER)ÚnameZnewcls© r   úQ/var/www/website-v5/atlas_env/lib/python3.8/site-packages/llvmlite/binding/ffi.pyÚ_make_opaque_ref
   s    r   ZLLVMContextZ
LLVMModuleZ	LLVMValueZLLVMTypeZLLVMExecutionEngineZLLVMPassManagerBuilderZLLVMPassManagerZLLVMTargetDataZLLVMTargetLibraryInfoZ
LLVMTargetZLLVMTargetMachineZLLVMMemoryBufferÚLLVMAttributeListIteratorÚLLVMAttributeSetIteratorÚLLVMGlobalsIteratorÚLLVMFunctionsIteratorÚLLVMBlocksIteratorÚLLVMArgumentsIteratorÚLLVMInstructionsIteratorÚLLVMOperandsIteratorÚLLVMTypesIteratorZLLVMObjectCacheZLLVMObjectFileZLLVMSectionIteratorc                   @   s8   e Zd ZdZdd„ Zdd„ Zdd„ Zdd	„ Zd
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                 C   s   t  ¡ | _g | _d S r   )Ú	threadingÚRLockÚ_lockÚ_cblist©Úselfr   r   r   Ú__init__2   s    
z_LLVMLock.__init__c                 C   s   | j  ||f¡ dS )z®Register callbacks that are invoked immediately after the lock is
        acquired (``acq_fn()``) and immediately before the lock is released
        (``rel_fn()``).
        N)r   Úappend©r   Úacq_fnÚrel_fnr   r   r   Úregister8   s    z_LLVMLock.registerc                 C   s   | j  ||f¡ dS )z)Remove the registered callbacks.
        N)r   Úremover    r   r   r   Ú
unregister?   s    z_LLVMLock.unregisterc                 C   s$   | j  ¡  | jD ]\}}|ƒ  qd S r   )r   Úacquirer   r    r   r   r   Ú	__enter__D   s    
z_LLVMLock.__enter__c                 G   s$   | j D ]\}}|ƒ  q| j ¡  d S r   )r   r   Úrelease)r   Úexc_detailsr!   r"   r   r   r   Ú__exit__J   s    z_LLVMLock.__exit__N)	Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r#   r%   r'   r*   r   r   r   r   r   )   s   r   c                   @   sB   e Zd ZdZdddgZdd„ Zdd„ Zed	d
„ ƒZed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   c                 C   s   || _ i | _tƒ | _d S r   )r0   r1   r   r   )r   Úlibr   r   r   r   Y   s    z_lib_wrapper.__init__c                 C   sL   z| j | W S  tk
rF   t| j|ƒ}t| j|ƒ}|| j |< | Y S X d S r   )r1   ÚKeyErrorÚgetattrr0   Ú_lib_fn_wrapperr   )r   r
   ÚcfnÚwrappedr   r   r   Ú__getattr__^   s    
z_lib_wrapper.__getattr__c                 C   s   | j jS )zgThe name of the library passed in the CDLL constructor.

        For duck-typing a ctypes.CDLL
        )r0   Ú_namer   r   r   r   r9   h   s    z_lib_wrapper._namec                 C   s   | j jS )z]The system handle used to access the library.

        For duck-typing a ctypes.CDLL
        )r0   Ú_handler   r   r   r   r:   p   s    z_lib_wrapper._handleN)
r+   r,   r-   r.   Ú	__slots__r   r8   Úpropertyr9   r:   r   r   r   r   r/   Q   s   


r/   c                   @   s\   e Zd ZdZddgZdd„ Zedd„ ƒZejdd„ ƒZed	d
„ ƒZ	e	jdd
„ ƒZ	dd„ Z
dS )r5   zîWraps 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   Ú_cfnc                 C   s   || _ || _d S r   ©r   r=   )r   Úlockr6   r   r   r   r   ‚   s    z_lib_fn_wrapper.__init__c                 C   s   | j jS r   ©r=   Úargtypesr   r   r   r   rA   †   s    z_lib_fn_wrapper.argtypesc                 C   s   || j _d S r   r@   )r   rA   r   r   r   rA   Š   s    c                 C   s   | j jS r   ©r=   Úrestyper   r   r   r   rC   Ž   s    z_lib_fn_wrapper.restypec                 C   s   || j _d S r   rB   )r   rC   r   r   r   rC   ’   s    c              
   O   s*   | j � | j||ŽW  5 Q R £ S Q R X d S r   r>   )r   ÚargsÚkwargsr   r   r   Ú__call__–   s    z_lib_fn_wrapper.__call__N)r+   r,   r-   r.   r;   r   r<   rA   ÚsetterrC   rF   r   r   r   r   r5   y   s   



r5   c                 C   s   t  t  | ¡| ¡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Úresourcer   r   r   Ú_importlib_resources_path_repl›   s    rM   é   )é   é	   Ú.éÿÿÿÿz(Could not find/load shared object file: z
 Error was: c                 C   s   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)r2   r   r#   ©r!   r"   r   r   r   Úregister_lock_callbackÁ   s    rT   c                 C   s   t j | |¡ dS )z‘Remove the registered callback functions for lock acquire and release.
    The arguments are the same as used in `register_lock_callback()`.
    N)r2   r   r%   rS   r   r   r   Úunregister_lock_callbackÈ   s    rU   c                   @   s   e Zd ZdZdS )Ú_DeadPointerz:
    Dummy class to make error messages more helpful.
    N)r+   r,   r-   r.   r   r   r   r   rV   Ï   s   rV   c                   @   sp   e Zd ZdZeƒ Zedd„ ƒZddd„Zdd	„ Z	d
d„ Z
dd„ Zefdd„Zdd„ Zdd„ ZeZedd„ ƒZdS )ÚOutputStringz<
    Object for managing the char* output of LLVM APIs.
    c                 C   s   | t  |t j¡d�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Úptrr   r   r   Úfrom_returnÛ   s    
zOutputString.from_returnTNc                 C   s0   |d k	r|nt  d ¡| _t  | j¡| _|| _d S r   )r   rZ   Ú_ptrÚbyrefÚ_as_parameter_Ú_owned)r   ZownedrX   r   r   r   r   ç   s    zOutputString.__init__c                 C   s*   | j d k	r&| jrt | j ¡ d | _ | `d S r   )r^   ra   r2   ZLLVMPY_DisposeStringr`   r   r   r   r   Úcloseì   s
    
zOutputString.closec                 C   s   | S r   r   r   r   r   r   r'   ó   s    zOutputString.__enter__c                 C   s   |   ¡  d S r   ©rb   ©r   Úexc_typeÚexc_valÚexc_tbr   r   r   r*   ö   s    zOutputString.__exit__c                 C   s   |ƒ s| j d k	r|   ¡  d S r   rc   ©r   r   r   r   r   Ú__del__ù   s    
zOutputString.__del__c                 C   s*   | j d krdS | j j}|d k	s"t‚t|ƒS )Nz<dead OutputString>)r^   ÚvalueÚAssertionErrorr   )r   Úsr   r   r   Ú__str__   s
    
zOutputString.__str__c                 C   s
   t | jƒS r   ©Úboolr^   r   r   r   r   Ú__bool__  s    zOutputString.__bool__c                 C   s   | j jS )z:Get the raw bytes of content of the char pointer.
        )r^   rj   r   r   r   r   Úbytes  s    zOutputString.bytes)TN)r+   r,   r-   r.   rV   r`   Úclassmethodr]   r   rb   r'   r*   r   ri   rm   rp   Ú__nonzero__r<   rq   r   r   r   r   rW   Õ   s   

rW   c                 C   s   | dk	rt t | ¡ƒS dS )z,To wrap string return-value from C-API.
    N)ÚstrrW   r]   ©r\   r   r   r   Ú
ret_string  s    rv   c                 C   s   | dk	rt  | ¡jS dS )z+To wrap bytes return-value from C-API.
    N)rW   r]   rq   ru   r   r   r   Ú	ret_bytes  s    rw   c                   @   s‚   e Zd ZdZdZeƒ ZdZdd„ Zdd„ Z	dd„ Z
d	d
„ Zedd„ ƒZdd„ Zdd„ Zefdd„Zdd„ Zdd„ ZeZdd„ ZdS )Ú	ObjectRefzJ
    A wrapper around a ctypes pointer to a LLVM object ("resource").
    Fc                 C   s&   |d krt dƒ‚|| _|| _t| _d S )NzNULL pointer)Ú
ValueErrorr^   r`   r2   Z_capi)r   r\   r   r   r   r   *  s
    zObjectRef.__init__c                 C   s(   z| js| js|  ¡  W 5 |   ¡  X dS )zI
        Close this object and do any required clean-up actions.
        N)ÚdetachÚ_closedra   Ú_disposer   r   r   r   rb   1  s    zObjectRef.closec                 C   s   | j s| `d| _ d| _dS )zN
        Detach the underlying LLVM resource without disposing of it.
        TN)r{   r`   r^   r   r   r   r   rz   ;  s    zObjectRef.detachc                 C   s   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).
        Nr   r   r   r   r   r|   D  s    zObjectRef._disposec                 C   s   | j S )zf
        Whether this object has been closed.  A closed object can't
        be used anymore.
        )r{   r   r   r   r   ÚclosedK  s    zObjectRef.closedc                 C   s(   t | dƒst‚| jr$td| jf ƒ‚| S )Nrb   z%s instance already closed)Úhasattrrk   r{   ÚRuntimeErrorÚ	__class__r   r   r   r   r'   S  s    zObjectRef.__enter__c                 C   s   |   ¡  d S r   rc   rd   r   r   r   r*   Y  s    zObjectRef.__exit__c                 C   s   |ƒ s| j d k	r|   ¡  d S r   rc   rh   r   r   r   ri   \  s    
zObjectRef.__del__c                 C   s
   t | jƒS r   rn   r   r   r   r   rp   a  s    zObjectRef.__bool__c                 C   s.   t |dƒsdS t | jd ¡t |jd ¡kS )Nr^   Fr   )r~   r   Ú	addressofr^   )r   Úotherr   r   r   Ú__eq__d  s
    
ÿzObjectRef.__eq__c                 C   s   t t | jtj¡jƒS r   )Úhashr   rY   r^   Úc_void_prj   r   r   r   r   Ú__hash__m  s    zObjectRef.__hash__N)r+   r,   r-   r.   r{   rV   r`   ra   r   rb   rz   r|   r<   r}   r'   r*   r   ri   rp   rƒ   rs   r†   r   r   r   r   rx   !  s"   
	
rx   )@Úsysr   r   Úimportlib.resourcesÚ	resourcesrH   Úllvmlite.binding.commonr   r   Zllvmlite.utilsr   r   ZLLVMContextRefZLLVMModuleRefZLLVMValueRefZLLVMTypeRefZLLVMExecutionEngineRefZLLVMPassManagerBuilderRefZLLVMPassManagerRefZLLVMTargetDataRefZLLVMTargetLibraryInfoRefZLLVMTargetRefZLLVMTargetMachineRefZLLVMMemoryBufferRefr   r   r   r   r   r   r   r   r   ZLLVMObjectCacheRefZLLVMObjectFileRefZLLVMSectionIteratorRefr   Úobjectr/   r5   rM   Úversion_infoÚpathZ_importlib_resources_pathZ	_lib_nameÚjoinr+   ÚsplitZpkgnameZ_lib_handleÚCDLLrt   r'   r2   ÚOSErrorÚeÚmsgrT   rU   rV   rW   rv   rw   rx   r   r   r   r   Ú<module>   sp   (("ÿþ
ÿ>