o
    $حcW                     @   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mZmZmZ dd	 Z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rz/ddl*Z*d"e)d#< d$e)d%< d&e)d'< e+e*d(dZ,e,dure-d)e,. \Z/Z0e,e)e/e0 < [/[0[,W n e1y   e
2d* Y nw d+d, Z3d;d.d/Z4d0d1 Z5d2d3 Z6d4d5 Z7d6d7 Z8d8d9 Z9d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)UnixCCompiler)get_build_version)UnknownFileError)msvc_runtime_librarymsvc_runtime_versionmsvc_runtime_majorget_build_architecturec                  C   s   t  } | du r	g S | gS )zBReplacement for outdated version of get_msvcr from cygwinccompilerN)r   )Zmsvcr r
   @lib/python3.10/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                   @   sH   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| ||| t  t }tdd}|s|r| d t }|r+| dd|  t dkr:| j	ddd	d
d n	| j	ddddd dg| _
d S )NT)debugZNPY_MINGW_USE_CUSTOM_MSVCRZ__MSVCRT_VERSION__z0x%04iAMD64z#gcc -g -DDEBUG -DMS_WIN64 -O0 -Wallz7gcc -g -DDEBUG -DMS_WIN64 -O0 -Wall -Wstrict-prototypeszgcc -gzgcc -g -shared)compilerZcompiler_soZ
linker_exeZ	linker_sozgcc -O2 -Wallz!gcc -O2 -Wall -Wstrict-prototypeszg++ zg++ -sharedzg++)	distutilscygwinccompilerCygwinCCompiler__init__build_import_librarybuild_msvcr_libraryZdefine_macror   r	   Zset_executablesZcompiler_cxx)selfverbosedry_runZforceZmsvcr_successZmsvcr_dbg_successZmsvcr_versionr
   r
   r   r   5   s4   


zMingw32CCompiler.__init__Nc                 C   sZ   t  }|r|s	g }|| | |||||||d |	|
|||f}tj}||d |jj   d S )N)r   appendr   link__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   m   s.   
zMingw32CCompiler.link c           	      C   s   |d u rd}g }|D ]\}t jt j|\}}t j|\}}|r(|dd  }|| jddg vr9td||f |rAt j|}|dksI|dkrY|t j	||| | j
  q
|t j	||| j
  q
|S )Nr!      .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base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   r-   r
   r
   r
   r   r   .   s"    
@
)r   c            
      C   sF  t jg} tt drt jt jkr| t j ntt dr't jt jkr'| t j g d}g }| D ]}|D ]}|tj|| q3q/dtj	v rS|tjtj	d d t
t jd d \}}t }|dkrod| | d	}n|d
krzd| d}nd| }td|  |D ]}tj||}	tj|	r|	  S qtd||f )Nbase_prefixreal_prefix)r!   libbinZ
SYSTEMROOTZSystem32   ZCPythonZpython.dllZPyPyZlibpypyz-c.dllzUnknown platform zLooking for %sz%s not found in %s)sysprefixhasattrr2   r   r3   r$   r%   r*   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   s6   

rI   c                 C   s   t dd| g}|dS )Nzobjdump.exez-p   
)
subprocesscheck_outputsplit)rH   str
   r
   r   
dump_table   s   
rO   c           	      C   s(  t | }tt|D ]}t||  r nq
tdg }t|d t|D ]}t||  }|rF|t	|
d |
df q' t|dkrTtd|   t|d1}|dtj|   |d |d	 |d
 |D ]}|d|d   qvW d   dS 1 sw   Y  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"   r6   r   zNo symbols found in %swzLIBRARY        %s
z,;CODE          PRELOAD MOVEABLE DISCARDABLE
z;DATA          PRELOAD SINGLE
z	
EXPORTS
z%s
N)rO   rangelen_STARTmatchdecoderA   _TABLEr   intgroupstripr   warnopenwriter$   r%   r)   )	rH   dfiledumpiZsymsjmdsr
   r
   r   generate_def   s.   $


"rd   c                    s4   dddt     fdd}dd }|| p|| S )NZamd64Zx86)r   Intelc                    sd   t jt jddd}t j|sd S t |D ]\}}}| |v r/ |v r/t j||   S qd S )NZWINDIRz
C:\WINDOWSZwinsxs)r$   r%   r*   r;   getr@   walk)dll_nameZwinsxs_pathrootdirsfilesZarchr
   r   _find_dll_in_winsxs	  s   z%find_dll.<locals>._find_dll_in_winsxsc                 S   sL   t jgtjd d D ]}tj|| }tj|r#tj|  S qd S )NPATH;)	r8   r9   r$   r;   rM   r%   r*   r@   abspath)rh   r%   filepathr
   r
   r   _find_dll_in_path  s   z#find_dll.<locals>._find_dll_in_path)r	   )rh   rm   rr   r
   rl   r   find_dll  s   rs   Fc                 C   s  t jdkrdS t }|d u rtd dS |dk r td dS t }| r)|d7 }d| }t jtj	d|}t j
