a
XC?h�v � @ sJ d dl Z d dlZd dlZd dlZd dlmZmZmZmZm Z d dl
m Z m Z d dl
mZ d dlmZ d dlZejZejZejZejZejejej 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 �r6ed e- ��G d!d"� d"e �Z.dS )$� N)�DistutilsExecError�DistutilsPlatformError� CompileError�LibError� LinkError)� CCompiler�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: t D ](}| �||�}|r||v r|| S qt|��d S �N)�HKEYS� read_values�KeyError)�cls�path�key�base�d� r � r9 �NET_BASEr r � WINSDK_BASEr r r r r r r7 )r; r< �pr �hr r r r r r: � s2 � �
zMacroExpander.load_macrosc C s$ | j �� D ]\}}|�||�}q
|S r )r7 �items�replace)r; r0 r$ �vr r r �sub� s zMacroExpander.subN)r1 r2 r3 r= r> r: rH r r r r r6 y s r6 c C s� d} t j�| �}|dkr dS |t| � }t j|d � �dd�\}}t|d d� �d }|dkrf|d7 }t|dd � �d
}|dkr�d }|dkr�|| S d S ) NzMSC v.������ � r ������
� � g $@r )�sysr< �find�len�split�int)�prefixr# r0 �restZ majorVersionZ minorVersionr r r �get_build_version� s rW c C s0 g }| D ]"}t j�|�}||vr|�|� q|S r )�osr �normpathr )�pathsZ
reduced_pathsrC Znpr r r �normalize_and_reduce_paths� s r[ c C s< | � tj�}g }|D ]}||vr|�|� qtj�|�}|S r )rS rX �pathsepr �join)ZvariableZoldListZnewListr# Z newVariabler r r �removeDuplicates� s r^ c C s t | }zt�d| d�}W n ty< t�d� d }Y n0 |rNtj�|�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.bat� Unable to find vcvarsall.bat)r8 r r r r �debugrX r �isdir�environ�getr] �pardir�abspath�isfile)r< r9 r? ZtoolskeyZtoolsdir� vcvarsallr r r �find_vcvarsall� s4
�
rh c C sH t | �}h d�}i }|d u r$td��t�d|| � tjd||f tjtjd�}z�|�� \}}|�� dkrvt|� d���|� d�}|�
d�D ]d}t � |�}d |vr�q�|�
� }|�
d d
�\} }
| �� } | |v r�|
�tj�r�|
d d � }
t|
�|| < q�W |j�� |j�� n|j�� |j�� 0 t|�t|�k�rDttt|�� ����|S ) N> r �libZlibpath�includer_ z'Calling 'vcvarsall.bat %s' (version=%s)z
"%s" %s & set)�stdout�stderrr r- �
�=r rI )rh r r r` �
subprocess�Popen�PIPEZ communicate�waitr, rS r r( �stripr' �endswithrX r\ r^ rk �closerl rR �
ValueError�str�list�keys) r< Zarchrg Z interesting�result�popenrk rl �liner r* r r r �query_vcvarsall� sB �
�
r} r@ z(VC %0.1f is not supported by this modulec
@ s� e Zd ZdZi ZdgZg d�ZdgZdgZee e e Z dZ
dZ dZ d Z
d
ZZd Zd+d
d�Zd,dd�Zd-dd�Zd.dd�Zd/dd�Zd0dd�Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Z d1d'd(�Z d)d*� Z dS )2� MSVCCompilerZmsvcz.c)z.ccz.cppz.cxx�.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= H s zMSVCCompiler.__init__Nc C s� |d u rt � }d}||vr(td|f ��dtjv rfdtjv rf| �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 �| _ z(tjd �
d�D ]}| j�|� �q:W n t�yd Y n0 t| j�| _d�| j�tjd < d | _| jdk�r�g d�| _g d�| _ng d�| _g d�| _g d�| _| jdk�r�g d�| _ dg| _ d| _ d S ) Nr
z --plat-name must be one of %sZDISTUTILS_USE_SDKZMSSdkzcl.exezlink.exezlib.exezrc.exezmc.exer �_r ri rj r zxPython was built with %s, and extensions need to be built with the same version of the compiler, but it isn't installed.�;r )�/nologo�/Ox�/MD�/W3�/DNDEBUG)r� �/Od�/MDdr� �/Z7�/D_DEBUG)r� r� r� r� �/GS-r� )r� r� r� r� r� r� r� )�/DLLr� z/INCREMENTAL:NO� )r� r� z/INCREMENTAL:noz/DEBUGr� T) r
r rX rb �find_exe�cc�linkerri �rc�mc�PLAT_TO_VCVARSr} r� rS r\ r� rR Z_MSVCCompiler__productr r r[ r] Zpreprocess_optionsr� �compile_options�compile_options_debug�ldflags_sharedr� �ldflags_shared_debugZldflags_staticr� )r; r� Zok_platsZ plat_specZvc_envrC r r r �
initializeS sb �
�
�
zMSVCCompiler.initialize� c C s� |d u r d}g }|D ]�}t j�|�\}}t j�|�d }|t j�|�d � }|| jvrbtd| ��|rrt j�|�}|| jv r�|� t j�
||| j �� q|| j v r�|� t j�
||| j �� q|� t j�
||| j
�� q|S )Nr� r z Don't know how to compile %s)rX r �splitext�
splitdrive�isabs�src_extensionsr �basename�_rc_extensionsr r] �
res_extension�_mc_extensions�
obj_extension)r; Zsource_filenamesZ strip_dir�
output_dirZ obj_namesZsrc_namer �extr r r �object_filenames� s,
�
� �z MSVCCompiler.object_filenamesc C st | j s| �� | �||||||�} | \}}
}} } |p6g }
|
�d� |rT|
�| j� n |
�| j� |
D �]}z| | \}}W n ty� Y qdY n0 |r�tj �
|�}|| j v r�d| }�nX|| j v r�d| }�nB|| j
v �r<|}d| }z"| �| jg| |g |g � W qd t�y6 } zt|��W Y d }~qdd }~0 0 qdn�|| jv �r tj �|�}tj �|�}zl| �| jgd|d|g |g � tj �tj �|��\}}tj �||d �}| �| jgd| g |g � W qd t�y� } zt|��W Y d }~qdd }~0 0 qdntd||f ��d | }z&| �| jg|
| ||g | � W qd t�yl } zt|��W Y d }~qdd }~0 0 qd|
S )
Nz/cz/Tcz/Tpz/foz-hz-rr z"Don't know how to compile %s to %sz/Fo)r� r� Z_setup_compiler �extendr� r� r rX r re �
_c_extensions�_cpp_extensionsr� �spawnr� r r r� �dirnamer� r� r� r] r� )r; Zsourcesr� r7 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� s�
�
��
��
��
���
zMSVCCompiler.compilec
C s� | j s| �� | �||�\}}| j||d�}| �||�r�|d| g }|rJz| �| jg| � W q� ty� } zt|��W Y d }~q�d }~0 0 n t �
d|� d S )N)r� �/OUT:�skipping %s (up-to-date)) r� r� �_fix_object_args�library_filename�
_need_linkr� ri 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 urptj� ||�}| �
||��rH|t j kr�| r�| j
dd � }q�| jdd � }n| r�| j
}n| j}g }|p�g D ]}|�d| � q�|| | | d| g }tj�|d �} |d u�rLtj�tj�|��\}}tj� | | �|��}|�d| � | �|| |� |
�rl|
|d d�< | �r||�| � | �tj�|�� z| �| jg| � W n. t�y� } zt|��W Y d }~n
d }~0 0 | �||�}|d u�rT|\}}d||f }z| �dd d
||g� W n. t�yD } zt|��W Y d }~n
d }~0 0 n t � d |� d S ) Nz5I don't know what to do with 'runtime_library_dirs': r z/EXPORT:r� r z/IMPLIB:z-outputresource:%s;%szmt.exez-nologoz -manifestr� ) r� r� r� Z
_fix_lib_args�warnrw r rX 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 �link5 s� �
��
��
��
�
zMSVCCompiler.linkc C s, t j�|t j�|�d �}|�d| � d S )Nz .manifest�/MANIFESTFILE:)rX r r] r� r )r; r� r� r� �
temp_manifestr r r r� � s
�z"MSVCCompiler.manifest_setup_ldargsc C s^ |D ]"}|� d�r|�dd�d } q,qd S |tjkr|)r� z*\s*zI|)�w)
�open�readru �rer� �DOTALLrH �search�write�OSError)r; Z
manifest_fileZ
manifest_fZ manifest_buf�patternr r r r� � s6
��
� 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 )Nz sN ��>. #
)