3 ��2a�w�@sPddlZddlZddlZddlZddlmZmZmZmZm Z ddl mZmZm Z ddlmZddlmZddlZejZejZejZejZejejejejfZej dko�ej!d%kZ"e"r�d Z#d Z$dZ%ndZ#d Z$dZ%dddd�Z&Gdd�d�Z'Gdd�d�Z(dd�Z)dd�Z*dd�Z+dd�Z,d&dd �Z-e)�Z.e.d!k�rr?r;rHrrrrr7{sr7cCs�d}tjj|�}|d krdS|t|�}tj|d�jdd�\}}t|dd��d}|dkrf|d7}t|dd��d}|dkr�d }|dkr�||SdS)NzMSC v.r�� r � �g$@r��)�sysr=�find�len�split�int)�prefixr$r1�restZmajorVersionZminorVersionrrr�get_build_version�srVcCs4g}x*|D]"}tjj|�}||kr |j|�q W|S)N)�osr�normpathr!)�pathsZ reduced_pathsrCZnprrr�normalize_and_reduce_paths�s rZcCs@|jtj�}g}x|D]}||kr|j|�qWtjj|�}|S)N)rRrW�pathsepr!�join)�variableZoldListZnewListr$ZnewVariablerrr�removeDuplicates�s r^c Cst|}ytjd|d�}Wn"tk r>tjd�d}YnX|sTtjj|�r�d|}tj j |d�}|r�tjj|�r�tjj|tjtjd�}tjj |�}tjj|�s�tjd|�dSntjd|�|s�tjd�dStjj|d �}tjj|�r�|Stjd �dS)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.batzUnable to find vcvarsall.bat)r9rrrr �debugrWr�isdir�environ�getr\�pardir�abspath�isfile)r=r:r@ZtoolskeyZtoolsdir� vcvarsallrrr�find_vcvarsall�s2 rgcCs8t|�}td�}i}|dkr$td��tjd||�tjd||ftjtjd�}z�|j�\}}|j �d krvt|j d ��|j d �}xr|jd�D]d}tj |�}d|kr�q�|j�}|jdd �\} } | j�} | |kr�| jtj�r�| dd�} t| �|| <q�WWd|jj�|jj�Xt|�t|�k�r4ttt|j��|S)N�include�lib�libpathrzUnable to find vcvarsall.batz'Calling 'vcvarsall.bat %s' (version=%s)z "%s" %s & set)�stdout�stderrrr.� �=r)rhrirjrrM)rg�setrr r_� subprocess�Popen�PIPEZcommunicate�waitr-rRrr)�stripr(�endswithrWr[r^rk�closerlrQ� ValueError�str�list�keys)r=ZarchrfZinteresting�result�popenrkrl�linerr+rrr�query_vcvarsall�s< r~g @z(VC %0.1f is not supported by this modulec @s�eZdZdZiZdgZdddgZdgZdgZeeeeZ dZ d Zd ZdZ dZZd Zd-dd�Zd.dd�Zd/dd�Zd0dd�Zd1dd�Zd2dd�Zdd�Zdd �Zd!d"�Zd#d$�Zd%d&�Zd'd(�Zd3d)d*�Zd+d,�ZdS)4�MSVCCompilerZmsvcz.cz.ccz.cppz.cxxz.rcz.mcz.resz.objz.libz.dllz%s%sz.exercCs8tj||||�t|_d|_g|_d|_d|_d|_dS)NzSoftware\Microsoft\VisualStudioF) rr>�VERSION�_MSVCCompiler__versionZ_MSVCCompiler__root�_MSVCCompiler__paths� plat_name�_MSVCCompiler__arch�initialized)r<�verboseZdry_run�forcerrrr>JszMSVCCompiler.__init__NcCs|dkrt�}d$}||kr(td|f��dtjkrfdtjkrf|jd�rfd|_d|_d |_d |_d|_ n�|t�ksx|dkr�t |}nt t�dt |}tt|�}|d j tj�|_|dtjd<|dtjd<t|j�dkr�td|j��|jd�|_|jd�|_|jd �|_|jd �|_|jd�|_ y,x&tjd j d�D]}|jj|��q||j k�r>|}d|}y"|j|jg||g|g�Wqhtk �r8}zt|��WYdd}~XqhXqhn�||jk�rtj j|�}tj j|�}yl|j|jgd|d|g|g�tj jtj j|��\}}tj j||d�}|j|jgd|g|g�Wqhtk �r�}zt|��WYdd}~XqhXqhntd||f��d |}y&|j|jg| |||g|�Wqhtk �rj}zt|��WYdd}~XqhXqhW| 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�rrWrrd� _c_extensions�_cpp_extensionsr��spawnr�rrr��dirnamer�r�r�r\r�)r<Zsourcesr�r8Zinclude_dirsr_� extra_preargs�extra_postargsZdependsZcompile_info�objectsZpp_optsZbuildZcompile_opts�obj�srcr�Z input_optZ output_opt�msgZh_dirZrc_dirrr�Zrc_filerrr�compile�sj zMSVCCompiler.compilec Cs�|js|j�|j||�\}}|j||d�}|j||�r�|d|g}|rJy|j|jg|�Wq�tk r�}zt|��WYdd}~Xq�Xnt j d|�dS)N)r�z/OUT:zskipping %s (up-to-date))r�r��_fix_object_args�library_filename� _need_linkr�rirrr r_) r<r�Zoutput_libnamer�r_�target_lang�output_filenameZlib_argsr�rrr�create_static_libszMSVCCompiler.create_static_libcCsX|js|j�|j||�\}}|j|||�}|\}}}|rL|jdt|��t||||�}|dk rptjj ||�}|j ||��rH|tjkr�| r�|j dd�}q�|jdd�}n| r�|j }n|j}g}x|p�gD]}|jd|�q�W||||d|g}tjj|d�}|dk �rPtjjtjj|��\}}tjj ||j|��}|jd|�|j|||�| �rp| |dd�<|�r�|j|�|jtjj|��y|j|jg|�Wn,tk �r�}zt|��WYdd}~XnX|j||�}|dk �rT|\}}d||f}y|jdd d ||g�Wn,tk �rD}zt|��WYdd}~XnXntjd|�dS)Nz5I don't know what to do with 'runtime_library_dirs': rz/EXPORT:z/OUT:rz/IMPLIB:z-outputresource:%s;%szmt.exez-nologoz -manifestzskipping %s (up-to-date))r�r�r�Z _fix_lib_args�warnrxr rWrr\r�r� EXECUTABLEr�r�r!r�r�r�r��manifest_setup_ldargsr�Zmkpathr�r�rr�manifest_get_embed_infor r_)r<�target_descr�r�r�Z librariesZlibrary_dirsZruntime_library_dirsZexport_symbolsr_r�r�� build_tempr�Z fixed_argsZlib_optsZldflagsZexport_optsZsym�ld_argsZdll_nameZdll_extZimplib_filer�ZmfinfoZ mffilename�mfidZout_argrrr�link8sl zMSVCCompiler.linkcCs,tjj|tjj|�d�}|jd|�dS)Nz .manifestz/MANIFESTFILE:)rWrr\r�r!)r<r�r�r�� temp_manifestrrrr��sz"MSVCCompiler.manifest_setup_ldargscCs`x,|D] }|jd�r|jdd�d}PqWdS|tjkr>d}nd}|j|�}|dkrXdS||fS)Nz/MANIFESTFILE:�:rr )� startswithrRrr��_remove_visual_c_ref)r<r�r��argr�r�rrrr��s z$MSVCCompiler.manifest_get_embed_infocCs�y�t|�}z|j�}Wd|j�Xtjdtj�}tj|d|�}d}tj|d|�}tjdtj�}tj||�dkrrdSt|d�}z|j|�|S|j�XWnt k r�YnXdS)NzU|)r�z*\s*zI|)�w) �open�readrv�rer��DOTALLrH�search�write�OSError)r<Z manifest_fileZ manifest_fZmanifest_buf�patternrrrr��s. z!MSVCCompiler._remove_visual_c_refcCsd|S)Nz /LIBPATH:r)r<�dirrrr�library_dir_option�szMSVCCompiler.library_dir_optioncCstd��dS)Nztjdjd�D]*}tjjtjj|�|�}tjj|�rH|SqHW|S)NZPathr�)r�rWrr\rdrerarR)r<ZexerC�fnrrrr�s zMSVCCompiler.find_exe)rrr)N)rr�)NNNrNNN)NrN) NNNNNrNNNN)r)r2r3r4Z compiler_typeZexecutablesr�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�rrrrr,sP X V R+ rl)r)0rWrprOr�Zdistutils.errorsrrrrrZdistutils.ccompilerrrr Z distutilsr Zdistutils.utilr�winregZ OpenKeyExrZEnumKeyr Z EnumValuer'�errorrZ HKEY_USERS�HKEY_CURRENT_USER�HKEY_LOCAL_MACHINEZHKEY_CLASSES_ROOTr�platform�maxsizeZNATIVE_WIN64r9rBrAr�rr7rVrZr^rgr~r�rrrrr�sL>.# )