o
    ʩZho                     @   s  d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZ d dlm	Z	 d dl
mZ d dlmZmZmZmZmZ d dlmZmZmZmZmZ d dlmZ d dlmZ d d	lmZ d d
lmZm Z  d dl!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z' d dl(Z(da)e(* Z+e, Z-dd Z.dd Z/dd Z0e/ede0 dPddZ1e/ede1 dQddZ2e/ede2 			dRddZ3e/ede3 dSddZ4e/ed e4 d!d" Z5d#d$ Z6e/ed%e6 dTd&d'Z7e/ed(e7 dUd*d+Z8d,d gfd-d.Z9e/ed/e9 d0d1 Z:e/ed2e: d3ed4< d5ed6< d7ed8< d9ed:< d;ed<< d=ed>< d?ed@< dAedB< e j;dC7  _;ejdDkr8dEedF< e$ r8e<dG dHej; e_;eZ=				 	 dVdIdJZee_eZ>dKdL Zee_dMD ]Z?ej@AdNe? dO ZBeBdurieCeBdLe qRdS )W    N)copy)Path)	ccompiler)compiler_classgen_lib_optionsget_default_compilernew_compiler	CCompiler)DistutilsExecErrorDistutilsModuleErrorDistutilsPlatformErrorCompileErrorUnknownFileError)customize_compiler)LooseVersion)log)filepath_from_subprocess_outputforward_bytes_to_stdout)	cyg2win32is_sequencemingw32get_num_build_jobs_commandline_dep_stringsanitize_cxx_flagsc                 C   s   | d }t j|sdS t|}| }W d   n1 sw   Y  t|||}|d }||kr4dS d|dd }	dd tj|	ddD }
zt 	| j
}|
D ]}t 	|j
|kr` W dS qRW d	S  tym   Y dS w )
z
    Check if an objects needs to be rebuild based on its dependencies

    Parameters
    ----------
    obj : str
        object file

    Returns
    -------
    bool
    .dTN c                 S   s"   g | ]}|d kr| ds|qS )
:)endswith).0x r"   P/var/www/html/lang_env/lib/python3.10/site-packages/numpy/distutils/ccompiler.py
<listcomp>G   s    z _needs_build.<locals>.<listcomp>)posixF)ospathexistsopen	readlinesr   joinshlexsplitstatst_mtimeOSError)objcc_argsextra_postargspp_optsZdep_fileflinesZcmdlineZlast_cmdlinecontentsdepsZt_objr"   r"   r#   _needs_build'   s.   

r9   c                    s    fdd}t | || d S )Nc                    s    | g|R i |S Nr"   )selfargskwfuncr"   r#   <lambda>[   s    z replace_method.<locals>.<lambda>)setattr)klassmethod_namer?   mr"   r>   r#   replace_methodY   s   rE   c                 C   s   dS )z
    Does nothing here, but is called by the get_version method and can be
    overridden by subclasses. In particular it is redefined in the `FCompiler`
    class where more documentation can be found.

    Nr"   r;   r"   r"   r#   CCompiler_find_executablesd   s   rG   find_executablesc           	      C   sV  |dur|nt tj}|du r|}t|rdt|}t| z| jr/t	j
||d W dS t	j
|t	j|d W dS  t	jyS } z|j}|j}W Y d}~n6d}~w ty } z&d| d}z	|tjj}W n tyw   |d}Y nw d}W Y d}~nd}~ww t|rdt|}| jrt| td	|rd
}nd}td|||f )a  
    Execute a command in a sub-process.

    Parameters
    ----------
    cmd : str
        The command to execute.
    display : str or sequence of str, optional
        The text to add to the log file kept by `numpy.distutils`.
        If not given, `display` is equal to `cmd`.
    env : a dictionary for environment variables, optional

    Returns
    -------
    None

    Raises
    ------
    DistutilsExecError
        If the command failed, i.e. the exit status was not 0.

    N )env)stderrrJ   z

z


utf8   s   Too many open filesz2
Try rerunning setup command until build succeeds.r   z)Command "%s" failed with exit status %d%s)dictr&   environr   r+   listr   infoverbose
subprocesscheck_outputSTDOUTCalledProcessErroroutput
returncoder0   encodesysstdoutencodingAttributeErrorr   researchr
   )	r;   cmddisplayrJ   excosemsgr"   r"   r#   CCompiler_spawnr   sH   
rg   Zspawnr   c                 C   s   |du rd}g }|D ]h}t jt j|\}}t j|d }|t j|d }|drN|dd }|d| }	t jt j	|	}	|	||d  }|| j
vr[td||f |rct j|}t j||| j }
||
 q
|S )a  
    Return the name of the object files for the given source files.

    Parameters
    ----------
    source_filenames : list of str
        The list of paths to source files. Paths can be either relative or
        absolute, this is handled transparently.
    strip_dir : bool, optional
        Whether to strip the directory from the returned paths. If True,
        the file name prepended by `output_dir` is returned. Default is False.
    output_dir : str, optional
        If given, this path is prepended to the returned paths to the
        object files.

    Returns
    -------
    obj_names : list of str
        The list of paths to the object files corresponding to the source
        files in `source_filenames`.

    Nr      z..   z"unknown file type '%s' (from '%s'))r&   r'   splitextnormpath
splitdriveisabs
startswithrfindbasenameabspathZsrc_extensionsr   r+   Zobj_extensionappend)r;   Zsource_filenamesZ	strip_dir
output_dirZ	obj_namesZsrc_namebaseextidobj_namer"   r"   r#   CCompiler_object_filenames   s&   

ry   Zobject_filenamesc	                    sn  t  }	t tdu rt|	aW d   n1 sw   Y  |s"g S ddlm}
m}m} t	|
rVg }dD ]}t
d| }|du rCq5|d|d|f  q5d|}nj}d	d|f }t| |||||\}}}|| d
d  }r|dd 7 }t|  fdd}t	|
rt| }g g }}|D ]9}||v r|| \}}jdkrt|}t|}t|j |v r||s||||ff q||||ff q|}|D ]}|| qn| }t|dkr+|	dkr+ddlm} ||	}|||}W d   n	1 s w   Y  t| |S |D ]}|| q-|S )a  
    Compile one or more source files.

    Please refer to the Python distutils API reference for more details.

    Parameters
    ----------
    sources : list of str
        A list of filenames
    output_dir : str, optional
        Path to the output directory.
    macros : list of tuples
        A list of macro definitions.
    include_dirs : list of str, optional
        The directories to add to the default include file search path for
        this compilation only.
    debug : bool, optional
        Whether or not to output debug symbols in or alongside the object
        file(s).
    extra_preargs, extra_postargs : ?
        Extra pre- and post-arguments.
    depends : list of str, optional
        A list of file names that all targets depend on.

    Returns
    -------
    objects : list of str
        A list of object file names, one per source file `sources`.

    Raises
    ------
    CompileError
        If compilation fails.

    Nr   )	FCompilerFORTRAN_COMMON_FIXED_EXTENSIONShas_f90_header)Zf77f90ZfixZ	compiler_zFortran %s compiler: %srI   r   zC compiler: %s
zcompile options: '%s'z
extra options: '%s'c                    s  | \}\}}t | sd S 	 t |tvr$t| 	 W d    nW d    n1 s.w   Y  td qz7t |||  W d    n1 sQw   Y  W t t| W d    d S 1 sjw   Y  d S t t| W d    w 1 sw   Y  w )NTg?)	r9   _global_lock_processing_filesaddtimesleep_job_semaphore_compileremove)r<   r1   srcru   r2   r3   r4   r;   r"   r#   single_compile4  s.   


(z)CCompiler_compile.<locals>.single_compileZabsoftrh   )ThreadPoolExecutor)r   r~   r   	threading	SemaphoreZnumpy.distutils.fcompilerrz   r{   r|   
isinstancegetattrrr   r+   compiler_sor   rQ   Z_setup_compileZ_get_cc_argsrP   keyscompiler_typer   r   suffixloweritemslenconcurrent.futuresr   map)r;   sourcesrs   macrosinclude_dirsdebugextra_preargsr3   Zdependsjobsrz   r{   r|   ra   ZfcZfcompZccompobjectsbuildr   Zobjects_to_buildZf77_objectsZother_objectsr1   r   ru   Zbuild_itemsrc   r   poolresr"   r   r#   CCompiler_compile   sx   (








r   compiler"   c                    s(  t d| jj jjf  t| dr0d| jd v r0t dkr$tjdks0| j	d | j
	d  fdd	}|d
rA|  j |drS jD ]
\}}| || qH|drb jD ]}| | qZ|dro| | j j  |dr|| | j j  |dr|  j |dr|  j dS dS )a  
    Customize compiler using distutils command.

    Parameters
    ----------
    cmd : class instance
        An instance inheriting from `distutils.cmd.Command`.
    ignore : sequence of str, optional
        List of `CCompiler` commands (without ``'set_'``) that should not be
        altered. Strings that are checked for are:
        ``('include_dirs', 'define', 'undef', 'libraries', 'library_dirs',
        'rpath', 'link_objects')``.

    Returns
    -------
    None

    zcustomize %s using %scompilerclangr   arm64darwinz-ftrapping-mathc                    s   t  | d d uo| vS r:   )r   )attrr`   ignorer"   r#   allow  s   z&CCompiler_customize_cmd.<locals>.allowr   defineundef	librarieslibrary_dirsrpathlink_objectsN)r   rQ   	__class____name__hasattrr   platformmachinerZ   rr   r   Zset_include_dirsr   r   Zdefine_macror   Zundefine_macroZset_librariesr   Zset_library_dirsr   Zset_runtime_library_dirsr   Zset_link_objectsr   )r;   r`   r   r   namevaluemacror"   r   r#   CCompiler_customize_cmdx  s6   
