a
    åze“{  ã                   @   s²   d dl Z d dlZd dlZd dlZd dlmZ d dlT d dlmZm	Z	 d dlm
Z
 d dlmZ d dlmZ d dlmZ d d	lmZ d d
lmZ e d¡Zdd„ ZG dd„ deƒZdS )é    N)ÚCommand)Ú*)Úcustomize_compilerÚget_python_version)Úget_config_h_filename)Únewer_group)Ú	Extension)Úget_platform)Úlog)Ú	USER_BASEz3^[a-zA-Z_][a-zA-Z_0-9]*(\.[a-zA-Z_][a-zA-Z_0-9]*)*$c                  C   s   ddl m}  | ƒ  d S )Nr   ©Úshow_compilers)Údistutils.ccompilerr   r   © r   ú1/usr/lib/python3.9/distutils/command/build_ext.pyr      s    r   c                   @   s  e Zd ZdZdej Zdddddeƒ  fdd	d
de fdddddde fddddddddddgZg d¢Z	ddde
fgZd d!„ Zd"d#„ Zd$d%„ Zd&d'„ Zd(d)„ Zd*d+„ Zd,d-„ Zd.d/„ Zd0d1„ Zejd2d3„ ƒZd4d5„ Zd6d7„ Zd8d9„ Zd:d;„ Zd<d=„ Zd>d?„ Zd@dA„ ZdBdC„ ZdS )DÚ	build_extz8build C/C++ extensions (compile/link to build directory)z (separated by '%s'))z
build-lib=Úbz(directory for compiled extension modules)zbuild-temp=Útz1directory for temporary files (build by-products)z
plat-name=Úpz>platform name to cross-compile for, if supported (default: %s))ÚinplaceÚiziignore build-lib and put compiled extensions into the source directory alongside your pure Python moduleszinclude-dirs=ÚIz.list of directories to search for header files)zdefine=ÚDzC preprocessor macros to define)zundef=ÚUz!C preprocessor macros to undefine)z
libraries=Úlz!external C libraries to link withzlibrary-dirs=ÚLz.directories to search for external C libraries)zrpath=ÚRz7directories to search for shared C libraries at runtime)zlink-objects=ÚOz2extra explicit link objects to include in the link)ÚdebugÚgz'compile/link with debugging information)ÚforceÚfz2forcibly build everything (ignore file timestamps))z	compiler=Úczspecify the compiler type)z	parallel=Újznumber of parallel build jobs)úswig-cppNz)make SWIG create C++ files (default is C))z
swig-opts=Nz!list of SWIG command line options)zswig=Nzpath to the SWIG executable)ÚuserNz#add user include, library and rpath)r   r   r    r$   r%   zhelp-compilerNzlist available compilersc                 C   s‚   d | _ d | _d | _d | _d| _d | _d | _d | _d | _d | _	d | _
d | _d | _d | _d | _d | _d | _d | _d | _d | _d | _d S )Nr   )Ú
extensionsÚ	build_libÚ	plat_nameÚ
build_tempr   ÚpackageÚinclude_dirsÚdefineÚundefÚ	librariesÚlibrary_dirsÚrpathÚlink_objectsr   r    ÚcompilerÚswigÚswig_cppÚ	swig_optsr%   Úparallel©Úselfr   r   r   Úinitialize_optionsj   s*    zbuild_ext.initialize_optionsc           
   
   C   sú  ddl m} |  ddddddd	d
