a
    XgQ                     @   s   d Z ddlZddlZddlZddlZddlZddlZddgZddg ddddZg g fddZ	d	d
 Z
dd Zdd Zg g g fddZdd Zdd ZdS )zb
Build a c-extension module on-the-fly in tests.
See build_and_import_extensions for usage hints

    Nbuild_and_import_extensioncompile_extension_module )prologue	build_dirinclude_dirs	more_initc             
   C   s   |t ||  }d}|s td}|r4|d7 }||7 }|d7 }t| ||}zt| |||}	W n6 ty }
 ztd| d|
W Y d}
~
n
d}
~
0 0 ddl}|j	| |	}|j
|}|j| |S )	a  
    Build and imports a c-extension module `modname` from a list of function
    fragments `functions`.


    Parameters
    ----------
    functions : list of fragments
        Each fragment is a sequence of func_name, calling convention, snippet.
    prologue : string
        Code to precede the rest, usually extra ``#include`` or ``#define``
        macros.
    build_dir : pathlib.Path
        Where to build the module, usually a temporary directory
    include_dirs : list
        Extra directories to find include files when compiling
    more_init : string
        Code to appear in the module PyMODINIT_FUNC

    Returns
    -------
    out: module
        The module will have been loaded and is ready for use

    Examples
    --------
    >>> functions = [("test_bytes", "METH_O", """
        if ( !PyBytesCheck(args)) {
            Py_RETURN_FALSE;
        }
        Py_RETURN_TRUE;
    """)]
    >>> mod = build_and_import_extension("testme", functions)
    >>> assert not mod.test_bytes(u'abc')
    >>> assert mod.test_bytes(b'abc')
    z8PyObject *mod = PyModule_Create(&moduledef);
           .z.#define INITERROR return NULL
                z
return mod;zcould not compile in :Nr   )_make_methodspathlibPath_make_sourcer   	ExceptionRuntimeErrorimportlib.utilutilspec_from_file_locationmodule_from_specloaderexec_module)modname	functionsr   r   r   r   bodyinitsource_stringZmod_soe	importlibspecZfoo r   m/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/numpy/testing/_private/extbuild.pyr      s(    '
(c           	      C   sR   |  dd }||  }|jdd t||}|tdg }t||| |g g dS )aH  
    Build an extension module and return the filename of the resulting
    native code file.

    Parameters
    ----------
    name : string
        name of the module, possibly including dots if it is a module inside a
        package.
    builddir : pathlib.Path
        Where to build the module, usually a temporary directory
    include_dirs : list
        Extra directories to find include files when compiling
    libraries : list
        Libraries to link into the extension module
    library_dirs: list
        Where to find the libraries, ``-L`` passed to the linker
    r	   Texist_ok	INCLUDEPY)outputfilenamer   	librarieslibrary_dirs)splitmkdir_convert_str_to_file	sysconfigget_config_var
_c_compile)	nameZbuilddirr   r   r&   r'   r   dirnamecfiler   r   r    r   P   s    
c                 C   sD   |d }| d}|t|  W d   n1 s60    Y  |S )zHelper function to create a file ``source.c`` in `dirname` that contains
    the string in `source`. Returns the file name
    zsource.cwN)openwritestr)sourcer/   filenamefr   r   r    r*   q   s    ,r*   c                 C   s   g }g }| D ]V\}}}d||f }d|v r0d}nd}| d|||f  dj|||d}	| |	 qd|d	td||d
  }
|
S )z Turns the name, signature, code in functions into complete functions
    and lists them in a methods_table. Then turns the methods_table into a
    ``PyMethodDef`` structure and returns the resulting code fragment ready
    for compilation
    z%s_%sZMETH_KEYWORDSz2(PyObject *self, PyObject *args, PyObject *kwargs)z (PyObject *self, PyObject *args)z{"%s", (PyCFunction)%s, %s},z^
        static PyObject* {cfuncname}{signature}
        {{
        {code}
        }}
        )	cfuncname	signaturecode
a6  
    static PyMethodDef methods[] = {
    %(methods)s
    { NULL }
    };
    static struct PyModuleDef moduledef = {
        PyModuleDef_HEAD_INIT,
        "%(modname)s",  /* m_name */
        NULL,           /* m_doc */
        -1,             /* m_size */
        methods,        /* m_methods */
    };
    )methodsr   )appendformatjoindict)r   r   Zmethods_tablecodesfuncnameflagsr:   r8   r9   	func_coder   r   r   r    r   {   s$    
r   c                 C   s   dt | ||d }|S )zG Combines the code fragments into source code ready to be compiled
    zn
    #include <Python.h>

    %(body)s

    PyMODINIT_FUNC
    PyInit_%(name)s(void) {
    %(init)s
    }
    )r.   r   r   )r@   )r.   r   r   r:   r   r   r    r      s    	r   c                 C   s   t jdkr(dg}dtjt jd g}n"t jdrBg d}g }ng  }}t jdkr^|dg }t jdkrd	D ]X}|d
 |vrtj|d
 r||d
  |d |vrltj|d rl||d  ql|	t
 }t| |||||| |S )Nwin32z/we4013z	/LIBPATH:Zlibslinux)z-O0z-gz%-Werror=implicit-function-declarationz-fPICz/DEBUGdarwin)z/sw/z/opt/local/includelib)sysplatformospathr?   base_prefix
startswithexistsr=   with_suffixget_so_suffixbuild)r0   r%   r   r&   r'   compile_extra
link_extrasr   r   r    r-      s0    



r-   c                 C   s  | j d }tj|dd |jd }t| j d dn}	dd |D }
d	d |D }|	td
| d| jd  d|
 d| d| d| d| d W d   n1 s0    Y  tj	dkrt
jg d|d nt
jg d|d t
jddg|d tt|| d | j |  dS )zuse meson to buildrS   Tr"   r!   zmeson.buildwtc                 S   s   g | ]}d | qS )z-Ir   .0dr   r   r    
<listcomp>       zbuild.<locals>.<listcomp>c                 S   s   g | ]}d | qS )z-Lr   rX   r   r   r    r[      r\   z;            project('foo', 'c')
            shared_module('z', 'z',
                c_args: z + z,
                link_args: z,
                link_with: z_,
                name_prefix: '',
                name_suffix: 'dummy',
            )
        NrE   )mesonsetupz--buildtype=release--vsenv..)cwd)r]   r^   r_   r`   r]   compilez.dummy)parentrL   makedirspartsr2   r3   textwrapdedentrJ   rK   
subprocess
check_callrenamer4   )r0   r%   rT   rU   r   r&   r'   r   Zso_nameZfidZincludesZ	link_dirsr   r   r    rS      s<    


*



rS   c                  C   s   t d} | sJ | S )N
EXT_SUFFIX)r+   r,   )retr   r   r    rR      s    
rR   )__doc__rL   r   rh   rJ   r+   rf   __all__r   r   r*   r   r   r-   rS   rR   r   r   r   r    <module>   s(   A
!
(
!