o g@sdZddlZddlZddlZddlZddlZddlZddgZddgddddZggfddZ d d Z d d Z d dZ gggfddZ ddZddZdS)zb Build a c-extension module on-the-fly in tests. See build_and_import_extensions for usage hints Nbuild_and_import_extensioncompile_extension_module)prologue build_dir include_dirs more_initc Cs|t||}d}|std}|r|d7}||7}|d7}t|||}z t||||} WntyB} z td|d| d} ~ wwddl} | 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('abc') >>> assert mod.test_bytes(b'abc') z PyObject *mod = PyModule_Create(&moduledef); #ifdef Py_GIL_DISABLED PyUnstable_Module_SetGIL(mod, Py_MOD_GIL_NOT_USED); #endif .z.#define INITERROR return NULL z return mod;zcould not compile in :Nr) _make_methodspathlibPath _make_sourcer Exception RuntimeErrorimportlib.utilutilspec_from_file_locationmodule_from_specloader exec_module)modname functionsrrrrbodyinit source_stringmod_soe importlibspecfoor!Y/var/www/html/api-tag/env/lib/python3.10/site-packages/numpy/testing/_private/extbuild.pyrs,'    c CsR|dd}||}|jddt||}|tdg}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 Texist_ok INCLUDEPY)outputfilenamer libraries library_dirs)splitmkdir_convert_str_to_file sysconfigget_config_var _c_compile) namebuilddirrrr(r)rdirnamecfiler!r!r"rTs  cCsF|d}|d}|t|Wd|S1swY|S)zHelper function to create a file ``source.c`` in `dirname` that contains the string in `source`. Returns the file name zsource.cwN)openwritestr)sourcer2filenamefr!r!r"r,us  r,c Csg}g}|D]+\}}}d||f}d|vrd}nd}|d|||fdj|||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_%s 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 signaturecode 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)appendformatjoindict) rr methods_tablecodesfuncnameflagsr>r<r= func_coderr!r!r"r s$      r cCsdt|||d}|S)zG Combines the code fragments into source code ready to be compiled zn #include %(body)s PyMODINIT_FUNC PyInit_%(name)s(void) { %(init)s } )r0rr)rD)r0rrr>r!r!r"rs   rcCstjdkrdg}dtjtjdg}ntjdr!gd}g}ng}} tjdkr0|dg}tjdkrdd D],}|d |vrNtj|d rN||d |d |vrctj|d rc||d q7| t }t ||||||||S) Nwin32z/we4013z /LIBPATH:libslinux)z-O0z-gz%-Werror=implicit-function-declarationz-fPICz/DEBUGdarwin)z/sw/z /opt/local/includelib) sysplatformospathrC base_prefix startswithexistsrA with_suffix get_so_suffixbuild)r3r'rr(r) compile_extra link_extrasr!r!r"r/s4       r/c Cs|jd}tj|dd|jd}t|jdd7} dd|D} d d|D} | td |d |jdd | d |d| d |d|dWdn1sSwYtj dkrgt j gd|dn t j gd|dt j ddg|dt t ||d|j|dS)zuse meson to buildrYTr$r#z meson.buildwtcSg|]}d|qS)z-Ir!.0dr!r!r" zbuild..cSr^)z-Lr!r_r!r!r"rbrcz; project('foo', 'c') shared_module('z', 'z', c_args: z + z, link_args: z, link_with: z_, name_prefix: '', name_suffix: 'dummy', ) NrJ)mesonsetupz--buildtype=release--vsenv..)cwd)rdrerfrgrdcompilez.dummy)parentrRmakedirspartsr5r6textwrapdedentrPrQ subprocess check_callrenamer7) r3r'rZr[rr(r)rso_namefidincludes link_dirsr!r!r"rYs>     "rYcCstd}|s J|S)N EXT_SUFFIX)r-r.)retr!r!r"rXs rX)__doc__rRr rorPr-rm__all__rrr,r rr/rYrXr!r!r!r"s* E ! (  !