o
    à8VaP  ã                   @   s´  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 d dl	m
Z
mZ ddlmZmZmZ ddlmZmZmZmZmZmZmZmZmZmZ e
dƒZejdkrXd	Znejd
kr`dZne d ej¡¡ d	Z		d/dd„Z d0dd„Z!		d1dd„Z"		d2dd„Z#d3dd„Z$edfedfedfedfedfedfedfedfedfedfdœ
Z%d4dd„Z&		d5dd„Z'd d!„ Z(d"d#„ Z)d$d%„ Z*		d6d'd(„Z+d)d*„ Z,d7d+d,„Z-d8d-d.„Z.dS )9é    N)ÚCompileError)Úget_config_varÚget_config_varsé   )ÚCCompilerRunnerÚCppCompilerRunnerÚFortranCompilerRunner)
Úget_abspathÚ	make_dirsÚcopyÚGlobÚArbitraryDepthGlobÚglob_at_depthÚimport_module_from_fileÚpyx_is_cplusÚsha256_of_stringÚsha256_of_fileZ
EXT_SUFFIXÚposixz.oÚntz.objzUnknown os.name: {}Fc                 K   sD  i }|dur9|  ¡ D ].\}}	t|tƒr!t |j¡D ]}
|	||
< qq
t|tƒr4t|j|ƒD ]}
|	||
< q,q
|	||< q
|p<d}tj	 
|¡sTtj	 |¡rPtd |¡ƒ‚t|ƒ |du rgd}| D ]
}t||ddd q\g }| D ]4}|rxtj	 |¡\}}ntj	 tj	 |¡¡\}}| ¡ }| | |i ¡¡ | t||fd|i|¤Ž¡ qk|S )aL   Compile source code files to object files.

    Parameters
    ==========

    files : iterable of str
        Paths to source files, if ``cwd`` is given, the paths are taken as relative.
    Runner: CompilerRunner subclass (optional)
        Could be e.g. ``FortranCompilerRunner``. Will be inferred from filename
        extensions if missing.
    destdir: str
        Output directory, if cwd is given, the path is taken as relative.
    cwd: str
        Working directory. Specify to have compiler run in other directory.
        also used as root of relative paths.
    keep_dir_struct: bool
        Reproduce directory structure in `destdir`. default: ``False``
    per_file_kwargs: dict
        Dict mapping instances in ``files`` to keyword arguments.
    \*\*kwargs: dict
        Default keyword arguments to pass to ``Runner``.

    NÚ.z{} is not a directoryT)Zonly_updateZdest_is_dirÚcwd)ÚitemsÚ
isinstancer   ÚglobÚpathnamer   r   ÚfilenameÚosÚpathÚisdirÚexistsÚOSErrorÚformatr
   r   ÚsplitextÚbasenameÚupdateÚgetÚappendÚsrc2obj)ÚfilesÚRunnerÚdestdirr   Zkeep_dir_structZper_file_kwargsÚkwargsZ_per_file_kwargsÚkÚvr   ÚfZdstpathsÚnameÚextZfile_kwargs© r1   úJ/usr/lib/python3/dist-packages/sympy/utilities/_compilation/compilation.pyÚcompile_sources!   s<   

ÿ

ÿ
r3   c                 C   sn   | pt j dd¡} |  ¡ dkr"|rtdddgi| fS tddgi| fS |  ¡ dks(|r2tdd	gi| fS ti | fS )
NZSYMPY_COMPILER_VENDORZgnuZintelÚflagsz-nofor_mainz-cxxlibZllvmZlib_optionsZfortran)r   Úenvironr%   Úlowerr   r   Ú
ValueError©ÚvendorÚcplusr   r1   r1   r2   Úget_mixed_fort_c_linker`   s"   ÿ
ÿ
ÿÿr;   c                 K   s   |du rt j t j | d ¡¡\}}|r|t7 }|sK|rDt| dd¡||d\}}	}
|	 ¡ D ]\}}||v r>||  |¡ q.|||< q.n|rIt	}nt
}| dg ¡}|r\d|vr\| d¡ | dd¡}|shtd	ƒ‚t||d
}|| ||fd|i|¤Ž}| ¡  |S )aK   Link object files.

    Parameters
    ==========

    obj_files: iterable of str
        Paths to object files.
    out_file: str (optional)
        Path to executable/shared library, if ``None`` it will be
        deduced from the last item in obj_files.
    shared: bool
        Generate a shared library?
    Runner: CompilerRunner subclass (optional)
        If not given the ``cplus`` and ``fort`` flags will be inspected
        (fallback is the C compiler).
    cwd: str
        Path to the root of relative paths and working directory for compiler.
    cplus: bool
        C++ objects? default: ``False``.
    fort: bool
        Fortran objects? default: ``False``.
    \*\*kwargs: dict
        Keyword arguments passed to ``Runner``.

    Returns
    =======

    The absolute path to the generated shared object / executable.

    Néÿÿÿÿr9   r8   r4   z-sharedÚ