r   Zcustomize_cmdc                    s   g }d}t | j }dD ]}||vr|| q|D ]}t| |r7t| |}t|t|}||t|f qdt|d  d   fdd|D }d	|S )	Nr   )
versionr   r   Zobject_switchZcompile_switchr   r   r   r   r   z%-rh   zs = %sc                    s   g | ]} | qS r"   r"   )r    propfmtr"   r#   r$     s    z'_compiler_to_string.<locals>.<listcomp>r   )
rP   Zexecutablesr   rr   r   r   maxr   reprr+   )r   propsZmxr   keyvr6   r"   r   r#   _compiler_to_string  s    



r   c                 C   sZ   z|    W n	 ty   Y nw tjjdk r+td t| j tt|  td dS dS )z
    Print the compiler customizations to stdout.

    Parameters
    ----------
    None

    Returns
    -------
    None

    Notes
    -----
    Printing is only done if the distutils log threshold is < 2.

    ri   zP********************************************************************************N)get_version	Exceptionr   Z_global_log	thresholdprintr   r   rF   r"   r"   r#   CCompiler_show_customization  s   
r   Zshow_customizationc           
   	   C   s  t d| jj  t|  |rrz| jd W n ttfy"   Y nw t	| drVd| j
d v rV| jsU| j
d dr?d\}}nd\}}| j
d ||g| j
d	d
  | _nt	| drdt d| j
