Your IP : 52.14.204.52
U
e5dRN � @ s� d dl Z d dlZd dlZd dlZd dlZd dlmZmZmZm Z m
Z
d dl m Z m
Z
d dlmZ d dlmZ d dlmZ dd� Zd d
� Zd d d
dd�Zdd� Zdd� Zd dd�Zd dddd�ZG dd� de �ZdS ) � 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 tk
r> t�d� Y dS X d}d }| �� t� D ]�}zt � | |�\}}}W n tk
r� Y q�Y nX |rT|t j
krTt j �
|�rTztt|��}W n ttfk
r� Y qTY nX |dkrT||krT|| }}qTW 5 Q R X ||fS )Nz'Software\Microsoft\VisualStudio\SxS\VC7)�accessz Visual C++ is not registered�NNr � )�winregZ OpenKeyEx�HKEY_LOCAL_MACHINEZKEY_READZKEY_WOW64_32KEY�OSErrorr �debugr Z EnumValueZREG_SZ�os�path�isdir�int�float�
ValueError� TypeError)�key� best_version�best_dir�i�vZvc_dirZvt�version� r �//usr/lib64/python3.8/distutils/_msvccompiler.py� _find_vc2015 s2
�
r" c
C s� dd l } tj�d�p tj�d�}|s(dS z8tjtj�|ddd�dd d
d d d
ddg ddd��� }W n tj t
t fk
r~ Y dS X tj�|ddd�}tj� |�r�d|fS dS )Nr zProgramFiles(x86)Z ProgramFilesr
zMicrosoft Visual StudioZ Installerz vswhere.exez-latestz -prereleasez -requiresz1Microsoft.VisualStudio.Component.VC.Tools.x86.x64z -propertyZinstallationPathz -products�*�mbcs�strict)�encoding�errorsZVCZ AuxiliaryZBuild� )
�jsonr �environ�get�
subprocess� check_outputr �join�strip�CalledProcessErrorr �UnicodeDecodeErrorr )r) �rootr r r r! � _find_vc2017: s2
� � r3 �x86Zx64ZarmZarm64)r4 � 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)r3 r" r r r r r. �isfile)� plat_spec�_r r � vcvarsallr r r! �_find_vcvarsallc s
r< c
C s� t �d�r dd� t j�� D �S t| �\}}|s6td��z&tjd�|| �tj d�j
ddd �}W n@ tj k
r� } z t �
|j� td
�|j���W 5 d }~X Y nX d d� d d
� |�� D �D �}|S )NZDISTUTILS_USE_SDKc S s i | ]\}}|� � |�qS r ��lower)�.0r �valuer r r! �
<dictcomp>w s � z _get_vc_env.<locals>.<dictcomp>z Unable to find vcvarsall.batzcmd /u /c "{}" {} && set)�stderrzutf-16le�replace)r' zError executing {}c S s$ i | ] \}}}|r|r|� � |�qS r r= )r? r r: r@ r r r! rA � s
� c s s | ]}|� d �V qdS )�=N)� partition)r? �liner r r! � <genexpr>� s z _get_vc_env.<locals>.<genexpr>)r �getenvr* �itemsr< r r, r- �formatZSTDOUT�decoder0 r �error�output�cmd�
splitlines)r9 r; r: �out�exc�envr r r! � _get_vc_envu s0
�
� �
��rS c C sN |st �d��t j�}|D ].}t j�t j�|�| �}t j�|�r| S q| S �Nr )r rH �split�pathsepr r. �abspathr8 )Zexe�paths�p�fnr r r! � _find_exe� s
r[ r5 r6 r7 )Zwin32z win-amd64z win-arm32z win-arm64c
s� e Zd ZdZi ZdgZdddgZdgZdgZee e e Z dZ
d Z d
Z d Z
d 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 ).� MSVCCompilerZmsvcz.cz.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)�self�verboseZdry_runZforcer r r! r^ � s zMSVCCompiler.__init__Nc C s� |d krt � }|tkr(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 | _d
dddddg| _d
dddddg| _d
ddg}d
dddg}|d �| _|d!�| _|d"�| _|d#�| _ |�| _ |�| _ t j d f| jt j d f| jt j d f| jt j!d f| jt j!d f| jt j!d f| j t j"d f| j t j"d f| j t j"d f| 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�libz/nologoz/Oxz/W3z/GLz/DNDEBUGz/MDz/Odz/MDdz/Ziz/D_DEBUGz/INCREMENTAL:NOz/LTCGz /DEBUG:FULL�/MANIFEST:EMBED,ID=1�/DLL�/MANIFEST:EMBED,ID=2�/MANIFESTUAC:NOFT)re )re )rf rg rh )rf rg rh )%r
�PLAT_TO_VCVARSr rJ �tuplerS r+ �_pathsrU r rV r[ �cc�linkerrd �rc�mcZmtZadd_include_dir�rstrip�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` )ra r_ r9 Zvc_envrX �dir�ldflagsZ
ldflags_debugr r r! �
initialize� s� � � � � �
� zMSVCCompiler.initializerc c sT �fdd��j D ��fdd��j�j D ��� �p4d�� ��fdd�}tt||��S )Nc s i | ] }|� j �qS r )�
obj_extension�r? �ext�ra r r! rA & s z1MSVCCompiler.object_filenames.<locals>.<dictcomp>c s i | ] }|� j �qS r )�
res_extensionry r{ r r! rA ' s rc c s� t j�| �\}}�r"t j�|�}n2t j�|�\}}|�t jjt jjf�rT|dd � }zt j��|� | �W S t k
r� t
d� | ���Y nX d S )N� z Don't know how to compile {}) r r �splitext�basename�
splitdrive�
startswithrq �altsepr. � LookupErrorr rJ )rY �baserz r: )�ext_map�
output_dir� strip_dirr r! �
make_out_path, s z4MSVCCompiler.object_filenames.<locals>.make_out_path)�src_extensions�_rc_extensions�_mc_extensions�list�map)ra Zsource_filenamesr� r� r� r )r� r� ra r� r! �object_filenames! s �z MSVCCompiler.object_filenamesc C s� | j s| �� | �||||||�} | \}}
}} } |p6g }
|
�d� |rT|
�| j� n |
�| j� d}|
D �]}z| | \}}W n tk
r� Y qhY nX |r�tj �
|�}|| j kr�d| }�nD|| j kr�d| }d}�n*|| j
k�r@|}d| }z | �| jg| ||g � W qh tk
�r: } z t|��W 5 d }~X Y qhX qhn�|| jk�r�tj �|�}tj �|�}z\| �| jd|d||g� tj �tj �|��\}}tj �||d �}| �| jd| |g� W qh tk
�r� } z t|��W 5 d }~X Y qhX qhntd
�||���| jg|
| }|�r"|�d � |�|� |�d | � |�|� z| �|� W qh tk
�r~ } z t|��W 5 d }~X Y qhX qh|
S )
Nz/cFz/Tcz/TpTz/foz-hz-rr] z"Don't know how to compile {} to {}z/EHscz/Fo)r` rw Z_setup_compile�append�extendrs rr �KeyErrorr r rW �
_c_extensions�_cpp_extensionsr� �spawnrn r r r� �dirnamero r~ r r. rJ rl )ra Zsourcesr� ZmacrosZ include_dirsr �
extra_preargs�extra_postargsZdependsZ compile_info�objectsZpp_optsZbuildZ compile_optsZ add_cpp_opts�obj�srcrz Z input_optZ
output_opt�msgZh_dirZrc_dirr� r: Zrc_file�argsr r r! �compileB sx
�
�
zMSVCCompiler.compilec
C s� | j s| �� | �||�\}}| j||d�}| �||�r�|d| g }|rJz,t�d| jd�|�� | � | jg| � W q� t
k
r� } z t |��W 5 d }~X Y q�X n t�d|� d S )N)r� �/OUT:�Executing "%s" %s� �skipping %s (up-to-date)) r` rw �_fix_object_args�library_filename�
_need_linkr r rd r. r� r r ) ra r� Zoutput_libnamer� r � target_lang�output_filenameZlib_argsr� r r r! �create_static_lib� s � z MSVCCompiler.create_static_libc
C s� | j s| �� | �||�\}}| �|||�}|\}}}|rL| �dt|� � t| |||�}|d k rptj� ||�}| �
||��r�| j || f }dd� |p�g D �}|| | | d| g }tj� |d �} |d k �r tj�
tj�|��\}}tj� | | �|��}|�d| � |
�r |
|d d�< | �r.|�| � tj� tj�|��}| �|� z,t�d| jd� |�� | �| jg| � W n, tk
�r� } z t|��W 5 d }~X Y nX 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! �
<listcomp>� s z%MSVCCompiler.link.<locals>.<listcomp>r� r z/IMPLIB:r� r� r� )r` rw r� Z
_fix_lib_args�warn�strr r r r. r� rt r� r~ r r� r� r� rW Zmkpathr r rm r� r r )ra Z target_descr� r� r� Z librariesZ library_dirsZruntime_library_dirsZexport_symbolsr r� r� Z
build_tempr� Z
fixed_argsZlib_optsrv Z export_optsZld_argsZdll_nameZdll_extZ implib_filer� r r r! �link� s` �
� �
��
��
zMSVCCompiler.linkc s8 t �d�}z | jt jd< t� �|�W �S |t jd< X d S rT )r rH r* rk �superr� )ra rN Zold_path�� __class__r r! r� � s
zMSVCCompiler.spawnc C s d| S )Nz /LIBPATH:r �ra ru r r r! �library_dir_option s 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 r� r r r! �runtime_library_dir_option
s �z'MSVCCompiler.runtime_library_dir_optionc C s
| � |�S )N)r� )ra rd r r r! �library_option s 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. r� r8 )ra �dirsrd r Z try_namesru �nameZlibfiler r r! �find_library_file s z MSVCCompiler.find_library_file)r r r )N)r rc )NNNr NNN)Nr N)
NNNNNr NNNN)r ) �__name__�
__module__� __qualname__Z
compiler_typeZ executablesr� r� r� r� r� r| rx Zstatic_lib_extensionZshared_lib_extensionZstatic_lib_formatZshared_lib_formatZ
exe_extensionr^ rw r� r� r� r� r� r� r� r� r� �
__classcell__r r r� r! r\ � sd
��
P �
" �
] �
�
E r\ )N) r Zshutil�statr, r Zdistutils.errorsr r r r r Zdistutils.ccompilerr r Z distutilsr Zdistutils.utilr
� itertoolsr r" r3 ZPLAT_SPEC_TO_RUNTIMEr< rS r[ ri r\ r r r r! �<module> s2 #�
