3
hb                 @   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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jdZej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ry`ddl-Z-d#e+d$< d%e+d&< d'e+d(< d)e+d*< e.e-d+re-j/j0d,d-\Z1Z2Z3e-j/e+e1e2 < [1[2[3W n  e4k
r   e	j5d. 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   A/tmp/pip-build-fibhr3ey/numpy/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.

    mingw32r   c       
      C   s  t jjj| ||| | jd krnytjddg}W n ttfk
rL   d}Y nX t	j
d|}|rnt|jd| _| jdkr~d}nd}| jdkrd| _n| jdkrd	| _t  t }td
d}|s|r| jd t }	|	r| j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 )&N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)compilercompiler_so
linker_exe	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OSErrorCalledProcessErrorresearchr   groupZ
linker_dllZlinkerbuild_import_librarybuild_msvcr_librarydefine_macror   r
   set_executablescompiler_cxx)
selfverbosedry_runforceZ
out_stringresultZentry_pointZmsvcr_successZmsvcr_dbg_successZmsvcr_versionr   r   r   r   8   sp    









zMingw32CCompiler.__init__Nc             C   sp   t  }|r|sg }|j| | |||||||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*   target_descobjectsoutput_filename
output_dir	librarieslibrary_dirsruntime_library_dirsexport_symbolsr   extra_preargsextra_postargs
build_temptarget_langZruntime_libraryargsfuncr   r   r   r0      s0    

zMingw32CCompiler.linkr   c       	      C   s   |d krd}g }x|D ]}t jjt jj|\}}t jj|\}}|rR|dd  }|| jddg krttd||f |rt jj|}|dks|dkr|jt jj	||| | j
  q|jt jj	||| j
  qW |S )Nr   r   z.rcz.resz"unknown file type '%s' (from '%s'))ospathsplitextnormcase
splitdrivesrc_extensionsr   basenamer/   joinobj_extension)	r*   source_filenames	strip_dirr6   	obj_namessrc_namebaseextdrvr   r   r   object_filenames   s(     
z!Mingw32CCompiler.object_filenames)r   r   r   )Nr   NNNN)r   r   )__name__
__module____qualname____doc__compiler_typer   r0   rQ   r   r   r   r   r   1   s     
r     
 r   c              C   s2  t jg} tt dr,t jt jkr,| jt j n"tt drNt jt jkrN| jt j dddg}g }x.| D ]&}x |D ]}|jtjj|| qlW qbW dtj	kr|jtjjtj	d d t
t jd d \}}d	g}xP|D ]H}|||f }	td
|	  x*|D ]"}tjj||	}
tjj|
r|
S qW qW td|	|f d S )Nbase_prefixreal_prefixr   libbinZ
SYSTEMROOTZSystem32   zpython%d%d.dllzLooking for %sz%s not found in %s)sysprefixhasattrrW   r/   rX   rA   rB   rH   environtupleversion_infoprintexists
ValueError)stemssub_dirsZlib_dirsZstemfoldermajor_versionminor_versionpatternspatdllnamedllr   r   r   find_python_dll   s,    





rn   c             C   s   t jdd| g}|j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j|| j rP qW tdg }xVt|d t|D ]@}tj|| j }|r|jt	|j
dj |j
df qRP qRW t|dkrtjd|   t|dZ}|jdtjj|   |jd |jd	 |jd
 x|D ]}|jd|d   qW W dQ R X dS )zGiven 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decoderd   _TABLEr/   intr$   stripr   warnopenwriterA   rB   rG   )	rm   dfiledumpiZsymsjmdsr   r   r   generate_def  s(    $



r   c                s4   dddt     fdd}dd }|| p2|| S )Namd64x86)r   Intelc                sd   t jjt jjddd}t jj|s(d S x6t j|D ](\}}}| |kr4 |kr4t jj|| S q4W d S )NZWINDIRz
C:\WINDOWSZwinsxs)rA   rB   rH   r_   getrc   walk)dll_nameZwinsxs_pathrootdirsfiles)archr   r   _find_dll_in_winsxsA  s    z%find_dll.<locals>._find_dll_in_winsxsc             S   sL   xFt jgtjd jd D ]*}tjj|| }tjj|rtjj|S qW d S )NPATH;)	r\   r]   rA   r_   rp   rB   rH   rc   abspath)r   rB   filepathr   r   r   _find_dll_in_pathL  s    z#find_dll.<locals>._find_dll_in_path)r
   )r   r   r   r   )r   r   find_dll<  s
    r   Fc             C   s  t jdkrdS t }|d kr*tjd dS |dk r@tjd dS t }| rR|d7 }d| }t jjtj	d|}t jj
|rtjd	|f  d
S |d }t|}|stjd|  dS d| }t jjtj	d|}tjd||f  t|| dd|d|g}	tj|	}
t j| |
 S )NntFz?Skip building import library: Runtime is not compiled with MSVCP   z=Skip building msvcr library: custom functionality not presentr   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)rA   namer	   r   r   r   rB   rH   r\   r]   isfiler   r|   infor   r   callremove)r   Z	msvcr_verZ
msvcr_nameZout_nameout_fileZmsvcr_dll_namedll_filedef_namedef_filecmdretcoder   r   r   r&   V  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 )Nr   r   r   zUnhandled arch %s)rA   r   r
   _build_import_library_amd64_build_import_library_x86rd   )r   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|jtj n"ttdrntjtjkrn|jtj dd	g}g }xJ|D ]B}|| |f }x0|D ](}x"|D ]}	|jtj	j
||	| qW qW qW x |D ]}
tj	j|
rd
|
fS qW 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.arW   rX   r   rY   TFr   )r`   r\   ra   r]   r^   rW   r/   rX   rA   rB   rH   r   )rh   ri   rj   re   rf   
candidatesrk   filenameZstem_dirrg   fullnamer   r   r   _check_for_import_lib  s(    


"
r   c              C   s   t  \} }| rtjd| d S t }tjd||f  dttjd d  }tj	j
tjd|}t|| dd|d|g}tj| d S )	Nz)Skip building import library: "%s" existsz4Building import library (arch=AMD64): "%s" (from %s)zpython%d%d.defr[   r   r   z-dz-l)r   r   r   rn   r   r`   r\   ra   rA   rB   rH   r]   r   r   
check_call)
out_existsr   r   r   r   r   r   r   r   r     s    

r   c              C   sv  t  \} }| rtjd| dS dttjdd  }tjjtj	d|}tjj
|sttdrptjjtjd|}n"ttdrtjjtjd|}nd}tjj
|r|}ntjd	| dS tjd
| ddlm} dttjdd  }tjjtj	d|}|j|j|g dd}|j|\}	}
t|d}|j|	|
|j| W dQ R X t }dd|d|d|g}tj|}|rrtjd dS )z; Build the import libraries for Mingw32-gcc on Windows
    z)Skip building import library: "%s" existsNzpython%d%d.libr[   r   rW   rX   r   z+Cannot build import library: "%s" not foundz(Building import library (ARCH=x86): "%s"r   )lib2defzpython%d%d.defF)shellrs   r   z	--dllnamez--defz--output-libz:Failed to build import library for gcc. Linking will fail.)r   r   r   r`   r\   ra   rA   rB   rH   r]   r   r^   rW   rX   r|   r   numpy.distutilsr   ZgetnmZ
