B
    òVbJc  ã               @   sê  d Z ddlZddlZddlZddlZddlZddlZddlZddl	m
Z
 ddlZddlmZ ddlmZ ddlmZ ddlmZ ddlmZmZmZmZ d	d
„ Zeej_e d¡Ze d¡Z G dd„ dejj!ƒZ"dd„ Z#dd„ Z$dd„ Z%dd„ Z&d>dd„Z'dd„ Z(dd„ Z)dd„ Z*d d!„ Z+i Z,ejd"kr¬y`ddl-Z-d#e,d$< d%e,d&< d'e,d(< d)e,d*< e.e-d+ƒrˆe-j/ 0d,d-¡\Z1Z2Z3e-j/e,e1e2 < [1[2[3W n  e4k
rª   e
 5d.¡ Y nX d/d0„ Z6d?d2d3„Z7d4d5„ Z8d6d7„ Z9d8d9„ Z:d:d;„ Z;d<d=„ Z<dS )@a  
Support code for building Python extensions on Windows.

    # NT stuff
    # 1. Make sure libpython<version>.a exists for gcc.  If not, build it.
    # 2. Force windows to use gcc (we're struggling with MSVC and g77 support)
    # 3. Force windows to use g77

é    N)Úlog)ÚStrictVersion)ÚUnixCCompiler)Úget_build_version)ÚUnknownFileError)Úmsvc_runtime_libraryÚmsvc_runtime_versionÚmsvc_runtime_majorÚget_build_architecturec              C   s   t ƒ } | dkrg S | gS )zBReplacement for outdated version of get_msvcr from cygwinccompilerN)r   )Zmsvcr© r   ú?lib/python3.7/site-packages/numpy/distutils/mingw32ccompiler.pyÚget_msvcr_replacement$   s    r   z\[Ordinal/Name Pointer\] Tablez$^\s+\[([\s*[0-9]*)\] ([a-zA-Z0-9_]*)c               @   s2   e Zd ZdZdZddd„Zddd„Zdd
d„ZdS )ÚMingw32CCompilerzH A modified MingW32 compiler compatible with an MSVC built Python.

    Zmingw32r   c       
   	   C   sŽ  t jj | |||¡ | jd krnyt ddg¡}W n ttfk
rL   d}Y nX t	 
d|¡}|rnt| d¡ƒ| _| jdkr~d}nd}| jdkr”d| _n| jdkr¤d	| _tƒ  tƒ }td
d}|sÂ|rÌ|  d¡ tƒ }	|	ræ|  dd|	 ¡ tƒ dkr&| jdk r| jddddd n| jddddd n\| jdkrP| jdddd| j|f d n2| jdk rp| jdd dd!d n| jd"d#d$d%d d	g| _d S )&NZgccz-dumpversionÚ z
(\d+\.\d+)é   z2.91.57z--entry _DllMain@12Zdllwrapzg++T)ÚdebugZNPY_MINGW_USE_CUSTOM_MSVCRZ__MSVCRT_VERSION__z0x%04iÚAMD64z4.0z/gcc -g -DDEBUG -DMS_WIN64 -mno-cygwin -O0 -WallzCgcc -g -DDEBUG -DMS_WIN64 -mno-cygwin -O0 -Wall -Wstrict-prototypeszgcc -g -mno-cygwinzgcc -g -mno-cygwin -shared)ÚcompilerZcompiler_soZ
linker_exeZ	linker_soz#gcc -g -DDEBUG -DMS_WIN64 -O0 -Wallz7gcc -g -DDEBUG -DMS_WIN64 -O0 -Wall -Wstrict-prototypeszgcc -gzgcc -g -sharedz3.0.0zgcc -mno-cygwin -O2 -wz0gcc -mno-cygwin -mdll -O2 -w -Wstrict-prototypeszg++ -mno-cygwinz%s -mno-cygwin -mdll -static %szgcc -mno-cygwin -O2 -Wallz-gcc -mno-cygwin -O2 -Wall -Wstrict-prototypeszg++ -mno-cygwin -sharedzgcc -O2 -Wallz!gcc -O2 -Wall -Wstrict-prototypeszg++ zg++ -shared)Ú	distutilsÚcygwinccompilerÚCygwinCCompilerÚ__init__Úgcc_versionÚ
subprocessÚcheck_outputÚOSErrorZCalledProcessErrorÚreÚsearchr   ÚgroupZ
linker_dllZlinkerÚbuild_import_libraryÚbuild_msvcr_libraryZdefine_macror   r
   Zset_executablesZcompiler_cxx)
ÚselfÚverboseZdry_runZforceZ
out_stringÚresultZentry_pointZmsvcr_successZmsvcr_dbg_successZmsvcr_versionr   r   r   r   9   sp    









zMingw32CCompiler.__init__Nc             C   sp   t ƒ }|r|sg }| |¡ | |||||||d |	|
|||f}| jdk rRtjjj}ntj}||d |jj	… Ž  d S )Nz3.0.0)
r   Úappendr   r   r   r   Úlinkr   Ú__code__Úco_argcount)r!   Ztarget_descZobjectsZoutput_filenameÚ
output_dirZ	librariesZlibrary_dirsZruntime_library_dirsZexport_symbolsr   Zextra_preargsZextra_postargsZ
build_tempZtarget_langZruntime_libraryÚargsÚfuncr   r   r   r%   ¦   s0    

zMingw32CCompiler.linkr   c       	      C   sÖ   |d krd}g }xÀ|D ]¸}t j t j |¡¡\}}t j |¡\}}|rR|dd … }|| jddg krttd||f ƒ‚|r„t j |¡}|dks”|dkr´| t j 	||| | j
 ¡¡ q| t j 	||| j
 ¡¡ qW |S )Nr   r   z.rcz.resz"unknown file type '%s' (from '%s'))ÚosÚpathÚsplitextÚnormcaseÚ
splitdriveZsrc_extensionsr   Úbasenamer$   ÚjoinZobj_extension)	r!   Zsource_filenamesZ	strip_dirr(   Z	obj_namesZsrc_nameÚbaseZextZdrvr   r   r   Úobject_filenamesÐ   s(     
z!Mingw32CCompiler.object_filenames)r   r   r   )Nr   NNNN)r   r   )Ú__name__Ú
__module__Ú__qualname__Ú__doc__Zcompiler_typer   r%   r3   r   r   r   r   r   2   s     
r     
 r   c        
      C   sR  t jg} tt dƒr,t jt jkr,|  t j¡ n"tt dƒrNt jt jkrN|  t j¡ dddg}g }x.| D ]&}x |D ]}| tj ||¡¡ qlW qbW dtj	kr°| tj tj	d d¡¡ t
t jd d… ƒ\}}t ¡ }|d	krèd
|› |› d}n|dkrþd|› d}nd}td| ƒ x.|D ]&}tj ||¡}	tj |	¡r|	S qW td||f ƒ‚d S )NÚbase_prefixÚreal_prefixr   ÚlibÚbinZ
SYSTEMROOTZSystem32é   ZCPythonZpythonz.dllZPyPyZlibpypyz-c.dllz!Unknown platform {implementation}zLooking for %sz%s not found in %s)ÚsysÚprefixÚhasattrr8   r$   r9   r+   r,   r1   ÚenvironÚtupleÚversion_infoÚplatformZpython_implementationÚprintÚexistsÚ
ValueError)
ÚstemsÚsub_dirsZlib_dirsZstemÚfolderÚmajor_versionÚminor_versionÚimplementationZdllnameÚdllr   r   r   Úfind_python_dlló   s2    





rN   c             C   s   t  dd| g¡}| d¡S )Nzobjdump.exez-pó   
)r   r   Úsplit)rM   Ústr   r   r   Ú
dump_table  s    rR   c       	   	   C   s  t | ƒ}x0tt|ƒƒD ]}t ||  ¡ ¡rP qW tdƒ‚g }xVt|d t|ƒƒD ]@}t ||  ¡ ¡}|r| t	| 
d¡ ¡ ƒ| 
d¡f¡ qRP qRW t|ƒdkr°t d|  ¡ t|dƒZ}| dtj | ¡ ¡ | d¡ | d	¡ | d
¡ x|D ]}| d|d  ¡ qöW W dQ R X dS )zŠGiven a dll file location,  get all its exported symbols and dump them
    into the given def file.

    The .def file will be overwrittenzSymbol table not foundr   r<   r   zNo symbols found in %sÚwzLIBRARY        %s
z,;CODE          PRELOAD MOVEABLE DISCARDABLE
z;DATA          PRELOAD SINGLE
z	
EXPORTS
z%s
N)rR   ÚrangeÚlenÚ_STARTÚmatchÚdecoderF   Ú_TABLEr$   Úintr   Ústripr   ÚwarnÚopenÚwriter+   r,   r0   )	rM   ÚdfileÚdumpÚiZsymsÚjÚmÚdÚsr   r   r   Úgenerate_def   s(    $



rf   c                s4   dddœt ƒ  ‰ ‡ fdd„}dd„ }|| ƒp2|| ƒS )NZamd64Zx86)r   ÚIntelc                sd   t j t j dd¡d¡}t j |¡s(d S x6t  |¡D ](\}}}| |kr4ˆ |kr4t j || ¡S q4W d S )NZWINDIRz
C:\WINDOWSZwinsxs)r+   r,   r1   r@   ÚgetrE   Úwalk)Údll_nameZwinsxs_pathÚrootÚdirsÚfiles)Úarchr   r   Ú_find_dll_in_winsxsE  s    z%find_dll.<locals>._find_dll_in_winsxsc             S   sL   xFt jgtjd  d¡ D ]*}tj || ¡}tj |¡rtj |¡S qW d S )NÚPATHú;)	r=   r>   r+   r@   rP   r,   r1   rE   Úabspath)rj   r,   Úfilepathr   r   r   Ú_find_dll_in_pathP  s    z#find_dll.<locals>._find_dll_in_path)r
   )rj   ro   rt   r   )rn   r   Úfind_dll@  s
    ru   Fc             C   s  t jdkrdS tƒ }|d kr*t d¡ dS |dk r@t d¡ dS tƒ }| rR|d7 }d| }t j tj	d|¡}t j 
|¡rŒt d	|f ¡ d
S |d }t|ƒ}|s²t d| ¡ dS d| }t j tj	d|¡}t d||f ¡ t||ƒ dd|d|g}	t |	¡}
t  |¡ |
 S )NÚntFz?Skip building import library: Runtime is not compiled with MSVCéP   z=Skip building msvcr library: custom functionality not presentrd   zlib%s.aÚlibsz(Skip building msvcr library: "%s" existsTz.dllz*Cannot build msvcr library: "%s" not foundz	lib%s.defz&Building msvcr library: "%s" (from %s)Údlltoolz-dz-l)r+   Únamer	   r   r   r   r,   r1   r=   r>   Úisfileru   r\   Úinforf   r   ZcallÚremove)r   Z	msvcr_verZ
msvcr_nameZout_nameÚout_fileZmsvcr_dll_nameÚdll_fileÚdef_nameÚdef_fileÚcmdZretcoder   r   r   r    Z  sB    






r    c              C   s@   t jdkrd S tƒ } | dkr"tƒ S | dkr0tƒ S td|  ƒ‚d S )Nrv   r   rg   zUnhandled arch %s)r+   rz   r
   Ú_build_import_library_amd64Ú_build_import_library_x86rF   )rn   r   r   r   r   •  s    
r   c           
   C   sô   t tjdd… ƒ\} }dddg}tjg}ttdƒrLtjtjkrL| tj¡ n"ttdƒrntjtjkrn| tj¡ dd	g}g }xJ|D ]B}|| |f }x0|D ](}x"|D ]}	| tj	 
||	|¡¡ q W q–W q€W x |D ]}
tj	 |
¡rÌd
|
fS qÌW d|d fS )zACheck if an import library for the Python runtime already exists.Nr<   zlibpython%d%d.azlibpython%d%d.dll.azlibpython%d.%d.dll.ar8   r9   rx   r:   TFr   )rA   r=   rB   r>   r?   r8   r$   r9   r+   r,   r1   r{   )rJ   rK   ÚpatternsrG   rH   Z
candidatesÚpatÚfilenameZstem_dirrI   Úfullnamer   r   r   Ú_check_for_import_lib¡  s(    


"
r‰   c              C   s„   t ƒ \} }| rt d|¡ d S tƒ }t d||f ¡ dttjd d… ƒ }tj	 
tjd|¡}t||ƒ dd|d|g}t |¡ d S )	Nz)Skip building import library: "%s" existsz4Building import library (arch=AMD64): "%s" (from %s)zpython%d%d.defr<   rx   ry   z-dz-l)r‰   r   r   rN   r|   rA   r=   rB   r+   r,   r1   r>   rf   r   Z
check_call)Ú
out_existsr~   r   r€   r   r‚   r   r   r   rƒ   Å  s    

rƒ   c           	   C   sv  t ƒ \} }| rt d|¡ dS dttjdd… ƒ }tj tj	d|¡}tj 
|¡s´ttdƒrptj tjd|¡}n"ttdƒrŽtj tjd|¡}nd}tj 
|¡r¤|}nt d	|¡ dS t d
|¡ ddlm} dttjdd… ƒ }tj tj	d|¡}|j|j|g dd}| |¡\}	}
t|dƒ}| |	|
|j|¡ W dQ R X tƒ }dd|d|d|g}t |¡}|rrt d¡ dS )z; Build the import libraries for Mingw32-gcc on Windows
    z)Skip building import library: "%s" existsNzpython%d%d.libr<   rx   r8   r9   r   z+Cannot build import library: "%s" not foundz(Building import library (ARCH=x86): "%s"r   )Úlib2defzpython%d%d.defF)ÚshellrS   ry   z	--dllnamez--defz--output-libz:Failed to build import library for gcc. Linking will fail.)r‰   r   r   rA   r=   rB   r+   r,   r1   r>   r{   r?   r8   r9   r\   r|   Únumpy.distutilsr‹   ZgetnmZ
DEFAULT_NMZparse_nmr]   Z
output_defZ
DEF_HEADERrN   r   r   )rŠ   r~   Zlib_nameZlib_fileZbase_libr‹   r€   r   Z	nm_outputZdlistZflistZfidrj   r‚   Zstatusr   r   r   r„   Ù  sD    




r„   Zwin32z8.0.50727.42Z80z9.0.21022.8Z90z10.0.30319.460Z100z14.15.26726.0Z140ÚCRT_ASSEMBLY_VERSIONÚ.r<   z9Cannot import msvcrt: using manifest will not be possiblec             C   sZ   yt t| d | ƒ }W n& tk
r>   td| |f ƒd‚Y nX t d¡}||| |dœ S )zUGiven a major and minor version of the MSVCR, returns the
    corresponding XML file.é
   z)Version %d,%d of MSVCRT not supported yetNaÃ          <assembly xmlns="urn:schemas-microsoft-com:asm.v1" manifestVersion="1.0">
          <trustInfo xmlns="urn:schemas-microsoft-com:asm.v3">
            <security>
              <requestedPrivileges>
                <requestedExecutionLevel level="asInvoker" uiAccess="false"></requestedExecutionLevel>
              </requestedPrivileges>
            </security>
          </trustInfo>
          <dependency>
            <dependentAssembly>
              <assemblyIdentity type="win32" name="Microsoft.VC%(maj)d%(min)d.CRT" version="%(fullver)s" processorArchitecture="*" publicKeyToken="1fc8b3b9a1e18e3b"></assemblyIdentity>
            </dependentAssembly>
          </dependency>
        </assembly>)ÚfullverÚmajÚmin)Ú_MSVCRVER_TO_FULLVERÚstrÚKeyErrorrF   ÚtextwrapÚdedent)r’   r“   r‘   Útemplater   r   r   Úmsvc_manifest_xml2  s    rš   rM   c             C   s4   |dkrd}n|dkrd}nt d| ƒ‚d|| f S )aW  Return the rc file used to generate the res file which will be embedded
    as manifest for given manifest file name, of given type ('dll' or
    'exe').

    Parameters
    ----------
    name : str
            name of the manifest file to embed
    type : str {'dll', 'exe'}
            type of the binary which will embed the manifest

    rM   r<   Zexer   zType %s not supportedz&#include "winuser.h"
%d RT_MANIFEST %s)rF   )rz   ÚtypeZrctyper   r   r   Úmanifest_rcR  s    rœ   c             C   s.   t ƒ }|r*|t| ƒks*tdt| ƒ|f ƒ‚dS )z6msver is the ms runtime version used for the MANIFEST.zKDiscrepancy between linked msvcr (%d) and the one about to be embedded (%d)N)r	   rZ   rF   )Úmsverr’   r   r   r   Ú!check_embedded_msvcr_match_linkedj  s    rž   c             C   s&   t j |  dg d¡¡}t j |¡d S )NZyoÚcr   )r+   r,   r0   Z_gen_temp_sourcefiler-   )Úconfigr2   r   r   r   Úconfigtest_namev  s    r¡   c             C   s   t | ƒ}| jj}|| d S )Nz	.manifest)r¡   r   Zexe_extension)r    rk   Zexextr   r   r   Úmanifest_namez  s    r¢   c             C   s   t | ƒ}|d S )Nz.rc)r¡   )r    rk   r   r   r   Úrc_name€  s    r£   c          	   C   sz   t ƒ }|d k	rv|dkrvt|ƒ t|ƒ d¡\}}tt|ƒt|ƒƒ}tt| ƒdƒ }| j 	t| ƒ¡ | 
|¡ W d Q R X d S )Né   r   rS   )Úget_build_msvc_versionrž   r•   rP   rš   rZ   r]   r¢   Z
temp_filesr$   r^   )r    r   Zma_strZmi_strZmanxmlZmanr   r   r   Úgenerate_manifest…  s    r¦   )F)rM   )=r7   r+   rC   r=   r   r   r—   Znumpy.distutils.ccompilerZnumpyr   r   Zdistutils.cygwinccompilerr   Zdistutils.versionr   Zdistutils.unixccompilerr   Zdistutils.msvccompilerr   r¥   Zdistutils.errorsr   Znumpy.distutils.misc_utilr   r   r	   r
   r   r   Z	get_msvcrÚcompilerV   rY   r   r   rN   rR   rf   ru   r    r   r‰   rƒ   r„   r”   Zmsvcrtr?   rŽ   rP   ÚmajorÚminorÚrestÚImportErrorr\   rš   rœ   rž   r¡   r¢   r£   r¦   r   r   r   r   Ú<module>	   sd   

 B) 
;$B
 
