3
���h�O � @ sT 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 jd�Zd d
� Ze� r�g Zng ZG d d� de�ZG d d� de�Zd
d� Zedk�rPd dlm Z d dl m Z e j d� e e d�j!� � ye e d�j!� � W n, e"k
�rN Z# ze e#� W Y ddZ#[#X nX dS )� N)�Popen�PIPE�STDOUT)� filepath_from_subprocess_output)� FCompiler)� LooseVersion� GnuFCompiler�Gnu95FCompilerzTarget: ([a-zA-Z0-9_\-]*)c C s t jdkotj� d dkS )N�win32r �64bit)�sys�platform� architecture� r r �>/tmp/pip-build-5_djhm0z/numpy/numpy/distutils/fcompiler/gnu.py�is_win64 s r c
@ s e Zd ZdZd1ZdZdd� Zdd� ZddgZd d
gd d d d
gd d d d d gddgdgd d d gd�Z d Z
d Z e j
dkr�ejdkr�dgZejdkr�xd2D ]Ze e jd� q�W dZd Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Z d-d.� Z d/d0� Z d S )3r �gnu�g77zGNU Fortran 77 compilerc C s� x$|j d�r$||jd�d d� }qW t|�dkrvtjd|�}|r�|j d�rZd|jd�fS |j� d kr�d
|jd�fS nftjd |�}|r�d
|jd�fS tjd |�}|r�|jd�}|j d
�s�|j d�s�|j 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�len�re�search�group�start�
ValueError)�self�version_string�m�v�errr r r �gnu_version_match& s*
z GnuFCompiler.gnu_version_matchc C s( | j |�}| s |d dkr d S |d S )Nr r r )r( )r# r$ r& r r r �
version_matchR s
zGnuFCompiler.version_matchZf77Nz -dumpversionz-gz-Wallz-fno-second-underscore�arz-cr�ranlib)� version_cmd� compiler_f77� compiler_f90� compiler_fix� linker_so�archiverr+ �
linker_exe�nt�cygwinz-fPICr
r, r- r0 r2 z -mno-cygwin�g2c�gnu95c C s� | j dd � }tjdkrxtjjdd �}|sfdd l}|jd�}|s\d}d|� �}tj |dd� |tjd< |j
d d
d g� n
|j d � tjj d
�r�|j d� |S )Nr �darwin�MACOSX_DEPLOYMENT_TARGETr z10.9z.Env. variable MACOSX_DEPLOYMENT_TARGET set to � )�
stacklevelz
-undefinedZdynamic_lookupz-bundlez-shared�sunosz
-mimpure-text)
r0 r r
�os�environ�get� sysconfig�get_config_var�warnings�warn�extend�appendr )r# �opt�targetr? �sr r r �get_flags_linker_sos s
z GnuFCompiler.get_flags_linker_soc C sJ yt j| jdg �}W n tt jfk
r0 Y nX t|�}tjj|�S d S )Nz-print-libgcc-file-name) �
subprocess� check_outputr- �OSError�CalledProcessErrorr r< �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 }|r�dj|�g}ytj| j| �}W n ttjfk
rn Y nX t|�}t j
j |�}|S )N� �linuxzlibgfortran.sor7 zlibgfortran.dylibz-print-file-name={0}) r r
�formatrI rJ r- rK rL r r< rM rN )r# Zlibgfortran_nameZlibgfortran_dirZ find_lib_argrO r r r �get_libgfortran_dir� s
z GnuFCompiler.get_libgfortran_dirc C s� g }t jd d� dkr�| j� }|r�t jdkr�|jd� r�tjj|�}tjj|d| j �}tjj |�s�tjj|ftj
fd �� }tjj tjj|d��}tjj|d| j �}tjj |�r�|j |� |j |� | j
� }|r�|j |� |S )NrQ rR r
z/usr/libzlib%s.a� �lib)r r
rP r r< rM �normpath�joinr5 �exists�pardir�abspathrD rT )r# rE �drM �rootZd2�lib_gfortran_dirr r r �get_library_dirs� s$
z GnuFCompiler.get_library_dirsc C s� g }| j � }|d k rL| jd }| j|| jf }tjjtjj||��sR| j}n| j}|d k rd|j|� | j }t
j dkr�|r�|j dkr�|jd� t
j dkr�|jd� |S )Nz-picr
�msvc�gccr7 �
cc_dynamic)
rP r5 Zstatic_lib_formatZstatic_lib_extensionr< rM �isfilerX rD �
c_compilerr r
�
compiler_type)r# rE r\ r5 �frd r r r �
get_libraries� s"
zGnuFCompiler.get_librariesc C s dgS )Nz-gr )r# r r r �get_flags_debug� s z GnuFCompiler.get_flags_debugc C s0 | j � }|r |dkr dg}ndg}|jd� |S )Nz3.3.3z-O2z-O3z-funroll-loops)� get_versionrD )r# r& rE r r r �
get_flags_opt� s
zGnuFCompiler.get_flags_optc C sd ddl m} y|j� d }W n tk
r0 g S X tjd�}g }x |j|�D ]}|d|g7 }qLW |S )z( Return detected arch flags from CFLAGS r )r? �CFLAGSz
-arch\s+(\w+)z-arch)� distutilsr? �get_config_vars�KeyErrorr �compile�findall)r# r? �cflagsZarch_re�
arch_flags�archr r r �
_c_arch_flags� s
zGnuFCompiler._c_arch_flagsc C s g S )Nr )r# r r r �get_flags_arch� s zGnuFCompiler.get_flags_archc C sX t jdkrt�d|kst�t jdkr.d|� �S t jd d� dkrJd|� �S d|� �S d S ) Nr
�,r7 z -Wl,-rpath,� �aixz-Wl,-blibpath:z -Wl,-rpath=)r r
�NotImplementedError�AssertionError)r# �dirr r r �runtime_library_dir_option� s
z'GnuFCompiler.runtime_library_dir_option)r )r, r- r0 r2 ) �__name__�
__module__� __qualname__re �compiler_aliases� descriptionr( r) �possible_executables� executables�module_dir_switch�module_include_switchr< �namer r
Z pic_flags�keyrD r5 Zsuggested_f90_compilerrH rP rT r_ rg rh rj rt ru r| r r r r r ! sB ,
#
c @ s* e Zd ZdZd4ZdZdd� ZddgZddgd d
d d d
ge d d
d d
ge d d
d d d
ge dd
d gddgdgd d
gd�Z dZ
dZ e j
d d� dkr�e d jd� e
j� d d d� dkr�xd5D ]Ze e jd � q�W dZd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Zd S )6r r6 r zGNU Fortran 95 compilerc C sb | j |�}| s |d dkr d S |d }t|�dkr6n(tjdkr^x d
D ]}| j| jd � qFW |S )Nr r r �4r
r, r- r. r/ r0 r2 z -mno-cygwin)r, r- r. r/ r0 r2 )r( r r r
r� rD )r# r$ r& r� r r r r) s
z Gnu95FCompiler.version_matchZf95z<F90>z -dumpversionNz-Wallz-gz -ffixed-formz-fno-second-underscorer* z-crr+ )r, r- r. r/ r0 r1 r+ r2 z-Jz-Irw rx r0 z -lpthreadr r9 Z64r- r. r/ r2 z-maix64c C sb t jdksg S g }| j� }d|kr0d||jd�< x,dD ]$}t||�r6||kr6|jd|g� q6W |S ) z>Return a list of -arch flags for every supported architecture.r7 �i386�i686�ppc�x86_64�ppc64z-arch)r� r� r� r� )r r
rt �index� _can_targetrC )r# �cmdrr Zc_archsrs r r r �_universal_flagsB s
z Gnu95FCompiler._universal_flagsc C s* t j| �}| j| j�}|r&||d d�< |S )Nr )r � get_flagsr� r. )r# �flagsrr r r r r� R s
zGnu95FCompiler.get_flagsc C s* t j| �}| j| j�}|r&||d d�< |S )Nr )r rH r� r0 )r# r� rr r r r rH Y s
z"Gnu95FCompiler.get_flags_linker_soc C s� t j| �}tjdkr�| j}|r�|jdkr�| j� }|r�tjj | j
� �}tjj |ftj fd �� }tjj |d�}tjj |�}tjj
tjj |d��r�|j|� | j� }|r�|j|� |S )Nr
r` rU rV z libmingwex.a)r r_ r r
rd re �
get_targetr< rM rW rP rX rZ rY rD rT ) r# rE rd rF r\ r] rM Zmingwdirr^ r r r r_ ` s
z Gnu95FCompiler.get_library_dirsc C s� t j| �}tjdkr |jd� tjdkr�| j}|rn|jdkrnd|krn|jd�}|j|d d� |j|d d� | j}|r�|jdkr�g S |S ) Nr7 rb r
r` ra r ZmingwexZmingw32) r rg r r
�removerd re r� �insert)r# rE rd �ir r r rg s s
z Gnu95FCompiler.get_librariesc C sV yt j| jdg �}W n tt jfk
r0 Y n"X t|�}tj|�}|rR|jd�S dS )Nz-vr � ) rI rJ r- rK rL r �TARGET_Rr r )r# rO r% r r r r� � s
zGnu95FCompiler.get_targetc
C sn t j� }xB|D ]:}t|d��&}x |jd�}|s0P |j|� q W W d Q R X qW tj|j� �}|jd�}|j d�S )N�rbi �ascii�=)
�hashlib�sha1�open�read�update�base64� b32encode�digest�decode�rstrip)r# � filenames�h�fnrf �block�textr r r � _hash_files� s
zGnu95FCompiler._hash_filesc C s` | j }|jdkrtd��| jt|�t|� �}t� r:d}nd}dtjjtjj |d ��d dd� } | d | d
| }
|
d } |
d } |
d
}
tjj
|| �}tjj
|| �}tjj
||
�}tjj |�r�||fS |r�dgt|� dg }| j || |t|�dd| ddddg d� t� �r d}nd}d| d| |g}|j
�sF|j� |j|jg| � ||fS )zeCreate a wrapper shared library for the given objects
Return an MSVC-compatible lib
r` z This method only supports MSVCZ win_amd64r
rV 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:)rd re r" r� �listr r< rM �splitext�basenamerX rc Zlink_shared_objectZ initializedZ
initialize�spawnrV )r# �objectsr� �
extra_dll_dir� chained_dlls�
is_archiverd Z object_hash�tagr� Z root_nameZdll_nameZdef_nameZlib_nameZdll_pathZdef_pathZlib_path� specifierZlib_argsr r r �_link_wrapper_lib� sP
z Gnu95FCompiler._link_wrapper_libc C s
|j dkS )Nr` )r` )re )r# �compilerr r r �can_ccompiler_link� s z!Gnu95FCompiler.can_ccompiler_linkc C s� | j jdkr�g }g }x0|D ](}|j� jd�r8|j|� q|j|� qW g }g }xH|ddd
� D ]6} | j| g|||dd�\}
} |jd|
� |jd| � q^W |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.
r` z.aNr T)r� r� r FzUnsupported C compiler���)rd re �lower�endswithrD r� r� r" ) r# r� r� r� ZarchivesZ
plain_objects�objZ chained_libsr� �archiverV �dllr r r �wrap_unlinkable_objects� s8
z&Gnu95FCompiler.wrap_unlinkable_objects)r )r- r. r/ r0 r2 ) r} r~ r re r� r� r) r� � _EXTRAFLAGSr� r� r� r r
rD r r� r5 r� r� rH r_ rg r� r� r� r� r� r r r r r sD
>c C s� | dd� }t jdd�\}}tj|� zrtjj|�}tjj|�d d }z4|jd|d|g� t|t t
|d�}|j � |j dkS tjj
|�r�tj|� X W dtj|� X d S )
z7Return true if the architecture supports the -arch flagNz.f)�suffixr z.oz-archz-c)�stderr�stdout�cwdF)�tempfile�mkstempr< �closerM rN r� rC r r r � communicate�
returncoderY r� )r� rs ZnewcmdZfid�filenamer\ rO �pr r r r� s
r� �__main__)�log)�customized_fcompilerr9 r Zg95)$r r< r rA r
r� r� r� rI r r r Z numpy.distutils.exec_commandr Znumpy.distutils.fcompilerr Zdistutils.versionr Z compilersro r� r r� r r r� r} rl r� Znumpy.distutilsr� �
set_verbosity�printri � Exception�er r r r �<module> sB
p
|