U
    9%e?                     @   s   d dl Z d dlZd dlm  m  mZ d dlm	Z	m
Z
 d dlmZ dd Zdd Zdd	 Zd
d Zd aadd Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zd d! ZdS )"    N)import_dynamictemp_directory)
complex128c                  C   s<   ddl m}  d}|  }|| ddlm} |||jfS )zP
    Create an inline module, return the corresponding ffi and dll objects.
    r   FFIz
    double _numba_test_sin(double x);
    double _numba_test_cos(double x);
    double _numba_test_funcptr(double (*func)(double));
    bool _numba_test_boolean(void);
    )
_helperlib)cffir   cdefZnumbar   Zdlopen__file__)r   defsffir    r   X/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/numba/tests/cffi_usecases.pyload_inline_module
   s    
r   c               	   C   s   ddl m}  d}d}|d }|| d }|  }|d| |j|dd	 td
}|j|d tj| z4t
d}t| t|jdt |j|fW S tj	| X dS )z]
    Compile an out-of-line module, return the corresponding ffi and
    module objects.
    r   r   zh
    typedef struct _numba_complex {
        double real;
        double imag;
    } numba_complex;
    z
    #ifdef _MSC_VER
        #define false 0
        #define true 1
        #define bool int
    #else
        #include <stdbool.h>
    #endif
    aE  
    bool boolean(void);
    double sin(double x);
    double cos(double x);
    int foo(int a, int b, int c);
    void vsSin(int n, float* x, float* y);
    void vdSin(int n, double* x, double* y);
    void vector_real(numba_complex *c, double *real, int n);
    void vector_imag(numba_complex *c, double *imag, int n);
    a  
    static bool boolean(void)
    {
        return true;
    }

    static int foo(int a, int b, int c)
    {
        return a + b * c;
    }

    void vsSin(int n, float* x, float* y) {
        int i;
        for (i=0; i<n; i++)
            y[i] = sin(x[i]);
    }

    void vdSin(int n, double* x, double* y) {
        int i;
        for (i=0; i<n; i++)
            y[i] = sin(x[i]);
    }

    static void vector_real(numba_complex *c, double *real, int n) {
        int i;
        for (i = 0; i < n; i++)
            real[i] = c[i].real;
    }

    static void vector_imag(numba_complex *c, double *imag, int n) {
        int i;
        for (i = 0; i < n; i++)
            imag[i] = c[i].imag;
    }
    Zcffi_usecases_oolT)overrideZ	test_cffi)tmpdirzstruct _numba_complexN)r   r   Z
set_sourcer	   r   compilesyspathappendremover   cffi_supportZregister_moduleZregister_typer   Ztypeofr   )r   Znumba_complexZbool_definer   sourcer   r   modr   r   r   load_ool_module!   s&    
$
r   c                  C   s*   t dkr&t \a } | ja| ja| ja~ dS )z
    Initialize module globals.  This can invoke external utilities, hence not
    being executed implicitly at module import.
    N)r   r   Z_numba_test_sincffi_sinZ_numba_test_coscffi_cosZ_numba_test_booleanZ	cffi_bool)Zdllr   r   r   initx   s    
r   c                  C   sX   t dkrTt \a } | jja| jja| jja| jj	a
| jja| jja| jja| jja~ dS )z&
    Same as init() for OOL mode.
    N)ffi_oolr   libsincffi_sin_oolcoscffi_cos_oolZfoocffi_foobooleancffi_bool_oolvsSinvdSinvector_realvector_imag)r   r   r   r   init_ool   s    
r+   c                 C   s   t | d S N   )r   xr   r   r   use_cffi_sin   s    r0   c                 C   s   t | t|  S N)r   r   r.   r   r   r   use_two_funcs   s    r2   c                 C   s   t | d S r,   )r!   r.   r   r   r   use_cffi_sin_ool   s    r3   c                   C   s   t  S r1   )r&   r   r   r   r   use_cffi_boolean_true   s    r4   c                 C   s   t | t|  S r1   )r!   r#   r.   r   r   r   use_two_funcs_ool   s    r5   c                 C   s   |dkr| |S ||S d S )Nr   r   )faZfbr/   r   r   r   use_func_pointer   s    r7   c                   C   s   t dddS )N   r-      )r$   r   r   r   r   use_user_defined_symbols   s    r:   c                 C   s    t t| t| t| d S r1   )r'   lenr   from_bufferr   r/   yr   r   r   vector_sin_float32   s    r?   c                 C   s    t t| t| t| d S r1   )r(   r;   r   r<   r   r=   r   r   r   vector_sin_float64   s    r@   c                 C   s    t t| t|t|  d S r1   )r)   r   r<   r;   r=   r   r   r   vector_extract_real   s    rA   c                 C   s    t t| t|t|  d S r1   )r*   r   r<   r;   r=   r   r   r   vector_extract_imag   s    rB   )r   numpynpZnumba.core.typing.cffi_utilscoretypingZ
cffi_utilsr   Znumba.tests.supportr   r   Znumba.core.typesr   r   r   r   r+   r   r   r0   r2   r3   r4   r5   r7   r:   r?   r@   rA   rB   r   r   r   r   <module>   s(   W