¡ | jd u r8| jj| _| jj| _| ¡ }|jdd}| j	d u rn| jj	pjg | _	t
| j	tƒrŠ| j	 tj¡| _	tjtjkr®| j	 tj tjd¡¡ | j	 | tjj¡¡ ||krâ| j	 | tjj¡¡ |  d¡ |  d¡ | jd u rg | _| jd u rg | _nt
| jtƒr:| j tj¡| _| jd u rNg | _nt
| jtƒrl| j tj¡| _tjdkrh| j tj tjd¡¡ tjtjkr¶| j tj tjd¡¡ | jrÒtj | jd¡| _ntj | jd¡| _| j	 tj tƒ ¡¡ t tdd ƒ}|r| j |¡ | j!dkr*d}n| j!dd … }tj tjd¡}|r\tj ||¡}| j |¡ tj"d d… dkrÆtj# $tj tjd¡¡rº| j tj tjddt%ƒ  d¡¡ n| j d¡ | &d¡rú|j'sî| j | &d ¡¡ n| j d¡ | j(r| j( d!¡}d"d#„ |D ƒ| _(| j)r4| j) d!¡| _)| j*d u rHg | _*n| j* d$¡| _*| j+rºtj t,d¡}tj t,d¡}	tj -|¡r”| j	 |¡ tj -|	¡rº| j |	¡ | j |	¡ t
| j.tƒrözt/| j.ƒ| _.W n t0yô   t1d%ƒ‚Y n0 d S )&Nr   )Ú	sysconfigZbuild)r'   r'   )r)   r)   )r2   r2   )r   r   )r    r    )r6   r6   )r(   r(   é   )Zplat_specificZincluder.   r1   ÚntZlibsZDebugZReleaseÚ_homeÚwin32é   ZPCbuildé   ÚcygwinÚbinÚlibÚpythonZconfigÚ.ÚPy_ENABLE_SHAREDZLIBDIRú,c                 S   s   g | ]}|d f‘qS )Ú1r   )Ú.0Úsymbolr   r   r   Ú
<listcomp>ù   ó    z.build_ext.finalize_options.<locals>.<listcomp>ú zparallel should be an integer)2Ú	distutilsr:   Zset_undefined_optionsr*   ÚdistributionZext_packageZext_modulesr&   Zget_python_incr+   Ú
isinstanceÚstrÚsplitÚosÚpathsepÚsysÚexec_prefixÚbase_exec_prefixÚappendÚpathÚjoinÚextendZensure_string_listr.   r/   r0   ÚnameÚprefixr   r)   Údirnamer   Úgetattrr(   ÚplatformÚ
executableÚ
startswithr   Úget_config_varZpython_buildr,   r-   r5   r%   r   Úisdirr6   ÚintÚ
ValueErrorZDistutilsOptionError)
r8   r:   Z
py_includeZplat_py_includeZ	_sys_homeÚsuffixZnew_libZdefinesZuser_includeZuser_libr   r   r   Úfinalize_options‚   s°    ù




ÿ

þzbuild_ext.finalize_optionsc                 C   sj  ddl m} | jsd S | j ¡ rL|  d¡}| j | ¡ p:g ¡ | j	 
|j¡ || j| j| j| jd| _t| jƒ tjdkr”| jtƒ kr”| j | j¡ | jd ur¬| j | j¡ | jd urÔ| jD ]\}}| j ||¡ q¼| jd urö| jD ]}| j |¡ qä| jd ur| j | j¡ | j	d ur*| j | j	¡ | jd urD| j | j¡ | j d ur^| j !| j ¡ |  "¡  d S )Nr   )Únew_compilerÚ
build_clib)r2   ÚverboseÚdry_runr    r<   )#r   ri   r&   rO   Zhas_c_librariesÚget_finalized_commandr.   r[   Zget_library_namesr/   rX   rj   r2   rk   rl   r    r   rS   r\   r(   r	   Z
initializer+   Zset_include_dirsr,   Zdefine_macror-   Zundefine_macroZset_librariesZset_library_dirsr0   Zset_runtime_library_dirsr1   Zset_link_objectsÚbuild_extensions)r8   ri   rj   r\   ÚvalueÚmacror   r   r   Úrun  s@    

ý




zbuild_ext.runc           
      C   sv  t |tƒstdƒ‚t|ƒD ]T\}}t |tƒr0qt |tƒrFt|ƒdkrNtdƒ‚|\}}t d|¡ t |t	ƒrvt
 |¡s~tdƒ‚t |tƒstdƒ‚t||d ƒ}dD ]"}| |¡}|d ur¢t|||ƒ q¢| d	¡|_d
|v rät d¡ | d¡}|rhg |_g |_|D ]b}	t |	tƒr"t|	ƒdv s*tdƒ‚t|	ƒdkrJ|j |	d ¡ nt|	ƒdkr|j |	¡ q|||< qd S )Nz:'ext_modules' option must be a list of Extension instancesé   zMeach element of 'ext_modules' option must be an Extension instance or 2-tuplezvold-style (ext_name, build_info) tuple found in ext_modules for extension '%s' -- please convert to Extension instancezRfirst element of each tuple in 'ext_modules' must be the extension name (a string)zOsecond element of each tuple in 'ext_modules' must be a dictionary (build info)Úsources)r+   r/   r.   Úextra_objectsÚextra_compile_argsÚextra_link_argsr0   Zdef_filez9'def_file' element of build info dict no longer supportedÚmacros)r;   rr   z9'macros' element of build info dict must be 1- or 2-tupler;   r   )rP   ÚlistÚDistutilsSetupErrorÚ	enumerater   ÚtupleÚlenr
   ÚwarnrQ   Úextension_name_reÚmatchÚdictÚgetÚsetattrÚruntime_library_dirsÚdefine_macrosÚundef_macrosrX   )
