B
YI~d�M � @ 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�ZG dd� de �ZdS )� 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< t�d� dS X d}d }| �� x�t� D ]�}yt � | |�\}}}W n tk
r� P Y nX |rT|t j
krTt 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 registered)NNr � )�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�vZvc_dirZvt�version� r �p s z _get_vc_env..z Unable to find vcvarsall.batzcmd /u /c "{}" {} && set)�stderrzutf-16le�replace)r& zError executing {}c S s$ i | ] \}}}|r|r||� � �qS r )r8 )r9 r r5 r: r r r r; � s c s s | ]}|� d �V qdS )�=N)� partition)r9 �liner r r � � s z _get_vc_env..)r �getenvr) �itemsr7 r r+ r, �formatZSTDOUT�decoder/ r �error�output�cmd�
splitlines)r4 r6 r5 �out�exc�envr r r � _get_vc_envn s$
rM c C sN |st �d��t j�}x2|D ]*}t j�t j�|�| �}t j�|�r |S q W | S )Nr )r rB �split�pathsepr r- �abspathr3 )Zexe�paths�p�fnr r r � _find_exe� s
rT Zx86Z x86_amd64)�win32z win-amd64c
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 �Z� fd d �Zd d � Zd!d"� Zd#d$� Zd-d%d&�Z� Z S ).� MSVCCompilerZmsvcz.cz.ccz.cppz.cxxz.rcz.mcz.resz.objz.libz.dllz%s%sz.exer c C s t �| |||� d | _d| _d S )NF)r �__init__� plat_name� initialized)�self�verboseZdry_run�forcer r r rW � s zMSVCCompiler.__init__Nc C s |d krt � }|tkr(td�tt����t| }t|�}|sDtd��|�dd�| _| j�t j
�}t d|�| _ t d|�| _
t d|�| _t d|�| _t d |�| _t d
|�| _x2|�d d��t j
�D ]}|r�| �|�t j�� q�W x2|�d d��t j
�D ]}|r�| �|�t j�� q�W d | _d
dddddg| _d
dddddg| _d
ddg}d
dddg}|d �| _|d!�| _|d"�| _|d#�| _ |�| _ |�| _ t j d f| jt j d f| jt j d f| jt j!d f| jt j!d f| jt 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.exeZinclude�libz/nologoz/Oxz/W3z/GLz/DNDEBUGz/MDz/Odz/MDdz/Ziz/D_DEBUGz/INCREMENTAL:NOz/LTCGz /DEBUG:FULL�/MANIFEST:EMBED,ID=1�/DLL�/MANIFEST:EMBED,ID=2�/MANIFESTUAC:NOFT)r_ )r_ )r` ra rb )r` ra rb )%r
�PLAT_TO_VCVARSr rD �tuplerM r* �_pathsrN r rO rT �cc�linkerr^ �rc�mc�mtZadd_include_dir�rstrip�sepZadd_library_dirZpreprocess_options�compile_options�compile_options_debugZ ldflags_exeZldflags_exe_debugZldflags_sharedZldflags_shared_debugZldflags_staticZldflags_static_debugr Z
EXECUTABLEZ
SHARED_OBJECTZSHARED_LIBRARY�_ldflagsrY )rZ rX r4 Zvc_envrQ �dir�ldflagsZ
ldflags_debugr r r �
initialize� sX
zMSVCCompiler.initializer] c sT �fdd��j D ��fdd��j�j D ��� �p4d�� ��fdd�}tt||��S )Nc s i | ] }� j |�qS r )�
obj_extension)r9 �ext)rZ r r r; s z1MSVCCompiler.object_filenames..c s i | ] }� j |�qS r )�
res_extension)r9 rt )rZ r r r; s r] c s� t j�| �\}}�r"t j�|�}n2t j�|�\}}|�t jjt jjf�rT|dd � }yt j��|� | �S t k
r� t
d� | ���Y nX d S )N� z Don't know how to compile {}) r r �splitext�basename�
splitdrive�
startswithrl �altsepr- � LookupErrorr rD )rR �basert r5 )�ext_map�
output_dir� strip_dirr r �
make_out_path# s z4MSVCCompiler.object_filenames..make_out_path)�src_extensions�_rc_extensions�_mc_extensions�list�map)rZ Zsource_filenamesr� r r� r )r~ r rZ r� r �object_filenames s
z MSVCCompiler.object_filenamesc C s� | j s| �� | �||||||�} | \}}
}} } |p6g }
|
�d� |rT|
�| j� n |
�| j� d}�x |
D �]}y| | \}}W n tk
r� wlY nX |r�tj �
|�}|| j kr�d| }�nD|| j kr�d| }d}�n*|| j
k�rB|}d| }y | �| jg| ||g � W ql tk
�r< } z t|��W d d }~X Y qlX qln�|| jk�r�tj �|�}tj �|�}y\| �| jd|d||g� tj �tj �|��\}}tj �||d �}| �| jd| |g� W ql tk
�r� } z t|��W d d }~X Y qlX qlntd
�||���| jg|
| }|�r$|�d � |�|� |�d | � |�|� y| �|� W ql tk
�r� } z t|��W d d }~X Y qlX qlW |
S )
Nz/cFz/Tcz/TpTz/foz-hz-rz.rcz"Don't know how to compile {} to {}z/EHscz/Fo)rY rr Z_setup_compile�append�extendrn rm �KeyErrorr r rP �
_c_extensions�_cpp_extensionsr� �spawnrh r r r� �dirnameri rw rx r- rD rf )rZ Zsourcesr ZmacrosZ include_dirsr �
extra_preargs�extra_postargsZdependsZ compile_info�objectsZpp_optsZbuildZ compile_optsZ add_cpp_opts�obj�srcrt Z input_optZ
output_opt�msgZh_dirZrc_dirr} r5 Zrc_file�argsr r r �compile9 sn
zMSVCCompiler.compilec
C s� | j s| �� | �||�\}}| j||d�}| �||�r�|d| g }|rJy,t�d| jd�|�� | � | jg| � W q� t
k
r� } z t |��W d d }~X Y q�X n t�d|� d S )N)r z/OUT:zExecuting "%s" %s� zskipping %s (up-to-date)) rY rr �_fix_object_args�library_filename�
_need_linkr r r^ r- r� r r ) rZ 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| �� | �||�\}}| �|||�}|\}}}|rL| �dt|� � t| |||�}|d k rptj� ||�}| �
||��r�| j || f }dd� |p�g D �}|| | | d| g }tj� |d �} |d k �r tj�
tj�|��\}}tj� | | �|��}|�d| � |
�r |
|d d�< | �r.|�| � tj� tj�|��}| �|� y,t�d| jd� |�� | �| jg| � W n, tk
�r� } z t|��W d d }~X Y nX n t�d |� d S )
Nz5I don't know what to do with 'runtime_library_dirs': c S s g | ] }d | �qS )z/EXPORT:r )r9 Zsymr r r �
� s z%MSVCCompiler.link..z/OUT:r z/IMPLIB:zExecuting "%s" %sr� zskipping %s (up-to-date))rY rr r� Z
_fix_lib_args�warn�strr r r r- r� ro r� rw rx r� r� r� rP Zmkpathr r rg r� r r )rZ Z target_descr� r� r Z librariesZ library_dirsZruntime_library_dirsZexport_symbolsr r� r� Z
build_tempr� Z
fixed_argsZlib_optsrq Z export_optsZld_argsZdll_nameZdll_extZ implib_filer� r r r �link� sL
zMSVCCompiler.linkc s4 t �d�}z| jt jd< t� �|�S |t jd< X d S )Nr )r rB re r) �superr� )rZ rH Zold_path)� __class__r r r� � s
zMSVCCompiler.spawnc C s d| S )Nz /LIBPATH:r )rZ rp 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 )rZ rp r r r �runtime_library_dir_option s z'MSVCCompiler.runtime_library_dir_optionc C s
| � |�S )N)r� )rZ r^ r r r �library_option s zMSVCCompiler.library_optionc C s` |r|d |g}n|g}xB|D ]6}x0|D ](}t j�|| �|��}t j�|�r(|S q(W q W d S d S )NZ_d)r r r- r� r3 )rZ �dirsr^ r Z try_namesrp �nameZlibfiler r r �find_library_file s
z MSVCCompiler.find_library_file)r r r )N)r r] )NNNr NNN)Nr N)
NNNNNr NNNN)r ) �__name__�
__module__� __qualname__Z
compiler_typeZ executablesr� r� r� r� r� ru rs Zstatic_lib_extensionZshared_lib_extensionZstatic_lib_formatZshared_lib_formatZ
exe_extensionrW rr r� r� r� r� r� r� r� r� r� �
__classcell__r r )r� r rV � sJ
P
[
8 rV )N)r Zshutil�statr+ r Zdistutils.errorsr r r r r Zdistutils.ccompilerr r Z distutilsr Zdistutils.utilr
� itertoolsr r! r2 r7 rM rT rc rV r r r r � s" "