7jVdZddlZddlZddlZddlZddlZddlmZddgZdddddd Z dd Z d Z d Z d Z dZdZdZy)zb Build a c-extension module on-the-fly in tests. See build_and_import_extensions for usage hints N)run_subprocessbuild_and_import_extensioncompile_extension_module)prologue build_dir include_dirs more_initcb|g}|t||z}d}|stjd}|r |dz }||z }|dz }t|||}t ||||} ddl} | j j|| } | j j| } | jj| | 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') Nz 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;r) _make_methodspathlibPath _make_sourcerimportlib.utilutilspec_from_file_locationmodule_from_specloader exec_module) modname functionsrr r r bodyinit source_stringmod_so importlibspecfoos U/root/workspace/.venv/lib/python3.12/site-packages/numpy/testing/_private/extbuild.pyrrsN mIw7 7D D LL%    OD $5M %L-9F >> 1 1'6 BD .. ) )$ /CKKC Jc|jdd}||z }|jdt||}|xsg}|xsg}|xsg}t|||z |||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 Texist_ok)outputfilenamer libraries library_dirs)splitmkdir_convert_str_to_file _c_compile) namebuilddirr rr(r)rdirnamecfiles r!rrSsu*jjob!GoG MM4M  8E%2LRI%2L  g/!Y!  r"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)sourcer0filenamefs r!r,r,wsF#H s q F  O Os <Ac  g}g}|D]M\}}}|d|}d|vrd}nd}|jd|d|d|dd ||d |d } |j| Od j|} d j|d | d|dz} | 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* z { z }  z* static PyMethodDef methods[] = { zi { NULL } }; static struct PyModuleDef moduledef = { PyModuleDef_HEAD_INIT, "z", /* m_name */ NULL, /* m_doc */ -1, /* m_size */ methods, /* m_methods */ }; )appendjoin) rr methods_tablecodesfuncnameflagscode cfuncname signature func_code methods_strrs r!rrs M E!*  %iq + e #LI:Is8*, z PyMODINIT_FUNC PyInit_z (void) { z } )r.rrrDs r!rrs7 F  6 F  D Kr"c g}tjdk(rdg}|jdn&tjjdrgd}ng}t |||||||S)Nwin32z/we4013z/DEBUGlinux)z-O0z-gz%-Werror=implicit-function-declarationz-fPIC)sysplatformr> startswithbuild)r1r'r r(r) link_extra compile_extras r!r-r-sgJ ||w" (#   )K   ~zi //r"c|jdz }tj|dt|jdz d5}|D cgc]} d| z } } |j t j d|jdd |jdd |d | d |d ddd|jdz } t| d5}|j t j dtjddddtjdk(rtgd|n'tdddddtj| g||jdtz} tddg|tjt|| z |j| z |j| z Scc} w#1swYxYw#1swYxYw)zuse meson to buildrQTr%z meson.buildwtz-Lz project('foo', 'c') py = import('python').find_installation(pure: false) py.extension_module( 'r$z', 'z', c_args: z, link_args: z', include_directories: z, ) Nz.mesonpy-native-file.iniz- [binaries] python = 'z ' rL)mesonsetupz--buildtype=release--vsenv..rVrWrXrYz--native-file=compile)parentosmakedirsr4r5textwrapdedentpartsrN executablerOrfspath get_so_suffixrenamer6) r1r'rSrRr r(r)r fidd link_dirsnative_file_nameso_names r!rQrQs w&IKK D) ell]*D 1 S'34!TAX4 4 (//'!&&r*+,++b/"#&(%;'&&2^4  #   ||&@@  % (//'~~&' #     ||w)! " )ryy9I/J.KLN  """2&8GGY'3IIc)g%& w(>? <<' !!?5  s+F7 F2AF77G2F77GG c8tjd}|sJ|S)N EXT_SUFFIX) sysconfigget_config_var)rets r!rcrcs  " "< 0C J3 Jr")NN)__doc__r\rrNrlr^utilsr__all__rrr,rrr-rQrcrJr"r!rrsf   ! ')C D)+dR>F59! H%P  /$&"Rr"