o *ήcsP @sddlZddlZddlZddlZddlZddlZddlZddlmZm Z m Z ddl m Z m Z mZddlmZmZmZmZmZmZmZmZmZmZmZejdkrRdZnejdkrZd Zn ed ejdZ d-d d Zd.ddZ  d/ddZ!  d0ddZ"d1ddZ#e dfe dfe dfedfedfedfedfedfedfedfd Z$d2ddZ%  d3ddZ&ddZ'd d!Z(d"d#Z)  d4d%d&Z*d'd(Z+d5d)d*Z,d6d+d,Z-dS)7N)get_config_varget_config_varsget_path)CCompilerRunnerCppCompilerRunnerFortranCompilerRunner) get_abspath make_dirscopyGlobArbitraryDepthGlob glob_at_depthimport_module_from_file pyx_is_cplussha256_of_stringsha256_of_file CompileErrorposixz.ontz.objzUnknown os.name: {}FcKsDi}|dur9|D].\}} t|tr!t|jD]} | || <qq t|tr4t|j|D]} | || <q,q | ||<q |p        rWc Ks |pg}|dg}|dg}tjdkrtdnDtjdkr!n>tjdddkr+n4td r^t} |d g| d g|d <|| d g7}| d D]} | dr\|| ddg7}qLn |dg} d} | D] } | | vrt| | qit |fd| ||||||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_dirswin32zWindows not yet supported.darwinNaixPy_ENABLE_SHAREDlinkline PY_LDFLAGSLIBDIR BLDLIBRARYz-lr?)-pthreadT)rSr?rrHrTrX librariesrY) rOsysplatformwarningswarnrrr(split startswithr)rW)rQso_filerrerHrTr0rXrYcfgDictoptr? needed_flagsflagr9r9r: link_py_sos>         rqc Kslddlm}m}ddlm}|ds|dsJ|p!d}|p%d}|ddr.d nd }tj tj |d|}tj ||} |rLt } nd} t |z\||} | j|d |vrgd | jd <||g| } | jdkrvtd tjtj|tj|krtj| rt| ttj tj|||Wt | | SWt | | St | 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_optionsCompilationOptions)compile.pyxz.pyrrHF.cpp.clanguage_levelr\zCython compilation failed.)Cython.Compiler.MainrrrsrtrDendswithr(rr r%r&joingetcwdchdir__dict__r' num_errorsrEabspathdirnamer"unlinkshutilmove) srcr-r cy_kwargsrrrs cy_compiler7c_namedstfileori_dir cy_options cy_resultr9r9r:simple_cythonizes<               rf95f2003f2008) rwrvz.cxxz.fz.forz.ftnz.f90z.f95z.f03z.f08cKsHtjtj|\}}|dur#tj|rd}n tj|}|p"d}tj|r2tj||t}| dg}|rGt d} | |vrG| | | dkrYt |f|||d|S|durmt| \}} d|vrm| |d<| dg} d } | D] } | | vr| | qw| d d }|rtd ||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 NrrXincluderu)objpathrXrstdr?)-fPICrLFz#src2obj called with run_linker=True)rXrLrr?)rr r%r&isabsrr!r{objextrOrr)rDpyx2objextension_mappingrrP)srcpathr,rrinc_pyr0r6r7rX py_inc_dirrr?rorprLrVr9r9r:r*=sJ          r*c KsF|dsJ|p d}|pd}|ptj|}t||d}tj|r:tj|} tj| \} } tj|| t }|p=i}||d<|durJt |}||d<t |f||d|} |p\g}| dg} d }|D] }|| vrr| |qg| d g}| d d rtd |r| dd}n| dd}t| f|||| ||dd 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. rurrM output_dirNrHr-rr?)z-fwrapvrdroptionsstrict_aliasingFz/Cython requires strict aliasing to be disabled.rzc++98c99T)rrrXr?rrrr)rzrr rr r!r&r%r{rrrrOr)rr*)pyxpathrr-rrXrrHr0 abs_objpath pyx_fnamer6r7 interm_c_filer?rorprrr9r9r:rsD%         rcCsD|D]}tj|\}}|}|tvrt|d|krdSqdS)NrTF)rr r%rDr)srcsclsrr6r7keyr9r9r:_any_XsrcC t|tSN)rrrr9r9r:any_fortran_src rcCrr)rrrr9r9r: any_cplus_srcrrrc Cs|durtjtj|dd}|pi}|pi}z ttj|||}W|StyUttt t |f||d|}t |f|t |t |d|}t|}Y|Sw)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. NrJrrrrTrH)rr r%r&rr{ ImportErrorr;listmapr rqrr)sourcesextname build_dircompile_kwargs link_kwargsmodobjssor9r9r:compile_link_import_py_exts*  rc Cs@|pt}tj|std|g}|D]\}}tj||}d}t|d }tj |r^tj |drTt |d }| } Wdn1sNwYnt | } | |k}|rt |d }||Wdn1suwYt |dd }||Wdn1swY||q||fS)NzNon-existent directory: Tutf-8z.sha256wt)tempfilemkdtemprr r!r#r{rencode hexdigestr"openreadrwriter)) rr source_filesr6rdestdiffers sha256_in_memfhsha256_on_diskr9r9r:_write_sources_to_build_dirs2           rcKs4t||\}}t|fd|i|}t|d}||fS)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)rrdict)rrr0rrinfor9r9r:compile_link_import_strings)s rc Cs|r |dur tdzVt||\}}tttt|f||d|p!i}t|f|t|t|d|p3i}t j |gt j t j d}| } dd| D\} } W|r`tj|r`t|d}n|rptj|rpt|d}wt| |d} | | f| fS) 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.rr)stdoutstderrcSsg|]}|dqS)r)decode).0txtr9r9r: jsz'compile_run_strings..) exit_statusr)rErr;rrr rWrr subprocessPopenPIPEwait communicaterr r!rrmtreer) rrcleanrrrrprogprrrrr9r9r:compile_run_stringsDs6     r)NNNFN)NFN)NFNNFF)NNNFF)NN)NNNF)NNNNNN)NrNNr)NFNN).rrrrrfrrh sysconfigrrrrunnersrrrutilr r r r r rrrrrrr6rrir$r;rIrWrqrrr*rrrrrrrrr9r9r9r:sf4   ? F A7 B Q  ,