o GZŽhŒVã @s´ddlZddlZddlZddlZddlZddlZddlZddlmZddl m Z m Z m Z ddl mZmZmZddlmZmZmZmZmZmZmZmZmZmZmZejdkrXdZnejd kr`d Zn e d   ej¡¡dZ d.d d„Z!d/dd„Z"  d0dd„Z#  d1dd„Z$d2dd„Z%edfedfedfedfedfedfedfedfedfedfdœ Z&d3dd„Z'  d4dd„Z(dd „Z)d!d"„Z*d#d$„Z+  d5d&d'„Z,d(d)„Z-d6d*d+„Z.d7d,d-„Z/dS)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.objzUnknown os.name: {}FcKsDi}|dur9| ¡D].\}} t|tƒr!t |j¡D]} | || <qq t|tƒr4t|j|ƒD]} | || <q,q | ||<q |pr=r7z-sharedÚ run_linkerTz*run_linker was set to False (nonsensical).©rr)rrr$r%rr@r'rÚexpandrrÚpopr(r<r Úrun)Ú obj_filesÚout_fileÚsharedr+rr?ÚfortÚ extra_objsr-r2Ú extra_kwargsr>r.r/r7rCÚrunnerr3r3r4Úlinkvs>"  ýÿ ü     rOc KsÒ|pg}| dg¡}| dg¡} tjdkrt d¡n¦tjdkrƒtƒ} | dg¡| dg|d<| | dg7} d } | d  ¡D])} | rR| dg¡d | g|d<d } q?|  d ¡ra|  | d d…¡q?|  d ¡rhd} q?| d} d  |  d¡dd…¡dd…}|  |¡n>tjdd…dkrn4t dƒrÀtƒ} | dg¡| dg|d<| | dg7} | d ¡D]} |  d ¡r¾|| d d…g7}q®n | dg¡}d}|D] }||vrÖ|  |¡qËt |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``. 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Úwin32zWindows not yet supported.ÚdarwinZlinklineÚLDFLAGSÚLIBDIRFÚLIBSz -frameworkz-léNTÚLIBRARYrrAéÚaixÚPy_ENABLE_SHAREDÚ BLDLIBRARYr7)ú-pthread) rJr7rr?rKrPÚ librariesrQrL) rFÚsysÚplatformÚwarningsÚwarnrr'ÚsplitÚ startswithr(ÚjoinrrO)rHZso_filerr^r?rKrLr-rPrQZcfgDictZ is_frameworkÚoptZlibfileZlibnamer7Ú needed_flagsÚflagr3r3r4Ú link_py_so¾s\       €   €þ  €þþric Kslddlm}m}ddlm}| ¡ d¡s| ¡ d¡sJ‚|p!d}|p%d}| dd¡r.d nd }tj  tj  |¡¡d|}tj  ||¡} |rLt  ¡} nd} t  |¡z\||ƒ} | j |¡d |vrgd | jd <||g| ƒ} | jdkrvtd ƒ‚tj tj |¡¡tj |¡kr¨tj | ¡r‘t | ¡t tj  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_optionsÚCompilationOptions)Úcompileú.pyxz.pyrr?Fú.cppú.cZlanguage_levelrYzCython compilation failed.)ZCython.Compiler.Mainrjrkrlr;Úendswithr'rrr$r%reÚgetcwdÚchdirÚ__dict__r&Z num_errorsr<ÚrealpathÚdirnamer!ÚunlinkÚshutilÚmove) Úsrcr,rÚ cy_kwargsrjrkZ cy_compiler2Zc_nameZdstfileZori_dirZ cy_optionsZ cy_resultr3r3r4Úsimple_cythonizes<             ú  ÿr{Zf95Zf2003Zf2008) rornz.cxxz.fz.forz.ftnz.f90z.f95z.f03z.f08cKsHtj tj |¡¡\}}|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 NrrPÚincluderm)ÚobjpathrPrÚstdr7)ú-fPICrCFz#src2obj called with run_linker=True)rPrCrr7)rrr$r%Úisabsrur reÚobjextrFrr(r;Úpyx2objÚextension_mappingrrG)Zsrcpathr+r}rÚinc_pyr-r1r2rPZ py_inc_dirr~r7rgrhrCrNr3r3r4r)]sJ      ÿ  €  ÿÿr)c KsF| d¡sJ‚|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 ¡rƒtd ƒ‚|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. rmrrDÚ output_dirNr?©r,rr7)z-fwrapvr]rÚoptionsÚstrict_aliasingFz/Cython requires strict aliasing to be disabled.r~zc++98Zc99T)r}rrPr7r~r‡r„rˆ)rprrrur r r%r$rerrr{rFr(rr))Zpyxpathr}r,rrPrzr?r-Z abs_objpathZ pyx_fnamer1r2Z interm_c_filer7rgrhr‡r~r3r3r4r‚ŸsD%     €    þýr‚cCsD|D]}tj |¡\}}| ¡}|tvrt|d|krdSqdS)NrTF)rrr$r;rƒ)ÚsrcsÚclsryr1r2Úkeyr3r3r4Ú_any_Xðs€rŒcCó t|tƒS©N)rŒr ©r‰r3r3r4Úany_fortran_srcúó rcCrrŽ)rŒrrr3r3r4Ú any_cplus_srcþr‘r’rc Cs®|durtj tj |d¡¡d}|pi}|pi}z ttj ||¡|ƒ}W|StyVttt 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 : 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. NrArr†)rrKr?rL)rrr$r%rreÚ ImportErrorr5ÚlistÚmapr rirr’) ÚsourcesZextnameÚ build_dirÚcompile_kwargsÚ link_kwargsrLÚmodÚobjsÚsor3r3r4Úcompile_link_import_py_exts* úÿÿÿÿ úrc Cs|pt ¡}tj |¡std|ƒ‚g}|D]t\}}tj ||¡}d}t| d¡ƒ  ¡}tj  |¡rLtj  |d¡rBt |dƒ  ¡}nt |ƒ  ¡}||k}|r„t|dƒ } |  |¡Wdƒn1scwYt|ddƒ } |  |¡Wdƒn1swY| |¡q||fS)NzNon-existent directory: Túutf-8z.sha256Úwt)ÚtempfileÚmkdtemprrr r"rerÚencodeÚ hexdigestr!rÚ read_textrÚopenÚwriter() r–r—Ú source_filesr1ryÚdestZdiffersZ sha256_in_memZsha256_on_diskÚfhr3r3r4Ú_write_sources_to_build_dir0s,        ÿ ÿ rªcKs2t||ƒ\}}t|fd|i|¤Ž}d|i}||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ªr)r–r—r-r§ršÚinfor3r3r4Úcompile_link_import_stringsJsr¬c Csø|r |dur tdƒ‚zVt||ƒ\}}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}w| |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.r†)rrKr?)ÚstdoutÚstderrcSsg|]}| d¡‘qS)rž)Údecode)Ú.0Útxtr3r3r4Ú ‹sz'compile_run_strings..)Ú exit_statusr—)r<rªr5r”r•r rOrr’Ú subprocessÚPopenÚPIPEÚwaitÚ communicaterrr rwÚrmtree) r–r—Úcleanr˜r™r§r›ÚprogÚpr³r­r®r«r3r3r4Úcompile_run_stringses6 ÿÿþþ €þ   r½)NNNFN)NFN)NFNNFFN)NNNFFN)NN)NNNF)NNNNNN)NrNNNrŽ)NFNN)0rrrwr´r_r raÚpathlibrÚ sysconfigrrrÚrunnersrrr Úutilr r r r rrrrrrrr1rrbr#r5r@rOrir{rƒr)r‚rŒrr’rrªr¬r½r3r3r3r4Úsh 4   ÿ B ÿH ÿ W;ö B ÿQ  ÿ.