a
    lº”hŒV  ã                   @   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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mZ ejdkr°dZn&ejd	krÀd
Zne d  ej¡¡ dZd.dd„Z!d/dd„Z"d0dd„Z#d1dd„Z$d2d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&d3dd„Z'd4dd„Z(dd „ Z)d!d"„ Z*d#d$„ Z+d5d&d'„Z,d(d)„ Z-d6d*d+„Z.d7d,d-„Z/dS )8é    N)ÚPath)Úget_config_varÚget_config_varsÚget_pathé   )Ú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_fileÚCompileErrorÚposixz.oÚntz.objzUnknown os.name: {}Fc                 K   sF  i }|durr|  ¡ D ]\\}}	t|tƒrBt |j¡D ]}
|	||
< q2qt|tƒrht|j|ƒD ]}
|	||
< qXq|	||< q|pxd}tj	 
|¡sªtj	 |¡r¢td |¡ƒ‚nt|ƒ |du rÐd}| D ]}t||ddd qºg }| D ]h}|ròtj	 |¡\}}ntj	 tj	 |¡¡\}}| ¡ }| | |i ¡¡ | t||fd|i|¤Ž¡ qØ|S )a‘   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``.

    Returns
    =======
    List of strings (paths of object files).
    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© r3   úV/var/www/auris/lib/python3.9/site-packages/sympy/utilities/_compilation/compilation.pyÚcompile_sources   s8    


r5   c                 C   s~   | pt j dd¡} |  ¡ dkrF|r4tdddgi| fS tddgi| fS n4|  ¡ dksR|rftdd	gi| fS ti | fS ntd
ƒ‚d S )NZSYMPY_COMPILER_VENDORZgnuÚintelÚflagsz-nofor_mainz-cxxlibZllvmZlib_optionsZfortranzNo vendor found.)r   Úenvironr'   Úlowerr	   r   Ú
ValueError©ÚvendorÚcplusr   r3   r3   r4   Úget_mixed_fort_c_linkera   s$    ÿ
ÿ
ÿÿr>   c                 K   s  |du r4t j t j | d ¡¡\}}	|r4|tdƒ7 }|sš|rŒt| dd¡||d\}}
}|
 ¡ D ](\}}||v r€||  |¡ q`|||< q`n|r–t	}nt
}| dg ¡}|r¼d|vr¼| d¡ | dd	¡}|sÔtd
ƒ‚t||d}|| |pêg  ||fd|i|¤Ž}| ¡  |S )a    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``.
    extra_objs: list
        List of paths to extra object files / static libraries.
    \*\*kwargs: dict
        Keyword arguments passed to ``Runner``.

    Returns
    =======

    The absolute path to the generated shared object / executable.

    NéÿÿÿÿÚ
EXT_SUFFIXr<   r;   r7   z-sharedÚ
run_linkerTz*run_linker was set to False (nonsensical).©r   r   )r   r   r$   r%   r   r>   r'   r   Úexpandr   r   Úpopr(   r:   r
   Úrun)Ú	obj_filesÚout_fileÚsharedr+   r   r=   ÚfortÚ
extra_objsr-   r2   Úextra_kwargsr<   r.   r/   r7   rA   Úrunnerr3   r3   r4   Úlinkv   s<    "
ýÿ
 rM   c                 K   sà  |pg }|  dg ¡}|  dg ¡}	tjdkr8t d¡ nTtjdkr
tƒ }
| dg ¡|
d g |d< |	|
d g7 }	d	}|
d
  ¡ D ]R}|r¨| dg ¡d|g |d< d	}q‚| d¡rÆ| 	|dd… ¡ q‚| d¡r‚d}q‚|
d }d 
| d¡dd… ¡dd… }| 	|¡ n‚tjdd… dkr nltdƒrŒtƒ }
| dg ¡|
d g |d< |	|
d g7 }	|
d  ¡ D ]$}| d¡rd||dd… g7 }qdn |  dg ¡}d}|D ]}||vr | 	|¡ q t| 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``.
    extra_objs: list
        List of paths of extra object files / static libraries to link against.
    kwargs**: dict
        Keyword arguments passed to ``link(...)``.

    Returns
    =======

    Absolute path to the generate shared object.
    Úinclude_dirsÚlibrary_dirsÚwin32zWindows not yet supported.ÚdarwinZlinklineÚLDFLAGSÚLIBDIRFÚLIBSz
