U
    @{?^2                  
   @   s  d dl Z d dl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 d dlmZ d dlmZ d dlmZ d d	lmZ ejrd dlZd
d e D Znd dlmZ zd dlmZ ed W n ek
r   eZY nX ed d dl
mZ dd Z dZ!dZ"dZ#ej$dkrdZ"n>e j%dkrTzd dl&Z&e'e&d Z"Z!W n ek
rR   Y nX dd Z(dd Z)G dd deZe"se j%dkrd"ddZ*nd Z#d#d!dZ*dS )$    N)	build_ext)	copy_file)new_compiler)customize_compilerget_config_var)DistutilsError)log)Library)sixc                 C   s    g | ]\}}}|t jkr|qS  )impC_EXTENSION).0s_tpr   r   >/usr/lib/python3/dist-packages/setuptools/command/build_ext.py
<listcomp>   s    
 r   EXTENSION_SUFFIXESzCython.Compiler.MainLDSHARED)_config_varsc              	   C   sZ   t jdkrNt }z$dtd< dtd< dtd< t|  W 5 t  t| X nt|  d S )Ndarwinz0gcc -Wl,-x -dynamiclib -undefined dynamic_lookupr   z -dynamiclibZCCSHAREDz.dylibZSO)sysplatform_CONFIG_VARScopyclearupdater   )compilerZtmpr   r   r   _customize_compiler_for_shlib$   s    
r    FZsharedr   TntRTLD_NOWc                 C   s   t r| S dS )N )	have_rtld)r   r   r   r   if_dlF   s    r%   c                  C   s.   t D ]$} d| kr|   S | dkr|   S qdS )z;Return the file extension for an abi3-compliant Extension()z.abi3z.pydNr   )suffixr   r   r   get_abi3_suffixJ   s
    r'   c                   @   sv   e 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d Zdd ZdddZdS )r   c                 C   s.   | j d }| _ t|  || _ |r*|   dS )z;Build extensions in build directory, then copy if --inplacer   N)Zinplace
_build_extruncopy_extensions_to_source)selfZold_inplacer   r   r   r)   T   s
    
zbuild_ext.runc           
      C   s   |  d}| jD ]}| |j}| |}|d}d|d d }||}tj	|tj	
|}tj	| j|}	t|	|| j| jd |jr| |ptj|d qd S )Nbuild_py.)verbosedry_runT)get_finalized_command
extensionsget_ext_fullnamenameget_ext_filenamesplitjoinZget_package_dirospathbasename	build_libr   r/   r0   _needs_stub
write_stubcurdir)
r+   r,   extfullnamefilenamemodpathpackageZpackage_dirZdest_filenameZsrc_filenamer   r   r   r*   \   s&    





  z#build_ext.copy_extensions_to_sourcec                 C   s   t | |}|| jkr| j| }tj o6t|do6t }|r`td}|d t|  }|t  }t	|t
rtj|\}}| j|tS tr|jrtj|\}}tj|d| S |S )NZpy_limited_apiZ
EXT_SUFFIXzdl-)r(   r5   ext_mapr
   PY2getattrr'   r   len
isinstancer	   r8   r9   splitextshlib_compilerlibrary_filenamelibtype	use_stubs_links_to_dynamicr6   r7   )r+   r@   rA   r?   Zuse_abi3Zso_extfndr   r   r   r5   r   s&    




zbuild_ext.get_ext_filenamec                 C   s    t |  d | _g | _i | _d S N)r(   initialize_optionsrJ   shlibsrD   r+   r   r   r   rR      s    
zbuild_ext.initialize_optionsc                 C   s,  t |  | jpg | _| | j dd | jD | _| jrB|   | jD ]}| |j|_qH| jD ]}|j}|| j	|< || j	|
dd < | jr| |pd}|otot|t }||_||_| | }|_tjtj| j|}|r||jkr|j| |rbtrbtj|jkrb|jtj qbd S )Nc                 S   s   g | ]}t |tr|qS r   )rH   r	   r   r?   r   r   r   r      s    
z.build_ext.finalize_options.<locals>.<listcomp>r-   r.   F)r(   finalize_optionsr2   Zcheck_extensions_listrS   setup_shlib_compilerr3   r4   
_full_namerD   r6   links_to_dynamicrM   rH   r	   rN   r<   r5   
_file_namer8   r9   dirnamer7   r;   library_dirsappendr>   runtime_library_dirs)r+   r?   r@   ZltdnsrA   Zlibdirr   r   r   rV      s,    



