U
    O8ÚcP  ã                
   @   sD  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Zd dlZd dlm	Z	m
Z
mZ d dlmZ d dlmZ d dlmZ ddgZe  d¡Zd	d
„ ZG dd„ deƒZG dd„ deƒZdd„ Zedkr@d dlmZ d dlmZ e d¡ eedƒ  ¡ ƒ zeedƒ  ¡ ƒ W n, e!k
r> Z" zee"ƒ W 5 dZ"["X Y nX dS )é    N)ÚPopenÚPIPEÚSTDOUT)Úfilepath_from_subprocess_output)Ú	FCompiler)ÚLooseVersionÚGnuFCompilerÚGnu95FCompilerzTarget: ([a-zA-Z0-9_\-]*)c                   C   s   t jdkot ¡ d dkS )NÚwin32r   Ú64bit)ÚsysÚplatformÚarchitecture© r   r   úA/tmp/pip-unpacked-wheel-fd_gsd75/numpy/distutils/fcompiler/gnu.pyÚis_win64   s    r   c                
   @   s   e Zd ZdZdZdZdd„ Zdd„ Zdd	gZd
dgd
dddgd
d
d
ddgddgdgd
ddgdœZ	d
Z
d
Zejdkr‚ejdkr‚dgZejdkr¤dD ]Ze	e  d¡ qdZdZdd„ Zdd„ Zdd „ Zd!d"„ Zd#d$„ Zd%d&„ Zd'd(„ Zd)d*„ Zd+d,„ Zd-d.„ Zd
S )/r   Úgnu)Úg77zGNU Fortran 77 compilerc                 C   sð   |  d¡r&|| d¡d d…  ¡ }q t|ƒdkrvt d|¡}|rÜ|  d¡rZd| d¡fS | ¡ d	krÜd
| d¡fS nft d|¡}|r”d
| d¡fS t d|¡}|rÜ| d¡}|  d¡sÌ|  d¡sÌ|  d¡rÔd|fS d
|fS d}t|| ƒ‚dS )z6Handle the different versions of GNU fortran compilerszgfortran: warningÚ
é   Né   z	([0-9.]+)zGNU Fortranr   r   ÚgfortranzGNU Fortran\s+95.*?([0-9-.]+)z%GNU Fortran.*?\-?([0-9-.]+\.[0-9-.]+)Ú0Ú2Ú3z6A valid Fortran version was not found in this string:
)	Ú
startswithÚfindÚstripÚlenÚreÚsearchÚgroupÚstartÚ
ValueError)ÚselfÚversion_stringÚmÚvÚerrr   r   r   Úgnu_version_match   s0    
ÿ
 ÿ
zGnuFCompiler.gnu_version_matchc                 C   s&   |   |¡}|r|d dkrd S |d S )Nr   r   r   )r)   )r$   r%   r'   r   r   r   Úversion_matchL   s    
zGnuFCompiler.version_matchr   Zf77Nú-dumpversionú-gú-Wallú-fno-second-underscoreÚarú-crÚranlib©Úversion_cmdÚcompiler_f77Úcompiler_f90Úcompiler_fixÚ	linker_soÚarchiverr1   Ú
linker_exeÚntÚcygwinz-fPICr
   )r3   r4   r7   r9   ú-mno-cygwinÚg2cÚgnu95c                 C   s    | j dd … }tjdkr|tj dd ¡}|sjdd l}| d¡}|s\d}d|› }tj	|dd t
|ƒtjd< | d	d
dg¡ n
| d¡ tj d¡rœ| d¡ |S )Nr   ÚdarwinÚMACOSX_DEPLOYMENT_TARGETr   z10.9z.Env. variable MACOSX_DEPLOYMENT_TARGET set to é   )Ú
stacklevelz
-undefinedZdynamic_lookupz-bundlez-sharedÚsunosz-mimpure-text)r7   r   r   ÚosÚenvironÚgetÚ	sysconfigÚget_config_varÚwarningsÚwarnÚstrÚextendÚappendr   )r$   ÚoptÚtargetrG   Úsr   r   r   Úget_flags_linker_som   s     