f  t	| dsrt d| jj  t	| drd| j
d v sd| j
d v sd| j
d v rd| _d
S tjdkrdd
l}dd
l}| }zQz4tj|d}t|d}	|	d W d
   n1 sw   Y  | j|g|dd|d gd d| _W n ty   d| _Y n	w W || d
S W || d
S || w d
S )am  
    Do any platform-specific customization of a compiler instance.

    This method calls `distutils.sysconfig.customize_compiler` for
    platform-specific customization, as well as optionally remove a flag
    to suppress spurious warnings in case C++ code is being compiled.

    Parameters
    ----------
    dist : object
        This parameter is not used for anything.
    need_cxx : bool, optional
        Whether or not C++ has to be compiled. If so (True), the
        ``"-Wstrict-prototypes"`` option is removed to prevent spurious
        warnings. Default is False.

    Returns
    -------
    None

    Notes
    -----
    All the default options used by distutils can be extracted with::

      from distutils import sysconfig
      sysconfig.get_config_vars('CC', 'CXX', 'OPT', 'BASECFLAGS',
                                'CCSHARED', 'LDSHARED', 'SO')

    zcustomize %sz-Wstrict-prototypesr   ccr   gcc)r   g++)r   zc++rh   Nz#### %s #######compiler_cxxzMissing compiler_cxx fix for r   r   Tr%   zfile.cwzint a;
