fume-manage-python.git

U  
P±dã@sdZddlZddlZddlZddlZddgZddgddddZggfddZd    d
ZddZ    d dZ
gggfddZddZddZ dS)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
CsÎddlm}|t||}d}|s,t d¡}|r@|d7}||7}|d7}t|||}    zt||||    }
Wn4|k
r}ztd|d|W5d    }~XYnXdd    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')
    r)ÚCompileErrorz8PyObject *mod = PyModule_Create(&moduledef);
           Ú.z.#define INITERROR return NULL
                z
return mod;zcould not compile in ú:N)Údistutils.errorsr    Ú _make_methodsÚpathlibÚPathÚ_make_sourcerÚRuntimeErrorZimportlib.utilÚutilÚspec_from_file_locationÚmodule_from_specÚloaderÚexec_module)ÚmodnameÚ    functionsrrrrr    ÚbodyÚinitÚ source_stringZmod_soÚeÚ    importlibÚspecZfoo©rúVd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/testing/_private/extbuild.pyrs0'
ÿ$c    CsR| d¡d}||}|jddt||}|t d¡g}t||||ggdS)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)    ÚnameZbuilddirrrr%r&rÚdirnameÚcfilerrr rPs
þc    Cs0|d}| d¡}| t|¡W5QRX|S)zHelper function to create a file ``source.c`` in `dirname` that contains
    the string in `source`. Returns the file name
    zsource.cÚw)ÚopenÚwriteÚstr)Úsourcer.ÚfilenameÚfrrr r)qsr)cCsg}g}|D]V\}}}d||f}d|kr0d}nd}| d|||f¡dj|||d}    | |    ¡qd |¡d    td |¡|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)rrZ methods_tableÚcodesÚfuncnameÚflagsr9r7r8Ú    func_coderrrr r {s(ÿû
ô r cCsdt|||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-rr)r?)r-rrr9rrr r£s    ÿ÷rc
Cs0tjdkr(dg}dtj tjd¡g}n&tj d¡rFdddd    g}d}nd}}tjdkrb|d
g}tjdkrÊdD]X}|d |krtj |d ¡r| |d ¡|d|krptj |d¡rp| |d¡qp|     t
¡}tj ¡}zt|||||||W5|  ¡D]&\}    }
tj |    ¡|
kr|
tj|    <qX|S)NÚwin32z/we4013z    /LIBPATH:ZlibsÚlinuxz-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ÚenvironÚcopyÚitemsÚgetÚbuild)r/r$rr%r&Ú compile_extraÚ
link_extraÚsZ saved_environÚkeyÚvaluerrr r,µsJ
ÿ
 
 
ÿ
ýr,c    sddlm}|ddd}| d¡g}    t ¡}
t j¡z2|jtj    g||d}|    fdd    |D7}    W5t |
¡X|j
|    t||||d
dS)z@cd into the directory where the cfile is, use distutils to buildr)Únew_compileréé)ÚforceÚverboser)rÚ extra_preargscsg|]}tj|qSr)r3Úparent)Ú.0Úr©r/rr Ú
<listcomp>íszbuild.<locals>.<listcomp>)r%rar&N)Znumpy.distutils.ccompilerr\Z    customizerKÚgetcwdÚchdirrbÚcompiler3r-Zlink_shared_object)r/r$rWrXrr%r&r\ÚcompilerZobjectsÚoldÚresrrer rVÜs*
 
ýürVcCst d¡}|st|S)NÚ
EXT_SUFFIX)r*r+ÚAssertionError)Úretrrr rQøs
rQ)Ú__doc__rKrrIr*Ú__all__rrr)r rr,rVrQrrrr Ú<module>s*þCþ
!
(ÿ
'