a XC?h'N�@s�ddlZddlZddlZddlmZmZmZmZmZddl m Z mZddlm Z ddlmZddlmZdd�Zd d �Zddd dd�Zdd�Zdd�Zddd�Zddddd�ZGdd�de �ZdS)�N)�DistutilsExecError�DistutilsPlatformError�CompileError�LibError� LinkError)� CCompiler�gen_lib_options)�log)�get_platform)�countcCsztjtjdtjtjBd�}Wnty<t�d�YdS0d}d}|��t�D]�}zt� ||�\}}}Wnty�Yq�Yn0|rR|tj krRtj� |�rRztt|��}Wnttfy�YqRYn0|dkrR||krR||}}qRWd�n1�s�0Y||fS)Nz'Software\Microsoft\VisualStudio\SxS\VC7)�accesszVisual C++ is not registered�NNr�)�winregZ OpenKeyEx�HKEY_LOCAL_MACHINEZKEY_READZKEY_WOW64_32KEY�OSErrorr �debugrZ EnumValueZREG_SZ�os�path�isdir�int�float� ValueError� TypeError)�key�best_version�best_dir�i�vZvc_dirZvt�version�r �ss�z_get_vc_env..zUnable to find vcvarsall.batzcmd /u /c "{}" {} && set)�stderrzutf-16le�replace)r'zError executing {}cSs$i|]\}}}|r|r|��|�qSr r<)r>rr9r?r r r!r@�s�css|]}|�d�VqdS)�=N)� partition)r>�liner r r!� ��z_get_vc_env..)r�getenvr)�itemsr;rr+r,�formatZSTDOUT�decoder/r �error�output�cmd� splitlines)r8r:r9�out�exc�envr r r!�_get_vc_envqs. � �� rScCsN|st�d��tj�}|D].}tj�tj�|�|�}tj�|�r|Sq|S�Nr)rrH�split�pathseprr-�abspathr7)Zexe�paths�p�fnr r r!� _find_exe�s r[r4r5r6)Zwin32z win-amd64z win-arm32z win-arm64c s�eZdZdZiZdgZgd�ZdgZdgZeeeeZ dZ dZdZd Z d ZZdZd%d d�Zd&dd�Zd'dd�Zd(dd�Zd)dd�Zd*dd�Z�fdd�Zdd�Zdd �Zd!d"�Zd+d#d$�Z�ZS),�MSVCCompilerZmsvcz.c)z.ccz.cppz.cxx�.rcz.mcz.resz.objz.libz.dllz%s%sz.exercCs t�||||�d|_d|_dS)NF)r�__init__� plat_name�initialized)�self�verboseZdry_runZforcer r r!r^�szMSVCCompiler.__init__NcCs |durt�}|tvr(td�tt��t|}t|�}|sDtd��|�dd�|_|j�t j �}td|�|_td|�|_ td|�|_td|�|_td |�|_td |�|_|�dd��t j �D]}|r�|�|�t j��q�|�dd��t j �D]}|r�|�|�t j��q�d|_gd �|_gd�|_gd�}gd�}g|�d�|_g|�d�|_g|�d�d�d�|_g|�d�d�d�|_g|�|_g|�|_tj df|jtj df|jtj df|jtj!df|jtj!df|jtj!df|jtj"df|jtj"df|jtj"df|ji |_#d|_$dS)Nz--plat-name must be one of {}z7Unable to find a compatible Visual Studio installation.r�zcl.exezlink.exezlib.exezrc.exezmc.exezmt.exeZinclude�lib)�/nologoz/Ox�/W3z/GLz/DNDEBUGz/MD)rez/Odz/MDdz/Zirfz/D_DEBUG)re�/INCREMENTAL:NO�/LTCG)rergrhz/DEBUG:FULLz/MANIFEST:EMBED,ID=1z/DLLz/MANIFEST:EMBED,ID=2z/MANIFESTUAC:NOFT)%r �PLAT_TO_VCVARSrrJ�tuplerSr*�_pathsrUrrVr[�cc�linkerrd�rc�mcZmtZadd_include_dir�rstrip�sepZadd_library_dirZpreprocess_options�compile_options�compile_options_debugZldflags_exeZldflags_exe_debugZldflags_sharedZldflags_shared_debugZldflags_staticZldflags_static_debugrZ EXECUTABLEZ SHARED_OBJECTZSHARED_LIBRARY�_ldflagsr`)rar_r8Zvc_envrX�dir�ldflagsZ ldflags_debugr r r!� initialize�s\� �zMSVCCompiler.initializerccsXi�fdd��jD��fdd��j�jD��p8d��fdd�}tt||��S)Ncsi|]}|�j�qSr )� obj_extension�r>�ext�rar r!r@"rGz1MSVCCompiler.object_filenames..csi|]}|�j�qSr )� res_extensionryr{r r!r@#rGrccs�tj�|�\}}�r"tj�|�}n2tj�|�\}}|�tjjtjjf�rT|dd�}ztj��|�|�WSt y�t d�|��Yn0dS)N�zDon't know how to compile {})rr�splitext�basename� splitdrive� startswithrq�altsepr-�LookupErrorrrJ)rY�baserzr9)�ext_map� output_dir� strip_dirr r!� make_out_path(sz4MSVCCompiler.object_filenames..make_out_path)�src_extensions�_rc_extensions�_mc_extensions�list�map)raZsource_filenamesr�r�r�r )r�r�rar�r!�object_filenamess��zMSVCCompiler.object_filenamesc Cs�|js|��|�||||||�} | \}} }}}|p6g} | �d�|rT| �|j�n| �|j�d}| D�]}z||\}}Wnty�YqhYn0|r�tj � |�}||jvr�d|}�nH||jvr�d|}d}�n.||j v�r@|}d|}z|�|jg|||g�Wqht�y:}zt|��WYd}~qhd}~00qhn�||jv�r�tj �|�}tj �|�}z\|�|jd|d||g�tj �tj �|��\}}tj �||d �}|�|jd||g�Wqht�y�}zt|��WYd}~qhd}~00qhntd �||��|jg| |}|�r$|�d�|�|�|�d|�|�|�z|�|�Wqht�y�}zt|��WYd}~qhd}~00qh| S) Nz/cFz/Tcz/TpTz/foz-hz-rr]z"Don't know how to compile {} to {}z/EHscz/Fo)r`rwZ_setup_compile�append�extendrsrr�KeyErrorrrrW� _c_extensions�_cpp_extensionsr��spawnrnrrr��dirnameror~rr-rJrl)raZsourcesr�ZmacrosZinclude_dirsr� extra_preargs�extra_postargsZdependsZcompile_info�objectsZpp_optsZbuildZcompile_optsZadd_cpp_opts�obj�srcrzZ input_optZ output_opt�msgZh_dirZrc_dirr�r9Zrc_file�argsr r r!�compile>sr � � zMSVCCompiler.compilec Cs�|js|��|�||�\}}|j||d�}|�||�r�|d|g}|rJz,t�d|jd�|��|� |jg|�Wq�t y�}zt|��WYd}~q�d}~00nt�d|�dS)N)r��/OUT:�Executing "%s" %s� �skipping %s (up-to-date))r`rw�_fix_object_args�library_filename� _need_linkr rrdr-r�rr) rar�Zoutput_libnamer�r�target_lang�output_filenameZlib_argsr�r r r!�create_static_lib�s� zMSVCCompiler.create_static_libc Cs�|js|��|�||�\}}|�|||�}|\}}}|rL|�dt|��t||||�}|durptj� ||�}|� ||��r�|j|| f}dd�|p�gD�}||||d|g}tj�|d�}|du�rtj� tj�|��\}}tj� ||�|��}|�d|�| �r| |dd�<|�r.|�|�tj�tj�|��}|�|�z,t�d|jd� |��|�|jg|�Wn.t�y�}zt|��WYd}~n d}~00nt�d |�dS) Nz5I don't know what to do with 'runtime_library_dirs': cSsg|]}d|�qS)z/EXPORT:r )r>Zsymr r r!� �rGz%MSVCCompiler.link..r�rz/IMPLIB:r�r�r�)r`rwr�Z _fix_lib_args�warn�strrrrr-r�rtr�r~rr�r�r�rWZmkpathr rrmr�rr)raZtarget_descr�r�r�Z librariesZlibrary_dirsZruntime_library_dirsZexport_symbolsrr�r�Z build_tempr�Z fixed_argsZlib_optsrvZexport_optsZld_argsZdll_nameZdll_extZimplib_filer�r r r!�link�s^� �� zMSVCCompiler.linkc s@t�d�}z$|jtjd<t��|�W|tjd<S|tjd<0dSrT)rrHrkr)�superr�)rarNZold_path�� __class__r r!r��s �zMSVCCompiler.spawncCsd|S)Nz /LIBPATH:r �rarur r r!�library_dir_optionszMSVCCompiler.library_dir_optioncCstd��dS)Nz:don't know how to set runtime library search path for MSVC)rr�r r r!�runtime_library_dir_options�z'MSVCCompiler.runtime_library_dir_optioncCs |�|�S)N)r�)rardr r r!�library_option szMSVCCompiler.library_optioncCs\|r|d|g}n|g}|D]:}|D]0}tj�||�|��}tj�|�r$|Sq$qdS)NZ_d)rrr-r�r7)ra�dirsrdrZ try_namesru�nameZlibfiler r r!�find_library_file szMSVCCompiler.find_library_file)rrr)N)rrc)NNNrNNN)NrN) NNNNNrNNNN)r)�__name__� __module__�__qualname__Z compiler_typeZexecutablesr�r�r�r�r�r|rxZstatic_lib_extensionZshared_lib_extensionZstatic_lib_formatZshared_lib_formatZ exe_extensionr^rwr�r�r�r�r�r�r�r�r�� __classcell__r r r�r!r\�sZ�� P� "� ]� � Er\)N)rr+rZdistutils.errorsrrrrrZdistutils.ccompilerrrZ distutilsr Zdistutils.utilr � itertoolsrr"r2ZPLAT_SPEC_TO_RUNTIMEr;rSr[rir\r r r r!�s.!� �