U
|
O ±dAR ã
|
@ sD d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlm Z m
|
Z
|
m Z d dl m Z d dlmZ d dlmZ ddgZe d¡Zd d
|
ZG d d deZG d d deZd d Zedkr@d dlmZ d dl m Z e d¡ e e d ¡ ze e d ¡ W n, e!k
|
r> Z" z e e" W 5 dZ"["X Y nX dS )é N)ÚPopenÚPIPEÚSTDOUT)Ú filepath_from_subprocess_output)Ú FCompiler)Ú LooseVersionÚ GnuFCompilerÚGnu95FCompilerzTarget: ([a-zA-Z0-9_\-]*)c C s t jdkot ¡ d dkS )NÚwin32r Ú64bit)ÚsysÚplatformÚ architecture© r r úTd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/distutils/fcompiler/gnu.pyÚis_win64 s r c
|
@ s e Zd ZdZdZdZdd Zdd Zdd gZd
|
d gd
|
d d dgd
|
d
|
d
|
d d gddgdgd
|
d d gdZ d
|
Z
|
d
|
Z e j dkrejdkrdgZejdkr¤dD ]Ze e d¡ qdZdZdd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Z d+d, Z d-d. Z d
|
S )/r Úgnu)Úg77zGNU Fortran 77 compilerc C sð | d¡r&|| d¡d d
¡ }q t|dkrvt d|¡}|rÜ| d¡rZd| d¡fS | ¡ d krÜd
|
| d¡fS nft d |¡}|rd
|
| d¡fS t d |¡}|rÜ| d¡}| d ¡sÌ| d¡sÌ| d¡rÔd|fS d
|
|fS d}t|| dS )z6Handle the different versions of GNU fortran compilerszgfortran: 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 dkr d S |d S )Nr r r )r) )r$ r% r' r r r Ú version_matchL s
|
zGnuFCompiler.version_matchr Zf77Nú -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 dd
}tjdkr|tj dd ¡}|sjdd l}| d¡}|s\d}d| }tj |dd t
|
|tjd< | d d
|
d g¡ n
|
| d ¡ tj d ¡r| d¡ |S )Nr ÚdarwinÚMACOSX_DEPLOYMENT_TARGETr z10.9z.Env. variable MACOSX_DEPLOYMENT_TARGET set to é )Ú
|
stacklevelz
|
-undefinedZdynamic_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 zt | jdg ¡}W n tt jfk
|
r0 Y nX t|}tj |¡S d S )Nz-print-libgcc-file-name) Ú
|
subprocessÚ check_outputr4 ÚOSErrorÚCalledProcessErrorr rD ÚpathÚdirname)r$ Úoutputr r r Úget_libgcc_dir s ÿ
|
zGnuFCompiler.get_libgcc_dirc C s t jd d
dkrd}nt jdkr(d}nd }d }|rd |¡g}zt | j| ¡}W n ttjfk
|
rn Y nX t|}t j
|
|¡}|S )Né Úlinuxzlibgfortran.sor? zlibgfortran.dylibz-print-file-name={0}) r r ÚformatrR rS r4 rT rU r rD rV rW )r$ Zlibgfortran_nameZlibgfortran_dirZ find_lib_argrX r r r Úget_libgfortran_dir s"
|
ÿ z GnuFCompiler.get_libgfortran_dirc C sà g }t jd d
dkrÆ| ¡ }|rÆt jdkr¼| d¡s¼tj |¡}tj |d| j ¡}tj |¡s¼tjj|ftj
|
fd }tj tj |d¡¡}tj |d| j ¡}tj |¡r¼| |¡ | |¡ | ¡ }|rÜ| |¡ |S )NrZ r[ r
|
z/usr/libzlib%s.aé Úlib)r r rY r rD rV ÚnormpathÚjoinr= ÚexistsÚpardirÚabspathrM r] )r$ rN ÚdrV ÚrootZd2Úlib_gfortran_dirr r r Úget_library_dirs° s$
|
|
|
z GnuFCompiler.get_library_dirsc C s¤ g }| ¡ }|d k rL| jd }| j|| jf }tj tj ||¡¡sR| j}n| j}|d k rd| |¡ | j }t
|
j dkr|r|j dkr| d¡ t
|
j dkr | d¡ |S )Nz-picr
|
ÚmsvcÚgccr? Ú
|
cc_dynamic) rY r= Zstatic_lib_formatZstatic_lib_extensionrD rV Úisfilera rM Ú
|
c_compilerr r Ú compiler_type)r$ rN re r= Úfrm r r r Ú get_librariesÆ s$
|
|
ÿ
|
|
|
zGnuFCompiler.get_librariesc C s dgS )Nr, r ©r$ r r r Úget_flags_debugÛ s z GnuFCompiler.get_flags_debugc C s0 | ¡ }|r |dkr dg}ndg}| d¡ |S )Nz3.3.3z-O2z-O3z-funroll-loops)Ú get_versionrM )r$ r' rN r r r Ú get_flags_optÞ s
|
zGnuFCompiler.get_flags_optc C s` ddl }z| ¡ d }W n tk
|
r0 g Y S X t d¡}g }| |¡D ]}|d|g7 }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 ÚcflagsZarch_reÚ
|
arch_flagsÚarchr r r Ú _c_arch_flagsé s
|
|
zGnuFCompiler._c_arch_flagsc C s g S )Nr rq r r r Úget_flags_archö s zGnuFCompiler.get_flags_archc C s\ t jdkst jdkrtd|ks$tt jdkr8d| S t j d¡rNd| S d| S d S ) Nr
|
r; ú,r? z -Wl,-rpath,©ÚaixZos400z-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= Zsuggested_f90_compilerrQ rY r] rh rp rr rt r~ r r r r r r r sD -
|
ø
|
|
# c @ s e Zd ZdZdZdZdd ZddgZdd gd
|
d d d dgd
|
d d dgd
|
d d d dgdd d gddgdgd
|
d gdZdZ dZ
|
e j d¡r¼ed d¡ e ¡ d d
|
d
dkr¼dD ]Zee d ¡ q¨dZd d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 Zd
|
S )1r r> )r zGNU Fortran 95 compilerc C s\ | |¡}|r|d dkr d S |d }t|dkr4n$tjdkrXdD ]}| 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 Zf95z<F90>r+ Nr- r, z -ffixed-formr. r/ r0 r1 r2 z-Jz-Ir r7 z -lpthreadr rA Ú64)r4 r5 r6 r7 r9 z-maix64c C s^ t jdksg S g }| ¡ }d|kr0d|| d¡< dD ]$}t||r4||kr4| 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| Zc_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
|
|
zGnu95FCompiler.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 | ¡}tjdkr| j}|r|jdkr| ¡ }|rtj |
|
¡ ¡}tjj |ftj fd }tj |d¡}tj |¡}tj tj |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 Zmingwdirrg r r r rh Z s
|
|
|
|
z Gnu95FCompiler.get_library_dirsc C s t | ¡}tjdkr | d¡ tjdkr| j}|rn|jdkrnd|krn| d¡}| |d d¡ | |d d¡ | j}|r|jdkrg S |S ) Nr? rk r
|
ri rj r ZmingwexZmingw32) 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| jdg t jt jd}| ¡ \}}|p0d|p6d }W n tt jfk
|
rV Y n"X t|}t |¡}|rx|
|
d¡S dS )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
|
ý
|
|
zGnu95FCompiler.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
|
|
zGnu95FCompiler._hash_filesc C s` | j }|jdkrtd| t|t| ¡}t r:d}nd}dtj tj |d ¡¡d dd
} | d | d
|
| }
|
|
|
d } |
|
d } |
|
d } tj
|
|| ¡}tj
|
|| ¡}tj
|
|| ¡}tj |¡rÐ||fS |rèdgt| dg }| j || |t|dd| ddddg d t r d}nd}d| d| |g}|j sF| ¡ | |jg| ¡ ||fS )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_dirZextra_postargsz /MACHINE:X64z /MACHINE:X86z/def:z/OUT:)rm rn r# rÀ Úlistr rD rV ÚsplitextÚbasenamera rl Zlink_shared_objectZ initializedZ
|
initializeÚspawnr_ )r$ ÚobjectsrÃ Ú extra_dll_dirÚ chained_dllsÚ
|
is_archiverm Z object_hashÚtagrÆ Z root_nameZdll_nameZdef_nameZlib_nameZdll_pathZdef_pathZlib_pathÚ specifierZlib_argsr r r Ú_link_wrapper_lib s`
|
ÿÿÿ ÿúüz Gnu95FCompiler._link_wrapper_libc C s
|
|j dkS )N)ri )rn )r$ Úcompilerr r r Úcan_ccompiler_linkÜ s z!Gnu95FCompiler.can_ccompiler_linkc C sÆ | j jdkrºg }g }|D ](}| ¡ d¡r6| |¡ q| |¡ qg }g }|ddd
D ]6} | j| g|||dd\}
|
} | d|
|
¡ | d| ¡ qX|s|S | j||||dd\}
|
} |
|
g| S td dS )
|
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 FzUnsupported C compiler)rm rn ÚlowerÚendswithrM rÎ r¤ r# ) r$ rÈ rà rÉ ZarchivesZ 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² | dd
}t jdd\}}t |¡ zztj |¡}tj |¡d d }z<| d|d|g¡ t
|
|t t |d}| ¡ |jdkW ¢W ¢ S tj |¡rt |¡ X W 5 t |¡ X dS ) 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} ZnewcmdZfidÚfilenamere rX r¬ r r r r s
|
r Ú__main__)Úlog)Úcustomized_fcompilerrA r Zg95)#r rD r rI r rÚ r± r¶ rR r r r Z numpy.distutils.exec_commandr Znumpy.distutils.fcompilerr Zdistutils.versionr Z compilersry r« r r r r r Ú distutilsrà Znumpy.distutilsrá Ú set_verbosityÚprintrs Ú ExceptionÚer r r r Ú<module> s<
|
q
|
|
|