DEFAULT_NMZparse_nmr}   Z
output_defZ
DEF_HEADERrn   r   r   )r   r   lib_namelib_fileZbase_libr   r   r   Z	nm_outputZdlistflistZfidr   r   statusr   r   r   r     sD    




r   win32z8.0.50727.42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   sX   yt t| d |  }W n$ tk
r<   td| |f Y nX tj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 yeta          <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KeyErrorrd   textwrapdedent)r   r   r   templater   r   r   msvc_manifest_xml.  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)rd   )r   typeZrctyper   r   r   manifest_rcN  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   rd   )msverr   r   r   r   !check_embedded_msvcr_match_linkedf  s    r   c             C   s&   t jj| jdg d}t jj|d S )NZyocr   )rA   rB   rG   Z_gen_temp_sourcefilerC   )configrN   r   r   r   configtest_namer  s    r   c             C   s   t | }| jj}|| d S )Nz	.manifest)r   r   exe_extension)r   r   Zexextr   r   r   manifest_namev  s    r   c             C   s   t | }|d S )Nz.rc)r   )r   r   r   r   r   rc_name|  s    r   c             C   st   t  }|d k	rp|dkrpt| t|}t|| d }t||}tt| d}| jjt|  |j| |j	  d S )N   r   rs   )
get_build_msvc_versionr   rz   r   r}   r   Z
temp_filesr/   r~   close)r   r   mamiZmanxmlZmanr   r   r   generate_manifest  s    

r   )F)rm   )=rU   rA   r\   r   r"   r   Znumpy.distutils.ccompilernumpyr   r   Zdistutils.cygwinccompilerr   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   r   r   r&   r%   r   r   r   r   platformmsvcrtr^   r   rp   majorminorrestImportErrorr|   r   r   r   r   r   r   r   r   r   r   r   <module>	   sb   

 B& 
;$B
 
