���]�w � @ s� d d l Z d d l Z d d l Z d d l Z d d l m Z m Z m Z m Z m Z d d l
m Z m Z m
Z
d d l m Z d d l m Z d d l Z e j Z e j Z e j Z e j Z e j e j e j e j f Z e j d k o� e j! d& k Z" e" rd Z# d
Z$ d Z% n d Z# d
Z$ d Z% d d d d d d i Z& Gd d � d � Z' Gd d � d � Z( d d � Z) d d � Z* d d � Z+ d d � Z, d d d! � Z- e) � Z. e. d" k r�e d# e. � � Gd$ d% � d% e � Z/ d S)'� N)�DistutilsExecError�DistutilsPlatformError� CompileError�LibError� LinkError)� CCompiler�gen_preprocess_options�gen_lib_options)�log)� get_platform�win32� � 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\.NETFramework�x86z win-amd64�amd64zwin-ia64Zia64c @ sp e Z d Z d d � Z e e � Z d d � Z e e � Z d d � Z e e � Z d d � Z e e � Z d S)
�Regc C sM x: t D]2 } | j | | � } | r | | k r | | Sq Wt | � � d S)N)�HKEYS� read_values�KeyError)�cls�path�key�base�d� r �2/opt/alt/python35/lib64/python3.5/msvc9compiler.py� get_valueA s
z
Reg.get_valuec C s� y t | | � } Wn t k
r+ d SYn Xg } d } xD y t | | � } Wn t k
rc PYn X| j | � | d 7} q; W| S)Nr � )� RegOpenKeyEx�RegError�
RegEnumKey�append)r r r �handle�L�i�kr r r � read_keysI s
z
Reg.read_keysc C s� y t | | � } Wn t k
r+ d SYn Xi } d } xh y t | | � \ } } } Wn t k
rl PYn X| j � } | j | � | | j | � <| d 7} q; W| S)Nr r )r r � RegEnumValue�lower� convert_mbcs) r r r r" r r$ �name�value�typer r r r [ s
zReg.read_valuesc C sG t | d d � } | d k rC y | d � } Wn t k
rB Yn X| S)N�decode�mbcs)�getattr� UnicodeError)�sZdecr r r r) q s
zReg.convert_mbcsN) �__name__�
__module__� __qualname__r � classmethodr& r r) � staticmethodr r r r r = s r c @ s@ e Z d Z d d � Z d d � Z d d � Z d d � Z d S)
�
MacroExpanderc C s' i | _ t | | _ | j | � d S)N)�macros�VS_BASE�vsbase� load_macros)�self�versionr r r �__init__} s
zMacroExpander.__init__c C s! t j | | � | j d | <d S)Nz$(%s))r r r8 )r< Zmacror r r r r � set_macro� s zMacroExpander.set_macroc C sT | j d | j d d � | j d | j d d � | j d t d � y2 | d k rl | j d t d
� n t d
� � Wn t k
r� t d � � Yn X| d k r� | j d
| j d � | j d t d � n~ d } xu t D]m } y t | | � } Wn t k
rw� Yn Xt | d � } t
j | d | | f � } | d | j d <q� Wd S)NZ VCInstallDirz \Setup\VC�
productdirZ VSInstallDirz \Setup\VSZ FrameworkDirZ installrootg @ZFrameworkSDKDirzsdkinstallrootv2.0a Python was built with Visual Studio 2008;
extensions must be built with a compiler than can generate compatible binaries.
Visual Studio 2008 was not found on this system. If you have Cygwin installed,
you can try compiling with MingW32, by passing "-c mingw32" to setup.py.g "@ZFrameworkVersionz clr versionZ
WindowsSdkDirZcurrentinstallfolderz.Software\Microsoft\NET Framework Setup\Productr z%s\%sr= z$(FrameworkVersion))
r? r: �NET_BASEr r � WINSDK_BASEr r r r r r r8 )r< r= �pr �hr r r r r r; � s.
zMacroExpander.load_macrosc C s6 x/ | j j � D] \ } } | j | | � } q W| S)N)r8 �items�replace)r< r1 r% �vr r r �sub� s zMacroExpander.subN)r2 r3 r4 r> r? r; rH r r r r r7 { s r7 c C s� d } t j j | � } | d
k r( d S| t | � } t j | d � j d d � \ } } t | d d � � d } | d k r� | d 7} t | d d � � d } | d k r� d } | d k r� | | Sd S) NzMSC v.r � � r
�
� g $@r ������)�sysr= �find�len�split�int)�prefixr$ r1 �restZ majorVersionZ minorVersionr r r �get_build_version� s %
rV c C sF g } x9 | D]1 } t j j | � } | | k r
| j | � q
W| S)N)�osr �normpathr! )�pathsZ
reduced_pathsrC Znpr r r �normalize_and_reduce_paths� s
rZ c C sX | j t j � } g } x' | D] } | | k r | j | � q Wt j j | � } | S)N)rR rW �pathsepr! �join)�variableZoldListZnewListr$ Z newVariabler r r �removeDuplicates� s
r^ c
C s^ t | } y t j d | d � } Wn% t k
rK t j d � d } Yn X| sf t j j | � r d | } t j j
| d � } | r� t j j | � r� t j j | t j t j d � } t j j
| � } t j j | � s t j d | � d Sn t j d | � | s"t j d � d St j j | d � } t j j | � rM| St j d
� 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 �isdir�environ�getr\ �pardir�abspath�isfile)r= r: r@ ZtoolskeyZtoolsdir� vcvarsallr r r �find_vcvarsall� s2
!
rg c C s� t | � } t d � } i } | d k r6 t d � � t j d | | � t j d | | f d t j d t j �} z� | j � \ } } | j � d
k r� t | j
d � � � | j
d � } x� | j d � D]� } t j
| � } d
| k r� q� | j � } | j d
d � \ } }
| j � } | | k r� |
j t j � rQ|
d d � }
t |
� | | <q� WWd | j j � | j j � Xt | � t | � k r�t t t | j � � � � � | S)N�include�lib�libpathr z Unable to find vcvarsall.batz'Calling 'vcvarsall.bat %s' (version=%s)z
"%s" %s & set�stdout�stderrr r. �
�=r )rh ri rj r rM ) rg �setr r
r_ �
subprocess�Popen�PIPE� communicate�waitr- rR r r) �stripr( �endswithrW r[ r^ rk �closerl rQ �
ValueError�str�list�keys) r= Zarchrf Z interesting�result�popenrk rl �liner r+ r r r �query_vcvarsall� s<
r g @z(VC %0.1f is not supported by this modulec @ sy e Z d Z d Z i Z d g Z d d d g Z d g Z d g Z e e e e Z d Z
d Z d
Z d Z
d Z Z d
Z d d d d d � Z d d d � Z d d d d � Z d d d d d d d d d � Z d d d d d � Z d d d d d d d d d d d d �
Z d d � Z d d � Z d! d" � Z d# d$ � Z d% d& � Z d' d( � Z d d) d* � Z d+ d, � Z d S)-� MSVCCompilerZmsvcz.cz.ccz.cppz.cxxz.rcz.mcz.resz.objz.libz.dllz%s%sz.exer c C sP 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< �verbose�dry_run�forcer r r r> J s zMSVCCompiler.__init__Nc C s� | d k r t � } d$ } | | k r: t d | f � � d t j k r� d t j k r� | j d � r� d | _ d | _ d | _ d
| _ d | _ n
| t � k s� | d k r� t
| } n t
t � d t
| } t t | � } | d
j
t j � | _ | d t j d <| d t j d <t | j � d k rJt d | j � � | j d � | _ | j d � | _ | j d � | _ | j d
� | _ | j d � | _ y5 x. t j d
j
d � D] } | j j | � q�WWn t k
r�Yn Xt | j � | _ d j | j � t j d
<d | _ | j d k rgd d d d d g | _ d d d d d d g | _ n9 d d d d d d g | _ d d d d d d d g | _ d d d g | _ | j d k r�d d d! d" g | _ d g | _ d# | _ d S)%Nr � win-amd64�win-ia64z --plat-name must be one of %sZDISTUTILS_USE_SDKZMSSdkzcl.exezlink.exezlib.exezrc.exezmc.exe�_r ri rh 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� r� ) r r rW ra �find_exe�cc�linkerri �rc�mc�PLAT_TO_VCVARSr r� rR r[ r� rQ Z_MSVCCompiler__productr! r rZ r\ Zpreprocess_optionsr� �compile_options�compile_options_debug�ldflags_sharedr� �ldflags_shared_debugZldflags_staticr� )r< r� Zok_platsZ plat_specZvc_envrC r r r �
initializeU sd
-
zMSVCCompiler.initialize� c C s; | d k r d } g } x | D]} t j j | � \ } } t j j | � d } | t j j | � d � } | | j k r� t d | � � | r� t j j | � } | | j k r� | j t j j
| | | j � � q | | j k r| j t j j
| | | j � � q | j t j j
| | | j
� � q W| S)Nr� r z Don't know how to compile %s)rW r �splitext�
splitdrive�isabs�src_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 s | j s | j � | j | | | | | | � } | \ } }
} } } | pO g }
|
j d � | rx |
j | j � n |
j | j � x�|
D]�} y | | \ } } Wn t k
r� w� Yn X| r� t j j
| � } | | j k r� d | } n�| | j k rd | } n�| | j
k r�| } d | } y) | j | j g | | g | g � Wq� t k
r�} z t | � � WYd d } ~ Xq� Xq� n| | j k r�t j j | � } t j j | � } y� | j | j g d | d | g | g � t j j t j j | � � \ } } t j j | | d � } | j | j g d | g | g � Wq� t k
r�} z t | � � WYd d } ~ Xq� Xq� n t d | | f � � d | } y- | j | j g |
| | | g | � Wq� t k
r} z t | � � WYd d } ~ Xq� Xq� W|
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 rd �
_c_extensions�_cpp_extensionsr� �spawnr� r r r� �dirnamer� r� r� r\ r� )r< Zsourcesr� r8 Z include_dirsr_ �
extra_preargs�extra_postargsZdependsZ compile_info�objectsZpp_optsZbuildZ compile_opts�obj�srcr� Z input_optZ
output_opt�msgZh_dirZrc_dirr r� Zrc_filer r r �compile� sj
# $$#
#zMSVCCompiler.compilec C s� | j s | j � | j | | � \ } } | j | d | �} | j | | � r� | d | g } | ri y | j | j g | � Wq� t k
r� } z t | � � WYd d } ~ Xq� Xn t j
d | � d S)Nr� z/OUT:zskipping %s (up-to-date)) r� r� �_fix_object_args�library_filename�
_need_linkr� ri r r r
r_ ) r< 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 | j � | j | | � \ } } | j | | | � } | \ } } } | rl | j d t | � � t | | | | � } | d k r� t j j | | � } | j
| | � r| t j k r� | r� | j
d d � } q| j d d � } n | r| j
} n | j } g } x% | p"g D] } | j d | � q#W| | | | d | g } t j j | d � } | d k r�t j j t j j | � � \ } } t j j | | j | � � } | j d | � | j | | | � |
r�|
| d d � <| r | j | � | j t j j | � � y | j | j g | � Wn1 t k
rs} z t | � � WYd d } ~ Xn X| j | | � } | d k r| \ } } d | | f } y | j d d d
| | g � Wqt k
r} z t | � � WYd d } ~ XqXn 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_args�warnry 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_optsZsym�ld_argsZdll_nameZdll_extZ implib_filer� ZmfinfoZ
mffilename�mfidZout_argr r r �link8 sl
"zMSVCCompiler.linkc C s: t j j | t j j | � d � } | j d | � 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� x8 | D], } | j d � r | j d d � d } Pq Wd S| t j k rS d } n d } | j | � } | d k rx d S| | f S)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� y� t | � } z | j � } Wd | j � Xt j d t j � } t j | d | � } d } t j | d | � } t j d t j � } t j | | � d k r� d St | d � } z | j | � | SWd | j � XWn t k
r� Yn Xd S)NzU<assemblyIdentity.*?name=("|')Microsoft\.VC\d{2}\.CRT("|').*?(/>|</assemblyIdentity>)r� z*<dependentAssembly>\s*</dependentAssembly>zI<assemblyIdentity.*?name=(?:"|')(.+?)(?:"|').*?(?:/>|</assemblyIdentity>)�w)
�open�readrw �rer� �DOTALLrH �search�write�OSError)r< Z
manifest_fileZ
manifest_fZ manifest_buf�patternr 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 d � � d S)Nz<don't know how to set runtime library search path for MSVC++)r )r< r� r r r �runtime_library_dir_option� s z'MSVCCompiler.runtime_library_dir_optionc C s
| j | � S)N)r� )r< ri r r r �library_option� s zMSVCCompiler.library_optionc C s� | r | d | g } n | g } xW | D]K } xB | D]: } t j j | | j | � � } t j j | � r6 | Sq6 Wq) Wd Sd S)NZ_d)rW r r\ r� �exists)r< �dirsri r_ Z try_namesr� r* Zlibfiler r r �find_library_file� s
z MSVCCompiler.find_library_filec C s� xH | j D]= } t j j t j j | � | � } t j j | � r
| Sq
WxU t j d j d � D]= } t j j t j j | � | � } t j j | � rb | Sqb W| S)NZPathr� )r� rW r r\ rd re ra rR )r< ZexerC �fnr r r r� s ! !zMSVCCompiler.find_exe) 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 VR + r� l )0rW rp rO r� Zdistutils.errorsr r r r r Zdistutils.ccompilerr r r Z distutilsr
Zdistutils.utilr �winregZ OpenKeyExr ZEnumKeyr Z EnumValuer' �errorr Z
HKEY_USERS�HKEY_CURRENT_USER�HKEY_LOCAL_MACHINEZHKEY_CLASSES_ROOTr �platform�maxsizeZ NATIVE_WIN64r9 rB rA r� r r7 rV rZ r^ rg r r� r� r r r r �<module> sL ( >. #)