U ,dh$@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 )_configsgetsysplatformargr7/tmp/pip-unpacked-wheel-eu7e0c37/numba/pycc/platform.py get_configssrc cs^z(t}ttj|d|d}|VW5z|t|Wn Yn Xt|XdS)Nztemp%swt)closeosremovermdirropenpathjoin)suffixntfZtmpdirrrr _gentmpfiles r#c Cst}t|dD]h}zJt|8}d}|||||j|jgtdW5QRXWqt k rxYdSXqdS)N)z.cz.cxxzint main(void) { return 0; }) output_dirFT) rrr#writeflushrcompilenamer Exception)compilerr!r"Zsimple_crrr_check_external_compiler.s  " r+c@seZdZgZdS)_DummyExtensionN)__name__ __module__ __qualname__ librariesrrrrr,Fsr,c@sjeZdZddZeddZejddZddZdd d Zdd d Z ddZ ddZ ddZ ddZ d S) ToolchaincCsjts |ddlm}d|_t|_t|jt||_ |j |j j |_ |j j |_td|_dS)Nr) DistributionFZnpymath)_external_compiler_ok_raise_external_compiler_errordistutils.distr2_verboser _compilerrr _build_extfinalize_options library_dirs _py_lib_dirs include_dirs_py_include_dirsnp_miscget_info _math_info)selfr2rrr__init__Ls      zToolchain.__init__cCs|jSN)r6rArrrverbose[szToolchain.verbosecCs ||_t|rtjntjdSrC)r6r set_thresholdINFOWARN)rAvaluerrrrE_scCsd}d}tj}|drLtjdkr,ddg}nddg}d ||d |f}nN|d rvd d g}d ||d |f}n$|drd}d ||f}nd|}t|dS)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)rAZbasemsgZ conda_msgZpltZ compilersmsgZwinmsgrrrr4fs      z(Toolchain._raise_external_compiler_errorrNcCs |jj|||||pg|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)r7r')rAsourcesr$r<rRrSZ extra_cflagsobjectsrrrcompile_objects{szToolchain.compile_objectsc Cs2tj|\}}|jjtj|||||||ddS)zu Create a shared library *output* linking the given *objects* and *libraries* (all strings). )export_symbolsrTN)rrsplitr7linkrZ SHARED_OBJECT) rAoutputrVr0r:rXZ extra_ldflagsr$Zoutput_filenamerrr link_shareds zToolchain.link_sharedcCs0|jt}tjdkr"|dg}||jdS)zJ Get the library arguments necessary to link with Python. rMmsvcrtr0)r8Z get_librariesr,rrr@)rAZlibsrrrget_python_librariess  zToolchain.get_python_librariescCst|j|jdS)zL Get the library directories necessary to link with Python. r:)listr;r@rDrrrget_python_library_dirssz!Toolchain.get_python_library_dirscCst|j|jdS)zo Get the include directories necessary to compile against the Python and Numpy C APIs. r<)r_r=r@rDrrrget_python_include_dirssz!Toolchain.get_python_include_dirscCs |j|S)zR Given a C extension's module name, return its intended filename. )r8get_ext_filename)rAZext_namerrrrbszToolchain.get_ext_filename)rrrN)rrrN)r-r.r/rBpropertyrEsetterr4rWr\r^r`rarbrrrrr1Js(     r1cCs"ddlmm}|j}t|_dS)Nr)Znumpy.distutils.exec_command distutilsZ exec_command _exec_command)modZorig_exec_commandrrr_patch_exec_commandsrhc Ks.|dkrtjdk}|dkr$tjdk}d}tjdkrn|rntjdd}t|r^|dd|g}n |d|g}d}n&tjdkrt|rdd d |D}|pd}ztj|||tjtjd d }Wnt k rYd SX| \}}|j r|r|d7}||7}|dddkr|dd}|r$t ||j |fS)zg Internal workhorse for exec_command(). Code from https://github.com/numpy/numpy/pull/7862 NposixSHELLz/bin/shz-crKFntcss|]}t|VqdSrC) _quote_arg).0rrrr sz _exec_command..T)shellenvstdoutstderruniversal_newlines) ) rr(environr _is_sequencer subprocessPopenPIPEOSError communicate returncodeprint) commandZ use_shellZuse_teerp executableshproctexterrrrrrfs@       rfcCsd|krd|krd|S|S)zB Quote the argument for safe use in a shell command line. "rKz"%s"rrrrrrlsrlcCs<t|ttfrdSzt|WdStk r6YdSXdS)NFT) isinstancestrbyteslenr)rrrrry sry)NN)%distutils.ccompilerrrZdistutils.command.build_extrdistutils.sysconfigrrerZnumpy.distutils.misc_utilZ misc_utilr> functoolsrrzrtempfilerrr contextlibr rrpartialZfind_shared_endingZfind_pyext_endingr#r+r3objectr,r1rhrfrlryrrrrs4       { 3