PtjV ddlZddlZddlZddlZddlZddlZddlZddlmZddl m Z m Z m Z ddl mZmZmZddlmZmZmZmZmZmZmZmZmZmZmZejdkrdZn8ejd krd Zn*ejd ejdZ d d Z!d!dZ" d"dZ# d#dZ$d$dZ%edfedfedfedfedfedfedfedfedfedfd Z&d%dZ' d&dZ(dZ)dZ*dZ+ d'dZ,dZ-d(dZ.d)dZ/dS)*N)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: {}Fc i}||D]t\}} t|tr"tj|jD]} | || <>' " " 188AABB B g    { A AA G4 @ @ @ @ @H DD  >((++ID##(()9)9!)<)<==ID#kkmm +//2667776BBsBkBBCCCC Oc|ptjdd}|dkr|rtdddgi|fStddgi|fS|dks |r t dd gi|fSti|fS) NSYMPY_COMPILER_VENDORgnuintelflagsz -nofor_mainz-cxxlibllvm lib_optionsfortran)r"environr+lowerr r ValueErrorvendorcplusrs r<get_mixed_fort_c_linkerrMas  Erz~~&=uEEF ||~~   8)}i896C C*}o.8 8 5  %"YK0&: :*  r>c |Ytjtj|d\}} |r|t dz }|s}|rjt |dd||\}} } | D]*\} } | |vr|| | %| || <+n|rt}nt}| dg}|rd|vr| d| dd }|std t|| }|||pgz||fd |i|}||S) a Link object files. Parameters ========== obj_files: iterable of str Paths to object files. out_file: str (optional) Path to executable/shared library, if ``None`` it will be deduced from the last item in obj_files. shared: bool Generate a shared library? Runner: CompilerRunner subclass (optional) If not given the ``cplus`` and ``fort`` flags will be inspected (fallback is the C compiler). cwd: str Path to the root of relative paths and working directory for compiler. cplus: bool C++ objects? default: ``False``. fort: bool Fortran objects? default: ``False``. extra_objs: list List of paths to extra object files / static libraries. \*\*kwargs: dict Keyword arguments passed to ``Runner``. Returns ======= The absolute path to the generated shared object / executable. N EXT_SUFFIXrKrJrCz-shared run_linkerTz*run_linker was set to False (nonsensical).rr)r"r#r(r)rrMr+rexpandr rpopr,rIr run) obj_filesout_filesharedr/rrLfort extra_objsr3r: extra_kwargsrKr5r6rCrQrunners r<linkr]vsD(()9)9)B-)H)HII #  5 |44 4H )  )'!::h55 )FL& %**,, " "1;;1I$$Q'''' !F1II  "  )*( JJw # #E $ E ! ! LL # # #L$//J GEFFF8---H VIz/R0(E U Us Uf U UF JJLLL Or>c |pg}|dg}|dg} tjdkrtjdntjdkrt } |dg| dgz|d<| | dgz } d } | d D]o} | r!|dgd | gz|d<d } %| d r| | d dX| d rd} p| d} d | ddddd}| |ntjdddkrntdr~t } |dg| dgz|d<| | dgz } | dD]%} | d r|| d dgz }&n |dg}d}|D]}||vr| |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_dirswin32zWindows not yet supported.darwinlinklineLDFLAGSLIBDIRFLIBSz -frameworkz-lNTLIBRARYrrOaixPy_ENABLE_SHARED BLDLIBRARYrC)-pthread) rXrCrrLrYr_ librariesr`rZ) rTsysplatformwarningswarnrr+split startswithr,joinrr])rVso_filerrnrLrYrZr3r_r`cfgDict is_frameworkoptlibfilelibnamerC needed_flagsflags r< link_py_sor~s>RI::nb11L::nb11L  |w 23333  ! !!###ZZ B7779;M:NNz*++  6?((** $ $C $%+ZZ B%?%?R!RF:  WX./ /L|,2244 + +>>$''+#abb'*I +  JJw # #E L u   LL      L$eEPT)Y)j L LDJ L LLr>c ddlm}m}ddlm}|ds)|dsJ|pd}|pd}|ddrd nd }tj tj |d|z}tj ||} |rtj } nd} tj | ||} | j|d |vr d | jd <||g| } | jdkrt#d tjtj|tj|krtj| rtj| t-jtj tj|||tj | n#tj | wxYw| S)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.pyrrLF.cpp.clanguage_levelrizCython compilation failed.)Cython.Compiler.MainrrrrHendswithr+r"r#r(r)rugetcwdchdir__dict__r* num_errorsrIrealpathdirnamer%unlinkshutilmove) srcr0r cy_kwargsrr cy_compiler:c_namedstfileori_dir cy_options cy_results r<simple_cythonizersJ ;::::: 99;;   ' 'F399;;+?+?+F+FFF F *CnGMM'511 ;&&tC W  bg..s33 4 4Q 7# =Fgll7F++G )++HSMMM''88 ""9--- 9 , ,45J  0 1Juj11  ! # #9:: : 7  BGOOC00 1 1RW5E5Eg5N5N N Nw~~g&& # '""" K RW__S%9%96BBG L L L  Ns D:I""I8f95f2003f2008) rrz.cxxz.fz.forz.ftnz.f90z.f95z.f03z.f08c tjtj|\}}|Etj|rd}n#tj|}|pd}tj|r(tj||tz}| dg}|r(td} | |vr| | | dkrt|f|||d|S|+t| \}} d|vr| |d<| dg} d } | D]} | | vr| | | 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 Nrr_includer)objpathr_rstdrC)-fPICrQFz#src2obj called with run_linker=True)r_rQrrC)r"r#r(r)isabsrr$ruobjextrTrr,rHpyx2objextension_mappingrrU)srcpathr/rrinc_pyr3r9r:r_ py_inc_dirrrCr|r}rQr\s r<r-r-]s .  !1!1'!:!:;;ID# 7== ! ! %GGgoog..GnG w}}W7',,wv 66::nb11L ,i(( \ ) )    + + + yy{{fw!lPS!!!! !~' 4    F5M JJw # #EL u   LL   L%00JB@AAA VWIwK\)s%KKCIKKF JJLLL Nr>c |dsJ|pd}|pd}|ptj|}t ||}tj|ritj|} tj| \} } tj|| tz}|pi}||d<|t|}||d<t|f||d|} |pg}| dg} d }|D]}|| vr| || 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. rrrR output_dirNrLr0rrC)z-fwrapvrmroptionsstrict_aliasingFz/Cython requires strict aliasing to be disabled.rzc++98c99T)rrr_rCrrrr)rr"r#rr r$r)r(rurrrrTr,rr-)pyxpathrr0rr_rrLr3 abs_objpath pyx_fnamer9r: interm_c_filerCr|r}rrs r<rrsJ   F # ### # *CnG111Gg3///K w}}[!!7G$$W-- G$$Y// c',,wv 66RI!Il }W%%Ig$WTg3TT)TTM%2L JJw # #E3L u   LL   jjB''G zz#U++NLMMM 'jj((jj&& = 's ,Es"4    r>c|D][}tj|\}}|}|tvrt|d|krdS\dS)NrTF)r"r#r(rHr)srcsclsrr9r:keys r<_any_XrseG$$S)) ciikk # # # %a(C//tt 5r>c,t|tSN)rr rs r<any_fortran_srcrs $- . ..r>c,t|tSr)rr rs r< any_cplus_srcrs $) * **r>rc |Htjtj|dd}|pi}|pi} t tj|||}n{#t $rntttt|f||d|}t|f|t|t||d|}t |}YnwxYw|S)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. NrOrr)rrYrLrZ)r"r#r(r)rru ImportErrorr=listmapr r~rr) sourcesextname build_dircompile_kwargs link_kwargsrZmodobjssos r<compile_link_import_py_extrs$:'""27#3#3GBK#@#@AA!D#)rN#K*%bgll9g&F&FPP ***tC W$=$=>>@ #,@@0>@@ \)/'2J2J+G44\\OZ\\%b)) * Js.BA5C;:C;c|ptj}tj|st d|g}|D]}\}}tj||}d}t|d }tj |rntj |dzr%t|dz }n!t| }||k}|r}t|d5} | |dddn #1swxYwYt|dzd5} | |dddn #1swxYwY||||fS)NzNon-existent directory: Tutf-8z.sha256wt)tempfilemkdtempr"r#r$r&rurencode hexdigestr%r read_textropenwriter,) rr source_filesr9rdestdiffers sha256_in_memsha256_on_diskfhs r<_write_sources_to_build_dirr0s/X-//I 7== # #=0)<<<L"" cw||It,,(G)<)<==GGII 7>>$   6w~~dY.// B!%dY&6!7!7!A!A!C!C!/!5!5!?!?!A!A$ 5G  (dD!! R                dY&-- (''' ( ( ( ( ( ( ( ( ( ( ( ( ( ( (D!!!!  ""s$E&&E* -E* F&&F* -F* c Vt||\}}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)rrr3rrinfos r<compile_link_import_stringsrJsF*:'9MML) $\ Q QY Q& Q QC  #D 9r>c|r|td t||\}}ttt t |f||d|pi}t |f|t|t|d|pi}tj |gtj tj }| } d| D\} } |r5tj|rt#j|d}n<#|r5tj|rt#j|d}wxYw| |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)rrYrL)stdoutstderrc8g|]}|dS)r)decode).0txts r< z'compile_run_strings..s$III##**W--IIIr>) exit_statusr)rIrr=rrr r]rr subprocessPopenPIPEwait communicater"r#r$rrmtree) rrcleanrrrrprogprrrrs r<compile_run_stringsres6 f&deee "=gy"Q"Q itC \$B$BCCHY#,HH1?1E2HHDNi(66' 55NN:E:KNN  dVJOJO T T Tffhh IIIII  RW]]9--  M) $ $ $I  RW]]9--  M) $ $ $I    &Y ? ?D F T !!s CD 9E)NNNFN)NFN)NFNNFFN)NNNFFN)NN)NNNF)NNNNNN)NrNNNr)NFNN)0rr"rrrorrqpathlibr sysconfigrrrrunnersrr r utilr r r rrrrrrrrr9rrrr'r=rMr]r~rrr-rrrrrrrrr>r<rs,  ??????????  7g FFW__ FFHM'..rw77888 FQV$(????D----*9=7;EEEEP=A37TLTLTLTLn7777v D !  %  % $ ' "D ) "D ) "D ) "E * "G , "G ,  ????D6:59NNNNb///+++UY<@++++\###46,",",",",","r>