o
    4ήc9                  
   @   sL  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g dZeg d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   @   t  } t j|  t| W  d   S 1 sw   Y  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-target-vg8gfxp4/lib/python/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  =} t }t j| s|W  d   S ddd}t| }|r8|dD ]}|r7||d  ||dd < q'|W  d   S 1 sDw   Y  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.
    NTF)r   r   r   r      )r	   r   r   r   LLVMPY_GetHostCPUFeaturesr   split)r   outdictr   contentfeatr   r   r   get_host_cpu_features.   s   

$r3   c                  C   r   )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   
$r5   c                  C   r   )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   
$r7   COFFELFMachO)r-         c                 C   s&   | du 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)r5   r	   r   LLVMPY_GetTripleObjectFormatr   _object_formats)tripleresr   r   r   get_object_formata   s   rA   c                 C   s   t tjt| S )zE
    Create a TargetData instance for the given *layout* string.
    )
TargetDatar	   r   LLVMPY_CreateTargetDatar   )layoutr   r   r   create_target_datam   s   rE   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 )rB   z{
    A TargetData provides structured access to a data layout.
    Use :func:`create_target_data` to create instances.
    c                 C   sL   | j rdS t }tj| | t|W  d    S 1 sw   Y  d S )Nz<dead TargetData>)_closedr	   r   r    LLVMPY_CopyStringRepOfTargetDatar   r%   r   r   r   r   __str__z   s   
$zTargetData.__str__c                 C      | j |  d S N)_capiLLVMPY_DisposeTargetDatar%   r   r   r   _dispose      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%   rR   positionoffsetr   r   r   get_element_offset   s   zTargetData.get_element_offsetc                 C   (   t j| |}|dkrtd|f |S )zI
        Get ABI size of pointee type of LLVM pointer type *ty*.
        rT   Not a pointer type: %s)r	   r   LLVMPY_ABISizeOfElementTypeRuntimeErrorr%   rR   sizer   r   r   get_pointee_abi_size      zTargetData.get_pointee_abi_sizec                 C   rZ   )zV
        Get minimum ABI alignment of pointee type of LLVM pointer type *ty*.
        rT   r[   )r	   r    LLVMPY_ABIAlignmentOfElementTyper]   r^   r   r   r   get_pointee_abi_alignment   ra   z$TargetData.get_pointee_abi_alignmentN)
r)   r*   r+   r,   rI   rO   rS   rY   r`   rc   r   r   r   r   rB   t   s    	rB   )defaultstaticpicdynamicnopic)rd   
jitdefaultsmallkernelmediumlargec                   @   sd   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.
        )r5   from_triple)clsr?   r   r   r   from_default_triple   s   
zTarget.from_default_triplec                 C   sb   t  #}t j|d|}|stt|| |}||_|W  d   S 1 s*w   Y  dS )zK
        Create a Target instance for the given triple (a string).
        utf8N)r	   r   r   LLVMPY_GetTargetFromTripleencoder]   r   _triple)rp   r?   outerrtargetr   r   r   ro      s   
$zTarget.from_triplec                 C      t j| }t|S rK   )r	   r   LLVMPY_GetTargetNamer
   r%   sr   r   r   name      zTarget.namec                 C   rx   rK   )r	   r   LLVMPY_GetTargetDescriptionr
   rz   r   r   r   description   r}   zTarget.descriptionc                 C   s   | j S rK   )ru   rN   r   r   r   r?      s   zTarget.triplec                 C   s   d | j| jS )Nz<Target {0} ({1})>)r   r|   r   rN   r   r   r   rI      rP   zTarget.__str__r;   rd   rh   Fc	                 C   s   d|  krdksJ  J |t v sJ |tv sJ | j}	tjdkr*|dkr*|	d7 }	tj| t|	t|t||t|t|t	|t	|t|
}
|
rOt
|
S td)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   r<   ntrh   z-elfzCannot create target machine)RELOC	CODEMODELru   osr|   r	   r   LLVMPY_CreateTargetMachiner   intTargetMachiner]   )r%   cpufeaturesoptreloc	codemodelprintmcjitabinamer?   tmr   r   r   create_target_machine   s(   zTarget.create_target_machineN)rn   rn   r;   rd   rh   FFrn   )r)   r*   r+   ru   classmethodrq   ro   propertyr|   r   r?   rI   r   r   r   r   r   rm      s"    




rm   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 )r   c                 C   rJ   rK   )rL   LLVMPY_DisposeTargetMachinerN   r   r   r   rO   	  rP   zTargetMachine._disposec                 C      t j| | dS )zW
        Register analysis passes for this target machine with a pass manager.
        N)r	   r   LLVMPY_AddAnalysisPasses)r%   pmr   r   r   add_analysis_passes  s   z!TargetMachine.add_analysis_passesc                 C   r   )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   s   t  }t j| |t||}|stt|W d   n1 s"w   Y  t j|}t j|}zt	||W t j
| S t j
| w )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_TargetMachineEmitToMemoryr   r]   r   LLVMPY_GetBufferStartLLVMPY_GetBufferSizer   LLVMPY_DisposeMemoryBuffer)r%   r   r   rv   mbbufptrbufszr   r   r   r   (  s   


zTargetMachine._emit_to_memoryc                 C   s   t tj| S rK   )rB   r	   r   LLVMPY_CreateTargetMachineDatarN   r   r   r   target_data>  s   zTargetMachine.target_datac                 C   sB   t  }t j| | t|W  d    S 1 sw   Y  d S rK   )r	   r   r   LLVMPY_GetTargetMachineTripler   rH   r   r   r   r?   B  s   
$zTargetMachine.tripleN)F)r)   r*   r+   rO   r   r   r   r   r   r   r   r?   r   r   r   r   r     s    

r   c                   C   s   t j dkr	dS dS )zG
    Returns True if SVML was enabled at FFI support compile time.
    r   FT)r	   r   LLVMPY_HasSVMLSupportr   r   r   r   has_svmlI  s   r   rK   )Cr   ctypesr   r   r   r   r   r   r   llvmlite.bindingr	   llvmlite.binding.commonr
   r   r   dictr   r3   r5   r7   r>   rA   rE   	ObjectRefrB   	frozensetr   r   rm   r   r   r   r   argtypesr.   restyper4   r6   r=   rC   LLVMTargetDataRefrG   rM   LLVMTypeRefrQ   rU   r\   rb   rs   LLVMTargetRefry   r~   r   LLVMTargetMachineRefr   r   r   LLVMPassManagerRefr   LLVMModuleRefr   LLVMMemoryBufferRefr   r   r   r   r   r   r   r   r   <module>   s    $	
5YB






















