ehdZddlZddlZddlZddlZddlZddlZddgZddgdddZggfdZ d Z d Z d Z gggfd Z d ZdZy)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|t||z}d}|stjd}|r |dz }||z }|dz }t|||} t ||||} ddl} | jj|| } | jj| } | jj| | S#t $r} t d|d| d} ~ wwxYw) 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 functionsrrrr bodyinit source_stringmod_soe importlibspecfoos ]/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/numpy/testing/_private/extbuild.pyrrsN mIw7 7D D LL%    OD $5MH) Y m=  >> 1 1'6 BD .. ) )$ /CKKC J H29+Q?@aGHsB.. C 7CC c|jdd}||z }|jdt||}|tjdgz}t |||z |ggS)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+rdirnamecfiles r#rrTso*jjob!GoG MM4M  8E9#;#;K#H"IIL  g/!Rb  c|dz }|jd5}|jt|ddd|S#1swY|SxYw)zHelper function to create a file ``source.c`` in `dirname` that contains the string in `source`. Returns the file name zsource.cwN)openwritestr)sourcer4filenamefs r#r.r.usF#H s q F  O Os <Ac &g}g}|D]V\}}}|d|}d|vrd}nd}|jd|d|d|dd j||| } |j| Xd j|d td j|| zz} | 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 _ METH_KEYWORDSz2(PyObject *self, PyObject *args, PyObject *kwargs)z (PyObject *self, PyObject *args)z{"z", (PyCFunction)z, z},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_tablecodesfuncnameflagsrDrBrC func_coders r#r r s M E!* %&1 e #LI:I08)U K M FY)$F G   Y 99U   tyy/ A B BD Kr6c(dt|||z}|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 } )r2rr)rJ)r2rrrDs r#rrs&   4  D Kr6c tjdk(r6dg}dtjj tj dzg}n+tjj drgd}g}ngx}} tjdk(r|dgz}tjdk(rd D]~}|d z|vr6tjj|d zr|j|d z|d z|vsHtjj|d zsk|j|d z|jt}t||||||||S) Nwin32z/we4013z /LIBPATH:libslinux)z-O0z-gz%-Werror=implicit-function-declarationz-fPICz/DEBUGdarwin)z/sw/z /opt/local/includelib) sysplatformospathrI base_prefix startswithexistsrG with_suffix get_so_suffixbuild)r5r)rr*r+ compile_extra link_extrass r#r1r1s, ||w" !BGGLL&$IIJ   )K  %''   ||w8*,  ||x( /AI \1q9}5##A M25y ,E 1J##AI.  /$// @N  ~zi/ r6c|jdz }tj|d|jd}t |jdz d5} |D cgc]} d| z } } |D cgc]} d| z } } | j t jd |d |jdd | d |d | d |d|ddddtjdk(rtjgd|ntjgd|tjddg|tjt||z dz|j|z ycc} wcc} w#1swYxYw)zuse meson to buildraTr&r%z meson.buildwtz-Iz-Lz; project('foo', 'c') shared_module('z', 'z', c_args: z + z, link_args: z, link_with: z_, name_prefix: '', name_suffix: 'dummy', ) NrR)mesonsetupz--buildtype=release--vsenv..)cwd)rgrhrirjrgcompilez.dummy)parentrZmakedirspartsr9r:textwrapdedentrXrY subprocess check_callrenamer;) r5r)rbrcrr*r+rso_namefiddincludes link_dirss r#rarasm w&IKK D)""2&G ell]*D 1 S&23D1H33'34!TAX4 4 (//'#9DR(9:! #m_5%;c*6%;' #    ||w0#, A"+ 7I.I>IIc)g%&15<<'3IJ/44  s+E E E& E2A E EE(c8tjd}|sJ|S)N EXT_SUFFIX)r/r0)rets r#r`r`s  " "< 0C J3 Jr6)__doc__rZrrrrXr/rp__all__rrr.r rr1rar`r6r#rst    ')C D)+d2@J"$" B%P$46>KBr6