U
    +d9                  
   @   sX  d dl Z d dlmZmZmZmZmZmZmZ d dl	m
Z
 d dlmZmZ dd ZG dd deZd	d
 Zdd Zdd ZddddZd(ddZdd ZG dd de
jZeddddgZeddddd d!gZG d"d# d#e
jZG d$d% d%e
jZd&d' Zeege
jj _!eege
jj"_!ee
jj"_#eege
jj$_!eege
jj%_!ege
jj&_!ee
jj&_#ege
jj'_!e
j(e
jj'_#e
j(eege
jj)_!e
j(ge
jj*_!e
j(e
j+ge
jj,_!ee
jj,_#e
j(e
j+ege
jj-_!ee
jj-_#e
j(e
j+ge
jj._!ee
jj._#e
j(e
j+ge
jj/_!ee
jj/_#eeege
jj0_!e
j1e
jj0_#e
j1ge
jj2_!ee
jj2_#e
j1ge
jj3_!ee
jj3_#e
j1eeeeeeeeeg
e
jj4_!e
j5e
jj4_#e
j5ge
jj6_!e
j5eege
jj7_!e
j5ege
jj8_!e
j5e
j9ge
jj:_!e
j5e
j;eeege
jj<_!e
j=e
jj<_#e
j=ge
jj>_!ee
jj>_#e
j=ge
jj?_!ee
jj?_#e
j=ge
jj@_!e
j5ge
jjA_!e
j(e
jjA_#g e
jjB_!ee
jjB_#dS ))    N)POINTERc_char_p
c_longlongc_intc_size_tc_void_p	string_at)ffi)_decode_string_encode_stringc               
   C   s4   t  "} t j|  t| W  5 Q R  S Q R X dS )a  
    Return a target triple suitable for generating code for the current process.
    An example when the default triple from ``get_default_triple()`` is not be
    suitable is when LLVM is compiled for 32-bit but the process is executing
    in 64-bit mode.
    N)r	   OutputStringlibLLVMPY_GetProcessTriplestrout r   </tmp/pip-unpacked-wheel-stw2luzp/llvmlite/binding/targets.pyget_process_triple	   s    
r   c                   @   s   e Zd ZdZdddZdS )
FeatureMapz
    Maps feature name to a boolean indicating the availability of the feature.
    Extends ``dict`` to add `.flatten()` method.
    Tc                    s>   |rt |  n
t|  }ddd d fdd|D S )ap  
        Args
        ----
        sort: bool
            Optional.  If True, the features are sorted by name; otherwise,
            the ordering is unstable between python session due to hash
            randomization.  Defaults to True.

        Returns a string suitable for use as the ``features`` argument to
        ``Target.create_target_machine()``.

        +-)TF,c                 3   s"   | ]\}}d   | |V  qdS )z{0}{1}N)format).0kvflag_mapr   r   	<genexpr>*   s   z%FeatureMap.flatten.<locals>.<genexpr>)sorteditemsiterjoin)selfsortiteratorr   r   r   flatten   s
    
zFeatureMap.flattenN)T)__name__
__module____qualname____doc__r'   r   r   r   r   r      s   r   c               
   C   s   t  v} t }t j| s,|W  5 Q R  S ddd}t| }|rn|dD ] }|rL||d  ||dd < qL|W  5 Q R  S Q R X dS )ac  
    Returns a dictionary-like object indicating the CPU features for current
    architecture and whether they are enabled for this CPU.  The key-value pairs
    are the feature name as string and a boolean indicating whether the feature
    is available.  The returned value is an instance of ``FeatureMap`` class,
    which adds a new method ``.flatten()`` for returning a string suitable for
    use as the "features" argument to ``Target.create_target_machine()``.

    If LLVM has not implemented this feature or it fails to get the information,
    this function will raise a RuntimeError exception.
    TF)r   r   r   r      N)r	   r   r   r   LLVMPY_GetHostCPUFeaturesr   split)r   Zoutdictr   contentZfeatr   r   r   get_host_cpu_features.   s    

r0   c               
   C   s4   t  "} t j|  t| W  5 Q R  S Q R X dS )zR
    Return the default target triple LLVM is configured to produce code for.
    N)r	   r   r   LLVMPY_GetDefaultTargetTripler   r   r   r   r   get_default_tripleG   s    
r2   c               
   C   s4   t  "} t j|  t| W  5 Q R  S Q R X dS )zm
    Get the name of the host's CPU, suitable for using with
    :meth:`Target.create_target_machine()`.
    N)r	   r   r   LLVMPY_GetHostCPUNamer   r   r   r   r   get_host_cpu_nameP   s    
