U
if/w @ sN 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!dkZ"e"rdZ#d Z$d
Z%n d Z#d Z$d
Z%dddZ&G dd dZ'G dd dZ(dd Z)dd Z*dd Z+dd Z,d#d d Z-e) Z.e.d k r:ed e. G d!d" d"e Z/dS )$ N)DistutilsExecErrorDistutilsPlatformError CompileErrorLibError LinkError) CCompilergen_preprocess_optionsgen_lib_options)log) get_platformwin32l z1Software\Wow6432Node\Microsoft\VisualStudio\%0.1fz5Software\Wow6432Node\Microsoft\Microsoft SDKs\Windowsz,Software\Wow6432Node\Microsoft\.NETFrameworkz%Software\Microsoft\VisualStudio\%0.1fz)Software\Microsoft\Microsoft SDKs\Windowsz Software\Microsoft\.NETFrameworkx86Zamd64r z win-amd64c @ sL e Zd Zdd ZeeZdd ZeeZdd ZeeZdd ZeeZd S )
Regc C s: t D ](}| ||}|r||kr|| S qt|d S N)HKEYS read_valuesKeyError)clspathkeybased r r? r; rI r r r r r7 z s r7 c C s d} t j| }|dkr dS |t| }t j|d dd\}}t|d d d }|dkrf|d7 }t|dd d
}|dkrd }|dkr|| S d S ) NzMSC v. r
g $@r )sysr= findlensplitint)prefixr$ r1 restZ majorVersionZ minorVersionr r r get_build_version s rX c C s0 g }| D ]"}t j|}||kr|| q|S r )osr normpathr! )pathsZ
reduced_pathsrD Znpr r r normalize_and_reduce_paths s r\ c C s< | tj}g }|D ]}||kr|| qtj|}|S r )rT rY pathsepr! join)ZvariableZoldListZnewListr$ Z newVariabler r r removeDuplicates s r_ c C s t | }ztd| d}W n" tk
r> td d }Y nX |rPtj|sd| }tj
|d }|rtj|rtj |tj tj d}tj
|}tj|std| d S ntd| |std d S tj |d }tj|r|S td
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.bat Unable to find vcvarsall.bat)r9 r r r r
debugrY r isdirenvirongetr^ pardirabspathisfile)r= r: r@ ZtoolskeyZtoolsdir vcvarsallr r r find_vcvarsall s4
ri c C s8 t | }ddddh}i }|d kr(tdtd|| tjd||f tjtjd}z|
\}}| d krzt| d
| d
}|
d D ]d}t|}d |krq| }|
d d
\} }
| } | |kr|
tjr|
d d }
t|
|| < qW 5 |j |j X t|t|kr4ttt| |S )NincludelibZlibpathr r` z'Calling 'vcvarsall.bat %s' (version=%s)z
"%s" %s & set)stdoutstderrr r.
=r rJ )ri r r
ra
subprocessPopenPIPErl closerm Z communicatewaitr- rT r r) stripr( endswithrY r] r_ rS
ValueErrorstrlistkeys) r= Zarchrh Z interestingresultpopenrl rm liner r+ r r r query_vcvarsall s>
r~ rA 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.cxx.rcz.mcz.resz.objz.libz.dllz%s%sz.exer c C s8 t | ||| 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> I s zMSVCCompiler.__init__Nc C s |d krt }d}||kr(td|f dtjkrfdtjkrf| drfd| _d| _d| _d| _d | _ n|t ksx|d
krt
| }nt
t d t
| }t t |}|d
tj| _|d
tjd
< |d tjd< t| jdkrtd| j | d| _| d| _| d| _| d| _| d | _ z(tjd
dD ]}| j| q:W n tk
rf Y nX t| j| _d| jtjd < d | _| jdkrdddddg| _ddddddg| _n&ddddd dg| _ddddd ddg| _d dd g| _| jd krd dd d!g| _ dg| _ d"| _ d S )#Nr z --plat-name must be one of %sZDISTUTILS_USE_SDKZMSSdkzcl.exezlink.exezlib.exezrc.exezmc.exer _r rk rj r zxPython was built with %s, and extensions need to be built with the same version of the compiler, but it isn't installed.;r
z/nologoz/Oxz/MDz/W3z/DNDEBUGz/Odz/MDdz/Z7z/D_DEBUGz/GS-z/DLLz/INCREMENTAL:NO z/INCREMENTAL:noz/DEBUGT) r r rY rc find_execclinkerrk rcmcPLAT_TO_VCVARSr~ r rT r] r rS Z_MSVCCompiler__productr! r r\ r^ Zpreprocess_optionsr compile_optionscompile_options_debugldflags_sharedr ldflags_shared_debugZldflags_staticr )r< r Zok_platsZ plat_specZvc_envrD r r r
initializeT s~
zMSVCCompiler.initialize c C s |d kr d}g }|D ]}t j|\}}t j|d }|t j|d }|| jkrbtd| |rrt j|}|| jkr| t j
||| j q|| j kr| t j
||| j q| t j
||| j
q|S )Nr r z Don't know how to compile %s)rY r splitext
splitdriveisabssrc_extensionsr basename_rc_extensionsr! r^
res_extension_mc_extensions
obj_extension)r< Zsource_filenamesZ strip_dir
output_dirZ obj_namesZsrc_namer extr r r object_filenames s.
z MSVCCompiler.object_filenamesc C sp | j s| | ||||||} | \}}
}} } |p6g }
|
d |rT|
| j n |
| j |
D ]}z| | \}}W n tk
r Y qdY nX |rtj
|}|| j krd| }nT|| j krd| }n>|| j
kr<|}d| }z"| | jg| |g |g W qd tk
r6 } z t|W 5 d }~X Y qdX qdn|| jkrtj |}tj |}zl| | jgd|d|g |g tj tj |\}}tj ||d }| | jgd| g |g W qd tk
r } z t|W 5 d }~X Y qdX qdntd||f d | }z&| | jg|
| ||g | W qd tk
rh } z t|W 5 d }~X Y qdX qd|
S )
Nz/cz/Tcz/Tpz/foz-hz-rr z"Don't know how to compile %s to %sz/Fo)r r Z_setup_compiler! extendr r r rY r rf
_c_extensions_cpp_extensionsr spawnr r r r dirnamer r r r^ r )r< Zsourcesr r8 Z include_dirsra
extra_preargsextra_postargsZdependsZ compile_infoobjectsZpp_optsZbuildZ compile_optsobjsrcr Z input_optZ
output_optmsgZh_dirZrc_dirr r Zrc_filer r r compile s
zMSVCCompiler.compilec
C s | j s| | ||\}}| j||d}| ||r|d| g }|rJz| | jg| W q tk
r } z t|W 5 d }~X Y qX n t
d| d S )N)r /OUT:skipping %s (up-to-date)) r r _fix_object_argslibrary_filename
_need_linkr rk r r r
ra ) r< r Zoutput_libnamer ra target_langoutput_filenameZlib_argsr r r r create_static_lib s z MSVCCompiler.create_static_libc
C sT | j s| | ||\}}| |||}|\}}}|rL| dt| t| |||}|d k rptj ||}|
||rD|t j kr| r| j
dd }q| jdd }n| r| j
}n| j}g }|pg D ]}|d| q|| | | d| g }tj|d } |d k rLtjtj|\}}tj | | |}|d| | || | |
rl|
|d d< | r||| | tj| z| | jg| W n, tk
r } z t|W 5 d }~X Y nX | ||}|d k rP|\}}d||f }z| dd d
||g W n, tk
r@ } z t|W 5 d }~X Y nX n t d | d S ) Nz5I don't know what to do with 'runtime_library_dirs': r z/EXPORT:r r z/IMPLIB:z-outputresource:%s;%szmt.exez-nologoz -manifestr ) r r r Z
_fix_lib_argswarnrx r rY r r^ r r
EXECUTABLEr r r! r r r r manifest_setup_ldargsr Zmkpathr r r r manifest_get_embed_infor
ra ) r< target_descr r r Z librariesZ library_dirsZruntime_library_dirsZexport_symbolsra r r
build_tempr Z
fixed_argsZlib_optsZldflagsZ export_optsZsymld_argsZdll_nameZdll_extZ implib_filer ZmfinfoZ
mffilenamemfidZout_argr r r link6 s
zMSVCCompiler.linkc C s, t j|t j|d }|d| d S )Nz .manifest/MANIFESTFILE:)rY r r^ r r! )r< r r r
temp_manifestr r r r s
z"MSVCCompiler.manifest_setup_ldargsc C s^ |D ]"}| dr|ddd } q,qd S |tjkr|)r z*\s*zI|)w)
openrs readrer DOTALLrI searchwriteOSError)r< Z
manifest_fileZ
manifest_fZ manifest_bufpatternr r r r s2
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 )Nz r r r r r r r r r r r r r r r r r r + sj
W
X
_ +
r )r
)0rY rp rQ 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 rC rB r r r7 rX r\ r_ ri r~ r r r r r r sN >. #
)