o
    ék)hP  ã                
   @   sF  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 dS  e!y  Z" zee"ƒ W Y dZ"["dS dZ"["ww 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   Z64bit)ÚsysÚplatformÚarchitecture© r   r   ú=lib/python3.10/site-packages/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g d¢d
d
g d¢ddgdgg d¢dœZ	d
Z
d
Zejdkr=ejdkr=dgZejdkrNdD ]	Ze	e  d¡ qDd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…  ¡ }|  d¡st|ƒdkr?t d|¡}|r>|  d¡r1d| d¡fS | ¡ d	kr>d
| d¡fS n3t d|¡}|rNd
| d¡fS t d|¡}|rr| d¡}|  d¡sj|  d¡sj|  d¡rnd|fS d
|fS d}t|| ƒ‚)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   s2   
ÿ
ÿ
€ÿ
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)Nú-gú-Wallú-fno-second-underscore)Nr+   r,   Úarú-crÚranlib©Úversion_cmdÚcompiler_f77Úcompiler_f90Úcompiler_fixÚ	linker_soZarchiverr0   Ú
linker_exeÚntÚcygwinz-fPICr
   )r2   r3   r6   r7   ú-mno-cygwinÚg2cÚgnu95c                 C   sž   | j dd … }tjdkr=tj dd ¡}|s5dd l}| d¡}|s.d}d|› }tj	|dd t
|ƒtjd< | g d	¢¡ n| d
¡ tj d¡rM| d¡ |S )Nr   ÚdarwinZMACOSX_DEPLOYMENT_TARGETr   z10.9z.Env. variable MACOSX_DEPLOYMENT_TARGET set to é   )Ú
stacklevel)z
-undefinedZdynamic_lookupz-bundlez-sharedZsunosz-mimpure-text)r6   r   r   ÚosÚenvironÚgetÚ	sysconfigZget_config_varÚwarningsÚwarnÚstrÚextendÚappendr   )r#   ÚoptÚtargetrC   Úsr   r   r   Úget_flags_linker_som   s    




z GnuFCompiler.get_flags_linker_soc              	   C   sF   zt  | jdg ¡}W n tt jfy   Y d S w t|ƒ}tj |¡S )Nz-print-libgcc-file-name)	Ú
subprocessÚcheck_outputr3   ÚOSErrorÚCalledProcessErrorr   r@   Ú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 }|rBd |¡g}z
t | j| ¡}W n ttjfy7   Y |S w t|ƒ}t	j
 |¡}|S )Né   Úlinuxzlibgfortran.sor=   zlibgfortran.dylibz-print-file-name={0})r   r   ÚformatrM   rN   r3   rO   rP   r   r@   rQ   rR   )r#   Zlibgfortran_nameZlibgfortran_dirZfind_lib_argrS   r   r   r   Úget_libgfortran_dir›   s&   
ÿûz GnuFCompiler.get_libgfortran_dirc                 C   sâ   g }t jd d… dkrd|  ¡ }|rdt jdkr_| d¡s_tj |¡}tj |d| j ¡}tj 	|¡s_tjj|gtj
fd ¢R Ž }tj tj |d¡¡}tj |d| j ¡}tj 	|¡r_| |¡ | |¡ |  ¡ }|ro| |¡ |S )NrU   rV   r
   z/usr/libzlib%s.aé   Úlib)r   r   rT   r   r@   rQ   ÚnormpathÚjoinr;   ÚexistsÚpardirÚabspathrH   rX   )r#   rI   ÚdrQ   ÚrootZd2Úlib_gfortran_dirr   r   r   Úget_library_dirs°   s$   


zGnuFCompiler.get_library_dirsc                 C   s¤   g }|   ¡ }|d ur&| jd }| j|| jf }tj tj ||¡¡s%| j}n| j}|d ur2| |¡ | j	}t
jdkrF|rF|jdkrF| d¡ t
jdkrP| d¡ |S )Nz-picr
   ÚmsvcÚgccr=   Ú