-frameworkz-lé   NTÚLIBRARYr   r?   é   ÚaixÚPy_ENABLE_SHAREDÚ
BLDLIBRARYr7   )ú-pthread)	rH   r7   r   r=   rI   rN   Ú	librariesrO   rJ   )rD   ÚsysÚplatformÚwarningsÚwarnr   r'   ÚsplitÚ
startswithr(   Újoinr   rM   )rF   Zso_filer   r\   r=   rI   rJ   r-   rN   rO   ZcfgDictZis_frameworkÚoptZlibfileZlibnamer7   Úneeded_flagsÚflagr3   r3   r4   Ú
link_py_so¾   sR    


 

þþrg   c              	   K   sb  ddl m}m} ddl m} |  ¡  d¡s<|  ¡  d¡s<J ‚|pBd}|pJd}| dd¡r\d	nd
}tj 	tj 
| ¡¡d | }tj ||¡}	|r˜t ¡ }
nd}
t |¡ zª||ƒ}|j |¡ d|vrÎd|jd< || g|ƒ}|jdkrìtdƒ‚tj tj | ¡¡tj |¡krDtj |	¡r&t |	¡ t tj tj | ¡|¡|¡ W t |
¡ nt |
¡ 0 |	S )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language_levelrW   zCython compilation failed.)ZCython.Compiler.Mainrh   ri   rj   r9   Úendswithr'   r   r   r$   r%   rc   ÚgetcwdÚchdirÚ__dict__r&   Z
num_errorsr:   ÚrealpathÚdirnamer!   ÚunlinkÚshutilÚmove)Úsrcr,   r   Ú	cy_kwargsrh   ri   Z
cy_compiler2   Zc_nameZdstfileZori_dirZ
cy_optionsZ	cy_resultr3   r3   r4   Úsimple_cythonize  s4     



"
 ry   Zf95Zf2003Zf2008)
rm   rl   z.cxxz.fz.forz.ftnz.f90z.f95z.f03z.f08c                 K   sJ  t j t j | ¡¡\}}|du rFt j | ¡r2d}nt j | ¡}|pDd}t j |¡rdt j ||t ¡}| 	dg ¡}|rŽt
dƒ}	|	|vrŽ| |	¡ | ¡ dkr²t| f|||dœ|¤ŽS |du rÚt| ¡  \}}
d|vrÚ|
|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   rN   Úincluderk   )ÚobjpathrN   r   Ústdr7   )ú-fPICrA   Fz#src2obj called with run_linker=True)rN   rA   r   r7   )r   r   r$   r%   Úisabsrs   r    rc   ÚobjextrD   r   r(   r9   Úpyx2objÚextension_mappingr   rE   )Zsrcpathr+   r{   r   Úinc_pyr-   r1   r2   rN   Z
py_inc_dirr|   r7   re   rf   rA   rL   r3   r3   r4   r)   ]  sH    
ÿÿÿr)   c                 K   sJ  |   d¡sJ ‚|pd}|pd}|p,tj |¡}t||d}tj |¡rttj | ¡}	tj |	¡\}
}tj ||
t	 ¡}|pzi }||d< |du r”t
| ƒ}||d< t| f||dœ|¤Ž}|p¸g }| dg ¡}d	}|D ]}||vrÎ| |¡ qÎ| 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.

    rk   r   rB   Ú
output_dirNr=   ©r,   r   r7   )z-fwrapvr[   r}   ÚoptionsÚstrict_aliasingFz/Cython requires strict aliasing to be disabled.r|   zc++98Zc99T)r{   r   rN   r7   r|   r…   r‚   r†   )rn   r   r   rs   r
   r    r%   r$   rc   r   r   ry   rD   r(   r   r)   )Zpyxpathr{   r,   r   rN   rx   r=   r-   Zabs_objpathZ	pyx_fnamer1   r2   Zinterm_c_filer7   re   rf   r…   r|   r3   r3   r4   r€   Ÿ  sB    %
þýr€   c                 C   sD   | D ]:}t j |¡\}}| ¡ }|tv rt| d |kr dS qdS )Nr   TF)r   r   r$   r9   r   )ÚsrcsÚclsrw   r1   r2   Úkeyr3   r3   r4   Ú_any_Xð  s    rŠ   c                 C   s
   t | tƒS ©N)rŠ   r	   ©r‡   r3   r3   r4   Úany_fortran_srcú  s    r   c                 C   s
   t | tƒS r‹   )rŠ   r   rŒ   r3   r3   r4   Úany_cplus_srcþ  s    rŽ   r   c           	   
   C   s®   |du r$t j t j | d ¡¡d }|p*i }|p2i }ztt j ||¡| ƒ}W n\ ty¨   ttt	t
