o
0i'N @ s d dl Z d dlZd dlZd dlmZmZmZmZmZ d dl m
Z
m Z d dl m
Z
d dlmZ d dlmZ dd Zd d
Zd d d
ddZdd Zdd Zd ddZd ddddZG dd de
ZdS ) N)DistutilsExecErrorDistutilsPlatformError CompileErrorLibError LinkError) CCompilergen_lib_options)log) get_platform)countc C s zt jt jdt jt jB d} W n ty td Y dS w d}d }| _ t D ]F}z t | |\}}}W n
ty@ Y n:w |ro|t j
krot j
|roztt|}W n ttfya Y q)w |dkro||kro||}}q)W d ||fS W d ||fS 1 sw Y ||fS )Nz'Software\Microsoft\VisualStudio\SxS\VC7)accessz Visual C++ is not registeredNNr )winregZ OpenKeyExHKEY_LOCAL_MACHINEZKEY_READZKEY_WOW64_32KEYOSErrorr debugr Z EnumValueZREG_SZospathisdirintfloat
ValueError TypeError)key best_versionbest_dirivZvc_dirZvtversion r >/opt/alt/python310/lib64/python3.10/distutils/_msvccompiler.py _find_vc2015 sH
r" c
C s t jdp t jd} | sdS z tjt j| dddddd d
d d d
dg ddd }W n tjt t
fy; Y dS w t j|ddd}t j |rOd|fS dS )NzProgramFiles(x86)Z ProgramFilesr
zMicrosoft Visual StudioZ Installerz vswhere.exez-latestz -prereleasez -requiresz1Microsoft.VisualStudio.Component.VC.Tools.x86.x64z -propertyZinstallationPathz -products*mbcsstrict)encodingerrorsZVCZ AuxiliaryZBuild ) r environget
subprocess check_outputr joinstripCalledProcessErrorr UnicodeDecodeErrorr )rootr r r r! _find_vc20178 s0 r2 x86Zx64ZarmZarm64)r3 x86_amd64x86_arm x86_arm64c C s\ t \}}|s t \}}|std dS tj|d}tj|s*td| dS |d fS )Nz$No suitable Visual C++ version foundr
z
vcvarsall.batz%s cannot be found)r2 r" r r r r r- isfile) plat_spec_r r vcvarsallr r r! _find_vcvarsall_ s
r; c
C s t drdd t j D S t| \}}|stdztjd|| tj dj
ddd }W n tj yI } zt
|j td
|jd }~ww d d d d
| D D }|S )NZDISTUTILS_USE_SDKc S s i | ] \}}| |qS r lower).0r valuer r r!
s s z _get_vc_env..z Unable to find vcvarsall.batzcmd /u /c "{}" {} && set)stderrzutf-16lereplace)r' zError executing {}c S s$ i | ]\}}}|r|r| |qS r r< )r> r r9 r? r r r! r@ s c s s | ]}| d V qdS )=N) partition)r> liner r r! s z _get_vc_env..)r getenvr) itemsr; r r+ r, formatZSTDOUTdecoder/ r erroroutputcmd
splitlines)r8 r: r9 outexcenvr r r! _get_vc_envq s2
rR c C sP |s t dt j}|D ]}t jt j|| }t j|r%| S q| S Nr )r rG splitpathsepr r- abspathr7 )Zexepathspfnr r r! _find_exe s rZ r4 r5 r6 )Zwin32z win-amd64z win-arm32z win-arm64c s e Zd Z dZi ZdgZg dZdgZdgZee e e Z dZ
dZ dZ d Z
d
ZZd Zd%d
dZd&ddZ d'ddZ d(ddZ d)ddZ d*ddZ fdd Zd d Zd d Zd!d" Zd+d#d$Z Z S ), MSVCCompilerZmsvcz.c)z.ccz.cppz.cxx.rcz.mcz.resz.objz.libz.dllz%s%sz.exer c C s t | ||| d | _d| _d S )NF)r __init__ plat_name initialized)selfverboseZdry_runZforcer r r! r] s
zMSVCCompiler.__init__Nc C s
|d u rt }|tvrtdttt| }t|}|s"td|dd| _| jt j
}t d|| _ t d|| _
t d|| _t d|| _t d || _t d
|| _|d dt j
D ]
}|rk| |t j q^|d dt j
D ]
}|r| |t j qvd | _g d
| _g d| _g d}g d}g |d| _g |d| _g |ddd| _g |ddd| _ g || _ g || _ t j d f| jt j df| jt j df| jt j!d f| jt j!df| jt j!df| j t j"d f| j t j"df| j t j"df| j i | _#d| _$d S )Nz --plat-name must be one of {}z7Unable to find a compatible Visual Studio installation.r zcl.exezlink.exezlib.exezrc.exezmc.exezmt.exeZincludelib)/nologoz/Ox/W3z/GLz/DNDEBUGz/MD)rd z/Odz/MDdz/Zire z/D_DEBUG)rd /INCREMENTAL:NO/LTCG)rd rf rg z /DEBUG:FULLz/MANIFEST:EMBED,ID=1z/DLLz/MANIFEST:EMBED,ID=2z/MANIFESTUAC:NOFT)%r
PLAT_TO_VCVARSr rI tuplerR r* _pathsrT r rU rZ cclinkerrc rcmcZmtZadd_include_dirrstripsepZadd_library_dirZpreprocess_optionscompile_optionscompile_options_debugZ ldflags_exeZldflags_exe_debugZldflags_sharedZldflags_shared_debugZldflags_staticZldflags_static_debugr Z
EXECUTABLEZ
SHARED_OBJECTZSHARED_LIBRARY_ldflagsr_ )r` r^ r8 Zvc_envrW dirldflagsZ
ldflags_debugr r r!
initialize s`
zMSVCCompiler.initializerb c sX i fddj D fddjj D p d fdd}tt||S )Nc i | ]}| j qS r )
obj_extensionr> extr` r r! r@ " z1MSVCCompiler.object_filenames..c rw r )
res_extensionry r{ r r! r@ # r| rb c s t j| \}}rt j|}nt j|\}}|t jjt jjfr*|dd }z t j| | W S t yD t
d | w )N z Don't know how to compile {}) r r splitextbasename
splitdrive
startswithrp altsepr- LookupErrorr rI )rX baserz r9 )ext_map
output_dir strip_dirr r!
make_out_path( s z4MSVCCompiler.object_filenames..make_out_path)src_extensions_rc_extensions_mc_extensionslistmap)r` Zsource_filenamesr r r r )r r r` r r! object_filenames s z MSVCCompiler.object_filenamesc C sL | j s| | ||||||} | \}}
}} } |pg }
|
d |r*|
| j n|
| j d}|
D ]}z| | \}}W n tyG Y q4w |rPtj
|}|| j v rZd| }n|| j v rfd| }d}n|| j
v r|}d| }z| | jg| ||g W n ty } zt|d }~ww q4|| jv rtj |}tj |}z.| | jd|d||g tj tj |\}}tj ||d }| | jd| |g W n ty } zt|d }~ww q4td
||| jg|
| }|r|d || |d | || z| | W q4 ty# } zt|d }~ww |
S )
Nz/cFz/Tcz/TpTz/foz-hz-rr\ z"Don't know how to compile {} to {}z/EHscz/Fo)r_ rv Z_setup_compileappendextendrr rq KeyErrorr r rV
_c_extensions_cpp_extensionsr spawnrm r r r dirnamern r r r- rI rk )r` Zsourcesr ZmacrosZ include_dirsr
extra_preargsextra_postargsZdependsZ compile_infoobjectsZpp_optsZbuildZ compile_optsZ add_cpp_optsobjsrcrz Z input_optZ
output_optmsgZh_dirZrc_dirr r9 Zrc_fileargsr r r! compile> s
zMSVCCompiler.compilec
C s | j s| | ||\}}| j||d}| ||rN|d| g }|r& ztd| jd| | | jg| W d S t
yM } zt |d }~ww td| d S )N)r /OUT:Executing "%s" %s skipping %s (up-to-date)) r_ rv _fix_object_argslibrary_filename
_need_linkr r rc 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 s | j s| | ||\}}| |||}|\}}}|r&| dt| t| |||}|d ur8tj ||}|
||r| j || f }dd |pKg D }|| | | d| g }tj |d } |d urtj
tj|\}}tj | | |}|d| |
r|
|d d< | r|| tj tj|}| | ztd| jd | | | jg| W d S ty } zt|d }~ww td | d S )
Nz5I don't know what to do with 'runtime_library_dirs': c S s g | ]}d | qS )z/EXPORT:r )r> Zsymr r r!
r| z%MSVCCompiler.link..r r z/IMPLIB:r r r )r_ rv r Z
_fix_lib_argswarnstrr r r r- r rs r r r r r r rV Zmkpathr r rl r r r )r` Z target_descr r r Z librariesZ library_dirsZruntime_library_dirsZexport_symbolsr r r Z
build_tempr Z
fixed_argsZlib_optsru Z export_optsZld_argsZdll_nameZdll_extZ implib_filer r r r! link sb
zMSVCCompiler.linkc s< t d}z| jt jd< t |W |t jd< S |t jd< w rS )r rG rj r) superr )r` rM Zold_path __class__r r! r s
zMSVCCompiler.spawnc C s d| S )Nz /LIBPATH:r r` rt r r r! library_dir_option s z MSVCCompiler.library_dir_optionc C s t d)Nz:don't know how to set runtime library search path for MSVC)r r r r r! runtime_library_dir_option s z'MSVCCompiler.runtime_library_dir_optionc C s
| |S N)r )r` rc r r r! library_option
s
zMSVCCompiler.library_optionc C s\ |r |d |g}n|g}|D ] }|D ]}t j|| |}t j|r*| S qqd S )NZ_d)r r r- r r7 )r` dirsrc r Z try_namesrt nameZlibfiler r r! find_library_file
s z MSVCCompiler.find_library_file)r r r r )r rb )NNNr NNN)Nr N)
NNNNNr NNNN)r ) __name__
__module__ __qualname__Z
compiler_typeZ executablesr r r r r r} rx Zstatic_lib_extensionZshared_lib_extensionZstatic_lib_formatZshared_lib_formatZ
exe_extensionr] rv r r r r r r r r r
__classcell__r r r r! r[ s^
P
"
]
E r[ r )r r+ r Zdistutils.errorsr r r r r Zdistutils.ccompilerr r Z distutilsr Zdistutils.utilr
itertoolsr r" r2 ZPLAT_SPEC_TO_RUNTIMEr; rR rZ rh r[ r r r r! s2 !