cc_dynamic)rT   r;   Zstatic_lib_formatZstatic_lib_extensionr@   rQ   Úisfiler\   rH   Ú
c_compilerr   r   Úcompiler_type)r#   rI   r`   r;   Úfrh   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_versionrH   )r#   r&   rI   r   r   r   Úget_flags_optÞ   s   
zGnuFCompiler.get_flags_optc                 C   s^   ddl }z| ¡ d }W n ty   g  Y S w t d¡}g }| |¡D ]}|d|g7 }q$|S )z( Return detected arch flags from CFLAGS r   NZCFLAGSz-arch\s+(\w+)ú-arch)rC   Zget_config_varsÚKeyErrorr   ÚcompileÚfindall)r#   rC   ZcflagsZarch_reÚ
arch_flagsÚarchr   r   r   Ú_c_arch_flagsé   s   ÿ
zGnuFCompiler._c_arch_flagsc                 C   s   g S )Nr   rl   r   r   r   Úget_flags_archö   s   zGnuFCompiler.get_flags_archc                 C   sX   t jdks
t jdkrt‚d|vsJ ‚t jdkrd|› S t j d¡r'd|› S d|› S )	Nr
   r9   ú,r=   z-Wl,-rpath,©ZaixZos400z-Wl,-blibpath:z-Wl,-rpath=)r   r   ÚNotImplementedErrorr   )r#   Údirr   r   r   Úruntime_library_dir_optionù   s   



z'GnuFCompiler.runtime_library_dir_option)Ú__name__Ú
__module__Ú__qualname__ri   Úcompiler_aliasesÚdescriptionr(   r)   Úpossible_executablesÚexecutablesÚmodule_dir_switchÚmodule_include_switchr@   Únamer   r   Z	pic_flagsÚkeyrH   r;   Zsuggested_f90_compilerrL   rT   rX   rc   rk   rm   ro   rv   rw   r|   r   r   r   r   r      sF    -ø

#c                	   @   s  e Zd ZdZdZdZdd„ ZddgZdd	gg d
¢g d¢g d
¢g d¢ddgdgddgdœZdZ	dZ
ej d¡rUed  d¡ e ¡ d dd… dkrUdD ]	Zee  d¡ qKd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r	 |S tjdkr.dD ]
}| j|  d¡ q#|S )Nr   r   r   Ú4r
   )r2   r3   r4   r5   r6   r7   r:   )r(   r   r   r   rƒ   rH   )r#   r$   r&   r‡   r   r   r   r)     s   


úzGnu95FCompiler.version_matchr   Zf95ú<F90>r*   )Nr,   r+   z-ffixed-formr-   )Nr,   r+   r-   )r‰   r,   r+   r.   r/   r0   Nr,   r1   z-Jz-Iry   r6   z	-lpthreadr   r>   Z64)r3   r4   r5   r6   r7   z-maix64c                 C   s^   t jdksg S g }|  ¡ }d|v rd|| d¡< dD ]}t||ƒr,||v r,| d|g¡ q|S )z>Return a list of -arch flags for every supported architecture.r=   Zi386Úi686)ZppcrŠ   Zx86_64Zppc64Zs390xrp   )r   r   rv   ÚindexÚ_can_targetrG   )r#   Úcmdrt   Zc_archsru   r   r   r   Ú_universal_flags<  s   
€zGnu95FCompiler._universal_flagsc                 C   ó*   t  | ¡}|  | j¡}|r||d d…< |S ©Nr   )r   Ú	get_flagsrŽ   r4   ©r#   Úflagsrt   r   r   r   r‘   L  ó
   
zGnu95FCompiler.get_flagsc                 C   r   r   )r   rL   rŽ   r6   r’   r   r   r   rL   S  r”   z"Gnu95FCompiler.get_flags_linker_soc           	      C   s´   t  | ¡}tjdkrM| j}|rM|jdkrM|  ¡ }|rMtj 	|  
¡ ¡}tjj|gtjfd ¢R Ž }tj |d¡}tj 	|¡}tj tj |d¡¡rM| |¡ |  ¡ }|rX| |¡ |S )Nr
   rd   rY   rZ   zlibmingwex.a)r   rc   r   r   rh   ri   Ú
get_targetr@   rQ   r[   rT   r\   r^   r]   rH   rX   )	r#   rI   rh   rJ   r`   ra   rQ   Zmingwdirrb   r   r   r   rc   Z  s    



zGnu95FCompiler.get_library_dirsc                 C   sŒ   t  | ¡}tjdkr| d¡ tjdkrD| j}|r7|jdkr7d|v r7| d¡}| |d d¡ | |d d¡ | j}|rC|jdkrCg S 	 |S )	Nr=   rf   r
   rd   re   r   ZmingwexZmingw32)	r   rk   r   r   Úremoverh   ri   r‹   Úinsert)r#   rI   rh   Úir   r   r   rk   m  s   