z-MMDz-MFr   )rs   r   F)r   rQ   r   r   r   r   r   r]   
ValueErrorr   r   r   rn   replacewarnZ_auto_dependsr&   r   tempfileshutilmkdtempr'   r+   r)   writer   r   rmtree)
r;   distZneed_cxxabr   r   Ztmpdirfnr5   r"   r"   r#   CCompiler_customize  sd   








r   Z	customize[-.\d]+c                    s    fdd}|S )aU  
    Simple matching of version numbers, for use in CCompiler and FCompiler.

    Parameters
    ----------
    pat : str, optional
        A regular expression matching version numbers.
        Default is ``r'[-.\d]+'``.
    ignore : str, optional
        A regular expression matching patterns to skip.
        Default is ``''``, in which case nothing is skipped.
    start : str, optional
        A regular expression matching the start of where to start looking
        for version numbers.
        Default is ``''``, in which case searching is started at the
        beginning of the version string given to `matcher`.

    Returns
    -------
    matcher : callable
        A function that is appropriate to use as the ``.version_match``
        attribute of a `CCompiler` class. `matcher` takes a single parameter,
        a version string.

    c                    sz   | dd}d}rt|}|sd S | }	 t||d  }|s'd S  r7t |dr7| }q	 |dS )Nr   rI   r   )r   r^   matchendr_   group)r;   version_stringposrD   r   patstartr"   r#   matcherL  s    
z%simple_version_match.<locals>.matcherr"   )r   r   r   r   r"   r   r#   simple_version_match2  s   r   Fc           	         s.  |s
t | dr
| jS |   z| j}W n
 ty   Y dS w |r$|d s&dS z| j}W n  tyK   z| j W n tyB   Y Y dS w  fdd}Y nw z
tj|tj	d}W n% tj
yo } z|j}|j}W Y d}~nd}~w ty{   d}d}Y nw t|}d}d}||v r||}|rt|}|| _|S )	a  
    Return compiler version, or None if compiler is not available.

    Parameters
    ----------
    force : bool, optional
        If True, force a new determination of the version, even if the
        compiler already has a version attribute. Default is False.
    ok_status : list of int, optional
        The list of status values returned by the version look-up process
        for which a version string is returned. If the status value is not
        in `ok_status`, None is returned. Default is ``[0]``.

    Returns
    -------
    version : str or None
        Version string, in the format of `distutils.version.LooseVersion`.

    r   Nr   c                    s"   t  | }|s
d S |d}|S )Nr   )r^   r   r   )r   rD   r   r   r"   r#   r     s
   
z&CCompiler_get_version.<locals>.matcher)rK   rM       )r   r   rH   version_cmdr]   Zversion_matchZversion_patternrS   rT   rU   rV   rW   rX   r0   r   r   )	r;   forceZ	ok_statusr   r   rW   rb   statusr   r"   r   r#   CCompiler_get_versiona  sN   


