B
‰ qOA %ã @ 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„ Zddd„ZdddœZG dd„ de ƒZdS )é N)ÚDistutilsExecErrorÚDistutilsPlatformErrorÚ CompileErrorÚLibErrorÚ LinkError)Ú CCompilerÚgen_lib_options)Úlog)Ú get_platform)Úcountc C sü y t jt jdt jt jB d} W n tk
r< t d¡ dS X d}d }| ¤ xœtƒ D ]’}yt | |¡\}}}W n tk
r„ P Y nX |rT|t j
krTt 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 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 úp s z _get_vc_env..z Unable to find vcvarsall.batzcmd /u /c "{}" {} && set)Ústderrzutf-16leÚreplace)r& zError executing {}c S s$ i | ] \}}}|r|r|| ¡ “qS r )r8 )r9 r r5 r: r r r r; ƒ s c s s | ]}| d ¡V qdS )ú=N)Ú partition)r9 Úliner r r ú † s z _get_vc_env..)r Úgetenvr) Úitemsr7 r r+ r, ÚformatZSTDOUTÚdecoder/ r ÚerrorÚoutputÚcmdÚ
splitlines)r4 r6 r5 ÚoutÚexcÚenvr r r Ú _get_vc_envn s$
rM c C sN |st d¡ t j¡}x2|D ]*}t j t j |¡| ¡}t j |¡r |S q W | S )Nr )r rB ÚsplitÚpathsepr r- Úabspathr3 )ZexeÚpathsÚpÚfnr r r Ú _find_exeŒ s
rT Zx86Z x86_amd64)Úwin32z win-amd64c
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.cxxz.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_runÚforcer r r rW Ä 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
|ƒ| _x2| d d¡ t j
¡D ]}|r¾| | t j¡¡ q¾W x2| d d¡ t j
¡D ]}|rò| | t j¡¡ qòW 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)r_ )r_ )r` ra rb )r` ra rb )%r
ÚPLAT_TO_VCVARSr rD ÚtuplerM r* Ú_pathsrN r rO rT ÚccÚlinkerr^ ÚrcÚmcÚmtZadd_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Ú_ldflagsrY )rZ rX r4 Zvc_envrQ ÚdirÚldflagsZ
ldflags_debugr r r Ú
initializeÊ sX
zMSVCCompiler.initializer] c sT ‡fdd„ˆj D ƒ‡fdd„ˆjˆj D ƒ–‰ ˆp4d‰‡ ‡‡fdd„}tt||ƒƒS )Nc s i | ] }ˆ j |“qS r )Ú
obj_extension)r9 Úext)rZ r r r; s z1MSVCCompiler.object_filenames..c s i | ] }ˆ j |“qS r )Ú
res_extension)r9 rt )rZ r r r; s r] c s’ t j | ¡\}}ˆr"t j |¡}n2t j |¡\}}| t jjt jjf¡rT|dd … }yt j ˆ|ˆ | ¡S t k
rΠt
d | ¡ƒ‚Y nX d S )Né z Don't know how to compile {}) r r ÚsplitextÚbasenameÚ
splitdriveÚ
startswithrl Úaltsepr- Ú LookupErrorr rD )rR Úbasert r5 )Úext_mapÚ
output_dirÚ strip_dirr r Ú
make_out_path# s z4MSVCCompiler.object_filenames..make_out_path)Úsrc_extensionsÚ_rc_extensionsÚ_mc_extensionsÚlistÚmap)rZ Zsource_filenamesr€ r r r )r~ r rZ r€ r Úobject_filenames s
z MSVCCompiler.object_filenamesc C sŠ | j s| ¡ | ||||||¡} | \}}
}} } |p6g }
|
d¡ |rT|
| j¡ n |
| j¡ d}x |
D ]}y| | \}}W n tk
r˜ wlY nX |rªtj
|¡}|| j krÀd| }nD|| j krÚd| }d}n*|| j
krB|}d| }y | | jg| ||g ¡ W ql tk
r< } z t|ƒ‚W d d }~X Y qlX qlnÂ|| jkrôtj |¡}tj |¡}y\| | jd|d||g¡ tj tj |¡¡\}}tj ||d ¡}| | jd| |g¡ W ql tk
rî } z t|ƒ‚W d d }~X Y qlX qlntd
||¡ƒ‚| jg|
| }|r$| d ¡ | |¡ | d | ¡ | |¡ y| |¡ W ql tk
r€ } z t|ƒ‚W d d }~X Y qlX qlW |
S )
Nz/cFz/Tcz/TpTz/foz-hz-rz.rcz"Don't know how to compile {} to {}z/EHscz/Fo)rY rr Z_setup_compileÚappendÚextendrn rm ÚKeyErrorr r rP Ú
_c_extensionsÚ_cpp_extensionsrƒ Úspawnrh r r r„ Údirnameri rw rx r- rD rf )rZ Zsourcesr ZmacrosZ include_dirsr Ú
extra_preargsÚextra_postargsZdependsZ compile_infoÚobjectsZpp_optsZbuildZ compile_optsZ add_cpp_optsÚobjÚsrcrt Z input_optZ
output_optÚmsgZh_dirZrc_dirr} r5 Zrc_fileÚargsr r r Úcompile9 sn
zMSVCCompiler.compilec
C s´ | j s| ¡ | ||¡\}}| j||d}| ||¡r¤|d| g }|rJy,t d| jd |¡¡ | | jg| ¡ W q° t
k
r } z t |ƒ‚W d d }~X Y q°X n t d|¡ d S )N)r z/OUT:zExecuting "%s" %sú zskipping %s (up-to-date)) rY rr Ú_fix_object_argsÚlibrary_filenameÚ
_need_linkr r r^ r- r r r ) rZ 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 |¡¡}| |¡ y,t d| jd |¡¡ | | jg| ¡ W n, tk
r¤ } z t|ƒ‚W d 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 )r9 Zsymr r r ú
Ð s z%MSVCCompiler.link..z/OUT:r z/IMPLIB:zExecuting "%s" %sr— zskipping %s (up-to-date))rY rr r˜ Z
_fix_lib_argsÚwarnÚstrr r r r- rš ro rŽ rw rx r™ rˆ r‰ rP Zmkpathr r rg r r r )rZ Z target_descr‘ rœ r Z librariesZ library_dirsZruntime_library_dirsZexport_symbolsr r r Z
build_tempr› Z
fixed_argsZlib_optsrq Z export_optsZld_argsZdll_nameZdll_extZ implib_filer” r r r Úlink sL
zMSVCCompiler.linkc s4 t d¡}z| jt jd< tƒ |¡S |t jd< X d S )Nr )r rB re r) Úsuperr )rZ rH Zold_path)Ú __class__r r r ò s
zMSVCCompiler.spawnc C s d| S )Nz /LIBPATH:r )rZ rp 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 )rZ rp r r r Úruntime_library_dir_option s z'MSVCCompiler.runtime_library_dir_optionc C s
| |¡S )N)r™ )rZ r^ r r r Úlibrary_option s zMSVCCompiler.library_optionc C s` |r|d |g}n|g}xB|D ]6}x0|D ](}t j || |¡¡}t j |¡r(|S q(W q W d S d S )NZ_d)r r r- r™ r3 )rZ Údirsr^ r Z try_namesrp ÚnameZlibfiler r r Úfind_library_file s
z MSVCCompiler.find_library_file)r r r )N)r r] )NNNr NNN)Nr N)
NNNNNr NNNN)r ) Ú__name__Ú
__module__Ú __qualname__Z
compiler_typeZ executablesr‹ rŒ rƒ r„ r‚ ru rs Zstatic_lib_extensionZshared_lib_extensionZstatic_lib_formatZshared_lib_formatZ
exe_extensionrW rr r‡ r– r r¡ r r¤ r¥ r¦ r© Ú
__classcell__r r )r£ r rV ¥ sJ
P
[
8 rV )N)r ZshutilÚstatr+ r Zdistutils.errorsr r r r r Zdistutils.ccompilerr r Z distutilsr Zdistutils.utilr
Ú itertoolsr r! r2 r7 rM rT rc rV r r r r Ú s" "