B
YI~d/w � @ sN 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!dkZ"e"r�dZ#d Z$d
Z%n d Z#d Z$d
Z%ddd�Z&G dd� d�Z'G dd� d�Z(dd� Z)dd� Z*dd� Z+dd � Z,d#d d �Z-e)� Z.e.d k �r:ed e. ��G d!d"� d"e �Z/dS )$� N)�DistutilsExecError�DistutilsPlatformError� CompileError�LibError� LinkError)� CCompiler�gen_preprocess_options�gen_lib_options)�log)� get_platform�win32l z1Software\Wow6432Node\Microsoft\VisualStudio\%0.1fz5Software\Wow6432Node\Microsoft\Microsoft SDKs\Windowsz,Software\Wow6432Node\Microsoft\.NETFrameworkz%Software\Microsoft\VisualStudio\%0.1fz)Software\Microsoft\Microsoft SDKs\Windowsz Software\Microsoft\.NETFramework�x86Zamd64)r z win-amd64c @ sL e Zd Zdd� Zee�Zdd� Zee�Zdd� Zee�Zdd� Zee�Zd S )
�Regc C s: x,t D ]$}| �||�}|r||kr|| S qW t|��d S )N)�HKEYS� read_values�KeyError)�cls�path�key�base�d� r � t�d� d }Y nX |rPtj�|�s�d| }tj �
|d �}|r�tj�|�r�tj� |tj tj d�}tj�
|�}tj�|�s�t�d| � d S nt�d| � |s�t�d� d S tj� |d �}tj�|�r�|S t�d
� 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)r6 r r r r
�debugrU r �isdir�environ�getrZ �pardir�abspath�isfile)r: r7 r= ZtoolskeyZtoolsdir� vcvarsallr r r �find_vcvarsall� s2
re c C s< t | �}ddddh}i }|d kr(td��t�d|| � tjd||f tjtjd�}z�|�� \}}|�� d krzt|� d
���|� d
�}xr|�
d �D ]d}t � |�}d |kr�q�|�
� }|�
d d
�\} }
| �� } | |kr�|
�tj�r�|
d d� }
t|
�|| < q�W W d |j�� |j�� X t|�t|�k�r8ttt|�� ����|S )N�include�libZlibpathr z Unable to find vcvarsall.batz'Calling 'vcvarsall.bat %s' (version=%s)z
"%s" %s & set)�stdout�stderrr r+ �
�=r rF )re r r
r] �
subprocess�Popen�PIPEZ communicate�waitr* rP r r& �stripr% �endswithrU rY r\ rh �closeri rO �
ValueError�str�list�keys) r: Zarchrd Z interesting�result�popenrh ri �liner r( r r r �query_vcvarsall� s<
rz 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 �| |||� 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)r9 �verboseZdry_run�forcer r r r; I s zMSVCCompiler.__init__Nc C s |d krt � }d}||kr(td|f ��dtjkrfdtjkrf| �d�rfd| _d| _d| _d| _d | _ n�|t � ksx|d
kr�t
| }nt
t � d t
| }t t |�}|d �
tj�| _|d
tjd
< |d tjd< t| j�dkr�td| j ��| �d�| _| �d�| _| �d�| _| �d�| _| �d �| _ y,x&tjd �
d�D ]}| j�|� �q|| j
k�r>|}d| }y"| �| jg| |g |g � W qh tk
�r8 } z t|��W d d }~X Y qhX qhn�|| jk�r tj �|�}tj �|�}yl| �| jgd|d|g |g � tj �tj �|��\}}tj �||d �}| �| jgd| g |g � W qh tk
�r� } z t|��W d d }~X Y qhX qhntd||f ��d | }y&| �| jg|
| ||g | � W qh tk
�rj } z t|��W d d }~X Y 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 rU r rb �
_c_extensions�_cpp_extensionsr� �spawnr� r r r� �dirnamer� r� r� rZ r� )r9 Zsourcesr� r5 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||d�}| �||�r�|d| g }|rJy| �| jg| � W q� tk
r� } z t|��W d d }~X Y q�X n t �
d|� d S )N)r� z/OUT:zskipping %s (up-to-date)) r� r� �_fix_object_args�library_filename�
_need_linkr� rg r r r
r] ) r9 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| �� | �||�\}}| �|||�}|\}}}|rL| �dt|� � t| |||�}|d k rptj� ||�}| �
||��rH|t j kr�| r�| j
dd � }q�| jdd � }n| r�| j
}n| j}g }x |p�g D ]}|�d| � q�W || | | d| g }tj�|d �} |d k �rPtj�tj�|��\}}tj� | | �|��}|�d| � | �|| |� |
�rp|
|d d�< | �r�|�| � | �tj�|�� y| �| jg| � W n, tk
�r� } z t|��W d d }~X Y nX | �||�}|d k �rT|\}}d||f }y| �dd d
||g� W n, tk
�rD } z t|��W d d }~X Y nX n t � 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�warnrt r rU r rZ r� r �
EXECUTABLEr� r� r r� r� r� r� �manifest_setup_ldargsr� Zmkpathr� r� r r �manifest_get_embed_infor
r] ) r9 � 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 �link6 sl
zMSVCCompiler.linkc C s, t j�|t j�|�d �}|�d| � d S )Nz .manifestz/MANIFESTFILE:)rU r rZ r� r )r9 r� r� r� �
temp_manifestr r r r� � s z"MSVCCompiler.manifest_setup_ldargsc C s` x,|D ] }|� d�r|�dd�d }P qW d S |tjkr>d}nd}| �|�}|d krXd S ||fS )Nz/MANIFESTFILE:�:r rK )�
startswithrP r r� �_remove_visual_c_ref)r9 r� r� �argr� r� r r r r� � s
z$MSVCCompiler.manifest_get_embed_infoc C s� y�t |�}z |�� }W d |�� X t�dtj�}t�|d|�}d}t�|d|�}t�dtj�}t�||�d krrd S t |d�}z|�|� |S |�� X W n t k
r� Y nX d S )NzU|)r� z*\s*zI|)�w)
�open�readrr �rer� �DOTALLrE �search�write�OSError)r9 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 )r9 �dirr r r �library_dir_option� s z MSVCCompiler.library_dir_optionc C s t d��d S )Nztjd �d�D ]*}tj�tj�|�|�}tj�|�rH|S qHW |S )NZPathr� )r~ rU r rZ rb rc r_ rP )r9 Zexer@ �fnr r r r� s zMSVCCompiler.find_exe)r r r )N)r r� )NNNr NNN)Nr N)
NNNNNr NNNN)r ) r/ r0 r1 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
W
V
R +
r{ )r
)0rU rl rM 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_WIN64r6 r? r> r� r r4 rT rX r\ re rz r| r{ r r r r � sJ >. #
)