r8   r&   r   ÚextÚext_nameZ
build_infoÚkeyÚvalrw   rp   r   r   r   Úcheck_extensions_listV  s^    

ÿ
ÿþ
ÿÿ
ÿ


ÿzbuild_ext.check_extensions_listc                 C   s,   |   | j¡ g }| jD ]}| |j¡ q|S ©N)rŠ   r&   r[   rs   )r8   Ú	filenamesr†   r   r   r   Úget_source_files¤  s
    
zbuild_ext.get_source_filesc                 C   s2   |   | j¡ g }| jD ]}| |  |j¡¡ q|S r‹   )rŠ   r&   rX   Úget_ext_fullpathr\   )r8   Zoutputsr†   r   r   r   Úget_outputs­  s
    
zbuild_ext.get_outputsc                 C   s(   |   | j¡ | jr|  ¡  n|  ¡  d S r‹   )rŠ   r&   r6   Ú_build_extensions_parallelÚ_build_extensions_serialr7   r   r   r   rn   »  s    
zbuild_ext.build_extensionsc              
      sà   ˆj }ˆj du rt ¡ }zddlm} W n ty>   d }Y n0 |d u rTˆ ¡  d S ||dn‰ ‡ ‡fdd„ˆjD ƒ}tˆj|ƒD ]:\}}ˆ 	|¡ | 
¡  W d   ƒ q‚1 s²0    Y  q‚W d   ƒ n1 sÒ0    Y  d S )NTr   )ÚThreadPoolExecutor)Úmax_workersc                    s   g | ]}ˆ   ˆj|¡‘qS r   )ÚsubmitÚbuild_extension)rI   r†   ©Úexecutorr8   r   r   rK   Ñ  s   ÿz8build_ext._build_extensions_parallel.<locals>.<listcomp>)r6   rS   Ú	cpu_countÚconcurrent.futuresr’   ÚImportErrorr‘   r&   ÚzipÚ_filter_build_errorsÚresult)r8   Úworkersr’   Úfuturesr†   Zfutr   r–   r   r   Ã  s"    

ÿz$build_ext._build_extensions_parallelc              	   C   sD   | j D ]8}|  |¡ |  |¡ W d   ƒ q1 s40    Y  qd S r‹   )r&   rœ   r•   )r8   r†   r   r   r   r‘   ×  s    
z"build_ext._build_extensions_serialc              
   c   sV   z
d V  W nF t ttfyP } z(|js(‚ |  d|j|f ¡ W Y d }~n
d }~0 0 d S )Nz"building extension "%s" failed: %s)ZCCompilerErrorZDistutilsErrorZCompileErrorZoptionalr}   r\   )r8   r†   Úer   r   r   rœ   Ü  s    
ÿzbuild_ext._filter_build_errorsc           
      C   sP  |j }|d u st|ttfƒs*td|j ƒ‚t|ƒ}|  |j¡}||j }| j	slt
||dƒslt d|j¡ d S t d|j¡ |  ||¡}|jpŽg }|jd d … }|jD ]}| |f¡ q¤| jj|| j||j| j||jd}|d d … | _|jrø| |j¡ |jpg }|jp| j |¡}	| jj|||  |¡|j|j ||  !|¡| j| j|	d
 d S )Nzjin 'ext_modules' option (extension '%s'), 'sources' must be present and must be a list of source filenamesZnewerz$skipping '%s' extension (up-to-date)zbuilding '%s' extension)Z
output_dirrw   r+   r   Úextra_postargsÚdepends)r.   r/   rƒ   r¡   Úexport_symbolsr   r)   Ztarget_lang)"rs   rP   rx   r{   ry   r\   ÚsortedrŽ   r¢   r    r   r
   r   ÚinfoÚswig_sourcesru   r„   r…   rX   r2   Úcompiler)   r+   Z_built_objectsrt   r[   rv   ÚlanguageZdetect_languageZlink_shared_objectÚget_librariesr/   rƒ   Úget_export_symbols)
r8   r†   rs   Úext_pathr¢   Ú
extra_argsrw   r-   Zobjectsr¨   r   r   r   r•   æ  sV    þÿ


