3
\�z � @ s� d dl Z d dlZd dlZd dlZd dlmZ d dlT d dlmZm Z d dlm
Z
d dl m Z d dl
mZ d dlmZ d d lmZ d d
lmZ ejd �Zd d
� ZG dd� de�ZdS )� 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 ddl m} | � d S )Nr )�show_compilers)�distutils.ccompilerr )r � r �3/usr/lib64/python3.6/distutils/command/build_ext.pyr s r c @ s e Zd ZdZdej Zd`dad d
d e� fdcddde fdddedfd d d e fdgdhdidjdkdldmdndodpgZd d%d(d1d8gZ d:d2d;e
fgZ dd?� Z
d@dA� ZdBdC� ZdDdE� ZdFdG� ZdHdI� ZdJdK� ZdLdM� ZejdNdO� �ZdPdQ� ZdRdS� ZdTdU� ZdVdW� ZdXdY� Z dZd[� Z d\d]� Z d^d_� Z d2S )q� build_extz8build C/C++ extensions (compile/link to build directory)z (separated by '%s')�
build-lib=�b�(directory for compiled extension modules� build-temp=�t�1directory for temporary files (build by-products)z
plat-name=�pz>platform name to cross-compile for, if supported (default: %s)�inplace�iz=ignore build-lib and put compiled extensions into the source z,directory alongside your pure Python modulesz
include-dirs=�Iz.list of directories to search for header files�define=�D� C preprocessor macros to define�undef=�U�!C preprocessor macros to undefine�
libraries=�l�!external C libraries to link withz
library-dirs=�Lz.directories to search for external C libraries�rpath=�R�7directories to search for shared C libraries at runtime�
link-objects=�O�2extra explicit link objects to include in the link�debug�g�'compile/link with debugging information�force�f�2forcibly build everything (ignore file timestamps)� compiler=�c�specify the compiler type� parallel=�j� number of parallel build jobs�swig-cppN�)make SWIG create C++ files (default is C)�
swig-opts=�!list of SWIG command line options�swig=�path to the SWIG executable�user�#add user include, library and rpathz
help-compilerzlist 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� ddl m} | jdd'd(d)d*d+d,d-� | jd kr8| jj| _| jj| _|j� }|jd d �}| j d krn| jj pjg | _ t
| j t �r�| j j t
j�| _ tjtjkr�| j jt
jjtjd
�� | j j|� ||kr�| j j|� | jd� | jd� | jd k� r�g | _| jd k�rg | _n t
| jt ��r&| jj t
j�| _| jd k�r:g | _n t
| jt ��rX| jj t
j�| _t
jdk�rT| jjt
jjtjd�� tjtjk�r�| jjt
jjtjd�� | j�r�t
jj| j d�| _ nt
jj| j d�| _ | j jt
jj t � �� t tdd �}|�r| jj|� | j dk�rd}n| j dd � }t
jjtjd�}|�rHt
jj||�}| jj|� tj!d d� dk�s|tj!d d� dk�r�tj"j#t
jjtjd���r�| jjt
jjtjd d t$� d �� n | jjd � |j%d ��r�|j&�s�| jj|j%d!�� n | jjd � | j'�r | j'j d"�}d#d$� |D �| _'| j(�r4| j(j d"�| _(| j)d k�rHg | _)n| j)j d%�| _)| j*�r�t
jjt+d
�}t
jjt+d �} t
jj,|��r�| j j|� t
jj,| ��r�| jj| � | jj| � t
| j-t ��r�yt.| j-�| _-W n t/k
�r� t0d&��Y nX d S ).Nr )� sysconfigZbuildr@ rB rK r+ r. rO rA � )Z
plat_specific�includerG rJ �ntZlibsZDebugZRelease�_home�win32� ZPCbuild� �cygwin�atheos�bin�lib�python�config�.�Py_ENABLE_SHARED�LIBDIR�,c S s g | ] }|d f�qS )�1r )�.0Zsymbolr r r �
� s z.build_ext.finalize_options..� z parallel should be an integer)r@ r@ )rB rB )rK rK )r+ r+ )r. r. )rO rO )rA rA )1� distutilsrR Zset_undefined_optionsrC � distributionZ ext_packageZ ext_modulesr? Zget_python_incrD �
isinstance�str�split�os�pathsep�sys� exec_prefix�base_exec_prefix�append�path�joinZensure_string_listrG rH rI �name�prefixr+ rB �dirnamer �getattrrA �platform�
executable�
startswithr �get_config_varZ python_buildrE rF rN r= r �isdirrO �int�
ValueErrorZDistutilsOptionError)
rP rR Z
py_includeZplat_py_include� _sys_home�suffixZnew_libZdefinesZ user_includeZuser_libr r r �finalize_options� s�
(
zbuild_ext.finalize_optionsc C st ddl m} | jsd S | jj� rL| jd�}| jj|j� p:g � | j j
|j � || j | j
| j| jd�| _ t| j � tjdkr�| jt� kr�| j j| j� | jd k r�| j j| j� | jd k r�x | jD ]\}}| j j||� q�W | jd k �r x| jD ]}| j j|� q�W | jd k �r| j j | j� | j d k �r4| j j | j � | j d k �rN| j j | j � | j d k �rh| j j!| j � | j"� d S )Nr )� new_compiler�
build_clib)rK �verbose�dry_runr. rU )#r
r� r? ri Zhas_c_libraries�get_finalized_commandrG �extendZget_library_namesrH rr r� rK r� r� r. r rm ru rA r Z
initializerD Zset_include_dirsrE Z define_macrorF Zundefine_macroZ
set_librariesZset_library_dirsrI Zset_runtime_library_dirsrJ Zset_link_objects�build_extensions)rP r� r� ru �value�macror r r �run s>
z
build_ext.runc
C s� t |t�std���xnt|�D �]`\}}t |t�r4q t |t� sLt|�dkrTtd��|\}}tjd|� t |t �ozt
j |�s�td��t |t �s�td��t||d �}x*dD ]"}|j
|�}|d k r�t|||� q�W |j
d�|_d|k� r�tjd� |j
d�}|�rxg |_g |_xj|D ]b} t | t��o,t| �dk�s8td��t| �dk�rX|jj| d � nt| �dk�r|jj| � �qW |||< q W 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)�sourcesrD rH rG �
extra_objects�extra_compile_args�extra_link_argsrI Zdef_filez9'def_file' element of build info dict no longer supported�macrosrS z9'macros' element of build info dict must be 1- or 2-tupler )rD rH rG r� r� r� )rS r� )rj �list�DistutilsSetupError� enumerater �tuple�lenr
�warnrk �extension_name_re�match�dict�get�setattr�runtime_library_dirs�
define_macros� undef_macrosrr )
rP r? r �ext�ext_nameZ
build_info�key�valr� r� r r r �check_extensions_listU sT
z build_ext.check_extensions_listc C s0 | j | j� g }x| jD ]}|j|j� qW |S )N)r� r? r� r� )rP � filenamesr� r r r �get_source_files� s
zbuild_ext.get_source_filesc C s6 | j | j� g }x | jD ]}|j| j|j�� qW |S )N)r� r? rr �get_ext_fullpathru )rP Zoutputsr� r r r � get_outputs� s
zbuild_ext.get_outputsc C s( | j | j� | jr | j� n| j� d S )N)r� r? rO �_build_extensions_parallel�_build_extensions_serial)rP r r r r� � s
zbuild_ext.build_extensionsc s� �j }�j dkrtj� }yddlm} W n tk
r@ d }Y nX |d krV�j� d S ||d��T� � �fdd��jD �}x6t�j|�D ]&\}}�j |�� |j
� W d Q R X q�W W d Q R X d S )NTr )�ThreadPoolExecutor)Z max_workersc s g | ]}� j �j|��qS r )Zsubmit�build_extension)re r� )�executorrP r r rf � s z8build_ext._build_extensions_parallel..) rO rm � cpu_countZconcurrent.futuresr� � ImportErrorr� r? �zip�_filter_build_errors�result)rP Zworkersr� Zfuturesr� Zfutr )r� rP r r� � s
z$build_ext._build_extensions_parallelc
C s4 x.| j D ]$}| j|�� | j|� W d Q R X qW d S )N)r? r� r� )rP r� r r r r� � s z"build_ext._build_extensions_serialc c sT y
d V W nD t ttfk
rN } z"|js*� | jd|j|f � W Y d d }~X nX d S )Nz"building extension "%s" failed: %s)ZCCompilerErrorZDistutilsErrorZ CompileErrorZoptionalr� ru )rP r� �er r r r� � s
z build_ext._filter_build_errorsc
C sX |j }|d ks t|ttf� r,td|j ��t|�}| j|j�}||j }| jpZt ||d�snt
j d|j� d S t
j d|j� | j
||�}|jp�g }|jd d � }x|jD ]}|j|f� q�W | jj|| j||j| j ||jd�}|d d � | _|j�r |j|j� |j�p
g }|j�p | jj|�} | jj ||| 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 filenamesZnewerz$skipping '%s' extension (up-to-date)zbuilding '%s' extension)Z
output_dirr� rD r+ �extra_postargs�depends)rG rH r� r� �export_symbolsr+ rB Z target_lang)!r� rj r� r� r� ru r� r� r. r r
r+ �info� swig_sourcesr� r� r� rr rK �compilerB rD Z_built_objectsr� r� r� �languageZdetect_languageZlink_shared_object�
get_librariesrH r� �get_export_symbols)
rP r� r� �ext_pathr� Z
extra_argsr� rF Zobjectsr� r r r r� � sN
zbuild_ext.build_extensionc C s2 g }g }i }| j r tjd� | j s6d| jks6d|jkr
zbuild_ext.swig_sourcesc C s` t jdkrdS t jdkrNxBd
D ]&}t jjd| d�}t jj|�r |S q W dS ntd t j ��d S ) N�posixrL rU �1.3�1.2�1.1z c:\swig%szswig.exez>I don't know how to find (much less run) SWIG on platform '%s')r� r� r� )rm ru rs rt �isfileZDistutilsPlatformError)rP Zvers�fnr r r r� f s
zbuild_ext.find_swigc C s� | j |�}|jd�}| j|d �}| jsRtjj|d d� |g � }tjj| j|�S dj|dd� �}| jd�}tjj |j
|��}tjj||�S )Nr` rS r �build_pyr� r� r� ) �get_ext_fullnamerl �get_ext_filenamer rm rs rt r@ r� �abspathZget_package_dir)rP r� �fullnameZmodpath�filenamerC r� Z package_dirr r r r� ~ s
zbuild_ext.get_ext_fullpathc C s | j d kr|S | j d | S d S )Nr` )rC )rP r� r r r r� � s
zbuild_ext.get_ext_fullnamec C s. ddl m} |jd�}|d�}tjj|� | S )Nr )r| r` �
EXT_SUFFIX)�distutils.sysconfigr| rl rm rs rt )rP r� r| r� Z
ext_suffixr r r r� � s
zbuild_ext.get_ext_filenamec C s0 d|j jd�d }||jkr*|jj|� |jS )NZPyInit_r` rS r� )ru rl r� rr )rP r� Z
initfunc_namer r r r� � s
z build_ext.get_export_symbolsc C s� t jdkrfddlm} t| j|�s\d}| jr4|d }|t jd? t jd? d@ f }|j|g S |jS �nRt jd d � d
kr�d }|t jd? t jd? d@ f }|j|g S t jd d � d k�r>dd
l m
} d }|t jd? t jd? d@ f }g }xB|j d�j � D ]0}|j
d��r |j|dd � � n
|j|� � q�W |j|dg | S t jdk�rP|jS t jd d� dk�rj|jS dd
l m
} |j d��r�djt jd? t jd? d@ |j d��}|j|g S |jS d S )NrW r )� MSVCCompilerz
python%d%dZ_d� � �� rY rZ z python%d.%dr[ )rR �SHLIBSz-lr� �m�darwin� �aixra z
python{}.{}{}�ABIFLAGS)ro ry Zdistutils._msvccompilerr� rj rK r+ �
hexversionrG rh rR r| rl r{ rr �format)rP r� r� �templateZ pythonlibrR Zextrar] r r r r� � sJ
zbuild_ext.get_libraries)r r r )r r r �iignore build-lib and put compiled extensions into the source directory alongside your pure Python modules)r r r� )r r r )r r r )r! r" r# )r% r& r' )r( r) r* )r+ r, r- )r. r/ r0 )r1 r2 r3 )r4 r5 r6 )r7 Nr8 )r9 Nr: )r; Nr< )r= Nr> ) �__name__�
__module__� __qualname__Z descriptionrm rn Zsep_byr Z user_optionsZboolean_optionsr Z help_optionsrQ r� r� r� r� r� r� r� r� �
contextlib�contextmanagerr� r� r� r� r� r� r� r� r� r r r r r ! s�
@N
K6
r )r� rm �rero Zdistutils.corer Zdistutils.errorsr� r r r Zdistutils.dep_utilr Zdistutils.extensionr Zdistutils.utilr rh r
Zsiter r� r� r r r r r r � s