r4   COFFELFZMachO)r,         c                 C   s&   | dkrt  } tjt| }t| S )z~
    Get the object format for the given *triple* string (or the default
    triple if omitted).
    A string is returned
    N)r2   r	   r   LLVMPY_GetTripleObjectFormatr   _object_formats)tripleresr   r   r   get_object_formata   s    r=   c                 C   s   t tjt| S )zE
    Create a TargetData instance for the given *layout* string.
    )
TargetDatar	   r   LLVMPY_CreateTargetDatar   )Zlayoutr   r   r   create_target_datam   s    r@   c                   @   s@   e Zd ZdZdd Zdd Zdd Zdd	 Zd
d Zdd Z	dS )r>   z{
    A TargetData provides structured access to a data layout.
    Use :func:`create_target_data` to create instances.
    c              
   C   s@   | j r
dS t $}tj| | t|W  5 Q R  S Q R X d S )Nz<dead TargetData>)_closedr	   r   r    LLVMPY_CopyStringRepOfTargetDatar   r$   r   r   r   r   __str__z   s
    
zTargetData.__str__c                 C   s   | j |  d S N)_capiLLVMPY_DisposeTargetDatar$   r   r   r   _dispose   s    zTargetData._disposec                 C   s   t j| |S )z1
        Get ABI size of LLVM type *ty*.
        )r	   r   LLVMPY_ABISizeOfType)r$   tyr   r   r   get_abi_size   s    zTargetData.get_abi_sizec                 C   s0   t j| ||}|dkr,td|t||S )zL
        Get byte offset of type's ty element at the given position
        zQCould not determined offset of {}th element of the type '{}'. Is it a structtype?)r	   r   LLVMPY_OffsetOfElement
ValueErrorr   r   )r$   rK   positionoffsetr   r   r   get_element_offset   s     zTargetData.get_element_offsetc                 C   s(   t j| |}|dkr$td|f |S )zI
        Get ABI size of pointee type of LLVM pointer type *ty*.
        rM   Not a pointer type: %s)r	   r   LLVMPY_ABISizeOfElementTypeRuntimeErrorr$   rK   sizer   r   r   get_pointee_abi_size   s    zTargetData.get_pointee_abi_sizec                 C   s(   t j| |}|dkr$td|f |S )zV
        Get minimum ABI alignment of pointee type of LLVM pointer type *ty*.
        rM   rS   )r	   r    LLVMPY_ABIAlignmentOfElementTyperU   rV   r   r   r   get_pointee_abi_alignment   s    z$TargetData.get_pointee_abi_alignmentN)
r(   r)   r*   r+   rD   rI   rL   rR   rX   rZ   r   r   r   r   r>   t   s   	r>   defaultZstaticZpicZdynamicnopic
jitdefaultsmallkernelZmediumZlargec                   @   s^   e Zd ZdZedd Zedd Zedd Zedd	 Z	ed
d Z
dd ZdddZdS )Target c                 C   s   t  }| |S )zB
        Create a Target instance for the default triple.
        )r2   from_triple)clsr;   r   r   r   from_default_triple   s    zTarget.from_default_triplec              
   C   sV   t  D}t j|d|}|s.tt|| |}||_|W  5 Q R  S Q R X dS )zK
        Create a Target instance for the given triple (a string).
        utf8N)r	   r   r   LLVMPY_GetTargetFromTripleencoderU   r   _triple)rb   r;   outerrtargetr   r   r   ra      s    
zTarget.from_triplec                 C   s   t j| }t|S rE   )r	   r   LLVMPY_GetTargetNamer
   r$   sr   r   r   name   s    zTarget.namec                 C   s   t j| }t|S rE   )r	   r   LLVMPY_GetTargetDescriptionr
   rk   r   r   r   description   s    zTarget.descriptionc                 C   s   | j S rE   )rg   rH   r   r   r   r;      s    zTarget.triplec                 C   s   d | j| jS )Nz<Target {0} ({1})>)r   rm   ro   rH   r   r   r   rD      s    zTarget.__str__r7   r[   r\   Fc	                 C   s   d|  krdksn t |tks&t |tks2t | j}	tjdkrR|dkrR|	d7 }	tj| t	|	t	|t	||t	|t	|t
