B
� f/w � @ sR d Z ddlZddlZddlZddlZddlmZmZmZm Z m
Z
ddl m Z m
Z
mZ ddlmZ ddlmZ ddlZejZejZejZejZej ej ej ej fZ ej!dko�ej"dkZ#e#r�d Z$d
Z%d Z&n d Z$d
Z%dZ&ddd�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>ed!e/ ��G d"d#� d#e �Z0dS )%a distutils.msvc9compiler
Contains MSVCCompiler, an implementation of the abstract CCompiler class
for the Microsoft Visual Studio 2008.
The module is compatible with VS 2005 and VS 2008. You can find legacy support
for older versions of VS in distutils.msvccompiler.
� 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 @ sP e Zd ZdZdd� Zee�Zdd� Zee�Zdd� Zee�Zdd � Ze e�Zd
S ) �Regz2Helper class to read values from the registry
c 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| � dS nt�d| � |s�t�d � dS tj� |d
�}tj�|�r�|S t�d � dS ) z�Find the vcvarsall.bat file
At first it tries to find the productdir of VS 2008 in the registry. If
that fails it falls back to the VS90COMNTOOLS env var.
z %s\Setup\VCr> z%Unable to find productdir in registryNzVS%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)r7 r r r r
�debugrV r �isdir�environ�getr[ �pardir�abspath�isfile)r; r8 r> ZtoolskeyZtoolsdir� vcvarsallr r r �find_vcvarsall� s2
re c C s< t | �}ddddh}i }|dkr(td��t�d|| � tjd||f tjtjd �}z�|�� \}}|�� d
krzt|� d ���|� d �}xr|�
d �D ]d}t � |�}d
|kr�q�|�
� }|�
d
d�\} }
| �� } | |kr�|
�tj�r�|
dd� }
t|
�|| < q�W W d|j�� |j�� X t|�t|�k�r8ttt|�� ����|S )zDLaunch vcvarsall.bat and read the settings from its environment
�include�libZlibpathr Nz Unable to find vcvarsall.batz'Calling 'vcvarsall.bat %s' (version=%s)z
"%s" %s & set)�stdout�stderrr r+ �
�=r rG )re r r
r] �
subprocess�Popen�PIPEZ communicate�waitr* rQ r r& �stripr% �endswithrV rZ r\ rh �closeri rP �
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 ZdZdZi ZdgZdddgZdgZdgZ ee e e Z
d Z d
Z d Z
d Zd
ZZdZd.dd�Zd/dd�Zd0dd�Zd1dd�Zd2dd�Zd3d d �Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Z d(d)� Z d4d*d+�Z d,d-� Z dS )5� MSVCCompilerzwConcrete class that implements an interface to Microsoft Visual C++,
as defined by the CCompiler abstract class.Zmsvcz.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)r: �verboseZdry_runZforcer 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 rV r rb �
_c_extensions�_cpp_extensionsr� �spawnr� r r r� �dirnamer� r� r� r[ r� )r: Zsourcesr� r6 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] ) 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| �� | �||�\}}| �|||�}|\}}}|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 rV 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 �link6 sl
zMSVCCompiler.linkc C s, t j�|t j�|�d �}|�d| � d S )Nz .manifestz/MANIFESTFILE:)rV r r[ r� r )r: 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 rL )�
startswithrQ 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 |�� }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� �DOTALLrF �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 �d�D ]*}tj�tj�|�|�}tj�|�rH|S qHW |S )a� Return path to an MSVC executable program.
Tries to find the program in several places: first, one of the
MSVC program search paths from the registry; next, the directories
in the PATH environment variable. If any of those work, return an
absolute path that is known to exist. If none of them work, just
return the original program name, 'exe'.
ZPathr� )r~ rV r r[ rb rc r_ rQ )r: ZexerA �fnr r r r� s zMSVCCompiler.find_exe)r r r )N)r r� )NNNr NNN)Nr N)
NNNNNr NNNN)r ) r/ r0 r1 r2 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{ + sR
W
V
R +
r{ )r
)1r2 rV rl rN 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_WIN64r7 r@ r? r� r r5 rU rY r\ re rz r| r{ r r r r � sL >. #
)