��Yf�O � @ s* d Z d d l Z d d l Z d d l Z d d l Z d d l m Z m Z m Z m Z m
Z
d d l m Z m
Z
d d l m Z d d l m Z d d l Z d d l m Z d d � Z d
d � Z d d d
� Z d d d d i Z d d d d d d i Z e d g � Z Gd d � d e � Z d S)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� y) t j t j d d t j t j B�} Wn# t k
rN t j d � d SYn X| ��d } d } x� t � D]� } y t j | | � \ } } } Wn t k
r� PYn X| rl | t j
k rl t j j
| � rl y t t | � � } Wn t t f k
rwl Yn X| d k rl | | k rl | | } } ql W| sBt j d � d
St j j | d � } t j j | � s}t j d | � d Sd }
t j | � } | r�t j j | | j | � � }
t j j |
� s�t j d |
� d }
| |
f SWd QRXd S) Nz'Software\Microsoft\VisualStudio\SxS\VC7�accessz Visual C++ is not registeredr � z$No suitable Visual C++ version foundz
vcvarsall.batz%s cannot be found)NN)NN)NN)�winregZ OpenKeyEx�HKEY_LOCAL_MACHINEZKEY_READZKEY_WOW64_32KEY�OSErrorr �debugr Z EnumValueZREG_SZ�os�path�isdir�int�float�
ValueError� TypeError�join�isfile� _VCVARS_PLAT_TO_VCRUNTIME_REDIST�get�format) � plat_spec�keyZ best_versionZbest_dir�i�vZvc_dirZvt�version� vcvarsall� vcruntimeZvcruntime_spec� r% �</opt/alt/python35/lib64/python3.5/distutils/_msvccompiler.py�_find_vcvarsall sN
'
r' c C s t j d � r) d d � t j j � D� St | � \ } } | sM t d � � y7 t j d j | | � d t j �j
d d d �} WnP t j k
r� } z- t j
| j � t d
j | j � � � WYd d } ~ Xn Xd d � d d
� | j � D� D� } | r
| | d <| S)NZDISTUTILS_USE_SDKc S s% i | ] \ } } | | j � � q Sr% )�lower)�.0r �valuer% r% r& �
<dictcomp>N s z _get_vc_env.<locals>.<dictcomp>z Unable to find vcvarsall.batzcmd /u /c "{}" {} && set�stderrzutf-16le�errors�replacezError executing {}c S s4 i | ]* \ } } } | r | r | | j � � q Sr% )r( )r) r �_r* r% r% r& r+ a s c s s | ] } | j d � Vq d S)�=N)� partition)r) �liner% r% r& � <genexpr>d s z _get_vc_env.<locals>.<genexpr>�py_vcruntime_redist)r �getenv�environ�itemsr' r �
subprocessZ check_outputr ZSTDOUT�decodeZCalledProcessErrorr �error�output�cmd�
splitlines)r r# r$ �out�exc�envr% r% r& � _get_vc_envL s( "
rA c C sm | s! t j d � j t j � } xE | D]= } t j j t j j | � | � } t j j | � r( | Sq( W| 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 r5 �split�pathsepr r �abspathr )Zexe�paths�p�fnr% r% r& � _find_exel s
!rH �win32Zx86z win-amd64Z x86_amd64z1redist\x86\Microsoft.VC{0}0.CRT\vcruntime{0}0.dll�amd64z1redist\x64\Microsoft.VC{0}0.CRT\vcruntime{0}0.dllzvcruntime140.dllc sm e Z d Z d Z d Z i Z d g Z d d d g Z d g Z d g Z e e e e Z
d Z d
Z d Z
d Z d
Z Z d Z d d d d d � Z d d d � Z d d d d � Z d d d d d d d d d � Z d d d d d � Z d d d d d d d d d d d d �
Z d d � Z � f d d! � Z d" d# � Z d$ d% � Z d& d' � Z d d( d) � Z � S)*� 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 s, t j | | | | � d | _ d | _ d S)NF)r �__init__� plat_name� initialized)�self�verboseZdry_runZforcer% r% r& rL � s zMSVCCompiler.__init__Nc C s; | j s t d � � | d k r+ t � } | t k rR t d j t t � � � � t | } t | � } | sz t d � � | j d d � | _ | j j
t j � } t
d | � | _ t
d | � | _ t
d | � | _ t
d | � | _ t
d
| � | _ t
d | � | _ | j d d � | _ x9 | j d
d � j
t j � D] } | rD| j | � qDWx9 | j d d � j
t j � D] } | r�| j | � q�Wd | _ d d d d d g | _ | j j | j r�d n d � d d d d d d g | _ d d d g } | j s | j d% � d d d d g } | d& � | _ | d' � | _ | d( � | _ | d) � | _ | � | _ | � | _! t" j# d f | j t" j# d# f | j t" j# d$ f | j t" j$ d f | j t" j$ d# f | j t" j$ d$ f | j t" j% d f | j t" j% d# f | j t" j% d$ f | j! i | _& d$ | _ d S)*Nzdon't init multiple timesz --plat-name must be one of {}z7Unable to find a compatible Visual Studio installation.r � zcl.exezlink.exezlib.exezrc.exezmc.exezmt.exer4 �include�libz/nologoz/Oxz/W3z/GLz/DNDEBUGz/MDz/MTz/Odz/MDdz/Ziz/D_DEBUGz/INCREMENTAL:NOz/LTCG�/nodefaultlib:libucrt.lib�ucrt.libz /DEBUG:FULL�/MANIFEST:EMBED,ID=1�/DLL�/MANIFEST:EMBED,ID=2�/MANIFESTUAC:NOFT)rT rU )rV )rV )rW rX rY )rW rX rY )'rN �AssertionErrorr
�PLAT_TO_VCVARSr r �tuplerA r �_pathsrB r rC rH �cc�linkerrS �rc�mcZmt�_vcruntime_redistZadd_include_dirZadd_library_dirZpreprocess_options�compile_options�append�compile_options_debug�extendZ ldflags_exeZldflags_exe_debugZldflags_sharedZldflags_shared_debugZldflags_staticZldflags_static_debugr Z
EXECUTABLEZ
SHARED_OBJECTZSHARED_LIBRARY�_ldflags)rO rM r Zvc_envrE �dir�ldflagsZ
ldflags_debugr% r% r& �
initialize� sb
%%
zMSVCCompiler.initializerQ c sv � f d d � � j D� � f d d � � j � j D� � � � pH d � � � � f d d � } t t | | � � S)Nc s i | ] } � j | � q Sr% )�
obj_extension)r) �ext)rO r% r& r+ s z1MSVCCompiler.object_filenames.<locals>.<dictcomp>c s i | ] } � j | � q Sr% )�
res_extension)r) rl )rO r% r& r+ s rQ c s� t j j | � \ } } � r3 t j j | � } nI t j j | � \ } } | j t j j t j j f � r| | d d � } y t j j � | � | � SWn' t k
r� t
d j | � � � Yn Xd S)N� z Don't know how to compile {}) r r �splitext�basename�
splitdrive�
startswith�sep�altsepr � LookupErrorr r )rF �baserl 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)rO Zsource_filenamesry rx rz r% )rw rx rO ry r& �object_filenames
s
& z MSVCCompiler.object_filenamesc C s? | j s | j � | j | | | | | | � } | \ } }
} } } | pO g }
|
j d � | rx |
j | j � n |
j | j � d } x�|
D]�} y | | \ } } Wn t k
r� w� Yn X| r� t j j
| � } | | j k r� d | } n�| | j k r d | } d } n�| | j
k r�| } d | } y% | j | j g | | | g � Wq� t k
r�} z t | � � WYd d } ~ Xq� Xq� n| | j k r�t j j | � } t j j | � } y� | j | j d | d | | g � t j j t j j | � � \ } } t j j | | d � } | j | j d | | g � Wq� t k
r�} z t | � � WYd d } ~ Xq� Xq� n t d
j | | � � � | j g |
| } | r�| j d � | j | � | j d | � | j | � y | j | � Wq� t k
r6} z t | � � WYd d } ~ Xq� Xq� W|
S)
Nz/cFz/Tcz/TpTz/foz-hz-rz.rcz"Don't know how to compile {} to {}z/EHscz/Fo)rN rj Z_setup_compilerd rf re rc �KeyErrorr r rD �
_c_extensions�_cpp_extensionsr| �spawnr` r r r} �dirnamera ro rp r r r^ )rO Zsourcesrx ZmacrosZ include_dirsr �
extra_preargs�extra_postargsZdependsZ compile_info�objectsZpp_optsZbuildZ compile_optsZ add_cpp_opts�obj�srcrl Z input_optZ
output_opt�msgZh_dirZrc_dirrv r/ Zrc_file�argsr% r% r& �compile+ sn
% "$!
#zMSVCCompiler.compilec C s� | j s | j � | j | | � \ } } | j | d | �} | j | | � r� | d | g } | ri y: t j d | j d j | � � | j | j g | � Wq� t
k
r� } z t | � � WYd d } ~ Xq� Xn t j d | � d S)Nrx z/OUT:zExecuting "%s" %s� zskipping %s (up-to-date)) rN rj �_fix_object_args�library_filename�
_need_linkr r rS r r� r r ) rO r� Zoutput_libnamerx r � target_lang�output_filenameZlib_argsr� r% r% r& �create_static_lib� s
"z MSVCCompiler.create_static_libc C sX | j s | j � | j | | � \ } } | j | | | � } | \ } } } | rl | j d t | � � t | | | | � } | d k r� t j j | | � } | j
| | � rD| j | | f } d d � | p� g D� } | | | | d | g } t j j | d � } | d k rrt j j
t j j | � � \ } } t j j | | j | � � } | j d | � |
r�|
| d d � <| r�| j | � t j j t j j | � � } | j | � yG t j d | j d j | � � | j | j g | � | j | � WqTt k
r@} z t | � � WYd d } ~ XqTXn t j d | � d S)
Nz5I don't know what to do with 'runtime_library_dirs': c S s g | ] } d | � q S)z/EXPORT:r% )r) Zsymr% r% r& �
<listcomp>� s z%MSVCCompiler.link.<locals>.<listcomp>z/OUT:r z/IMPLIB:zExecuting "%s" %sr� zskipping %s (up-to-date))rN rj r� Z
_fix_lib_args�warn�strr r r r r� rg r� ro rp r� rd rf rD Zmkpathr r r_ r� �_copy_vcruntimer r )rO Z target_descr� r� rx Z librariesZ library_dirsZruntime_library_dirsZexport_symbolsr r� r� Z
build_tempr� Z
fixed_argsZlib_optsri Z export_optsZld_argsZdll_nameZdll_extZ implib_filer� r% r% r& �link� sN
"zMSVCCompiler.linkc C s� | j } | s# t j j | � r' d St j j | � j � t k rI d St j d | � t j
| | � } t j | t j
� d S)Nz Copying "%s")rb r r r rp r( �
_BUNDLED_DLLSr r �shutil�copy�chmod�stat�S_IWRITE)rO rx r$ r% r% r& r� � s z MSVCCompiler._copy_vcruntimec sH t j d � } z$ | j t j d <t � j | � SWd | t j d <Xd S)Nr )r r5 r] r6 �superr� )rO r< Zold_path)� __class__r% r& r� � s
zMSVCCompiler.spawnc C s d | S)Nz /LIBPATH:r% )rO rh 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 )rO rh r% r% r& �runtime_library_dir_option s z'MSVCCompiler.runtime_library_dir_optionc C s
| j | � S)N)r� )rO rS r% r% r& �library_option s zMSVCCompiler.library_optionc C s� | r | d | g } n | g } xW | D]K } xB | D]: } t j j | | j | � � } t j j | � r6 | Sq6 Wq) Wd Sd S)NZ_d)r r r r� r )rO �dirsrS r Z try_namesrh �nameZlibfiler% r% r& �find_library_file s
z MSVCCompiler.find_library_file) �__name__�
__module__� __qualname__�__doc__Z
compiler_typeZ executablesr� r� r| r} r{ rm rk Zstatic_lib_extensionZshared_lib_extensionZstatic_lib_formatZshared_lib_formatZ
exe_extensionrL rj r� r� r� r� r� r� r� r� r� r� r% r% )r� r& rK � sN
T [9 rK ) r� r r� r� r8 Zdistutils.errorsr r r r r Zdistutils.ccompilerr r Z distutilsr Zdistutils.utilr
r � itertoolsr r' rA rH r[ r � frozensetr� rK r% r% r% r& �<module> s( ( .