|t
|t	|
}
|
rt|
S tddS )am  
        Create a new TargetMachine for this target and the given options.

        Specifying codemodel='default' will result in the use of the "small"
        code model. Specifying codemodel='jitdefault' will result in the code
        model being picked based on platform bitness (32="small", 64="large").

        The `printmc` option corresponds to llvm's `-print-machineinstrs`.

        The `jit` option should be set when the target-machine is to be used
        in a JIT engine.

        The `abiname` option specifies the ABI. RISC-V targets with hard-float
        needs to pass the ABI name to LLVM.
        r   r8   ntr\   z-elfzCannot create target machineN)AssertionErrorRELOC	CODEMODELrg   osrm   r	   r   LLVMPY_CreateTargetMachiner   intTargetMachinerU   )r$   cpufeaturesoptZrelocZ	codemodelZprintmcZjitZabinamer;   tmr   r   r   create_target_machine   s(    zTarget.create_target_machineN)r`   r`   r7   r[   r\   FFr`   )r(   r)   r*   rg   classmethodrc   ra   propertyrm   ro   r;   rD   r|   r   r   r   r   r_      s(   




           r_   c                   @   sV   e Zd Zdd Zdd Zdd Zdd Zd	d
 ZdddZe	dd Z
e	dd ZdS )rw   c                 C   s   | j |  d S rE   )rF   LLVMPY_DisposeTargetMachinerH   r   r   r   rI   	  s    zTargetMachine._disposec                 C   s   t j| | dS )zW
        Register analysis passes for this target machine with a pass manager.
        N)r	   r   LLVMPY_AddAnalysisPasses)r$   Zpmr   r   r   add_analysis_passes  s    z!TargetMachine.add_analysis_passesc                 C   s   t j| | dS )z
        Set whether this target machine will emit assembly with human-readable
        comments describing control flow, debug information, and so on.
        N)r	   r   #LLVMPY_SetTargetMachineAsmVerbosity)r$   verboser   r   r   set_asm_verbosity  s    zTargetMachine.set_asm_verbosityc                 C   s   | j |ddS )z
        Represent the module as a code object, suitable for use with
        the platform's linker.  Returns a byte string.
        T
use_object)_emit_to_memoryr$   moduler   r   r   emit_object  s    zTargetMachine.emit_objectc                 C   s   t | j|ddS )z
        Return the raw assembler of the module, as a string.

        llvm.initialize_native_asmprinter() must have been called first.
        Fr   )r
   r   r   r   r   r   emit_assembly   s    zTargetMachine.emit_assemblyFc              	   C   st   t  ,}t j| |t||}|s0tt|W 5 Q R X t j|}t j|}zt
||W S t j	| X dS )zReturns bytes of object code of the module.

        Args
        ----
        use_object : bool
            Emit object code or (if False) emit assembly code.
        N)r	   r   r    LLVMPY_TargetMachineEmitToMemoryrv   rU   r   LLVMPY_GetBufferStartLLVMPY_GetBufferSizeLLVMPY_DisposeMemoryBufferr   )r$   r   r   rh   mbZbufptrZbufszr   r   r   r   (  s    

zTargetMachine._emit_to_memoryc                 C   s   t tj| S rE   )r>   r	   r   LLVMPY_CreateTargetMachineDatarH   r   r   r   target_data>  s    zTargetMachine.target_datac              
   C   s6   t  $}t j| | t|W  5 Q R  S Q R X d S rE   )r	   r   r   LLVMPY_GetTargetMachineTripler   rC   r   r   r   r;   B  s    
zTargetMachine.tripleN)F)r(   r)   r*   rI   r   r   r   r   r   r~   r   r;   r   r   r   r   rw     s   

rw   c                   C   s   t j dkrdS dS dS )zG
    Returns True if SVML was enabled at FFI support compile time.
    r   FTN)r	   r   LLVMPY_HasSVMLSupportr   r   r   r   has_svmlI  s    r   )N)Crt   ctypesr   r   r   r   r   r   r   Zllvmlite.bindingr	   Zllvmlite.binding.commonr
   r   r   dictr   r0   r2   r4   r:   r=   r@   Z	ObjectRefr>   	frozensetrr   rs   r_   rw   r   r   r   argtypesr-   restyper1   r3   r9   r?   ZLLVMTargetDataRefrB   rG   ZLLVMTypeRefrJ   rN   rT   rY   re   ZLLVMTargetRefrj   rn   ru   ZLLVMTargetMachineRefr   r   r   ZLLVMPassManagerRefr   ZLLVMModuleRefr   ZLLVMMemoryBufferRefr   r   r   r   r   r   r   r   r   <module>   s   $	
5YB















 