run_linkerTz*run_linker was set to False (nonsensical).©r   r   )r   r   r"   r#   Ú	sharedextr;   r%   r   Úexpandr   r   Úpopr&   r7   r	   Úrun)Ú	obj_filesZout_fileÚsharedr)   r   r:   Úfortr+   r0   Zextra_kwargsr9   r,   r-   r4   r=   Úrunnerr1   r1   r2   Úlinku   s>    
ýÿ
ü
rG   c                 K   s  |pg }|  dg ¡}|  dg ¡}tjdkrt d¡ nKtjdkr!nEtjdd… dkr+n;d	d
lm}	 |	 d¡retƒ }
| 	dg ¡|
d g |d< ||
d g7 }|
d  
¡ D ]}| d¡rc||dd… g7 }qSn	 |  dg ¡}d}|D ]}||vr{| |¡ qpt| fd|||||||dœ|¤ŽS )a,   Link python extension module (shared object) for importing

    Parameters
    ==========

    obj_files: iterable of str
        Paths to object files to be linked.
    so_file: str
        Name (path) of shared object file to create. If not specified it will
        have the basname of the last object file in `obj_files` but with the
        extension '.so' (Unix).
    cwd: path string
        Root of relative paths and working directory of linker.
    libraries: iterable of strings
        Libraries to link against, e.g. ['m'].
    cplus: bool
        Any C++ objects? default: ``False``.
    fort: bool
        Any Fortran objects? default: ``False``.
    kwargs**: dict
        Keyword arguments passed to ``link(...)``.

    Returns
    =======

    Absolute path to the generate shared object.
    Úinclude_dirsÚlibrary_dirsÚwin32zWindows not yet supported.ÚdarwinNé   Zaixr   )Ú	sysconfigZPy_ENABLE_SHAREDZlinklineZ
PY_LDFLAGSZLIBDIRZ
BLDLIBRARYz-lé   r4   )ú-pthreadT)rD   r4   r   r:   rE   rH   Ú	librariesrI   )rA   ÚsysÚplatformÚwarningsÚwarnZ	distutilsrM   r   r   r%   ÚsplitÚ
startswithr&   rG   )rC   Zso_filer   rP   r:   rE   r+   rH   rI   rM   ZcfgDictZoptr4   Úneeded_flagsÚflagr1   r1   r2   Ú
link_py_so»   s@   



