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
ddZdd Zdd Zd ddZd ddddZG dd de ZdS ) N)DistutilsExecErrorDistutilsPlatformError CompileErrorLibError LinkError) CCompilergen_lib_options)log) get_platform)countc C s z t jt jdt jt jB d} W n tk
r> td Y dS X d}d }| t D ]}zt | |\}}}W n tk
r Y qY 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 registeredNNr )winregZ OpenKeyExHKEY_LOCAL_MACHINEZKEY_READZKEY_WOW64_32KEYOSErrorr debugr Z EnumValueZREG_SZospathisdirintfloat
ValueError TypeError)key best_versionbest_dirivZvc_dirZvtversion r //usr/lib64/python3.8/distutils/_msvccompiler.py _find_vc2015 s2
r" c
C s dd l } tjdp tjd}|s(dS z8tjtj|ddddd d
d d d
ddg ddd }W n tj t
t fk
r~ Y dS X tj|ddd}tj |rd|fS dS )Nr zProgramFiles(x86)Z ProgramFilesr
zMicrosoft Visual StudioZ Installerz vswhere.exez-latestz -prereleasez -requiresz1Microsoft.VisualStudio.Component.VC.Tools.x86.x64z -propertyZinstallationPathz -products*mbcsstrict)encodingerrorsZVCZ AuxiliaryZBuild )
jsonr environget
subprocess check_outputr joinstripCalledProcessErrorr UnicodeDecodeErrorr )r) rootr r r r! _find_vc2017: s2
r3 x86Zx64ZarmZarm64)r4 x86_amd64x86_arm x86_arm64c C s\ t \}}|st \}}|s*td dS tj|d}tj|sTtd| 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 dr dd t j D S t| \}}|s6tdz&tjd|| tj dj
ddd }W n@ tj k
r } z t
|j td
|jW 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!
w s z _get_vc_env..z Unable to find vcvarsall.batzcmd /u /c "{}" {} && set)stderrzutf-16lereplace)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! s z _get_vc_env..)r getenvr* itemsr< r r, r- formatZSTDOUTdecoder0 r erroroutputcmd
splitlines)r9 r; r: outexcenvr r r! _get_vc_envu s0
rS c C sN |st dt j}|D ].}t jt j|| }t j|r| S q| S Nr )r rH splitpathsepr r. abspathr8 )Zexepathspfnr 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'ddZd(ddZd)ddZd*ddZd+ddZd,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)selfverboseZdry_runZforcer r r! r^ s zMSVCCompiler.__init__Nc C s |d krt }|tkr(tdttt| }t|}|sDtd|dd| _| jt j
}t d|| _ t d|| _
t d|| _t d|| _t d || _t d
|| _|d dt j
D ]}|r| |t j q|d dt j
D ]}|r| |t j qd | _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.exeZincludelibz/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[ cclinkerrd rcmcZmtZadd_include_dirrstripsepZadd_library_dirZpreprocess_optionscompile_optionscompile_options_debugZ ldflags_exeZldflags_exe_debugZldflags_sharedZldflags_shared_debugZldflags_staticZldflags_static_debugr Z
EXECUTABLEZ
SHARED_OBJECTZSHARED_LIBRARY_ldflagsr` )ra r_ r9 Zvc_envrX dirldflagsZ
ldflags_debugr r r!
initialize s
zMSVCCompiler.initializerc c sT fddj D fddjj D p4d fdd}tt||S )Nc s i | ] }| j qS r )
obj_extensionr? extra r r! rA & s z1MSVCCompiler.object_filenames..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 jjfrT|dd }zt j| | W S t k
r t
d | Y nX d S )N z Don't know how to compile {}) r r splitextbasename
splitdrive
startswithrq altsepr. LookupErrorr rJ )rY baserz r: )ext_map
output_dir strip_dirr r!
make_out_path, s z4MSVCCompiler.object_filenames..make_out_path)src_extensions_rc_extensions_mc_extensionslistmap)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 |rtj
|}|| j krd| }nD|| j krd| }d}n*|| j
kr@|}d| }z | | jg| ||g W qh tk
r: } z t|W 5 d }~X Y qhX qhn|| jkrtj |}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_compileappendextendrs rr KeyErrorr r rW
_c_extensions_cpp_extensionsr spawnrn r r r dirnamero r~ r r. rJ rl )ra Zsourcesr ZmacrosZ include_dirsr
extra_preargsextra_postargsZdependsZ compile_infoobjectsZpp_optsZbuildZ compile_optsZ add_cpp_optsobjsrcrz Z input_optZ
output_optmsgZh_dirZrc_dirr r: Zrc_fileargsr r r! compileB sx
zMSVCCompiler.compilec
C s | j s| | ||\}}| j||d}| ||r|d| g }|rJz,td| jd| | | jg| W q t
k
r } z t |W 5 d }~X Y qX n td| d S )N)r /OUT:Executing "%s" %s skipping %s (up-to-date)) r` rw _fix_object_argslibrary_filename
_need_linkr r rd r. r r r ) ra r Zoutput_libnamer r target_langoutput_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 |pg D }|| | | d| g }tj |d } |d k r tj
tj|\}}tj | | |}|d| |
r |
|d d< | r.|| tj tj|}| | z,td| jd | | | jg| W n, tk
r } z t|W 5 d }~X Y nX n td | 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..r r z/IMPLIB:r r r )r` rw r Z
_fix_lib_argswarnstrr 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 dd 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 Zshutilstatr, 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! s2 #