o 8ήch$@sddlmZmZddlmZddlmZddlmZddl mm Z ddl Z ddl Z ddlZddlZddlmZmZmZddlmZdd d Zd d Ze edZe ed ZeddZddZeZGdddeZGdddeZ ddZ!dddZ"ddZ#ddZ$dS)) CCompiler new_compiler) build_ext)customize_compiler)logN)NamedTemporaryFilemkdtemp gettempdir)contextmanager)z.dllz.pyd).sor )windefaultcCsttjddtd|S)Nr )_configsgetsysplatformargr:/tmp/pip-target-vg8gfxp4/lib/python/numba/pycc/platform.py get_configssrc csz,t}ttj|d|d}|VWz |t|WnYdSt|dSz |t|WnYwt|w)Nztemp%swt)ropenospathjoincloseremovermdir)suffixtmpdirntfrrr _gentmpfiles" r$c Cst}t|dD]<}z/t|!}d}|||||j|jgtdWdn1s4wYWq t yEYdSwdS)N)z.cz.cxxzint main(void) { return 0; }) output_dirFT) rrr$writeflushrcompilenamer Exception)compilerr!r#simple_crrr_check_external_compiler.s    r-c@seZdZgZdS)_DummyExtensionN)__name__ __module__ __qualname__ librariesrrrrr.Fsr.c@steZdZddZeddZejddZddZ  dd d Z   dd d Z ddZ ddZ ddZ ddZ d S) ToolchaincCsjts|ddlm}d|_t|_t|jt||_ |j |j j |_ |j j |_td|_dS)Nr) DistributionFnpymath)_external_compiler_ok_raise_external_compiler_errordistutils.distr4_verboser _compilerrr _build_extfinalize_options library_dirs _py_lib_dirs include_dirs_py_include_dirsnp_miscget_info _math_info)selfr4rrr__init__Ls      zToolchain.__init__cCs|jSN)r9rDrrrverbose[szToolchain.verbosecCs&||_t|r tjdStjdSrF)r9r set_thresholdINFOWARN)rDvaluerrrrH_s cCsd}d}tj}|dr)tjdkrddg}nddg}d ||d |f}t||d rAd d g}d ||d |f}t||drRd}d ||f}t|d|}t|)NzYAttempted to compile AOT function without the compiler used by `numpy.distutils` present.z(If using conda try: #> conda install %slinuxlz gcc_linux-32z gxx_linux-32z gcc_linux-64z gxx_linux-64z%s %s darwinz clang_osx-64zclangxx_osx-64win32zCannot find suitable msvc.zUnknown platform %s)rr startswithmaxsizer RuntimeError)rDbasemsg conda_msgplt compilersmsgwinmsgrrrr7fs&      z(Toolchain._raise_external_compiler_errorrNcCs |jj|||||p g|d}|S)a Compile the given source files into a separate object file each, all beneath the *output_dir*. A list of paths to object files is returned. *macros* has the same format as in distutils: a list of 1- or 2-tuples. If a 1-tuple (name,), the given name is considered undefined by the C preprocessor. If a 2-tuple (name, value), the given name is expanded into the given value by the C preprocessor. )r%r?dependsmacros extra_preargs)r:r()rDsourcesr%r?rZr[ extra_cflagsobjectsrrrcompile_objects{szToolchain.compile_objectsc Cs2tj|\}}|jjtj|||||||ddS)zu Create a shared library *output* linking the given *objects* and *libraries* (all strings). )export_symbolsr\N)rrsplitr:linkr SHARED_OBJECT) rDoutputr_r2r=ra extra_ldflagsr%output_filenamerrr link_shareds  zToolchain.link_sharedcCs0|jt}tjdkr|dg}||jdS)zJ Get the library arguments necessary to link with Python. rPmsvcrtr2)r; get_librariesr.rrrC)rDlibsrrrget_python_librariess  zToolchain.get_python_librariescCt|j|jdS)zL Get the library directories necessary to link with Python. r=)listr>rCrGrrrget_python_library_dirssz!Toolchain.get_python_library_dirscCrm)zo Get the include directories necessary to compile against the Python and Numpy C APIs. r?)rnr@rCrGrrrget_python_include_dirssz!Toolchain.get_python_include_dirscCs |j|S)zR Given a C extension's module name, return its intended filename. )r;get_ext_filename)rDext_namerrrrqs zToolchain.get_ext_filename)rrrN)r/r0r1rEpropertyrHsetterr7r`rhrlrorprqrrrrr3Js$      r3cCs"ddlmm}|j}t|_dS)Nr)numpy.distutils.exec_command distutils exec_command _exec_command)modorig_exec_commandrrr_patch_exec_commands r{c Ks(|dur tjdk}|durtjdk}d}tjdkr7|r7tjdd}t|r/|dd|g}n|d|g}d}ntjdkrJt|rJdd d |D}|pMd}ztj|||tjtjd d }Wn t ygYd Sw| \}}|j r{|rw|d7}||7}|dddkr|dd}|rt ||j |fS)zg Internal workhorse for exec_command(). Code from https://github.com/numpy/numpy/pull/7862 NposixSHELLz/bin/shz-crNFntcss|]}t|VqdSrF) _quote_arg).0rrrr sz _exec_command..T)shellenvstdoutstderruniversal_newlines) ) rr)environr _is_sequencer subprocessPopenPIPEOSError communicate returncodeprint) command use_shelluse_teer executableshproctexterrrrrrxsB        rxcCsd|vr d|vr d|S|S)zB Quote the argument for safe use in a shell command line. "rNz"%s"rrrrrrsrcCs6t|ttfr dSzt|WdStyYdSw)NFT) isinstancestrbyteslenr*rrrrr s r)NN)%distutils.ccompilerrrdistutils.command.build_extrdistutils.sysconfigrrvrnumpy.distutils.misc_util misc_utilrA functoolsrrrtempfilerrr contextlibr rrpartialfind_shared_endingfind_pyext_endingr$r-r6objectr.r3r{rxrrrrrrs6       { 3