€þ
€þþrY   c              	   K   sZ  ddl m}m} ddl m} |  ¡  d¡s|  ¡  d¡sJ ‚|p!d}|p%d}| dd¡r.d	nd
}tj 	tj 
| ¡¡d | }tj ||¡}	|rLt ¡ }
nd}
t |¡ zS||ƒ}|j |¡ || g|ƒ}|jdkrmtdƒ‚tj tj | ¡¡tj |¡krŸtj |	¡rˆt |	¡ t tj tj | ¡|¡|¡ W t |
¡ |	S W t |
¡ |	S t |
¡ w )a¬   Generates a C file from a Cython source file.

    Parameters
    ==========

    src: str
        Path to Cython source.
    destdir: str (optional)
        Path to output directory (default: '.').
    cwd: path string (optional)
        Root of relative paths (default: '.').
    **cy_kwargs:
        Second argument passed to cy_compile. Generates a .cpp file if ``cplus=True`` in ``cy_kwargs``,
        else a .c file.
    r   )Údefault_optionsÚCompilationOptions)Úcompileú.pyxz.pyr   r:   Fú.cppú.czCython compilation failed.)ZCython.Compiler.MainrZ   r[   r\   r6   Úendswithr%   r   r   r"   r#   ÚjoinÚgetcwdÚchdirÚ__dict__r$   Z
num_errorsr7   ÚabspathÚdirnamer   ÚunlinkÚshutilÚmove)Úsrcr*   r   Ú	cy_kwargsrZ   r[   Z
cy_compiler0   Zc_nameZdstfileZori_dirZ
cy_optionsZ	cy_resultr1   r1   r2   Úsimple_cythonizeý   s8    


 
 
ú
ÿrl   Zf95Zf2003Zf2008)
r_   r^   z.cxxz.fz.forz.ftnz.f90z.f95z.f03z.f08c                 K   sR  t j t j | ¡¡\}}|du r#t j | ¡rd}n
t j | ¡}|p"d}t j |¡r2t j ||t ¡}| 	dg ¡}|rLddl
m}	 |	ƒ }
|
|vrL| |
¡ | ¡ dkr^t| f|||dœ|¤ŽS |du rrt| ¡  \}}d|vrr||d< | 	d	g ¡}d
}|D ]}||vr‡| |¡ q|| 	dd¡}|r”tdƒ‚|| g|f||||dœ|¤Ž}| ¡  |S )a¸   Compiles a source code file to an object file.

    Files ending with '.pyx' assumed to be cython files and
    are dispatched to pyx2obj.

    Parameters
    ==========

    srcpath: str
        Path to source file.
    Runner: CompilerRunner subclass (optional)
        If ``None``: deduced from extension of srcpath.
    objpath : str (optional)
        Path to generated object. If ``None``: deduced from ``srcpath``.
    cwd: str (optional)
        Working directory and root of relative paths. If ``None``: current dir.
    inc_py: bool
        Add Python include path to kwarg "include_dirs". Default: False
    \*\*kwargs: dict
        keyword arguments passed to Runner or pyx2obj

    Nr   rH   r   )Úget_python_incr]   )ÚobjpathrH   r   Ústdr4   )ú-fPICr=   Fz#src2obj called with run_linker=True)rH   r=   r   r4   )r   r   r"   r#   Úisabsrf   r   ra   ÚobjextrA   Údistutils.sysconfigrm   r&   r6   Úpyx2objÚextension_mappingr   rB   )Zsrcpathr)   rn   r   Úinc_pyr+   r/   r0   rH   rm   Z
