o jg‘Vã @s¨ddlZddlZddlZddlZddlZddlZddlZddlmZm Z m Z ddl m Z m Z mZddlmZmZmZmZmZmZmZmZmZmZmZejdkrRdZnejdkrZd Zn e d  ej¡¡dZ d-d d „Zd.dd„Z  d/dd„Z!  d0dd„Z"d1dd„Z#e dfe dfe dfedfedfedfedfedfedfedfdœ Z$d2dd„Z%  d3dd„Z&dd„Z'd d!„Z(d"d#„Z)  d4d%d&„Z*d'd(„Z+d5d)d*„Z,d6d+d,„Z-dS)7éN)Ú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"  ýÿ ü     rXc 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ÚLIBRARYrrJéÚaixÚPy_ENABLE_SHAREDÚ BLDLIBRARYr?)ú-pthread) rSr?rrHrTrYÚ librariesrZrU) rOÚsysÚplatformÚwarningsÚwarnrr(ÚsplitÚ startswithr)ÚjoinrrX)rQÚso_filerrhrHrTrUr0rYrZÚcfgDictÚ is_frameworkÚoptÚlibfileÚlibnamer?Ú needed_flagsÚflagr9r9r:Ú link_py_so½s\       €   €þ  €þþrxc 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.pyrrHFú.cppú.cÚlanguage_levelrczCython compilation failed.)ÚCython.Compiler.Mainryrzr{rDÚendswithr(rr r%r&roÚgetcwdÚchdirÚ__dict__r'Ú num_errorsrEÚrealpathÚdirnamer"ÚunlinkÚshutilÚmove) Úsrcr-rÚ cy_kwargsryrzÚ cy_compiler7Úc_nameÚdstfileÚori_dirÚ cy_optionsÚ cy_resultr9r9r:Úsimple_cythonizes<             ú  ÿr“Úf95Úf2003Úf2008) r~r}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 NrrYÚincluder|)ÚobjpathrYrÚstdr?)ú-fPICrLFz#src2obj called with run_linker=True)rYrLrr?)rr r%r&Úisabsr‡r!roÚobjextrOrr)rDÚpyx2objÚextension_mappingrrP)Úsrcpathr,r˜rÚinc_pyr0r6r7rYÚ py_inc_dirr™r?rvrwrLrWr9r9r: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|rrMÚ output_dirNrH©r-rr?)z-fwrapvrgršÚoptionsÚstrict_aliasingFz/Cython requires strict aliasing to be disabled.r™zc++98Úc99T)r˜rrYr?r™r¤r r¥)rrr r‡r r!r&r%rorœrr“rOr)rr*)Úpyxpathr˜r-rrYrŒrHr0Ú abs_objpathÚ pyx_fnamer6r7Ú interm_c_filer?rvrwr¤r™r9r9r:ržsD%     €    þýrcCsD|D]}tj |¡\}}| ¡}|tvrt|d|krdSqdS)NrTF)rr r%rDrž)ÚsrcsÚclsr‹r6r7Úkeyr9r9r:Ú_any_Xïs€r®cCó t|tƒS©N)r®r©r«r9r9r:Úany_fortran_srcùó r²cCr¯r°)r®rr±r9r9r:Ú 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. NrJrr£)rrTrHrU)rr r%r&rroÚ ImportErrorr;ÚlistÚmapr rxr²r´) ÚsourcesÚextnameÚ build_dirÚcompile_kwargsÚ link_kwargsrUÚmodÚobjsÚsor9r9r:Úcompile_link_import_py_exts* úÿÿÿÿ úrÀc Cs@|pt ¡}tj |¡std|ƒ‚g}|D]†\}}tj ||¡}d}t| d¡ƒ  ¡}tj  |¡r^tj  |d¡rTt |dƒ }|  ¡} Wdƒn1sNwYnt |ƒ  ¡} | |k}|r–t |dƒ }| |¡Wdƒn1suwYt |ddƒ }| |¡Wdƒn1s‘wY| |¡q||fS)NzNon-existent directory: Túutf-8z.sha256Úwt)ÚtempfileÚmkdtemprr r!r#rorÚencodeÚ hexdigestr"ÚopenÚreadrÚwriter)) r¸rºÚ source_filesr6r‹ÚdestÚdiffersÚ sha256_in_memÚfhÚsha256_on_diskr9r9r:Ú_write_sources_to_build_dir/s2      ÿ€   ÿ ÿ 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ºr0rÊr½Úinfor9r9r:Ú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£)rrTrH)ÚstdoutÚstderrcSsg|]}| d¡‘qS)rÁ)Údecode)Ú.0Útxtr9r9r:Ú ‹sz'compile_run_strings..)Ú exit_statusrº)rErÐr;r¶r·r rXr²r´Ú subprocessÚPopenÚPIPEÚwaitÚ communicaterr r!r‰Úrmtree) r¸rºÚcleanr»r¼rÊr¾ÚprogÚprÙrÓrÔrÑr9r9r:Úcompile_run_stringses6 ÿÿþþ €þ   rã)NNNFN)NFN)NFNNFFN)NNNFFN)NN)NNNF)NNNNNN)NrNNNr°)NFNN).rrr‰rÚrirÃrkÚ sysconfigrrrÚrunnersrrrÚutilr r r r r rrrrrrr6rœrlr$r;rIrXrxr“ržr*rr®r²r´rÀrÐrÒrãr9r9r9r:Úsf4   ÿ B ÿH ÿ W;ö B ÿQ  ÿ.