3
\KT @ 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 Zd
d ZdddZdddZedgZG dd de Z dS ) N)DistutilsExecErrorDistutilsPlatformError CompileErrorLibError LinkError) CCompilergen_lib_options)log) get_platform)countc $ C s y t jt jdt jt jB d} W n tk
r< tjd dS X d}d }| xt D ]}yt j | |\}}}W n tk
r P Y nX |rT|t j
krTt j j
|rTytt|}W n ttfk
r wTY nX |dkrT||krT|| }}qTW W d Q R X ||fS )Nz'Software\Microsoft\VisualStudio\SxS\VC7)accessz Visual C++ is not registeredr )NN)winregZ OpenKeyExHKEY_LOCAL_MACHINEZKEY_READZKEY_WOW64_32KEYOSErrorr debugr Z EnumValueZREG_SZospathisdirintfloat
ValueError TypeError)key best_versionbest_dirivvc_dirZvtversion r //usr/lib64/python3.6/distutils/_msvccompiler.py _find_vc2015 s0
r" c
s dd l dd l} d }d }g fdd}| j|d}|j |j xD ]~\}}}}d|krLtjj|ddd} tjj| sqLy td d
|j d D }
W n t
t fk
r wLY nX |
|krL|
| }}qLW y |d }W n t k
r d }Y nX ||fS )
Nr c s j j d S )N)extendfindallr )_distutils_findvs all_packagesr r! _getallD s z _find_vc2017.._getall)targetz1Microsoft.VisualStudio.Component.VC.Tools.x86.x64ZVCZ AuxiliaryZBuildc s s | ]}t |V qd S )N)r ).0r r r r! P s z _find_vc2017...)r )
r% threadingZThreadstartjoinr r r tuplesplitr r
IndexError) r, r r r' tnameZ version_strr Zpackagesr r r )r% r& r! _find_vc2017: s2
r4 c C s
t \}}d }d| krdnd}|rztjj|ddddddd }y d
d l}|j|d d d }W n tttfk
rx d }Y nX |st \}}|rtjj|d|dd }|st j
d dS tjj|d}tjj |st j
d| dS | ptjj | rt j
d| d }||fS )Namd64Zx64x86z..ZredistZMSVCz**zMicrosoft.VC141.CRTzvcruntime140.dllr T) recursive zMicrosoft.VC140.CRTz$No suitable Visual C++ version foundz
vcvarsall.batz%s cannot be found)NN)NN) r4 r r r. glob ImportErrorr LookupErrorr" r r isfile) plat_specr r vcruntimeZvcruntime_platZvcredistr: vcvarsallr r r! _find_vcvarsall[ s6
rA c C s t jdr dd t jj D S t| \}}|s6tdy&tjdj|| tj dj
ddd }W n@ tj k
r } z"t j
|j td
j|jW Y d d }~X nX d d d d
|j D D }|r||d< |S )NZDISTUTILS_USE_SDKc S s i | ]\}}||j qS r )lower)r) r valuer r r!
s z _get_vc_env..z Unable to find vcvarsall.batzcmd /u /c "{}" {} && set)stderrzutf-16lereplace)errorszError executing {}c S s$ i | ] \}}}|r|r||j qS r )rB )r) r _rC r r r! rD s c s s | ]}|j d V qdS )=N) partition)r) liner r r! r* s z _get_vc_env..py_vcruntime_redist)r getenvenvironitemsrA r
subprocessZ check_outputformatZSTDOUTdecodeZCalledProcessErrorr erroroutputcmd
splitlines)r> r@ r? outexcenvr r r! _get_vc_env} s(
rZ c C sN |st jdjt j}x2|D ]*}t jjt jj|| }t jj|r |S q W | S )Nr )r rM r0 pathsepr r. abspathr= )Zexepathspfnr r r! _find_exe s
r` r6 Z x86_amd64)Zwin32z win-amd64zvcruntime140.dllc
s e Zd ZdZi ZdgZdddgZdgZdgZee e e Z dZ
d Z d
Z d Z
d ZZd
Zd)ddZd*ddZd+ddZd,ddZd-ddZd.dd Zd d Z fd d Zd!d" Zd#d$ Zd%d& Zd/d'd(Z Z S )0 MSVCCompilerZmsvcz.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)selfverboseZdry_runZforcer r r! rb s zMSVCCompiler.__init__Nc C s> |d krt }|tkr(tdjttt| }t|}|sDtd|jdd| _| jjt j
}t d|| _ t d|| _
t d|| _t d|| _t d || _t d
|| _|jd d| _x2|jd djt j
D ]}|r| j|jt j qW x6|jd
djt j
D ] }|r | j|jt j q W d | _dddddg| _| jj| jrJdnd ddddddg| _dddg}| js~|jd$ dddd 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 ))Nz --plat-name must be one of {}z7Unable to find a compatible Visual Studio installation.r zcl.exezlink.exezlib.exezrc.exezmc.exezmt.exerL includelibz/nologoz/Oxz/W3z/GLz/DNDEBUGz/MDz/MTz/Odz/MDdz/Ziz/D_DEBUGz/INCREMENTAL:NOz/LTCG/nodefaultlib:libucrt.libucrt.libz /DEBUG:FULL/MANIFEST:EMBED,ID=1/DLL/MANIFEST:EMBED,ID=2/MANIFESTUAC:NOFT)rj rk )rl )rl )rm rn ro )rm rn ro )(r
PLAT_TO_VCVARSr rQ r/ rZ get_pathsr0 r r[ r` cclinkerri rcmcZmt_vcruntime_redistZadd_include_dirrstripsepZadd_library_dirZpreprocess_optionscompile_optionsappendcompile_options_debugr# Z ldflags_exeZldflags_exe_debugZldflags_sharedZldflags_shared_debugZldflags_staticZldflags_static_debugr Z
EXECUTABLEZ
SHARED_OBJECTZSHARED_LIBRARY_ldflagsrd )re rc r> Zvc_envr] dirldflagsZ
ldflags_debugr r r!
initialize s`
zMSVCCompiler.initializerg c sT fddj D fddjj D p4d fdd}tt||S )Nc s i | ] } j |qS r )
obj_extension)r) ext)re r r! rD 8 s z1MSVCCompiler.object_filenames..c s i | ] } j |qS r )
res_extension)r) r )re r r! rD 9 s rg c s t jj| \}}r"t jj|}n2t jj|\}}|jt jjt jjfrT|dd }yt jj| | S t k
r t
dj | Y nX d S )Nr8 z Don't know how to compile {}) r r splitextbasename
splitdrive
startswithry altsepr. r< r rQ )r^ baser rH )ext_map
output_dir strip_dirr r!
make_out_path> s z4MSVCCompiler.object_filenames..make_out_path)src_extensions_rc_extensions_mc_extensionslistmap)re Zsource_filenamesr r r r )r r re r r! object_filenames3 s
z MSVCCompiler.object_filenamesc C s | j s| j | j||||||} | \}}
}} } |p6g }
|
jd |rT|
j| j n |
j| j d}x |
D ]}y| | \}}W n tk
r wlY nX |rtj j
|}|| j krd| }nD|| j krd| }d}n*|| j
krB|}d| }y | j| jg| ||g W ql tk
r< } zt|W Y d d }~X qlX qln|| jkrtj j|}tj j|}y\| j| jd|d||g tj jtj j|\}}tj j||d }| j| jd| |g W ql tk
r } zt|W Y d d }~X qlX qlntd
j||| jg|
| }|r$|jd |j| |jd | |j| y| j| W ql tk
r } zt|W Y d d }~X qlX qlW |
S )
Nz/cFz/Tcz/TpTz/foz-hz-rz.rcz"Don't know how to compile {} to {}z/EHscz/Fo)rd r Z_setup_compiler{ r# r| rz KeyErrorr r r\
_c_extensions_cpp_extensionsr spawnru r r r dirnamerv r r r. rQ rs )re Zsourcesr ZmacrosZ include_dirsr
extra_preargsextra_postargsZdependsZ compile_infoobjectsZpp_optsZbuildZ compile_optsZ add_cpp_optsobjsrcr Z input_optZ
output_optmsgZh_dirZrc_dirr rH Zrc_fileargsr r r! compileT sn
zMSVCCompiler.compilec C s | j s| j | j||\}}| j||d}| j||r|d| g }|rJy,tjd| jdj| | j | jg| W q t
k
r } zt |W Y d d }~X qX n tjd| d S )N)r z/OUT:zExecuting "%s" %s zskipping %s (up-to-date)) rd r _fix_object_argslibrary_filename
_need_linkr r ri r. r r r ) re r Zoutput_libnamer r target_langoutput_filenameZlib_argsr r r r! create_static_lib s z MSVCCompiler.create_static_libc C s | j s| j | j||\}}| j|||}|\}}}|rL| jdt| t| |||}|d k rptjj ||}| j
||r| j || f }dd |pg D }|| | | d| g }tjj |d } |d k r tjj
tjj|\}}tjj | | j|}|jd| |
r |
|d d< | r.|j| tjj tjj|}| j| y6tjd| jdj | | j| jg| | j| W n, tk
r } zt|W Y d d }~X nX n tjd | 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!
s z%MSVCCompiler.link..z/OUT:r z/IMPLIB:zExecuting "%s" %sr zskipping %s (up-to-date))rd r r Z
_fix_lib_argswarnstrr r r r. r r} r r r r r{ r# r\ Zmkpathr r rt r _copy_vcruntimer r )re Z target_descr r r Z librariesZ library_dirsZruntime_library_dirsZexport_symbolsr r r Z
build_tempr Z
fixed_argsZlib_optsr Z export_optsZld_argsZdll_nameZdll_extZ implib_filer r r r! link sN
zMSVCCompiler.linkc C s` | j }| stjj| r d S tjj|j tkr6d S tjd| t j
||}tj |t j
d S )Nz Copying "%s")rw r r r= r rB
_BUNDLED_DLLSr r shutilcopychmodstatS_IWRITE)re r r? r r r! r s z MSVCCompiler._copy_vcruntimec s4 t jd}z| jt jd< t j|S |t jd< X d S )Nr )r rM rr rN superr )re rU Zold_path) __class__r r! r s
zMSVCCompiler.spawnc C s d| S )Nz /LIBPATH:r )re r~ r r r! library_dir_option& s z MSVCCompiler.library_dir_optionc C s t dd S )Nz:don't know how to set runtime library search path for MSVC)r )re r~ r r r! runtime_library_dir_option) s z'MSVCCompiler.runtime_library_dir_optionc C s
| j |S )N)r )re ri r r r! library_option- s zMSVCCompiler.library_optionc C s` |r|d |g}n|g}xB|D ]6}x0|D ](}t jj|| j|}t jj|r(|S q(W q W d S d S )NZ_d)r r r. r r= )re dirsri r Z try_namesr~ r3 Zlibfiler r r! find_library_file0 s
z MSVCCompiler.find_library_file)r r r )N)r rg )NNNr NNN)Nr N)
NNNNNr NNNN)r ) __name__
__module__ __qualname__Z
compiler_typeZ executablesr r r r r r r Zstatic_lib_extensionZshared_lib_extensionZstatic_lib_formatZshared_lib_formatZ
exe_extensionrb r r r r r r r r r r r
__classcell__r r )r r! ra sL
T
[
9 ra )N) r r r rP r Zdistutils.errorsr r r r r Zdistutils.ccompilerr r Z distutilsr Zdistutils.utilr
itertoolsr r" r4 rA rZ r` rp frozensetr ra r r r r! s$ !"