E%+ | d Z ddlZddlZddlZddlZddgZddg dddZg g fdZd Zd
Z d Z
g g g fd Z d
Z d Z
dS )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 ddl m} |t || z }d}|st j d }|r
|dz
}||z
}|dz
}t | || } t
| ||| }
n!# |$ r} t d| d | d } ~ ww xY wdd 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_methodspathlibPath _make_sourcer RuntimeErrorimportlib.utilutilspec_from_file_locationmodule_from_specloader exec_module)modname functionsr r r r r bodyinit
source_stringmod_soe importlibspecfoos s/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/testing/_private/extbuild.pyr r s9 N .----- mIw77 7D D &L%% OD $55MH) Y m= = H H H?9???@@aGH > 1 1'6 B BD
.
)
)$
/
/CKC Js A( (B-BBc | d d }|| z }| d t || }|t j d gz }t |||z |g g 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)splitmkdir_convert_str_to_file sysconfigget_config_var
_c_compile) namebuilddirr r r* r+ r dirnamecfiles r$ r r P s * jjoob!GoG MM4M 8 8E 9#;K#H#H"IIL
g /!Rb
c |dz }| d 5 }| t | ddd n # 1 swxY w Y |S )zHelper function to create a file ``source.c`` in `dirname` that contains
the string in `source`. Returns the file name
zsource.cwN)openwritestr)sourcer4 filenamefs r$ r. r. q s #H s q F Os #A
AAc
R g }g }| D ]b\ }}}|d|}d|v rd}nd}| d|d|d|d d |||
} | | cd | d t d | |
z z }
|
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) r r
methods_tablecodesfuncnameflagsrD rB rC func_coder s r$ r r { s M E!* % &ww1 e # #LII:I 08)))UUU K M M M
FY)$F G G Y 99U tyy //
A
A
A B BD Kr6 c 0 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
}
)r2 r r )rJ )r2 r r rD s r$ r r s0
4 D Kr6 c t j dk r2dg}dt j t j d z g}n+t j d rg d}d }nd x}} t j dk r|dgz }t j dk rd D ]}|d
z |vr:t j |d
z r| |d
z |d z |vr:t j |d z r| |d z | t }t j } t | |||||| | D ]7\ } }
t j | |
k r|
t j | < 8nQ# | D ]7\ } }
t j | |
k r|
t j | < 8w xY w|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_suffixenvironcopybuilditemsget) r5 r) r r* r+
compile_extra
link_extras
saved_environkeyvalues r$ r1 r1 s
|w "
!BGLL &$I$IIJ
) )
K K K
%))
|w 8*,
|x ( / /AI
\11q9}55 2 ##A M2225y ,, E 1J1J, ##AI...#//
@@NJOO%%M
(
> : )\ 3 3 3 (--// ( (JCz~~c""e++"'
3 (---// ( (JCz~~c""e++"'
3 ( s
F= =AH c ddl m} |dd }| d g } t j }
t j j | t j g|| } | fd| D z
} t j |
n# t j |
w xY w|
| t | ||| d
S ) z@cd into the directory where the cfile is, use distutils to buildr ) new_compiler )forceverboser )r
extra_preargsc > g | ]}t j |z S )r; parent).0rr5 s r$
zbuild.. s( 777aC q())777r6 )r* rr r+ N) numpy.distutils.ccompilerrm customizerZ getcwdchdirru compiler; r2 link_shared_object) r5 r) rf rg r r* r+ rm compilerobjectsoldress ` r$ rc rc s 766666|!Q///H rG
)++CHU\
__ %'
777737777
^$$ ! # # # # #s r s
')C
D )+d2> > > > >F "$"
B % % %P $ 46 $ $ $ $N# # #8 r6