| ƒƒf||dœ|¤Ž}t|f|t| ƒt| ƒ|dœ|¤Ž}t|ƒ}Y n0 |S )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 : list of strings
        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``
    extra_objs: list
        List of paths to (prebuilt) object files / static libraries to link against.

    Returns
    =======

    The imported module from of the Python extension.
    Nr?   r   r„   )r   rI   r=   rJ   )r   r   r$   r%   r   rc   ÚImportErrorr5   ÚlistÚmapr
   rg   r   rŽ   )	ÚsourcesZextnameÚ	build_dirÚcompile_kwargsÚlink_kwargsrJ   ÚmodÚobjsÚsor3   r3   r4   Úcompile_link_import_py_ext  s&    ÿÿÿÿr™   c           
   	   C   s   |p
t  ¡ }tj |¡s"td|ƒ‚g }| D ]ì\}}tj ||¡}d}t| d¡ƒ 	¡ }tj 
|¡r˜tj 
|d ¡r„t|d ƒ ¡ }nt|ƒ 	¡ }||k}|rt|dƒ}	|	 |¡ W d   ƒ n1 sÈ0    Y  t|d dƒ}	|	 |¡ W d   ƒ n1 s0    Y  | |¡ q*||fS )NzNon-existent directory: Túutf-8z.sha256Úwt)ÚtempfileÚmkdtempr   r   r    r"   rc   r   ÚencodeÚ	hexdigestr!   r   Ú	read_textr   ÚopenÚwriter(   )
r’   r“   Úsource_filesr1   rw   ÚdestZdiffersZsha256_in_memZsha256_on_diskÚfhr3   r3   r4   Ú_write_sources_to_build_dir0  s(    
(*r¦   c                 K   s2   t | |ƒ\}}t|fd|i|¤Ž}d|i}||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’   r“   r-   r£   r–   Úinfor3   r3   r4   Úcompile_link_import_stringsJ  s    r¨   c              	   C   sø   |r|durt dƒ‚z¬t| |ƒ\}}tttt|ƒƒf||dœ|pBi ¤Ž}t|f|t|ƒt|ƒdœ|pfi ¤Ž}t	j
|gt	jt	jd}| ¡ }	dd„ | ¡ D ƒ\}
}W |râtj |¡rât |¡ d}n |ràtj |¡ràt |¡ d}0 |	|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.r„   )r   rI   r=   )ÚstdoutÚstderrc                 S   s   g | ]}|  d ¡‘qS )rš   )Údecode)Ú.0Útxtr3   r3   r4   Ú
<listcomp>‹  ó    z'compile_run_strings.<locals>.<listcomp>)Úexit_statusr“   )r:   r¦   r5   r   r‘   r
   rM   r   rŽ   Ú
subprocessÚPopenÚPIPEÚwaitÚcommunicater   r   r    ru   Úrmtree)r’   r“   Úcleanr”   r•   r£   r—   ÚprogÚpr°   r©   rª   r§   r3   r3   r4   Úcompile_run_stringse  s4    ÿÿþþ
þ

rº   )NNNFN)NFN)NFNNFFN)NNNFFN)NN)NNNF)NNNNNN)Nr   NNN)N)NFNN)0r   r   ru   r±   r]   rœ   r_   Úpathlibr   Ú	sysconfigr   r   r   Zrunnersr   r   r	   Úutilr
   r   r   r   r   r   r   r   r   r   r   r1   r   r`   r#   r5   r>   rM   rg   ry   r   r)   r€   rŠ   r   rŽ   r™   r¦   r¨   rº   r3   r3   r3   r4   Ú<module>   sf   4

  ÿ
B
  ÿ
H  ÿ
W
;ö
B  ÿ
Q
  ÿ
.