zGnu95FCompiler.get_librariesc              	   C   s|   zt j| jdg t jt jd}| ¡ \}}|pd|pd }W n tt jfy+   Y dS w t|ƒ}t 	|¡}|r<| 
d¡S dS )Nz-v)ÚstdinÚstderró    r   Ú )rM   r   r3   r   ÚcommunicaterO   rP   r   ÚTARGET_Rr   r    )r#   ÚpÚstdoutrš   rS   r%   r   r   r   r•     s"   
ýù

zGnu95FCompiler.get_targetc              	   C   s|   t  ¡ }|D ]&}t|dƒ}	 | d¡}|sn| |¡ qW d   ƒ n1 s'w   Y  qt | ¡ ¡}| d¡}| 	d¡S )NÚrbTi   Úasciiú=)
ÚhashlibZsha1ÚopenÚreadÚupdateÚbase64Z	b32encodeZdigestÚdecodeÚrstrip)r#   Ú	filenamesÚhÚfnrj   Ú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 |¡rh||fS |rtdgt|ƒ dg }| j|||t|ƒdd| ddddg d tƒ rd}nd}d| d| |g}|js¡| ¡  | |jg| ¡ ||fS )zeCreate a wrapper shared library for the given objects

        Return an MSVC-compatible lib
        rd   zThis method only supports MSVCZ	win_amd64r
   rZ   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:)rh   ri   r"   r°   Úlistr   r@   rQ   ÚsplitextÚbasenamer\   rg   Zlink_shared_objectZinitializedZ
initializeZspawnrZ   )r#   Úobjectsr³   Úextra_dll_dirÚchained_dllsÚ
is_archiverh   Zobject_hashÚtagr¶   Z	root_nameZdll_nameZdef_nameZlib_nameZdll_pathZdef_pathZlib_pathZ	specifierZlib_argsr   r   r   Ú_link_wrapper_libž  s`   
ÿÿÿÿúüz Gnu95FCompiler._link_wrapper_libc                 C   s
   |j dvS )N)rd   )ri   )r#   Zcompilerr   r   r   Úcan_ccompiler_linkÜ  s   
z!Gnu95FCompiler.can_ccompiler_linkc                 C   sÂ   | j jdkr]g }g }|D ]}| ¡  d¡r| |¡ q| |¡ qg }g }|ddd… D ]}	| j|	g|||dd\}
}| d|
¡ | d|¡ q,|sL|S | j||||dd\}
}|
g| S td	ƒ‚)
z†
        Convert a set of object files that are not compatible with the default
        linker, to a file that is compatible.
        rd   z.aNéÿÿÿÿT)r¹   rº   r   FzUnsupported C compiler)rh   ri   ÚlowerÚendswithrH   r¼   r—   r"   )r#   r·   r³   r¸   ZarchivesZplain_objectsÚobjZchained_libsr¹   ÚarchiverZ   Zdllr   r   r   Úwrap_unlinkable_objectsà  s<   
û
û
z&Gnu95FCompiler.wrap_unlinkable_objects)r}   r~   r   ri   r€   r   r)   r‚   rƒ   r„   r…   r   r   r   rH   r   r‡   r;   rŽ   r‘   rL   rc   rk   r•   r°   r¼   r½   rÃ   r   r   r   r   r	   
  sB    õ>c              	   C   sÖ   | dd… }t jdd\}}t |¡ zQtj |¡}tj |¡d d }z3| d|d|g¡ t|t	t
|d}| ¡  |jdkW tj |¡rQt |¡ W t |¡ S W t |¡ S tj |¡rdt |¡ w w t |¡ w )	z7Return true if the architecture supports the -arch flagNz.f)Úsuffixr   z.orp   z-c)rš   r    Úcwd)ÚtempfileZmkstempr@   ÚcloserQ   rR   rµ   rG   r   r   r   r   Ú
returncoder]   r–   )r   ru   ZnewcmdZfidÚfilenamer`   rS   rŸ   r   r   r   rŒ     s(   

ýýÿrŒ   Ú__main__)Úlog)Úcustomized_fcompilerr>   r   Zg95)#r   r@   r   rD   r   rÆ   r¤   r¨   rM   r   r   r   Znumpy.distutils.exec_commandr   Znumpy.distutils.fcompilerr   Zdistutils.versionr   Z	compilersrr   rž   r   r   r	   rŒ   r}   Ú	distutilsrË   Znumpy.distutilsrÌ   Zset_verbosityÚprintrn   Ú	ExceptionÚer   r   r   r   Ú<module>   sD    
 q  
€ÿø