z GnuFCompiler.get_flags_linker_soc              	   C   sJ   zt  | jdg ¡}W n tt jfk
r0   Y nX t|ƒ}tj |¡S d S )Nz-print-libgcc-file-name)	Ú
subprocessÚcheck_outputr4   ÚOSErrorÚCalledProcessErrorr   rD   ÚpathÚdirname)r$   Úoutputr   r   r   Úget_libgcc_dir   s    ÿ
zGnuFCompiler.get_libgcc_dirc              	   C   sˆ   t jd d… dkrd}nt jdkr(d}nd }d }|r„d |¡g}zt | j| ¡}W n ttjfk
rn   Y nX t|ƒ}t	j
 |¡}|S )Né   Úlinuxzlibgfortran.sor?   zlibgfortran.dylibz-print-file-name={0})r   r   ÚformatrR   rS   r4   rT   rU   r   rD   rV   rW   )r$   Zlibgfortran_nameZlibgfortran_dirZfind_lib_argrX   r   r   r   Úget_libgfortran_dir›   s"    
ÿz GnuFCompiler.get_libgfortran_dirc                 C   sà   g }t jd d… dkrÆ|  ¡ }|rÆt jdkr¼| d¡s¼tj |¡}tj |d| j ¡}tj 	|¡s¼tjj|ftj
fd žŽ }tj tj |d¡¡}tj |d| j ¡}tj 	|¡r¼| |¡ | |¡ |  ¡ }|rÜ| |¡ |S )NrZ   r[   r
   z/usr/libzlib%s.aé   Úlib)r   r   rY   r   rD   rV   ÚnormpathÚjoinr=   ÚexistsÚpardirÚabspathrM   r]   )r$   rN   ÚdrV   ÚrootZd2Úlib_gfortran_dirr   r   r   Úget_library_dirs°   s$    


zGnuFCompiler.get_library_dirsc                 C   s¤   g }|   ¡ }|d k	rL| jd }| j|| jf }tj tj ||¡¡sR| j}n| j}|d k	rd| |¡ | j	}t
jdkrŒ|rŒ|jdkrŒ| d¡ t
jdkr | d¡ |S )Nz-picr
   ÚmsvcÚgccr?   Ú
cc_dynamic)rY   r=   Zstatic_lib_formatZstatic_lib_extensionrD   rV   Úisfilera   rM   Ú
c_compilerr   r   Úcompiler_type)r$   rN   re   r=   Úfrm   r   r   r   Úget_librariesÆ   s$    

ÿ