r   r   c                 C   s   | j dv r| S t| }|j|_|jd gt|jdd  |_tjdr>d|jd v r>|jd |jd g|jdd  |_tjdrX|j	d	 |j	d
 |j	d
 |S |jd g|jdd  |_|S )z
    Return the C++ compiler.

    Parameters
    ----------
    None

    Returns
    -------
    cxx : class instance
        The C++ compiler, as a `CCompiler` instance.

    )msvcintelwintelemwr   rh   N)aixos400Z	ld_so_aixri   r   z-D__STDC_FORMAT_MACROSz-fno-extern-tls-init)
r   r   r   r   r   rZ   r   rn   Z	linker_sorr   )r;   Zcxxr"   r"   r#   CCompiler_cxx_compiler  s&   

r   Zcxx_compiler)intelccompilerZIntelCCompilerz(Intel C Compiler for 32-bit applicationsintel)r   ZIntelItaniumCCompilerz7Intel C Itanium Compiler for Itanium-based applicationsintele)r   ZIntelEM64TCCompilerz(Intel C Compiler for 64-bit applicationsintelem)r   ZIntelCCompilerWz3Intel C Compiler for 32-bit applications on Windowsr   )r   ZIntelEM64TCCompilerWz3Intel C Compiler for 64-bit applications on Windowsr   )ZpathccompilerZPathScaleCCompilerz2PathScale Compiler for SiCortex-based applicationspathcc)ZarmccompilerZArmCCompilerzArm C Compilerarm)ZfujitsuccompilerZFujitsuCCompilerzFujitsu C CompilerZfujitsu))linux.*r   )r   r   )r   r   )r   r   )ntr   )r   r   win32)Zmingw32ccompilerZMingw32CCompilerz>Mingw32 port of GNU C Compiler for Win32(for MSC built Python)r   z+Setting mingw32 as default compiler for nt.))r   r   c                 C   sp  |d u rt  t jk}| d u rtj} z|d u rt| }t| \}}}W n ty<   d|  }|d ur8|d|  }t|w d| }zt	| W nA t
y }	 z5t|	}t dt| |dd  }zt	| W n t
y} }	 z
t|	}td| d }	~	ww W Y d }	~	nd }	~	ww ztj| }
t|
| }W n ty   td||f w |d ||}||_t d|  |S )	Nz5don't know how to compile C/C++ code on platform '%s'z with '%s' compilerznumpy.distutils.z,%s in numpy.distutils; trying from distutils   z4can't compile C/C++ code: unable to load module '%s'zBcan't compile C/C++ code: unable to find class '%s' in module '%s'znew_compiler returns %s)r   Zget_thresholdINFOr&   r   r   r   KeyErrorr   
__import__ImportErrorstrrQ   r   rZ   modulesvarsrR   r   )platr   rR   dry_runr   module_name
class_nameZlong_descriptionrf   re   modulerB   r"   r"   r#   r     s^   
r   c                 C   sB   t | |||}g }|D ]}t|r|t| q|| q|S r:   )_distutils_gen_lib_optionsr   extendrP   rr   )r   r   Zruntime_library_dirsr   rZlib_optsrv   r"   r"   r#   r      s   r   )Zmsvc9r   Z_msvcZbcppZcygwincZemxcZunixcz
distutils.r   )NN)r   r   )NNNr   NNN)r"   )r   )r   r   r   )NNNr   r   )Dr&   r^   rZ   r   r,   r   rS   r   pathlibr   	distutilsr   Zdistutils.ccompilerr   r   r   r   r	   Zdistutils.errorsr
   r   r   r   r   distutils.sysconfigr   Zdistutils.versionr   Znumpy.distutilsr   Znumpy.distutils.exec_commandr   r   Znumpy.distutils.misc_utilr   r   r   r   r   r   r   r   Lockr~   setr   r9   rE   rG   rg   ry   r   r   r   r   r   r   r   r   Z_default_compilersrQ   Z_distutils_new_compilerr  Z_ccr  get_mrA   r"   r"   r"   r#   <module>   s     2

G
-
 
6
Q
/C$

,
