o ª3IhŒ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 |p"  ýÿ ü     rYc 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.ÚdarwinÚlinklineÚLDFLAGSÚLIBDIRFÚLIBSz -frameworkz-léNTÚLIBRARYrrKéÚaixÚPy_ENABLE_SHAREDÚ BLDLIBRARYr@)ú-pthread) rTr@rrIrUrZÚ librariesr[rV) rPÚsysÚplatformÚwarningsÚwarnrr)ÚsplitÚ startswithr*ÚjoinrrY)rRÚso_filerrirIrUrVr1rZr[ÚcfgDictÚ is_frameworkÚoptÚlibfileÚlibnamer@Ú needed_flagsÚflagr:r:r;Ú link_py_so¾s\       €   €þ  €þþryc 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.pyrrIFú.cppú.cÚlanguage_levelrdzCython compilation failed.)ÚCython.Compiler.Mainrzr{r|rEÚendswithr)r r!r&r'rpÚgetcwdÚchdirÚ__dict__r(Ú num_errorsrFÚrealpathÚdirnamer#ÚunlinkÚshutilÚmove) Úsrcr.rÚ cy_kwargsrzr{Ú cy_compiler8Úc_nameÚdstfileÚori_dirÚ cy_optionsÚ cy_resultr:r:r;Úsimple_cythonizes<             ú  ÿr”Úf95Úf2003Úf2008) rr~z.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 NrrZÚincluder})ÚobjpathrZrÚstdr@)ú-fPICrMFz#src2obj called with run_linker=True)rZrMrr@)r r!r&r'Úisabsrˆr"rpÚobjextrPrr*rEÚpyx2objÚextension_mappingrrQ)Úsrcpathr-r™rÚinc_pyr1r7r8rZÚ py_inc_dirršr@rwrxrMrXr:r:r;r+]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. r}rrNÚ output_dirNrI©r.rr@)z-fwrapvrhr›ÚoptionsÚstrict_aliasingFz/Cython requires strict aliasing to be disabled.ršzc++98Úc99T)r™rrZr@ršr¥r¡r¦)r‚r r!rˆr r"r'r&rprrr”rPr*rr+)Úpyxpathr™r.rrZrrIr1Ú abs_objpathÚ pyx_fnamer7r8Ú interm_c_filer@rwrxr¥ršr:r:r;ržŸsD%     €    þýržcCsD|D]}tj |¡\}}| ¡}|tvrt|d|krdSqdS)NrTF)r r!r&rErŸ)ÚsrcsÚclsrŒr7r8Úkeyr:r:r;Ú_any_Xðs€r¯cCó t|tƒS©N)r¯r ©r¬r:r:r;Úany_fortran_srcúó r³cCr°r±)r¯rr²r:r:r;Ú 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. NrKrr¤)rrUrIrV)r r!r&r'rrpÚ ImportErrorr<ÚlistÚmapr ryr³rµ) ÚsourcesÚextnameÚ build_dirÚcompile_kwargsÚ link_kwargsrVÚmodÚobjsÚsor:r:r;Úcompile_link_import_py_exts* úÿÿÿÿ úrÁc 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Úmkdtempr r!r"r$rprÚencodeÚ hexdigestr#rÚ read_textrÚopenÚwriter*) r¹r»Ú source_filesr7rŒÚdestÚdiffersÚ sha256_in_memÚsha256_on_diskÚfhr:r:r;Ú_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»r1rËr¾Úinfor:r:r;Ú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¤)rrUrI)ÚstdoutÚstderrcSsg|]}| d¡‘qS)rÂ)Údecode)Ú.0Útxtr:r:r;Ú ‹sz'compile_run_strings..)Ú exit_statusr»)rFrÑr<r·r¸r rYr³rµÚ subprocessÚPopenÚPIPEÚwaitÚ communicater r!r"rŠÚrmtree) r¹r»Úcleanr¼r½rËr¿ÚprogÚprÚrÔrÕrÒr:r:r;Úcompile_run_stringses6 ÿÿþþ €þ   rä)NNNFN)NFN)NFNNFFN)NNNFFN)NN)NNNF)NNNNNN)NrNNNr±)NFNN)0rr rŠrÛrjrÄrlÚpathlibrÚ sysconfigrrrÚrunnersrrr Úutilr r r r rrrrrrrr7rrmr%r<rJrYryr”rŸr+ržr¯r³rµrÁrÑrÓrär:r:r:r;Úsh 4   ÿ B ÿH ÿ W;ö B ÿQ  ÿ.