py_inc_dirro   r4   rW   rX   r=   rF   r1   r1   r2   r'   =  sL   
ÿ
€ÿÿr'   c                 K   sF  |   d¡sJ ‚|p
d}|pd}|ptj |¡}t||d}tj |¡r:tj | ¡}	tj |	¡\}
}tj ||
t	 ¡}|p=i }||d< |du rJt
| ƒ}||d< t| f||dœ|¤Ž}|p\g }| dg ¡}d	}|D ]}||vrr| |¡ qg| d
g ¡}| dd¡rƒtdƒ‚|rŒ| dd¡}n| dd¡}t|f||||||dddœ|¤ŽS )a9  
    Convenience function

    If cwd is specified, pyxpath and dst are taken to be relative
    If only_update is set to `True` the modification time is checked
    and compilation is only run if the source is newer than the
    destination

    Parameters
    ==========

    pyxpath: str
        Path to Cython source file.
    objpath: str (optional)
        Path to object file to generate.
    destdir: str (optional)
        Directory to put generated C file. When ``None``: directory of ``objpath``.
    cwd: str (optional)
        Working directory and root of relative paths.
    include_dirs: iterable of path strings (optional)
        Passed onto src2obj and via cy_kwargs['include_path']
        to simple_cythonize.
    cy_kwargs: dict (optional)
        Keyword arguments passed onto `simple_cythonize`
    cplus: bool (optional)
        Indicate whether C++ is used. default: auto-detect using ``.util.pyx_is_cplus``.
    compile_kwargs: dict
        keyword arguments passed onto src2obj

    Returns
    =======

    Absolute path of generated object file.

    r]   r   r>   Z