zGnuFCompiler.get_librariesc                 C   s   dgS )Nr,   r   ©r$   r   r   r   Úget_flags_debugÛ   s    zGnuFCompiler.get_flags_debugc                 C   s0   |   ¡ }|r|dkrdg}ndg}| d¡ |S )Nz3.3.3z-O2z-O3z-funroll-loops)Úget_versionrM   )r$   r'   rN   r   r   r   Úget_flags_optÞ   s    
zGnuFCompiler.get_flags_optc                 C   s`   ddl }z| ¡ d }W n tk
r0   g  Y S X t d¡}g }| |¡D ]}|d|g7 }qJ|S )z( Return detected arch flags from CFLAGS r   NÚCFLAGSz-arch\s+(\w+)ú-arch)rG   Úget_config_varsÚKeyErrorr   ÚcompileÚfindall)r$   rG   ÚcflagsZarch_reÚ
arch_flagsÚarchr   r   r   Ú_c_arch_flagsé   s    

zGnuFCompiler._c_arch_flagsc                 C   s   g S )Nr   rq   r   r   r   Úget_flags_archö   s    zGnuFCompiler.get_flags_archc                 C   s\   t jdkst jdkrt‚d|ks$t‚t jdkr8d|› S t j d¡rNd|› S d|› S d S )	Nr
   r;   ú,r?   z-Wl,-rpath,©ÚaixZos400z-Wl,-blibpath:z-Wl,-rpath=)r   r   ÚNotImplementedErrorÚAssertionErrorr   )r$   Údirr   r   r   Úruntime_library_dir_optionù   s    


z'GnuFCompiler.runtime_library_dir_option)Ú__name__Ú
__module__Ú__qualname__rn   Úcompiler_aliasesÚdescriptionr)   r*   Úpossible_executablesÚexecutablesÚmodule_dir_switchÚmodule_include_switchrD   Únamer   r   Z	pic_flagsÚkeyrM   r=   Zsuggested_f90_compilerrQ   rY   r]   rh   rp   rr   rt   r~   r   r†   r   r   r   r   r      sD   -
ø

#c                	   @   s  e Zd ZdZdZdZdd„ ZddgZdd	gd
ddddgd
dddgd
ddddgdddgddgdgd
dgdœZdZ	dZ
ej d¡r¼ed  d¡ e ¡ d d
d… dkr¼dD ]Zee  d¡ q¨dZdd„ Zdd „ Zd!d"„ Zd#d$„ Zd%d&„ Zd'd(„ Zd)d*„ Zd+d,„ Zd-d.„ Zd/d0„ Zd
S )1r	   r>   )r   zGNU Fortran 95 compilerc                 C   s\   |   |¡}|r|d dkrd S |d }t|ƒdkr4n$tjdkrXdD ]}| j|  d¡ qB|S )Nr   r   r   Ú4r
   )r3   r4   r5   r6   r7   r9   r<   )r)   r   r   r   r   rM   )r$   r%   r'   r‘   r   r   r   r*     s    

zGnu95FCompiler.version_matchr   Zf95z<F90>r+   Nr-   r,   z-ffixed-formr.   r/   r0   r1   r2   z-Jz-Ir   r7   z	-lpthreadr   rA   Z64)r4   r5   r6   r7   r9   z-maix64c                 C   s^   t jdksg S g }|  ¡ }d|kr0d|| d¡< dD ]$}t||ƒr4||kr4| d|g¡ q4|S )z>Return a list of -arch flags for every supported architecture.r?   Úi386Úi686)Úppcr”   Úx86_64Úppc64Ús390xrv   )r   r   r~   ÚindexÚ_can_targetrL   )r$   Úcmdr|   Zc_archsr}   r   r   r   Ú_universal_flags<  s    
zGnu95FCompiler._universal_flagsc                 C   s*   t  | ¡}|  | j¡}|r&||d d…< |S ©Nr   )r   Ú	get_flagsrœ   r5   ©r$   Úflagsr|   r   r   r   rž   L  s
    
zGnu95FCompiler.get_flagsc                 C   s*   t  | ¡}|  | j¡}|r&||d d…< |S r   )r   rQ   rœ   r7   rŸ   r   r   r   rQ   S  s
    
z"Gnu95FCompiler.get_flags_linker_soc           	      C   s²   t  | ¡}tjdkr˜| j}|r˜|jdkr˜|  ¡ }|r˜tj 	|  
¡ ¡}tjj|ftjfd žŽ }tj |d¡}tj 	|¡}tj tj |d¡¡r˜| |¡ |  ¡ }|r®| |¡ |S )Nr
   ri   r^   r_   zlibmingwex.a)r   rh   r   r   rm   rn   Ú
get_targetrD   rV   r`   rY   ra   rc   rb   rM   r]   )	r$   rN   rm   rO   re   rf   rV   Zmingwdirrg   r   r   r   rh   Z  s     



zGnu95FCompiler.get_library_dirsc                 C   sŠ   t  | ¡}tjdkr| d¡ tjdkr†| j}|rn|jdkrnd|krn| d¡}| |d d¡ | |d d¡ | j}|r†|jdkr†g S |S )	Nr?   rk   r
   ri   rj   r   ZmingwexZmingw32)	r   rp   r   r   Úremoverm   rn   r™   Úinsert)r$   rN   rm   Úir   r   r   rp   m  s    




zGnu95FCompiler.get_librariesc              	   C   s|   z<t j| jdg t jt jd}| ¡ \}}|p0d|p6d }W n tt jfk
rV   Y n"X t|ƒ}t 	|¡}|rx| 
d¡S dS )Nz-v)ÚstdinÚstderró    r   Ú )rR   r   r4   r   ÚcommunicaterT   rU   r   ÚTARGET_Rr    r!   )r$   ÚpÚstdoutr¦   rX   r&   r   r   r   r¡     s    
ý

zGnu95FCompiler.get_targetc              
   C   sf   t  ¡ }|D ]6}t|dƒ"}| d¡}|s,q8| |¡ qW 5 Q R X qt | ¡ ¡}| d¡}| 	d¡S )NÚrbi   Úasciiú=)
ÚhashlibÚsha1ÚopenÚreadÚupdateÚbase64Ú	b32encodeÚdigestÚdecodeÚrstrip)r$   Ú	filenamesÚhÚfnro   ÚblockÚtextr   r   r   Ú_hash_files‘  s    

