3
\�w � @ sP d dl Z d dlZd dlZd dlZd dlmZmZmZmZm Z d dl
m Z m Z m
Z
d dlmZ d dlmZ d dlZejZejZejZejZejej ej ej fZ ej dko�ej!d%kZ"e"r�d Z#d
Z$d Z%n d Z#d
Z$dZ%dddd�Z&G dd� d�Z'G dd� d�Z(dd� Z)dd� Z*dd � Z+d d � Z,d&d d �Z-e)� Z.e.d!k �r r? r; rH r r r r r7 { s r7 c C s� d} t jj| �}|d
kr dS |t| � }t j|d � jdd�\}}t|d d � �d }|dkrf|d7 }t|dd� �d }|dkr�d }|dkr�|| S d S ) NzMSC v.r � � r
�
� g $@r ����������)�sysr= �find�len�split�int)�prefixr$ r1 �restZ majorVersionZ minorVersionr r r �get_build_version� s rV c C s4 g }x*| D ]"}t jj|�}||kr
|j|� q
W |S )N)�osr �normpathr! )�pathsZ
reduced_pathsrC Znpr r r �normalize_and_reduce_paths� s
rZ c C s@ | j tj�}g }x |D ]}||kr|j|� qW tjj|�}|S )N)rR rW �pathsepr! �join)ZvariableZoldListZnewListr$ Z newVariabler r r �removeDuplicates� s
r] c
C s t | }ytjd| d�}W n" tk
r> tjd� d }Y nX | sTtjj|� r�d| }tj j
|d �}|r�tjj|�r�tjj |tj tj d�}tjj
|�}tjj|�s�tjd| � d S ntjd| � |s�tjd� d S tjj |d �}tjj|�r�|S tjd
� d S ) Nz %s\Setup\VCr@ z%Unable to find productdir in registryzVS%0.f0COMNTOOLSZVCz%s is not a valid directoryz Env var %s is not set or invalidzNo productdir foundz
vcvarsall.batz Unable to find vcvarsall.bat)r9 r r r r
�debugrW r �isdir�environ�getr\ �pardir�abspath�isfile)r= r: r@ ZtoolskeyZtoolsdir� vcvarsallr r r �find_vcvarsall� s2
rf c C s8 t | �}td�}i }|d kr$td��tjd|| � tjd||f tjtjd�}z�|j� \}}|j � d krvt|j
d
���|j
d
�}xr|j d �D ]d}t j
|�}d |kr�q�|j� }|j d d
�\} }
| j� } | |kr�|
jtj�r�|
d d� }
t|
�|| < q�W W d |jj� |jj� X t|�t|�k�r4ttt|j� ����|S )N�include�lib�libpathr z Unable to find vcvarsall.batz'Calling 'vcvarsall.bat %s' (version=%s)z
"%s" %s & set)�stdout�stderrr r. �
�=r )rg rh ri r rM ) rf �setr r
r^ �
subprocess�Popen�PIPEZ communicate�waitr- rR r r) �stripr( �endswithrW r[ r] rj �closerk rQ �
ValueError�str�list�keys) r= Zarchre Z interesting�result�popenrj rk �liner r+ r r r �query_vcvarsall� s<
r} g @z(VC %0.1f is not supported by this modulec
@ s� e Zd ZdZi ZdgZdddgZdgZdgZee e e Z dZ
d Z d
Z d Z
d ZZd
Zd-dd�Zd.dd�Zd/dd�Zd0dd�Zd1dd�Zd2dd �Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Z d3d)d*�Z d+d,� Z dS )4� MSVCCompilerZmsvcz.cz.ccz.cppz.cxxz.rcz.mcz.resz.objz.libz.dllz%s%sz.exer c C s8 t j| |||� t| _d| _g | _d | _d | _d| _d S )Nz Software\Microsoft\VisualStudioF) r r> �VERSION�_MSVCCompiler__versionZ_MSVCCompiler__root�_MSVCCompiler__paths� plat_name�_MSVCCompiler__arch� initialized)r< �verboseZdry_runZforcer r r r> J s zMSVCCompiler.__init__Nc C s |d krt � }d$}||kr(td|f ��dtjkrfdtjkrf| jd�rfd| _d| _d | _d
| _d | _ n�|t � ksx|dkr�t
| }nt
t � d t
| }t t |�}|d
j
tj�| _|d tjd< |d tjd< t| j�dkr�td| j ��| jd�| _| jd�| _| jd �| _| jd
�| _| jd �| _ y,x&tjd
j
d�D ]}| jj|� �q|| j
k�r>|}d| }y"| j| jg| |g |g � W qh tk
�r8 } zt|��W Y d d }~X qhX qhn�|| jk�r tj j|�}tj j|�}yl| j| jgd|d|g |g � tj jtj j|��\}}tj j||d �}| j| jgd| g |g � W qh tk
�r� } zt|��W Y d d }~X qhX qhntd||f ��d | }y&| j| jg|
| ||g | � W qh tk
�rj } zt|��W Y d d }~X qhX qhW |
S )
Nz/cz/Tcz/Tpz/foz-hz-rz.rcz"Don't know how to compile %s to %sz/Fo)r� r� Z_setup_compiler! �extendr� r� r rW r rc �
_c_extensions�_cpp_extensionsr� �spawnr� r r r� �dirnamer� r� r� r\ r� )r< Zsourcesr� r8 Z include_dirsr^ �
extra_preargs�extra_postargsZdependsZ compile_info�objectsZpp_optsZbuildZ compile_opts�obj�srcr� Z input_optZ
output_opt�msgZh_dirZrc_dirr r� Zrc_filer r r �compile� sj
zMSVCCompiler.compilec C s� | j s| j� | j||�\}}| j||d�}| j||�r�|d| g }|rJy| j| jg| � W q� tk
r� } zt|��W Y d d }~X q�X n t j
d|� d S )N)r� z/OUT:zskipping %s (up-to-date)) r� r� �_fix_object_args�library_filename�
_need_linkr� rh r r r
r^ ) r< r� Zoutput_libnamer� r^ � target_lang�output_filenameZlib_argsr� r r r �create_static_lib s z MSVCCompiler.create_static_libc C sX | j s| j� | j||�\}}| j|||�}|\}}}|rL| jdt|� � t| |||�}|d k rptjj ||�}| j
||��rH|t j kr�| r�| j
dd � }q�| jdd � }n| r�| j
}n| j}g }x |p�g D ]}|jd| � q�W || | | d| g }tjj|d �} |d k �rPtjjtjj|��\}}tjj | | j|��}|jd| � | j|| |� |
�rp|
|d d�< | �r�|j| � | jtjj|�� y| j| jg| � W n, tk
�r� } zt|��W Y d d }~X nX | j||�}|d k �rT|\}}d||f }y| jdd d
||g� W n, tk
�rD } zt|��W Y d d }~X nX n t j d |� d S ) Nz5I don't know what to do with 'runtime_library_dirs': r z/EXPORT:z/OUT:r z/IMPLIB:z-outputresource:%s;%szmt.exez-nologoz -manifestzskipping %s (up-to-date)) r� r� r� Z
_fix_lib_args�warnrw r rW r r\ r� r �
EXECUTABLEr� r� r! r� r� r� r� �manifest_setup_ldargsr� Zmkpathr� r� r r �manifest_get_embed_infor
r^ ) r< � target_descr� r� r� Z librariesZ library_dirsZruntime_library_dirsZexport_symbolsr^ r� r� �
build_tempr� Z
fixed_argsZlib_optsZldflagsZ export_optsZsym�ld_argsZdll_nameZdll_extZ implib_filer� ZmfinfoZ
mffilename�mfidZout_argr r r �link8 sl
zMSVCCompiler.linkc C s, t jj|t jj|�d �}|jd| � d S )Nz .manifestz/MANIFESTFILE:)rW r r\ r� r! )r< r� r� r� �
temp_manifestr r r r� � s z"MSVCCompiler.manifest_setup_ldargsc C s` x,|D ] }|j d�r|jdd�d }P qW d S |tjkr>d}nd}| j|�}|d krXd S ||fS )Nz/MANIFESTFILE:�:r r
)�
startswithrR r r� �_remove_visual_c_ref)r< r� r� �argr� r� r r r r� � s
z$MSVCCompiler.manifest_get_embed_infoc C s� y�t |�}z |j� }W d |j� X tjdtj�}tj|d|�}d}tj|d|�}tjdtj�}tj||�d krrd S t |d�}z|j|� |S |j� X W n t k
r� Y nX d S )NzU|)r� z*\s*zI|)�w)
�open�readru �rer� �DOTALLrH �search�write�OSError)r< Z
manifest_fileZ
manifest_fZ manifest_buf�patternr r r r� � s.
z!MSVCCompiler._remove_visual_c_refc C s d| S )Nz /LIBPATH:r )r< �dirr r r �library_dir_option� s z MSVCCompiler.library_dir_optionc C s t d��d S )Nztjd jd�D ]*}tjjtjj|�|�}tjj|�rH|S qHW |S )NZPathr� )r� rW r r\ rc rd r` rR )r< ZexerC �fnr r r r� s zMSVCCompiler.find_exe)r r r )N)r r� )NNNr NNN)Nr N)
NNNNNr NNNN)r ) r2 r3 r4 Z
compiler_typeZ executablesr� r� r� r� r� r� r� Zstatic_lib_extensionZshared_lib_extensionZstatic_lib_formatZshared_lib_formatZ
exe_extensionr> r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r~ , sP
X
V
R +
r~ l )r )0rW ro rO r� Zdistutils.errorsr r r r r Zdistutils.ccompilerr r r Z distutilsr
Zdistutils.utilr �winregZ OpenKeyExr ZEnumKeyr Z EnumValuer' �errorr Z
HKEY_USERS�HKEY_CURRENT_USER�HKEY_LOCAL_MACHINEZHKEY_CLASSES_ROOTr �platform�maxsizeZ NATIVE_WIN64r9 rB rA r� r r7 rV rZ r] rf r} r r~ r r r r � sL >. #
)