3
\w @ sP d dl Z d dlZd dlZd dlZd dlmZmZmZmZm Z d dl
m Z m Z m
Z
d dlmZ d dlmZ d dlZejZejZejZejZejej ej ej fZ ej dkoej!d%kZ"e"rd Z#d
Z$d Z%n d Z#d
Z$dZ%ddddZ&G dd dZ'G dd dZ(dd Z)dd Z*dd Z+d d Z,d&d d Z-e) Z.e.d!k r r? r; rH r r r r r7 { s r7 c C s d} t jj| }|d
kr dS |t| }t j|d jdd\}}t|d d d }|dkrf|d7 }t|dd d }|dkrd }|dkr|| S d S ) NzMSC v.r r
g $@r )sysr= findlensplitint)prefixr$ r1 restZ majorVersionZ minorVersionr r r get_build_version s rV c C s4 g }x*| D ]"}t jj|}||kr
|j| q
W |S )N)osr normpathr! )pathsZ
reduced_pathsrC Znpr r r normalize_and_reduce_paths s
rZ c C s@ | j tj}g }x |D ]}||kr|j| qW tjj|}|S )N)rR rW pathsepr! join)ZvariableZoldListZnewListr$ Z newVariabler r r removeDuplicates s
r] c
C s t | }ytjd| d}W n" tk
r> tjd d }Y nX | sTtjj| rd| }tj j
|d }|rtjj|rtjj |tj tj d}tjj
|}tjj|stjd| d S ntjd| |stjd d S tjj |d }tjj|r|S tjd
d S ) Nz %s\Setup\VCr@ z%Unable to find productdir in registryzVS%0.f0COMNTOOLSZVCz%s is not a valid directoryz Env var %s is not set or invalidzNo productdir foundz
vcvarsall.batz Unable to find vcvarsall.bat)r9 r r r r
debugrW r isdirenvirongetr\ pardirabspathisfile)r= r: r@ ZtoolskeyZtoolsdir vcvarsallr r r find_vcvarsall s2
rf c C s8 t | }td}i }|d kr$tdtjd|| tjd||f tjtjd}z|j \}}|j d krvt|j
d
|j
d
}xr|j d D ]d}t j
|}d |krq|j }|j d d
\} }
| j } | |kr|
jtjr|
d d }
t|
|| < qW W d |jj |jj X t|t|kr4ttt|j |S )Nincludeliblibpathr z Unable to find vcvarsall.batz'Calling 'vcvarsall.bat %s' (version=%s)z
"%s" %s & set)stdoutstderrr r.
=r )rg rh ri r rM ) rf setr r
r^
subprocessPopenPIPEZ communicatewaitr- rR r r) stripr( endswithrW r[ r] rj closerk rQ
ValueErrorstrlistkeys) r= Zarchre Z interestingresultpopenrj rk liner r+ r r r query_vcvarsall s<
r} g @z(VC %0.1f is not supported by this modulec
@ s e Zd ZdZi ZdgZdddgZdgZdgZee e e Z dZ
d Z d
Z d Z
d ZZd
Zd-ddZd.ddZd/ddZd0ddZd1ddZd2dd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Z d3d)d*Z d+d, Z dS )4 MSVCCompilerZmsvcz.cz.ccz.cppz.cxxz.rcz.mcz.resz.objz.libz.dllz%s%sz.exer c C s8 t j| ||| t| _d| _g | _d | _d | _d| _d S )Nz Software\Microsoft\VisualStudioF) r r> VERSION_MSVCCompiler__versionZ_MSVCCompiler__root_MSVCCompiler__paths plat_name_MSVCCompiler__arch initialized)r< verboseZdry_runZforcer r r r> J s zMSVCCompiler.__init__Nc C s |d krt }d$}||kr(td|f dtjkrfdtjkrf| jdrfd| _d| _d | _d
| _d | _ n|t ksx|dkrt
| }nt
t d t
| }t t |}|d
j
tj| _|d tjd< |d tjd< t| jdkrtd| j | jd| _| jd| _| jd | _| jd
| _| jd | _ y,x&tjd
j
dD ]}| jj| q|| j
kr>|}d| }y"| j| jg| |g |g W qh tk
r8 } zt|W Y d d }~X qhX qhn|| jkr tj j|}tj j|}yl| j| jgd|d|g |g tj jtj j|\}}tj j||d }| j| jgd| g |g W qh tk
r } zt|W Y d d }~X qhX qhntd||f d | }y&| j| jg|
| ||g | W qh tk
rj } zt|W Y d d }~X qhX qhW |
S )
Nz/cz/Tcz/Tpz/foz-hz-rz.rcz"Don't know how to compile %s to %sz/Fo)r r Z_setup_compiler! extendr r r rW r rc
_c_extensions_cpp_extensionsr spawnr r r r dirnamer r r r\ r )r< Zsourcesr r8 Z include_dirsr^
extra_preargsextra_postargsZdependsZ compile_infoobjectsZpp_optsZbuildZ compile_optsobjsrcr Z input_optZ
output_optmsgZh_dirZrc_dirr r Zrc_filer r r compile sj
zMSVCCompiler.compilec C s | j s| j | j||\}}| j||d}| j||r|d| g }|rJy| j| jg| W q tk
r } zt|W Y d d }~X qX n t j
d| d S )N)r z/OUT:zskipping %s (up-to-date)) r r _fix_object_argslibrary_filename
_need_linkr rh r r r
r^ ) r< r Zoutput_libnamer r^ target_langoutput_filenameZlib_argsr r r r create_static_lib s z MSVCCompiler.create_static_libc C sX | j s| j | j||\}}| j|||}|\}}}|rL| jdt| t| |||}|d k rptjj ||}| j
||rH|t j kr| r| j
dd }q| jdd }n| r| j
}n| j}g }x |pg D ]}|jd| qW || | | d| g }tjj|d } |d k rPtjjtjj|\}}tjj | | j|}|jd| | j|| | |
rp|
|d d< | r|j| | jtjj| y| j| jg| W n, tk
r } zt|W Y d d }~X nX | j||}|d k rT|\}}d||f }y| jdd d
||g W n, tk
rD } zt|W Y d d }~X nX n t j d | d S ) Nz5I don't know what to do with 'runtime_library_dirs': r z/EXPORT:z/OUT:r z/IMPLIB:z-outputresource:%s;%szmt.exez-nologoz -manifestzskipping %s (up-to-date)) r r r Z
_fix_lib_argswarnrw r rW r r\ r r
EXECUTABLEr r r! r r r r manifest_setup_ldargsr Zmkpathr r r r manifest_get_embed_infor
r^ ) r< target_descr r r Z librariesZ library_dirsZruntime_library_dirsZexport_symbolsr^ r r
build_tempr Z
fixed_argsZlib_optsZldflagsZ export_optsZsymld_argsZdll_nameZdll_extZ implib_filer ZmfinfoZ
mffilenamemfidZout_argr r r link8 sl
zMSVCCompiler.linkc C s, t jj|t jj|d }|jd| d S )Nz .manifestz/MANIFESTFILE:)rW r r\ r r! )r< r r r
temp_manifestr r r r s z"MSVCCompiler.manifest_setup_ldargsc C s` x,|D ] }|j dr|jddd }P qW d S |tjkr>d}nd}| j|}|d krXd S ||fS )Nz/MANIFESTFILE::r r
)
startswithrR r r _remove_visual_c_ref)r< r r argr r r r r r s
z$MSVCCompiler.manifest_get_embed_infoc C s yt |}z |j }W d |j X tjdtj}tj|d|}d}tj|d|}tjdtj}tj||d krrd S t |d}z|j| |S |j X W n t k
r Y nX d S )NzU|)r z*\s*zI|)w)
openreadru rer DOTALLrH searchwriteOSError)r< Z
manifest_fileZ
manifest_fZ manifest_bufpatternr r r r s.
z!MSVCCompiler._remove_visual_c_refc C s d| S )Nz /LIBPATH:r )r< dirr r r library_dir_option s z MSVCCompiler.library_dir_optionc C s t dd S )Nztjd jdD ]*}tjjtjj||}tjj|rH|S qHW |S )NZPathr )r rW r r\ rc rd r` rR )r< ZexerC fnr r r r s zMSVCCompiler.find_exe)r r r )N)r r )NNNr NNN)Nr N)
NNNNNr NNNN)r ) r2 r3 r4 Z
compiler_typeZ executablesr r r r r r r Zstatic_lib_extensionZshared_lib_extensionZstatic_lib_formatZshared_lib_formatZ
exe_extensionr> r r r r r r r r r r r r r r r r r r~ , sP
X
V
R +
r~ l )r )0rW ro rO r Zdistutils.errorsr r r r r Zdistutils.ccompilerr r r Z distutilsr
Zdistutils.utilr winregZ OpenKeyExr ZEnumKeyr Z EnumValuer' errorr Z
HKEY_USERSHKEY_CURRENT_USERHKEY_LOCAL_MACHINEZHKEY_CLASSES_ROOTr platformmaxsizeZ NATIVE_WIN64r9 rB rA r r r7 rV rZ r] rf r} r r~ r r r r sL >. #
)