output_dirNr:   ©r*   r   r4   )z-fwrapvrO   rp   ÚoptionsÚstrict_aliasingFz/Cython requires strict aliasing to be disabled.ro   zc++98Zc99T)rn   r   rH   r4   ro   rx   rv   ry   )r`   r   r   rf   r	   r   r#   r"   ra   rr   r   rl   rA   r&   r   r'   )Zpyxpathrn   r*   r   rH   rk   r:   r+   Zabs_objpathZ	pyx_fnamer/   r0   Zinterm_c_filer4   rW   rX   rx   ro   r1   r1   r2   rt   €  sD   %
€
þýrt   c                 C   sD   | D ]}t j |¡\}}| ¡ }|tv rt| d |kr dS qdS )Nr   TF)r   r   r"   r6   ru   )ÚsrcsÚclsrj   r/   r0   Úkeyr1   r1   r2   Ú_any_XÑ  s   €r}   c                 C   ó
   t | tƒS ©N)r}   r   ©rz   r1   r1   r2   Úany_fortran_srcÛ  ó   
r   c                 C   r~   r   )r}   r   r€   r1   r1   r2   Úany_cplus_srcß  r‚   rƒ   r   c              	   C   s¬   |du rt j t j | d ¡¡d }|pi }|pi }ztt j ||¡| ƒ}W |S  tyU   ttt	t
| ƒƒf||dœ|¤Ž}t|f|t| ƒt| ƒdœ|¤Ž}t|ƒ}Y |S w )a   Compiles sources to a shared object (python extension) and imports it

    Sources in ``sources`` which is imported. If shared object is newer than the sources, they
    are not recompiled but instead it is imported.

    Parameters
    ==========

    sources : string
        List of paths to sources.
    extname : string
        Name of extension (default: ``None``).
        If ``None``: taken from the last file in ``sources`` without extension.
    build_dir: str
        Path to directory in which objects files etc. are generated.
    compile_kwargs: dict
        keyword arguments passed to ``compile_sources``
    link_kwargs: dict
        keyword arguments passed to ``link_py_so``

    Returns
    =======

    The imported module from of the python extension.
    Nr<   r   rw   ©r   rE   r:   )r   r   r"   r#   r   ra   ÚImportErrorr3   ÚlistÚmapr	   rY   r   rƒ   )ÚsourcesZextnameÚ	build_dirÚcompile_kwargsÚlink_kwargsÚmodÚobjsZsor1   r1   r2   Úcompile_link_import_py_extã  s*   úÿÿÿÿ
úrŽ   c           
   	   C   sø   |pt  ¡ }tj |¡std|ƒ‚g }| D ]b\}}tj ||¡}d}t| d¡ƒ 	¡ }tj 
|¡rLtj 
|d ¡rBt|d ƒ ¡ }nt|ƒ 	¡ }||k}|rrt|dƒ}	|	 |¡ t|d dƒ |¡ W d   ƒ n1 smw   Y  | |¡ q||fS )NzNon-existent directory: Túutf-8z.sha256Zwt)ÚtempfileZmkdtempr   r   r   r    ra   r   ÚencodeZ	hexdigestr   ÚopenÚreadr   Úwriter&   )
rˆ   r‰   Úsource_filesr/   rj   ÚdestZdiffersZsha256_in_memZsha256_on_diskZfhr1   r1   r2   Ú_write_sources_to_build_dir  s(   

þr—   c                 K   s4   t | |ƒ\}}t|fd|i|¤Ž}t|d}||fS )aå   Compiles, links and imports extension module from source.

    Parameters
    ==========

    sources : iterable of name/source pair tuples
    build_dir : string (default: None)
        Path. ``None`` implies use a temporary directory.
    **kwargs:
        Keyword arguments passed onto `compile_link_import_py_ext`.

    Returns
    =======

    mod : module
        The compiled and imported extension module.
    info : dict
        Containing ``build_dir`` as 'build_dir'.

    r‰   )r‰   )r—   rŽ   Údict)rˆ   r‰   r+   r•   rŒ   Úinfor1   r1   r2   Úcompile_link_import_strings(  s   
rš   c              	   C   sú   |r
|dur
t dƒ‚zVt| |ƒ\}}tttt|ƒƒf||dœ|p!i ¤Ž}t|f|t|ƒt|ƒdœ|p3i ¤Ž}t	j
|gt	jt	jd}| ¡ }	dd„ | ¡ D ƒ\}
}W |r`tj |¡r`t |¡ d}n|rptj |¡rpt |¡ d}w t|	|d}|
|f|fS )	ay   Compiles, links and runs a program built from sources.

    Parameters
    ==========

    sources : iterable of name/source pair tuples
    build_dir : string (default: None)
        Path. ``None`` implies use a temporary directory.
    clean : bool
        Whether to remove build_dir after use. This will only have an
        effect if ``build_dir`` is ``None`` (which creates a temporary directory).
        Passing ``clean == True`` and ``build_dir != None`` raises a ``ValueError``.
        This will also set ``build_dir`` in returned info dictionary to ``None``.
    compile_kwargs: dict
        Keyword arguments passed onto ``compile_sources``
    link_kwargs: dict
        Keyword arguments passed onto ``link``

    Returns
    =======

    (stdout, stderr): pair of strings
    info: dict
        Containing exit status as 'exit_status' and ``build_dir`` as 'build_dir'

    NzIAutomatic removal of build_dir is only available for temporary directory.rw   r„   )ÚstdoutÚstderrc                 S   s   g | ]}|  d ¡‘qS )r   )Údecode)Ú.0Ztxtr1   r1   r2   Ú
<listcomp>i  s    z'compile_run_strings.<locals>.<listcomp>)Úexit_statusr‰   )r7   r—   r3   r†   r‡   r	   rG   r   rƒ   Ú
subprocessÚPopenÚPIPEÚwaitZcommunicater   r   r   rh   Úrmtreer˜   )rˆ   r‰   ZcleanrŠ   r‹   r•   r   ÚprogÚpr    r›   rœ   r™   r1   r1   r2   Úcompile_run_stringsC  s6   ÿÿþþ
€þ
r¨   )NNNFN)NFN)NFNNFF)NNNFF)NN)NNNF)NNNNNN)Nr   NNr   )NFNN)/r   r   rh   r¡   rQ   r   rS   Zdistutils.errorsr   rs   r   r   Zrunnersr   r   r   Úutilr	   r
   r   r   r   r   r   r   r   r   r?   r/   rr   rT   r!   r3   r;   rG   rY   rl   ru   r'   rt   r}   r   rƒ   rŽ   r—   rš   r¨   r1   r1   r1   r2   Ú<module>   sj    0


ÿ
?
ÿF
ÿ
B3ö
C
ÿQ

ÿ,