|rFtd	|f  d
S |d }t|}|sYt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 MSVCP   z=Skip building msvcr library: custom functionality not presentrb   zlib%s.alibsz(Skip building msvcr library: "%s" existsTr7   z*Cannot build msvcr library: "%s" not foundz	lib%s.defz&Building msvcr library: "%s" (from %s)dlltool-d-l)r$   namer   r   r   r   r%   r*   r8   r9   isfilers   rZ   inford   rK   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     sH   





r   c                  C   s<   t jdkrd S t } | dkrt S | dkrt S td|  )Nrt   r   re   zUnhandled arch %s)r$   rz   r	   _build_import_library_amd64_build_import_library_x86rA   rl   r
   r
   r   r   Y  s   
r   c               
   C   s   t tjdd \} }g d}tjg}ttdr%tjtjkr%|tj nttdr6tjtjkr6|tj ddg}g }|D ]}|| |f }|D ]}|D ]}	|tj	
||	| qLqHq>|D ]}
tj	|
rld|
f  S q^d	|d
 fS )zACheck if an import library for the Python runtime already exists.Nr6   )zlibpython%d%d.azlibpython%d%d.dll.azlibpython%d.%d.dll.ar2   r3   rv   r4   TFr   )r<   r8   r=   r9   r:   r2   r   r3   r$   r%   r*   r{   )rE   rF   patternsrB   rC   Z
candidatespatfilenameZstem_dirrD   fullnamer
   r
   r   _check_for_import_libe  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 )	N)Skip building import library: "%s" existsz4Building import library (arch=AMD64): "%s" (from %s)python%d%d.defr6   rv   rw   rx   ry   )r   r   r   rI   r|   r<   r8   r=   r$   r%   r*   r9   rd   rK   Z
check_call)
out_existsr~   r   r   r   r   r
   r
   r   r     s   

r   c                  C   s  t  \} }| rtd| dS dttjdd  }tjtj	d|}tj
|sZttdr8tjtjd|}nttdrGtjtjd|}nd}tj
|rR|}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   n1 sw   Y  t }dd|d|d|g}t|}|rtd dS )z; Build the import libraries for Mingw32-gcc on Windows
    r   Nzpython%d%d.libr6   rv   r2   r3   r!   z+Cannot build import library: "%s" not foundz(Building import library (ARCH=x86): "%s"r   )lib2defr   F)shellrP   rw   z	--dllnamez--defz--output-libz:Failed to build import library for gcc. Linking will fail.)r   r   r   r<   r8   r=   r$   r%   r*   r9   r{   r:   r2   r3   rZ   r|   numpy.distutilsr   ZgetnmZ
DEFAULT_NMZparse_nmr[   Z
output_defZ
DEF_HEADERrI   rK   rL   )r   r~   Zlib_nameZlib_fileZbase_libr   r   r   Z	nm_outputZdlistZflistZfidrh   r   statusr
   r
   r   r     sJ   




r   win32z8.0.50727.42Z80z9.0.21022.8Z90z10.0.30319.460Z100ZCRT_ASSEMBLY_VERSIONz(\d+)\.(\d)z9Cannot import msvcrt: using manifest will not be possiblec                 C   sT   zt t| d |  }W n ty   td| |f dw 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KeyErrorrA   textwrapdedent)r   r   r   templater
   r
   r   msvc_manifest_xml  s   
	r   rH   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

    rH   r6   Zexer"   zType %s not supportedz&#include "winuser.h"
%d RT_MANIFEST %s)rA   )rz   typeZrctyper
   r
   r   manifest_rc  s   r   c                 C   s2   t  }|r|t| kstdt| |f dS 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   rW   rA   )msverr   r
   r
   r   !check_embedded_msvcr_match_linked.  s   
r   c                 C   s&   t j| dg d}t j|d S )NZyocr   )r$   r%   r)   Z_gen_temp_sourcefiler&   )configr+   r
   r
   r   configtest_name:  s   r   c                 C   s   t | }| jj}|| d S )Nz	.manifest)r   r   Zexe_extension)r   ri   Zexextr
   r
   r   manifest_name>  s   r   c                 C   s   t | }|d S )Nr#   )r   )r   ri   r
   r
   r   rc_nameD  s   r   c                 C   s   t  }|d urH|dkrJt| t|d\}}tt|t|}tt| d}| j	t|  |
| W d    d S 1 sAw   Y  d S d S d S )N   .rP   )get_build_msvc_versionr   r   rM   r   rW   r[   r   Z
temp_filesr   r\   )r   r   Zma_strZmi_strZmanxmlZmanr
   r
   r   generate_manifestI  s   "r   )F)rH   ):r1   r$   r>   r8   rK   rer   Znumpy.distutils.ccompilerZnumpyr   r   Zdistutils.cygwinccompilerr   Zdistutils.unixccompilerr   Zdistutils.msvccompilerr   r   Zdistutils.errorsr   Znumpy.distutils.misc_utilr   r   r   r	   r   compilerS   rV   r   r   r   rI   rO   rd   rs   r   r   r   r   r   r   ZmsvcrtgetattrZcrt_verrT   groupsr   r   ImportErrorrZ   r   r   r   r   r   r   r   r
   r
   r
   r   <module>   sf    	

 
) 
;$B

 