zbuild_ext.finalize_optionsc                 C   s   t | j| j| jd }| _t| | jd k	r8|| j | jd k	r^| jD ]\}}|	|| qH| j
d k	r~| j
D ]}|| qn| jd k	r|| j | jd k	r|| j | jd k	r|| j | jd k	r|| j t||_d S )N)r   r0   force)r   r   r0   r`   rJ   r    Zinclude_dirsZset_include_dirsZdefineZdefine_macroZundefZundefine_macro	librariesZset_librariesr\   Zset_library_dirsZrpathZset_runtime_library_dirsZlink_objectsZset_link_objectslink_shared_object__get__)r+   r   r4   valueZmacror   r   r   rW      s.      







zbuild_ext.setup_shlib_compilerc                 C   s   t |tr|jS t| |S rQ   )rH   r	   export_symbolsr(   get_export_symbolsr+   r?   r   r   r   rf      s    
zbuild_ext.get_export_symbolsc                 C   s\   |   | j}z@t|tr"| j| _t| | |jrL| dj	}| 
|| W 5 || _X d S )Nr,   )Z_convert_pyx_sources_to_langr   rH   r	   rJ   r(   build_extensionr<   r1   r;   r=   )r+   r?   Z	_compilercmdr   r   r   rh      s    
zbuild_ext.build_extensionc                    sP   t dd | jD  d|jddd dg t fdd|jD S )	z?Return true if 'ext' links to a dynamic lib in the same packagec                 S   s   g | ]
}|j qS r   )rX   )r   libr   r   r   r      s     z.build_ext.links_to_dynamic.<locals>.<listcomp>r-   Nr.   r#   c                 3   s   | ]}|  kV  qd S rQ   r   )r   ZlibnameZlibnamespkgr   r   	<genexpr>   s     z-build_ext.links_to_dynamic.<locals>.<genexpr>)dictfromkeysrS   r7   rX   r6   anyra   rg   r   rk   r   rY      s     zbuild_ext.links_to_dynamicc                 C   s   t | |   S rQ   )r(   get_outputs_build_ext__get_stubs_outputsrT   r   r   r   rq      s    zbuild_ext.get_outputsc                    s6    fdd j D }t|  }tdd |D S )Nc                 3   s0   | ](}|j rtjj jf|jd  V  qdS )r-   N)r<   r8   r9   r7   r;   rX   r6   rU   rT   r   r   rm      s   z0build_ext.__get_stubs_outputs.<locals>.<genexpr>c                 s   s   | ]\}}|| V  qd S rQ   r   )r   baseZfnextr   r   r   rm      s     )r2   	itertoolsproduct!_build_ext__get_output_extensionslist)r+   Zns_ext_basesZpairsr   rT   r   Z__get_stubs_outputs   s
    
zbuild_ext.__get_stubs_outputsc                 c   s"   dV  dV  |  djrdV  d S )N.pyz.pycr,   z.pyo)r1   optimizerT   r   r   r   Z__get_output_extensions   s    z!build_ext.__get_output_extensionsFc                 C   s,  t d|j| tjj|f|jd d }|rJtj|rJt|d | j	st
|d}|dddd	td
 dtj|j dddtddddtdddtddddg |  |r(ddlm} ||gdd| j	d | dj}|dkr||g|d| j	d tj|r(| j	s(t| d S )Nz writing stub loader for %s to %sr-   rx   z already exists! Please delete.w
zdef __bootstrap__():z-   global __bootstrap__, __file__, __loader__z%   import sys, os, pkg_resources, impz, dlz:   __file__ = pkg_resources.resource_filename(__name__,%r)z   del __bootstrap__z    if '__loader__' in globals():z       del __loader__z#   old_flags = sys.getdlopenflags()z   old_dir = os.getcwd()z   try:z(     os.chdir(os.path.dirname(__file__))z$     sys.setdlopenflags(dl.RTLD_NOW)z(     imp.load_dynamic(__name__,__file__)z   finally:z"     sys.setdlopenflags(old_flags)z     os.chdir(old_dir)z__bootstrap__()r#   r   )byte_compileT)ry   r`   r0   Zinstall_lib)r   inforX   r8   r9   r7   r6   existsr   r0   openwriter%   r:   rZ   closeZdistutils.utilr|   r1   ry   unlink)r+   
output_dirr?   compileZ	stub_filefr|   ry   r   r   r   r=      sb    


 
 zbuild_ext.write_stubN)F)__name__
__module____qualname__r)   r*   r5   rR   rV   rW   rf   rh   rY   rq   rr   rv   r=   r   r   r   r   r   S   s   	r   c                 C   s(   |  | j|||||||||	|
|| d S rQ   )linkZSHARED_LIBRARY)r+   objectsoutput_libnamer   ra   r\   r^   re   debugextra_preargsextra_postargs
build_temptarget_langr   r   r   rb   &  s             rb   Zstaticc                 C   s^   |d kst tj|\}}tj|\}}| ddrH|dd  }| ||||| d S )Nxrj      )AssertionErrorr8   r9   r6   rI   rK   
startswithZcreate_static_lib)r+   r   r   r   ra   r\   r^   re   r   r   r   r   r   rA   r:   r?   r   r   r   rb   5  s        )
NNNNNr   NNNN)
NNNNNr   NNNN)+r8   r   rt   Zdistutils.command.build_extr   Z_du_build_extZdistutils.file_utilr   Zdistutils.ccompilerr   Zdistutils.sysconfigr   r   Zdistutils.errorsr   Z	distutilsr   Zsetuptools.extensionr	   Zsetuptools.externr
   rE   r   Zget_suffixesr   importlib.machineryZCython.Distutils.build_extr(   
__import__ImportErrorr   r   r    r$   rM   rL   r   r4   Zdlhasattrr%   r'   rb   r   r   r   r   <module>   s~   
	 Q                                