3
\KT � @ s� d dl Z d dlZd dlZd dlZd dlZd dlmZmZmZm Z m
Z
d dl m Z m
Z
d dlmZ d dlmZ d dlmZ dd� Zd d
� Zd d � Zd
d� Zddd�Zddd�Zedg�ZG dd� de �Z dS )� N)�DistutilsExecError�DistutilsPlatformError� CompileError�LibError� LinkError)� CCompiler�gen_lib_options)�log)� get_platform)�countc $ C s� y t jt jdt jt jB d�} W n tk
r< tjd� dS X d}d }| �� x�t� D ]�}yt j | |�\}}}W n tk
r� P Y nX |rT|t j
krTt j j
|�rTytt|��}W n ttfk
r� wTY nX |dkrT||krT|| }}qTW W d Q R X ||fS )Nz'Software\Microsoft\VisualStudio\SxS\VC7)�accessz Visual C++ is not registeredr � )NN)�winregZ OpenKeyEx�HKEY_LOCAL_MACHINEZKEY_READZKEY_WOW64_32KEY�OSErrorr �debugr Z EnumValueZREG_SZ�os�path�isdir�int�float�
ValueError� TypeError)�key� best_version�best_dir�i�v�vc_dirZvt�version� r �//usr/lib64/python3.6/distutils/_msvccompiler.py� _find_vc2015 s0
r" c
s� dd l � dd l} d }d }g �� �fdd�}| j|d�}|j� |j� x��D ]~\}}}}d|krLtjj|ddd�} tjj| �s�qLy td d
� |j d �D ��}
W n t
t fk
r� wLY nX |
|krL|
| }}qLW y |d }W n t k
r� d }Y nX ||fS )
Nr c s �j � j� � d S )N)�extend�findallr )�_distutils_findvs� all_packagesr r! �_getallD s z _find_vc2017.<locals>._getall)�targetz1Microsoft.VisualStudio.Component.VC.Tools.x86.x64ZVCZ AuxiliaryZBuildc s s | ]}t |�V qd S )N)r )�.0r r r r! � <genexpr>P s z _find_vc2017.<locals>.<genexpr>�.)r )
r% � threadingZThread�start�joinr r r �tuple�splitr r �
IndexError) r, r r r' �t�nameZ version_strr Zpackagesr r r )r% r& r! � _find_vc2017: s2
r4 c C s
t � \}}d }d| krdnd}|rztjj|ddddddd �}y d
d l}|j|d d �d }W n tttfk
rx d }Y nX |s�t� \}}|r�tjj|d|dd �}|s�t j
d� dS tjj|d�}tjj |�s�t j
d|� dS | p�tjj |� �rt j
d|� d }||fS )N�amd64Zx64�x86z..ZredistZMSVCz**zMicrosoft.VC141.CRTzvcruntime140.dllr T)� recursive� zMicrosoft.VC140.CRTz$No suitable Visual C++ version foundz
vcvarsall.batz%s cannot be found���)NN)NN) r4 r r r. �glob� ImportErrorr � LookupErrorr" r r �isfile)� plat_specr r � vcruntimeZvcruntime_platZvcredistr: � vcvarsallr r r! �_find_vcvarsall[ s6
rA c C s� t jd�r dd� t jj� D �S t| �\}}|s6td��y&tjdj|| �tj d�j
ddd �}W n@ tj k
r� } z"t j
|j� td
j|j���W Y d d }~X nX d d� d d
� |j� D �D �}|r�||d< |S )NZDISTUTILS_USE_SDKc S s i | ]\}}||j � �qS r )�lower)r) r �valuer r r! �
<dictcomp> s z _get_vc_env.<locals>.<dictcomp>z Unable to find vcvarsall.batzcmd /u /c "{}" {} && set)�stderrzutf-16le�replace)�errorszError executing {}c S s$ i | ] \}}}|r|r||j � �qS r )rB )r) r �_rC r r r! rD � s c s s | ]}|j d �V qdS )�=N)� partition)r) �liner r r! r* � s z _get_vc_env.<locals>.<genexpr>�py_vcruntime_redist)r �getenv�environ�itemsrA r �
subprocessZ check_output�formatZSTDOUT�decodeZCalledProcessErrorr �error�output�cmd�
splitlines)r> r@ r? �out�exc�envr r r! � _get_vc_env} s(
rZ c C sN |st jd�jt j�}x2|D ]*}t jjt jj|�| �}t jj|�r |S q W | S )Nr )r rM r0 �pathsepr r. �abspathr= )Zexe�paths�p�fnr r r! � _find_exe� s
r` r6 Z x86_amd64)Zwin32z win-amd64zvcruntime140.dllc
s� e Zd ZdZi ZdgZdddgZdgZdgZee e e Z dZ
d Z d
Z d Z
d ZZd
Zd)dd�Zd*dd�Zd+dd�Zd,dd�Zd-dd�Zd.dd �Zd d � Z� fd d �Zd!d"� Zd#d$� Zd%d&� Zd/d'd(�Z � Z S )0� MSVCCompilerZmsvcz.cz.ccz.cppz.cxxz.rcz.mcz.resz.objz.libz.dllz%s%sz.exer c C s t j| |||� d | _d| _d S )NF)r �__init__� plat_name� initialized)�self�verboseZdry_runZforcer r r! rb � s zMSVCCompiler.__init__Nc C s> |d krt � }|tkr(tdjtt����t| }t|�}|sDtd��|jdd�| _| jjt j
�}t d|�| _ t d|�| _
t d|�| _t d|�| _t d |�| _t d
|�| _|jd d�| _x2|jd d�jt j
�D ]}|r�| j|jt j�� q�W x6|jd
d�jt j
�D ] }|�r | j|jt j�� �q W d | _dddddg| _| jj| j�rJdnd� ddddddg| _dddg}| j�s~|jd$� dddd g}|d%�| _ |d&�| _ |d'�| _ |d(�| _ |�| _ |�| _!t"j#d f| j t"j#d"f| j t"j#d#f| j t"j$d f| j t"j$d"f| j t"j$d#f| j t"j%d f| j t"j%d"f| j t"j%d#f| j!i | _&d#| _'d S ))Nz --plat-name must be one of {}z7Unable to find a compatible Visual Studio installation.r � zcl.exezlink.exezlib.exezrc.exezmc.exezmt.exerL �include�libz/nologoz/Oxz/W3z/GLz/DNDEBUGz/MDz/MTz/Odz/MDdz/Ziz/D_DEBUGz/INCREMENTAL:NOz/LTCG�/nodefaultlib:libucrt.lib�ucrt.libz /DEBUG:FULL�/MANIFEST:EMBED,ID=1�/DLL�/MANIFEST:EMBED,ID=2�/MANIFESTUAC:NOFT)rj rk )rl )rl )rm rn ro )rm rn ro )(r
�PLAT_TO_VCVARSr rQ r/ rZ �get�_pathsr0 r r[ r` �cc�linkerri �rc�mcZmt�_vcruntime_redistZadd_include_dir�rstrip�sepZadd_library_dirZpreprocess_options�compile_options�append�compile_options_debugr# Z ldflags_exeZldflags_exe_debugZldflags_sharedZldflags_shared_debugZldflags_staticZldflags_static_debugr Z
EXECUTABLEZ
SHARED_OBJECTZSHARED_LIBRARY�_ldflagsrd )re rc r> Zvc_envr] �dir�ldflagsZ
ldflags_debugr r r! �
initialize� s`
zMSVCCompiler.initializerg c sT �fdd��j D ��fdd��j�j D ��� �p4d�� ��fdd�}tt||��S )Nc s i | ] }� j |�qS r )�
obj_extension)r) �ext)re r r! rD 8 s z1MSVCCompiler.object_filenames.<locals>.<dictcomp>c s i | ] }� j |�qS r )�
res_extension)r) r� )re r r! rD 9 s rg c s� t jj| �\}}�r"t jj|�}n2t jj|�\}}|jt jjt jjf�rT|dd � }yt jj�|� | �S t k
r� t
dj | ���Y nX d S )Nr8 z Don't know how to compile {}) r r �splitext�basename�
splitdrive�
startswithry �altsepr. r< r rQ )r^ �baser� rH )�ext_map�
output_dir� strip_dirr r! �
make_out_path> s z4MSVCCompiler.object_filenames.<locals>.make_out_path)�src_extensions�_rc_extensions�_mc_extensions�list�map)re Zsource_filenamesr� r� r� r )r� r� re r� r! �object_filenames3 s
z MSVCCompiler.object_filenamesc C s� | j s| j� | j||||||�} | \}}
}} } |p6g }
|
jd� |rT|
j| j� n |
j| j� d}�x |
D �]}y| | \}}W n tk
r� wlY nX |r�tj j
|�}|| j kr�d| }�nD|| j kr�d| }d}�n*|| j
k�rB|}d| }y | j| jg| ||g � W ql tk
�r< } zt|��W Y d d }~X qlX qln�|| jk�r�tj j|�}tj j|�}y\| j| jd|d||g� tj jtj j|��\}}tj j||d �}| j| jd| |g� W ql tk
�r� } zt|��W Y d d }~X qlX qlntd
j||���| jg|
| }|�r$|jd � |j|� |jd | � |j|� y| j|� W ql tk
�r� } zt|��W Y d d }~X qlX qlW |
S )
Nz/cFz/Tcz/TpTz/foz-hz-rz.rcz"Don't know how to compile {} to {}z/EHscz/Fo)rd r� Z_setup_compiler{ r# r| rz �KeyErrorr r r\ �
_c_extensions�_cpp_extensionsr� �spawnru r r r� �dirnamerv r� r� r. rQ rs )re Zsourcesr� ZmacrosZ include_dirsr �
extra_preargs�extra_postargsZdependsZ compile_info�objectsZpp_optsZbuildZ compile_optsZ add_cpp_opts�obj�srcr� Z input_optZ
output_opt�msgZh_dirZrc_dirr� rH Zrc_file�argsr r r! �compileT sn
zMSVCCompiler.compilec C s� | j s| j� | j||�\}}| j||d�}| j||�r�|d| g }|rJy,tjd| jdj|�� | j | jg| � W q� t
k
r� } zt |��W Y d d }~X q�X n tjd|� d S )N)r� z/OUT:zExecuting "%s" %s� zskipping %s (up-to-date)) rd r� �_fix_object_args�library_filename�
_need_linkr r ri r. r� r r ) re r� Zoutput_libnamer� r � target_lang�output_filenameZlib_argsr� r r r! �create_static_lib� s z MSVCCompiler.create_static_libc C s� | j s| j� | j||�\}}| j|||�}|\}}}|rL| jdt|� � t| |||�}|d k rptjj ||�}| j
||��r�| j || f }dd� |p�g D �}|| | | d| g }tjj |d �} |d k �r tjj
tjj|��\}}tjj | | j|��}|jd| � |
�r |
|d d�< | �r.|j| � tjj tjj|��}| j|� y6tjd| jdj |�� | j| jg| � | j|� W n, tk
�r� } zt|��W Y d d }~X nX n tjd |� d S )
Nz5I don't know what to do with 'runtime_library_dirs': c S s g | ] }d | �qS )z/EXPORT:r )r) Zsymr r r! �
<listcomp>� s z%MSVCCompiler.link.<locals>.<listcomp>z/OUT:r z/IMPLIB:zExecuting "%s" %sr� zskipping %s (up-to-date))rd r� r� Z
_fix_lib_args�warn�strr r r r. r� r} r� r� r� r� r{ r# r\ Zmkpathr r rt r� �_copy_vcruntimer r )re Z target_descr� r� r� Z librariesZ library_dirsZruntime_library_dirsZexport_symbolsr r� r� Z
build_tempr� Z
fixed_argsZlib_optsr Z export_optsZld_argsZdll_nameZdll_extZ implib_filer� r r r! �link� sN
zMSVCCompiler.linkc C s` | j }| stjj|� r d S tjj|�j� tkr6d S tjd|� t j
||�}tj |t j
� d S )Nz Copying "%s")rw r r r= r� rB �
_BUNDLED_DLLSr r �shutil�copy�chmod�stat�S_IWRITE)re r� r? r r r! r� s z MSVCCompiler._copy_vcruntimec s4 t jd�}z| jt jd< t� j|�S |t jd< X d S )Nr )r rM rr rN �superr� )re rU Zold_path)� __class__r r! r� s
zMSVCCompiler.spawnc C s d| S )Nz /LIBPATH:r )re r~ r r r! �library_dir_option& s z MSVCCompiler.library_dir_optionc C s t d��d S )Nz:don't know how to set runtime library search path for MSVC)r )re r~ r r r! �runtime_library_dir_option) s z'MSVCCompiler.runtime_library_dir_optionc C s
| j |�S )N)r� )re ri r r r! �library_option- s zMSVCCompiler.library_optionc C s` |r|d |g}n|g}xB|D ]6}x0|D ](}t jj|| j|��}t jj|�r(|S q(W q W d S d S )NZ_d)r r r. r� r= )re �dirsri r Z try_namesr~ r3 Zlibfiler r r! �find_library_file0 s
z MSVCCompiler.find_library_file)r r r )N)r rg )NNNr NNN)Nr N)
NNNNNr NNNN)r ) �__name__�
__module__� __qualname__Z
compiler_typeZ executablesr� r� r� r� r� r� r� Zstatic_lib_extensionZshared_lib_extensionZstatic_lib_formatZshared_lib_formatZ
exe_extensionrb r� r� r� r� r� r� r� r� r� r� r� �
__classcell__r r )r� r! ra � sL
T
[
9 ra )N) r r� r� rP r Zdistutils.errorsr r r r r Zdistutils.ccompilerr r Z distutilsr Zdistutils.utilr
� itertoolsr r" r4 rA rZ r` rp � frozensetr� ra r r r r! �<module> s$ !"