zGnu95FCompiler._hash_filesc                 C   s`  | j }|jdkrtdƒ‚|  t|ƒt|ƒ ¡}tƒ r:d}nd}dtj tj 	|d ¡¡d dd…  }	|	d	 | d
 | }
|
d }|
d }|
d }tj 
||¡}tj 
||¡}tj 
||¡}tj |¡rÐ||fS |rèdgt|ƒ dg }| j|||t|ƒdd| ddddg d tƒ r d}nd}d| d| |g}|jsF| ¡  | |jg| ¡ ||fS )zeCreate a wrapper shared library for the given objects

        Return an MSVC-compatible lib
        ri   zThis method only supports MSVCZ	win_amd64r
   r_   r   Né   Ú.z
.gfortran-z.dllz.defz.libz-Wl,--whole-archivez-Wl,--no-whole-archivez-Wl,--allow-multiple-definitionz-Wl,--output-def,z-Wl,--export-all-symbolsz-Wl,--enable-auto-importz-staticz-mlong-double-64)Ú
output_dirZextra_postargsz/MACHINE:X64z/MACHINE:X86z/def:z/OUT:)rm   rn   r#   r¿   Úlistr   rD   rV   ÚsplitextÚbasenamera   rl   Zlink_shared_objectZinitializedZ
initializeÚspawnr_   )r$   ÚobjectsrÂ   Úextra_dll_dirÚchained_dllsÚ
is_archiverm   Zobject_hashÚtagrÅ   Ú	root_nameZdll_nameZdef_nameZlib_nameZdll_pathZdef_pathZlib_pathÚ	specifierZlib_argsr   r   r   Ú_link_wrapper_libž  s`    
ÿÿÿÿúüz Gnu95FCompiler._link_wrapper_libc                 C   s
   |j dkS )N)ri   )rn   )r$   Úcompilerr   r   r   Úcan_ccompiler_linkÜ  s    z!Gnu95FCompiler.can_ccompiler_linkc                 C   sÆ   | j jdkrºg }g }|D ](}| ¡  d¡r6| |¡ q| |¡ qg }g }|ddd… D ]6}	| j|	g|||dd\}
}| d|
¡ | d|¡ qX|s˜|S | j||||dd\}
}|
g| S td	ƒ‚dS )
z†
        Convert a set of object files that are not compatible with the default
        linker, to a file that is compatible.
        ri   z.aNéÿÿÿÿT)rÉ   rÊ   r   FzUnsupported C compiler)rm   rn   ÚlowerÚendswithrM   rÎ   r£   r#   )r$   rÇ   rÂ   rÈ   ZarchivesZplain_objectsÚobjZchained_libsrÉ   Úarchiver_   Údllr   r   r   Úwrap_unlinkable_objectsà  s<    û
û

z&Gnu95FCompiler.wrap_unlinkable_objects)r‡   rˆ   r‰   rn   rŠ   r‹   r*   rŒ   r   rŽ   r   r   r   r   rM   r   r‘   r=   rœ   rž   rQ   rh   rp   r¡   r¿   rÎ   rÐ   r×   r   r   r   r   r	   
  sL   ÿÿÿõ>c              	   C   s²   | dd… }t jdd\}}t |¡ zztj |¡}tj |¡d d }z<| 	d|d|g¡ t
|tt|d}| ¡  |jdkW ¢W ¢S tj |¡rœt |¡ X W 5 t |¡ X dS )	z7Return true if the architecture supports the -arch flagNz.f)Úsuffixr   z.orv   z-c)r¦   r¬   Úcwd)ÚtempfileÚmkstemprD   Úcloser¢   rV   rW   rÄ   rb   rL   r   r   r   r©   Ú
returncode)r›   r}   ZnewcmdZfidÚfilenamere   rX   r«   r   r   r   rš     s    
rš   Ú__main__)Úlog)Úcustomized_fcompilerrA   r   Zg95)#r   rD   r   rI   r   rÚ   r°   rµ   rR   r   r   r   Znumpy.distutils.exec_commandr   Znumpy.distutils.fcompilerr   Zdistutils.versionr   Z	compilersry   rª   r   r   r	   rš   r‡   Ú	distutilsrà   Znumpy.distutilsrá   Úset_verbosityÚprintrs   Ú	ExceptionÚer   r   r   r   Ú<module>   s<   
 q  

