a
XC?h'N @ s d dl Z d dlZd dlZd dlmZmZmZmZmZ d dl m
Z
m Z d dl m
Z
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 ty< td Y dS 0 d}d }| t D ]}zt | |\}}}W n ty Y qY n0 |rR|t j
krRt j
|rRztt|}W n ttfy Y qRY n0 |dkrR||krR|| }}qRW d n1 s0 Y ||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 s 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 r9 r? r r r! r@ s c s s | ]}| d V qdS )=N) partition)r> liner r r! z _get_vc_env..)r getenvr) itemsr; r r+ r, formatZSTDOUTdecoder/ r erroroutputcmd
splitlines)r8 r: r9 outexcenvr r r! _get_vc_envq s.
rS c C sN |st dt j}|D ].}t jt j|| }t j|r| S q| S Nr )r rH splitpathsepr r- abspathr7 )Zexepathspfnr r r! _find_exe s
r[ r4 r5 r6 )Zwin32z win-amd64z win-arm32z win-arm64c
s e Zd ZdZi ZdgZg dZdgZdgZee e e Z dZ
dZ dZ d Z
d
ZZd Zd%d
dZd&ddZd'ddZd(ddZd)ddZd*ddZ fdd Zd d Zd d Zd!d" Zd+d#d$Z Z S ), MSVCCompilerZmsvcz.c)z.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 u rt }|tvr(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 | _g d
| _g d| _g d}g d}g |d| _g |d| _g |ddd| _g |ddd| _ g || _ g || _ t j d f| jt j df| jt j df| jt j!d f| jt j!df| jt j!df| j t j"d f| j t j"df| j t j"df| 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.exeZincludelib)/nologoz/Ox/W3z/GLz/DNDEBUGz/MD)re z/Odz/MDdz/Zirf z/D_DEBUG)re /INCREMENTAL:NO/LTCG)re rg rh z /DEBUG:FULLz/MANIFEST:EMBED,ID=1z/DLLz/MANIFEST:EMBED,ID=2z/MANIFESTUAC:NOFT)%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_ r8 Zvc_envrX dirldflagsZ
ldflags_debugr r r!
initialize s\
zMSVCCompiler.initializerc c sX i fddj D fddjj D p8d fdd}tt||S )Nc s i | ] }| j qS r )
obj_extensionr> extra r r! r@ " rG z1MSVCCompiler.object_filenames..c s i | ] }| j qS r )
res_extensionry r{ r r! r@ # rG rc c s t j| \}}r"t j|}n2t j|\}}|t jjt jjfrT|dd }zt j| | W S t y t
d | Y n0 d S )N z Don't know how to compile {}) r r splitextbasename
splitdrive
startswithrq altsepr- LookupErrorr rJ )rY baserz r9 )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 ty Y qhY n0 |rtj
|}|| j v rd| }nH|| j v rd| }d}n.|| j
v r@|}d| }z | | jg| ||g W qh ty: } zt|W Y d }~qhd }~0 0 qhn|| jv rtj |}tj |}z\| | jd|d||g tj tj |\}}tj ||d }| | jd| |g W qh ty } zt|W Y d }~qhd }~0 0 qhntd
||| jg|
| }|r$|d || |d | || z| | W qh ty } zt|W Y d }~qhd }~0 0 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 r9 Zrc_fileargsr r r! compile> sr
zMSVCCompiler.compilec
C s | j s| | ||\}}| j||d}| ||r|d| g }|rJz,td| jd| | | jg| W q t
y } zt |W Y d }~qd }~0 0 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 urptj ||}|
||r| j || f }dd |pg D }|| | | d| g }tj |d } |d ur tj
tj|\}}tj | | |}|d| |
r |
|d d< | r.|| tj tj|}| | z,td| jd | | | jg| W n. ty } zt|W Y d }~n
d }~0 0 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!
rG 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 s@ t d}z$| jt jd< t |W |t jd< S |t jd< 0 d S rT )r rH rk r) 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 r7 )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\ sZ
P
"
]
E r\ )N)r r+ r Zdistutils.errorsr r r r r Zdistutils.ccompilerr r Z distutilsr Zdistutils.utilr
itertoolsr r" r2 ZPLAT_SPEC_TO_RUNTIMEr; rS r[ ri r\ r r r r! s. !