a
Z��^'N c @ s� d Z ddlZddlZddlZddlmZmZmZmZm Z ddl
m Z m Z ddl
mZ ddlmZ ddlmZ dd � Zd
d � Zd d
ddd�Zdd� Zdd� Zd dd�Zd dddd�ZG dd � d e �ZdS ) a distutils._msvccompiler
Contains MSVCCompiler, an implementation of the abstract CCompiler class
for Microsoft Visual Studio 2015.
The module is compatible with VS 2015 and later. You can find legacy support
for older versions in distutils.msvc9compiler and distutils.msvccompiler.
� N)�DistutilsExecError�DistutilsPlatformError� CompileError�LibError� LinkError)� CCompiler�gen_lib_options)�log)� get_platform)�countc C s z t jt jdt jt jB d�} W n ty< t�d� Y dS 0 d}d }| �� t� D ]�}zt � | |�\}}}W n ty� Y q�Y n0 |rR|t j
krRt j �
|�rRztt|��}W n ttfy� Y qRY n0 |dkrR||krR|| }}qRW d � n1 � s�0 Y ||fS )Nz'Software\Microsoft\VisualStudio\SxS\VC7)Zaccessz Visual C++ is not registered�NNr i )�winregZ OpenKeyExZHKEY_LOCAL_MACHINEZKEY_READZKEY_WOW64_32KEY�OSErrorr �debugr
Z EnumValueZREG_SZ�os�path�isdirZintZfloatZ
ValueErrorZ TypeError)�key� best_version�best_dirZiZvZvc_dirZvtZversion� r �//usr/lib64/python3.9/distutils/_msvccompiler.py� _find_vc2015 s2
�
,r c
C s� t j�d�pt j�d�} | s dS z8tjt j�| ddd�ddd d
d d d
dg ddd��� }W n tjt t
fyt Y dS 0 t j�|ddd�}t j� |�r�d|fS dS )aJ Returns "15, path" based on the result of invoking vswhere.exe
If no install is found, returns "None, None"
The version is returned to avoid unnecessarily changing the function
result. It may be ignored when the path is not None.
If vswhere.exe is not available, by definition, VS 2017 is not
installed.
zProgramFiles(x86)Z ProgramFilesr zMicrosoft Visual StudioZ Installerz vswhere.exez-latestz -prereleasez -requiresz1Microsoft.VisualStudio.Component.VC.Tools.x86.x64z -propertyZinstallationPathz -productsZ*ZmbcsZstrict)Zencoding�errorsZVCZ AuxiliaryZBuildi ) r �environ�get�
subprocess� check_outputr �joinZstrip�CalledProcessErrorr
ZUnicodeDecodeErrorr )Zrootr r r r � _find_vc20178 s(
�� r �x86Zx64ZarmZarm64)r � x86_amd64�x86_arm� x86_arm64c C s\ t � \}}|st� \}}|s*t�d� dS tj�|d�}tj�|�sTt�d|� dS |d fS )Nz$No suitable Visual C++ version foundr z
vcvarsall.batz%s cannot be found)r r r r r r r �isfile)� plat_spec�_r r � vcvarsallr r r �_find_vcvarsall_ s
r( c
C s� t �d�r dd� t j�� D �S t| �\}}|s6td��z&tjd�|| �tj d�j
ddd �}W nB tj y� } z(t �
|j� td
�|j���W Y d }~n
d }~0 0 d d� d d
� |�� D �D �}|S )NZDISTUTILS_USE_SDKc S s i | ]\}}|� � |�qS r �Zlower)�.0r �valuer r r �
<dictcomp>s s �z _get_vc_env.<locals>.<dictcomp>z Unable to find vcvarsall.batzcmd /u /c "{}" {} && set)Zstderrzutf-16leZreplace)r zError executing {}c S s$ i | ] \}}}|r|r|� � |�qS r r) )r* r r& r+ r r r r, � s �c s s | ]}|� d �V qdS )z=N)Z partition)r* Zliner r r Z <genexpr>� � z _get_vc_env.<locals>.<genexpr>)r �getenvr Zitemsr( r r r �formatZSTDOUTZdecoder r ZerrorZoutput�cmdZ
splitlines)r% r' r& ZoutZexcZenvr r r � _get_vc_envq s.
�
��
� �r1 c C sN |st �d��t j�}|D ].}t j�t j�|�| �}t j�|�r| S q| S )at 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'.
r )r r. �split�pathsepr r �abspathr$ )Zexe�paths�pZfnr r r � _find_exe� s
r7 r! r" r# )Zwin32z win-amd64z win-arm32z win-arm64c
s� e Zd ZdZdZi ZdgZg d�ZdgZdgZ ee e e Z
dZ dZ d Z
d
Zd 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 )-� MSVCCompilerzwConcrete class that implements an interface to Microsoft Visual C++,
as defined by the CCompiler abstract class.Zmsvcz.c)z.ccz.cppz.cxx�.rcz.mcz.resz.objz.libz.dllz%s%sz.exer c C s t �| |||� d | _d| _d S )NF)r �__init__� plat_name� initialized)�selfZverboseZdry_runZforcer r r r: � s zMSVCCompiler.__init__Nc C s
|d u rt � }|tvr(td�tt����t| }t|�}|sDtd��|�dd�| _| j�t j
�}t d|�| _ t d|�| _
t d|�| _t d|�| _t d |�| _t d
|�| _|�d d��t j
�D ]}|r�| �|�t j�� q�|�d d��t j
�D ]}|r�| �|�t j�� q�d | _g d
�| _g d�| _g d�}g d�}g |�d�| _g |�d�| _g |�d�d�d�| _g |�d�d�d�| _ g |�| _ g |�| _ t j d f| jt j df| jt j df| jt j!d f| jt j!df| jt j!df| j t j"d f| j t j"df| j t j"df| 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�lib)�/nologoz/Ox�/W3z/GLz/DNDEBUGz/MD)r@ z/Odz/MDdz/ZirA z/D_DEBUG)r@ �/INCREMENTAL:NO�/LTCG)r@ rB rC z /DEBUG:FULLz/MANIFEST:EMBED,ID=1z/DLLz/MANIFEST:EMBED,ID=2z/MANIFESTUAC:NOFT)%r �PLAT_TO_VCVARSr r/ Ztupler1 r �_pathsr2 r r3 r7 �cc�linkerr? �rc�mcZmtZadd_include_dirZrstrip�sepZadd_library_dirZpreprocess_options�compile_options�compile_options_debugZ ldflags_exeZldflags_exe_debugZldflags_sharedZldflags_shared_debugZldflags_staticZldflags_static_debugr Z
EXECUTABLEZ
SHARED_OBJECTZSHARED_LIBRARY�_ldflagsr< )r= r; r% Zvc_envr5 �dir�ldflagsZ
ldflags_debugr r r �
initialize� s\ �
� zMSVCCompiler.initializer> c sX i �fdd��j D ���fdd��j�j D ��� �p8d�� ��fdd�}tt||��S )Nc � i | ] }|� j �qS r )�
obj_extension�r* �ext�r= r r r, " r- z1MSVCCompiler.object_filenames.<locals>.<dictcomp>c rQ r )�
res_extensionrS rU r r r, # r- r> c s� t j�| �\}}�r"t j�|�}n2t j�|�\}}|�t jjt jjf�rT|dd � }zt j��|� | �W S t y� t
d� | ���Y n0 d S )Ni z Don't know how to compile {}) r r �splitext�basenameZ
splitdriveZ
startswithrJ Zaltsepr Z LookupErrorr r/ )r6 �baserT r& )�ext_map�
output_dir� strip_dirr r �
make_out_path( s z4MSVCCompiler.object_filenames.<locals>.make_out_path)�src_extensions�_rc_extensions�_mc_extensionsZlistZmap)r= Zsource_filenamesr\ r[ r] r )rZ r[ r= r\ r �object_filenames s ��z MSVCCompiler.object_filenamesc C s� | j s| �� | �||||||�} | \}}
}} } |p6g }
|
�d� |rT|
�| j� n |
�| j� d}|
D �]}z| | \}}W n ty� Y qhY n0 |r�tj �
|�}|| j v r�d| }�nH|| j v r�d| }d}�n.|| j
v �r@|}d| }z | �| jg| ||g � W qh t�y: } zt|��W Y d }~qhd }~0 0 qhn�|| jv �r�tj �|�}tj �|�}z\| �| jd|d||g� tj �tj �|��\}}tj �||d �}| �| jd| |g� W qh t�y� } zt|��W Y d }~qhd }~0 0 qhntd
�||���| jg|
| }|�r$|�d � |�|� |�d | � |�|� z| �|� W qh t�y� } zt|��W Y d }~qhd }~0 0 qh|
S )
Nz/cFz/Tcz/TpTz/foz-hz-rr9 z"Don't know how to compile {} to {}z/EHscz/Fo)r< rP Z_setup_compile�append�extendrL rK ZKeyErrorr r r4 �
_c_extensions�_cpp_extensionsr_ �spawnrH r r r` �dirnamerI rW rX r r/ rF )r= Zsourcesr[ ZmacrosZ include_dirsr �
extra_preargs�extra_postargsZdependsZ compile_info�objectsZpp_optsZbuildZ compile_optsZ add_cpp_optsZobjZsrcrT Z input_optZ
output_opt�msgZh_dirZrc_dirrY r& Zrc_fileZargsr r r �compile> sr
�
�
zMSVCCompiler.compilec
C s� | j s| �� | �||�\}}| j||d�}| �||�r�|d| g }|rJz,t�d| jd�|�� | � | jg| � W q� t
y� } zt |��W Y d }~q�d }~0 0 n t�d|� d S )N)r[ �/OUT:�Executing "%s" %s� �skipping %s (up-to-date)) r< rP �_fix_object_args�library_filename�
_need_linkr r r? r rf r r ) r= rj Zoutput_libnamer[ r � target_lang�output_filenameZlib_argsrk r r r �create_static_lib� s � z MSVCCompiler.create_static_libc
C s� | j s| �� | �||�\}}| �|||�}|\}}}|rL| �dt|� � t| |||�}|d urptj� ||�}| �
||��r�| j || f }dd� |p�g D �}|| | | d| g }tj� |d �} |d u�r tj�
tj�|��\}}tj� | | �|��}|�d| � |
�r |
|d d�< | �r.|�| � tj� tj�|��}| �|� z,t�d| jd� |�� | �| jg| � W n. t�y� } zt|��W Y d }~n
d }~0 0 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 )r* Zsymr r r Z
<listcomp>� r- z%MSVCCompiler.link.<locals>.<listcomp>rm r z/IMPLIB:rn ro rp )r< rP rq Z
_fix_lib_argsZwarnZstrr r r r rs rM rg rW rX rr rb rc r4 Zmkpathr r rG rf r r )r= Z target_descrj ru r[ Z librariesZ library_dirsZruntime_library_dirsZexport_symbolsr rh ri Z
build_temprt Z
fixed_argsZlib_optsrO Z export_optsZld_argsZdll_nameZdll_extZ implib_filerk r r r �link� s^ �
��
��
��
zMSVCCompiler.linkc s@ t �d�}z$| jt jd< t� �|�W |t jd< S |t jd< 0 d S )Nr )r r. rE r Zsuperrf )r= r0 Zold_path�Z __class__r r rf � s
�zMSVCCompiler.spawnc C s d| S )Nz /LIBPATH:r �r= rN r r r �library_dir_option � 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 ry r r r �runtime_library_dir_option s �z'MSVCCompiler.runtime_library_dir_optionc C s
| � |�S �N)rr )r= r? r r r �library_option
r{ zMSVCCompiler.library_optionc C s\ |r|d |g}n|g}|D ]:}|D ]0}t j�|| �|��}t j�|�r$| S q$q d S )NZ_d)r r r rr r$ )r= Zdirsr? r Z try_namesrN ZnameZlibfiler r r �find_library_file
s z MSVCCompiler.find_library_file)r r r r} )r r> )NNNr NNN)Nr N)
NNNNNr NNNN)r ) Z__name__Z
__module__Z __qualname__�__doc__Z
compiler_typeZ executablesrd re r_ r` r^ rV rR Zstatic_lib_extensionZshared_lib_extensionZstatic_lib_formatZshared_lib_formatZ
exe_extensionr: rP ra rl rv rw rf rz r| r~ r Z
__classcell__r r rx r r8 � s\ ��
P �
" �
] �
�
E r8 r} )r� r r r Zdistutils.errorsr r r r r Zdistutils.ccompilerr r Z distutilsr Zdistutils.utilr Z itertoolsr
r r ZPLAT_SPEC_TO_RUNTIMEr( r1 r7 rD r8 r r r r Z<module> s0 !�
