o ݹiV @sddlZddlZddlZddlZddlZddlZddlZddlmZddl m Z m Z m Z ddl mZmZmZddlmZmZmZmZmZmZmZmZmZmZmZejdkrXdZnejd kr`d Zn ed ejdZ d.d dZ!d/ddZ"  d0ddZ#  d1ddZ$d2ddZ%edfedfedfedfedfedfedfedfedfedfd Z&d3ddZ'  d4ddZ(dd Z)d!d"Z*d#d$Z+  d5d&d'Z,d(d)Z-d6d*d+Z.d7d,d-Z/dS)8N)Path)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"        rYc Ks|pg}|dg}|dg} tjdkrtdntjdkrt} |dg| dg|d<| | dg7} d } | d D])} | rR|dgd | g|d<d } q?| d ra| | d dq?| d rhd} q?| d} d | ddddd}| |n>tjdddkrn4t drt} |dg| dg|d<| | dg7} | dD]} | d r|| d dg7}qn |dg}d}|D] }||vr| |qt |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_dirswin32zWindows not yet supported.darwinlinklineLDFLAGSLIBDIRFLIBSz -frameworkz-lNTLIBRARYrrKaixPy_ENABLE_SHARED BLDLIBRARYr@)-pthread) rTr@rrIrUrZ librariesr[rV) rPsysplatformwarningswarnrr)split startswithr*joinrrY)rRso_filerrirIrUrVr1rZr[cfgDict is_frameworkoptlibfilelibnamer@ needed_flagsflagr:r:r; link_py_sos\            ryc 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.pyrrIF.cpp.clanguage_levelrdzCython compilation failed.)Cython.Compiler.Mainrzr{r|rEendswithr)r r!r&r'rpgetcwdchdir__dict__r( num_errorsrFrealpathdirnamer#unlinkshutilmove) srcr.r cy_kwargsrzr{ cy_compiler8c_namedstfileori_dir cy_options cy_resultr:r:r;simple_cythonizes<               rf95f2003f2008) rr~z.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 NrrZincluder})objpathrZrstdr@)-fPICrMFz#src2obj called with run_linker=True)rZrMrr@)r r!r&r'isabsrr"rpobjextrPrr*rEpyx2objextension_mappingrrQ)srcpathr-rrinc_pyr1r7r8rZ py_inc_dirrr@rwrxrMrXr:r:r;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. r}rrN output_dirNrIr.rr@)z-fwrapvrhroptionsstrict_aliasingFz/Cython requires strict aliasing to be disabled.rzc++98c99T)rrrZr@rrrr)rr r!rr r"r'r&rprrrrPr*rr+)pyxpathrr.rrZrrIr1 abs_objpath pyx_fnamer7r8 interm_c_filer@rwrxrrr:r:r;rsD%         rcCsD|D]}tj|\}}|}|tvrt|d|krdSqdS)NrTF)r r!r&rEr)srcsclsrr7r8keyr:r:r;_any_XsrcC t|tSN)rr rr:r:r;any_fortran_src rcCrr)rrrr:r:r; any_cplus_srcrrrc Cs|durtjtj|dd}|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<listmapr ryrr) sourcesextname build_dircompile_kwargs link_kwargsrVmodobjssor:r:r;compile_link_import_py_exts*  rc Cs|pt}tj|std|g}|D]t\}}tj||}d}t|d }tj |rLtj |drBt |d }nt | }||k}|rt|d } | |Wdn1scwYt|dd } | |Wdn1swY||q||fS)NzNon-existent directory: Tutf-8z.sha256wt)tempfilemkdtempr r!r"r$rprencode hexdigestr#r read_textropenwriter*) rr source_filesr7rdestdiffers sha256_in_memsha256_on_diskfhr:r:r;_write_sources_to_build_dir0s,          rcKs2t||\}}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)rr)rrr1rrinfor:r:r;compile_link_import_stringsJsrc 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}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)stdoutstderrcSsg|]}|dqS)r)decode).0txtr:r:r; sz'compile_run_strings..) exit_statusr)rFrr<rrr rYrr subprocessPopenPIPEwait communicater r!r"rrmtree) rrcleanrrrrprogprrrrr:r:r;compile_run_stringses6     r)NNNFN)NFN)NFNNFFN)NNNFFN)NN)NNNF)NNNNNN)NrNNNr)NFNN)0rr rrrjrrlpathlibr sysconfigrrrrunnersrrr utilr r r r rrrrrrrr7rrmr%r<rJrYryrrr+rrrrrrrrr:r:r:r;sh 4   B H W; B Q  .