ú
÷zbuild_ext.build_extensionc                 C   s$  g }g }i }| j rt d¡ | j s6d| jv s6d|jv r<d}nd}|D ]P}tj |¡\}}	|	dkrŠ| |d | ¡ | |¡ |d ||< qD| |¡ qD|sž|S | jpª|  	¡ }
|
dg}| 
| j¡ | j rÐ| d¡ | jsì|jD ]}| |¡ qÜ|D ].}|| }t d	||¡ |  |d
||g ¡ qð|S )Nz/--swig-cpp is deprecated - use --swig-opts=-c++z-c++z.cppz.cz.iÚ_wrapéÿÿÿÿz-pythonzswigging %s to %sz-o)r4   r
   r}   r5   rS   rY   ÚsplitextrX   r3   Ú	find_swigr[   r¥   Úspawn)r8   rs   Ú	extensionZnew_sourcesr¦   Zswig_targetsZ
target_extÚsourceÚbaser†   r3   Zswig_cmdÚoÚtargetr   r   r   r¦   2  s@    
ÿ


zbuild_ext.swig_sourcesc                 C   s^   t jdkrdS t jdkrLdD ]*}t j d| d¡}t j |¡r|  S qdS tdt j ƒ‚d S )NÚposixr3   r<   )z1.3z1.2z1.1z	c:\swig%szswig.exez>I don't know how to find (much less run) SWIG on platform '%s')rS   r\   rY   rZ   ÚisfileZDistutilsPlatformError)r8   ZversÚfnr   r   r   r°   h  s    


ÿÿzbuild_ext.find_swigc                 C   sŽ   |   |¡}| d¡}|  |d ¡}| jsRtjj|d d… |g Ž }tj | j|¡S d |dd… ¡}|  d¡}tj 	| 
|¡¡}tj ||¡S )NrE   r®   r   Úbuild_py)Úget_ext_fullnamerR   Úget_ext_filenamer   rS   rY   rZ   r'   rm   ÚabspathZget_package_dir)r8   r‡   ÚfullnameÚmodpathÚfilenamer*   rº   Zpackage_dirr   r   r   rŽ   €  s    


zbuild_ext.get_ext_fullpathc                 C   s    | j d u r|S | j d | S d S )NrE   )r*   )r8   r‡   r   r   r   r»   ›  s    
zbuild_ext.get_ext_fullnamec                 C   s.   ddl m} | d¡}|dƒ}tjj|Ž | S )Nr   ©rc   rE   Z
EXT_SUFFIX)Údistutils.sysconfigrc   rR   rS   rY   rZ   )r8   r‡   rc   r«   Z
ext_suffixr   r   r   r¼   ¤  s    
zbuild_ext.get_ext_filenamec                 C   sv   d|j  d¡d  }z| d¡ W n. tyP   d| d¡ dd¡ d¡ }Y n0 d	| }||jvrp|j |¡ |jS )
NÚ_rE   r®   Úasciir   Zpunycodeó   -ó   _ZPyInit)r\   rR   ÚencodeÚUnicodeEncodeErrorÚreplaceÚdecoder£   rX   )r8   r†   rg   Zinitfunc_namer   r   r   rª   ®  s    "
zbuild_ext.get_export_symbolsc                 C   sä   t jdkr^ddlm} t| j|ƒsÞd}| jr4|d }|t jd? t jd? d@ f }|j|g S n€dd	l	m
} d
}|dƒrÂtt dƒr†d}n<t jdkr–d}n,dtjv rÂ|dƒdkr²d}n|dƒdkrÂd}|rÞ|dƒ}|jd| g S |jS )Nr>   r   )ÚMSVCCompilerz
python%d%dZ_dé   é   éÿ   rÁ   FrF   ÚgetandroidapilevelTrA   Z_PYTHON_HOST_PLATFORMZANDROID_API_LEVELZMACHDEPZ	LDVERSIONrD   )rU   r`   Zdistutils._msvccompilerrË   rP   r2   r   Ú
hexversionr.   rÂ   rc   ÚhasattrrS   Úenviron)r8   r†   rË   ÚtemplateZ	pythonlibrc   Zlink_libpythonZ	ldversionr   r   r   r©   Á  s4    

ÿ



zbuild_ext.get_libraries) Ú__name__Ú
__module__Ú__qualname__ÚdescriptionrS   rT   Zsep_byr	   Zuser_optionsZboolean_optionsr   Zhelp_optionsr9   rh   rq   rŠ   r   r   rn   r   r‘   Ú
contextlibÚcontextmanagerrœ   r•   r¦   r°   rŽ   r»   r¼   rª   r©   r   r   r   r   r   !   sn   
ÿÿÿÿÙ+ÿÿ @N	
	L6	
r   )rØ   rS   ÚrerU   Zdistutils.corer   Zdistutils.errorsrÂ   r   r   r   Zdistutils.dep_utilr   Zdistutils.extensionr   Zdistutils.utilr	   rN   r
   Zsiter   r§   r~   r   r   r   r   r   r   Ú<module>   s"   ÿ