Ë
    èÿæi?  ã                   óª   — d dl Z d dlZd dlmc mc mZ d dlm	Z	m
Z
 d dlmZ d„ Zd„ Zd„ Zd„ Zdxaad„ Zd	„ Zd
„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zy)é    N)Úimport_dynamicÚtemp_directory)Ú
complex128c                  óˆ   — ddl m}  d} | «       }|j                  |«       ddlm} ||j                  |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   ÚcdefÚnumbar	   ÚdlopenÚ__file__)r   ÚdefsÚffir	   s       ún/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/numba/tests/cffi_usecases.pyÚload_inline_moduler   
   s?   € õ ð
€Dñ ‹%€CØ‡H�HˆT„Nå Ø�—
‘
˜:×.Ñ.Ó/Ð/Ð/ó    c                  óJ  — ddl m}  d}d}|dz   }||z   dz   } | «       }|j                  d|«       |j                  |d¬	«       t	        d
«      }|j                  |¬«       t        j                  j                  |«       	 t        d«      }t        j                  |«       t        j                  |j                  j                  d«      t        «       |j                  |ft        j                  j!                  |«       S # t        j                  j!                  |«       w xY w)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;
    }
    Úcffi_usecases_oolT)ÚoverrideÚ	test_cffi)Útmpdirzstruct _numba_complex)r
   r   Ú
set_sourcer   r   ÚcompileÚsysÚpathÚappendr   Úcffi_supportÚregister_moduleÚregister_typer   Útypeofr   Úremove)r   Únumba_complexÚbool_definer   Úsourcer   r   Úmods           r   Úload_ool_moduler'   !   sø   € õ
 ð€Mð€Kð ð 	ñ 	€Dð ˜[Ñ(ð ",ñ "€FñH ‹%€CØ‡N�NÐ&¨Ô/Ø‡H�HˆT˜D€HÔ!Ü˜KÓ(€FØ‡K�K�v€KÔÜ‡H�H‡O�O�FÔð ÜÐ0Ó1ˆÜ×$Ñ$ SÔ)Ü×"Ñ" 3§7¡7§>¡>Ð2IÓ#JÜ#-ô	/à�w‰w˜ˆ|ä�‰�‰˜ÕøŒ�‰�‰˜Õús   ÂA D Ä!D"c                  óv   — t         €3t        «       \  a } | j                  a| j                  a| j                  a~ yy)zƒ
    Initialize module globals.  This can invoke external utilities, hence not
    being executed implicitly at module import.
    N)r   r   Ú_numba_test_sinÚcffi_sinÚ_numba_test_cosÚcffi_cosÚ_numba_test_booleanÚ	cffi_bool)Údlls    r   Úinitr0   x   s>   € ô €{Ü%Ó'‰ˆˆSØ×&Ñ&ˆØ×&Ñ&ˆØ×+Ñ+ˆ	Ùð r   c                  óŽ  — t         €¿t        «       \  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~ yy)z&
    Same as init() for OOL mode.
    N)Úffi_oolr'   ÚlibÚsinÚcffi_sin_oolÚcosÚcffi_cos_oolÚfooÚcffi_fooÚbooleanÚcffi_bool_oolÚvsSinÚvdSinÚvector_realÚvector_imag)r&   s    r   Úinit_oolr@   †   s‰   € ô €Ü&Ó(‰ˆ�Ø—w‘w—{‘{ˆØ—w‘w—{‘{ˆØ—7‘7—;‘;ˆØŸ™Ÿ™ˆØ—‘—‘ˆØ—‘—‘ˆØ—g‘g×)Ñ)ˆØ—g‘g×)Ñ)ˆÙð r   c                 ó   — t        | «      dz  S ©Né   )r*   ©Úxs    r   Úuse_cffi_sinrF   œ   s   € Ü�A‹;˜‰?Ðr   c                 ó0   — t        | «      t        | «      z
  S ©N)r*   r,   rD   s    r   Úuse_two_funcsrI   Ÿ   s   € Ü�A‹;œ !›Ñ$Ð$r   c                 ó   — t        | «      dz  S rB   )r5   rD   s    r   Úuse_cffi_sin_oolrK   ¢   s   € Ü˜‹?˜QÑÐr   c                  ó   — t        «       S rH   )r;   © r   r   Úuse_cffi_boolean_truerN   ¥   s
   € Ü‹?Ðr   c                 ó0   — t        | «      t        | «      z
  S rH   )r5   r7   rD   s    r   Úuse_two_funcs_oolrP   ¨   s   € Ü˜‹?œ\¨!›_Ñ,Ð,r   c                 ó,   — |dkD  r | |«      S  ||«      S )Nr   rM   )ÚfaÚfbrE   s      r   Úuse_func_pointerrT   «   s   € Øˆ1‚uÙ�!‹uˆá�!‹uˆr   c                  ó   — t        ddd«      S )Né   rC   é   )r9   rM   r   r   Úuse_user_defined_symbolsrX   ±   s   € Ü�A�q˜!ÓÐr   c                 ó|   — t        t        | «      t        j                  | «      t        j                  |«      «       y rH   )r<   Úlenr   Úfrom_bufferr2   ©rE   Úys     r   Úvector_sin_float32r^   ¸   ó&   € Ü	Œ#ˆa‹&”#—/‘/ !Ó$¤g×&9Ñ&9¸!Ó&<Õ=r   c                 ó|   — t        t        | «      t        j                  | «      t        j                  |«      «       y rH   )r=   rZ   r   r[   r2   r\   s     r   Úvector_sin_float64ra   »   r_   r   c                 ó|   — t        t        j                  | «      t        j                  |«      t        | «      «       y rH   )r>   r   r[   rZ   r\   s     r   Úvector_extract_realrc   Á   ó$   € Ü”—‘ Ó"¤C§O¡O°AÓ$6¼¸A»Õ?r   c                 ó|   — t        t        j                  | «      t        j                  |«      t        | «      «       y rH   )r?   r   r[   rZ   r\   s     r   Úvector_extract_imagrf   Ä   rd   r   )r   ÚnumpyÚnpÚnumba.core.typing.cffi_utilsÚcoreÚtypingÚ
cffi_utilsr   Únumba.tests.supportr   r   Únumba.core.typesr   r   r'   r0   r@   r   r2   rF   rI   rK   rN   rP   rT   rX   r^   ra   rc   rf   rM   r   r   Ú<module>ro      sr   ðÛ 
ã ç 3Ó 3ß >Ý 'ò0ò.T ònòð& Ð €€gòò%òòò-òòò>ò>ò@ó@r   