B \$@s$ddlmZmZmZddlmZmZddlmZddl m Z ddl m Z ddl m mZddlZddlZddlZddlZddlmZmZmZddlmZd d d Zd d ZeedZeedZeddZddZ e Z!Gddde"Z#Gddde"Z$ddZ%dddZ&ddZ'ddZ(dS) )print_functiondivisionabsolute_import) CCompiler new_compiler) build_ext)customize_compiler)logN)NamedTemporaryFilemkdtemp gettempdir)contextmanager)z.dllz.pyd)z.soz.so)windefaultcCsttjddtd|S)Nr)_configsgetsysplatform)argr2lib/python3.7/site-packages/numba/pycc/platform.py get_configssrc cs^z(t}ttj|d|d}|VWdy|t|Wn Yn Xt|XdS)Nztemp%sZwt)r openospathjoincloseremovermdir)suffixZtmpdirntfrrr _gentmpfiles r#c Cst}t|xldD]d}yJt|8}d}|||||j|jgtdWdQRXWqt k rvdSXqWdS)N)z.cz.cxxzint main(void) { return 0; }) output_dirFT) rrr#writeflushrcompilenamer Exception)Zcompilerr!r"Zsimple_crrr_check_external_compiler0s   " r*c@seZdZgZdS)_DummyExtensionN)__name__ __module__ __qualname__ librariesrrrrr+Hsr+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_errorZdistutils.distr1_verboser _compilerrr _build_extZfinalize_options library_dirs _py_lib_dirs include_dirs_py_include_dirsnp_miscZget_info _math_info)selfr1rrr__init__Ns      zToolchain.__init__cCs|jS)N)r4)r=rrrverbose]szToolchain.verbosecCs ||_t|rtjntjdS)N)r4r Z set_thresholdINFOZWARN)r=valuerrrr?ascCsd}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 %sZlinuxlz 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)r=ZbasemsgZ conda_msgZpltZ compilersmsgZwinmsgrrrr3hs      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$r9dependsmacros extra_preargs)r5r')r=Zsourcesr$r9rIrJZ extra_cflagsobjectsrrrcompile_objects}szToolchain.compile_objectsc Cs2tj|\}}|jjtj|||||||ddS)zu Create a shared library *output* linking the given *objects* and *libraries* (all strings). )export_symbolsrKN)rrsplitr5linkrZ SHARED_OBJECT) r=outputrLr/r7rNZ extra_ldflagsr$Zoutput_filenamerrr link_shareds  zToolchain.link_sharedcCs0|jt}tjdkr"|dg}||jdS)zJ Get the library arguments necessary to link with Python. rDZmsvcrtr/)r6Z get_librariesr+rrr<)r=Zlibsrrrget_python_librariess  zToolchain.get_python_librariescCst|j|jdS)zL Get the library directories necessary to link with Python. r7)listr8r<)r=rrrget_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. r9)rTr:r<)r=rrrget_python_include_dirssz!Toolchain.get_python_include_dirscCs |j|S)zR Given a C extension's module name, return its intended filename. )r6get_ext_filename)r=Zext_namerrrrWszToolchain.get_ext_filename)rrrN)rrrN)r,r-r.r>propertyr?setterr3rMrRrSrUrVrWrrrrr0Ls    r0cCs"ddlmm}|j}t|_dS)Nr)Znumpy.distutils.exec_command distutilsZ exec_command _exec_command)modZorig_exec_commandrrr_patch_exec_commandsr]c Ks,|dkrtjdk}|dkr$tjdk}d}tjdkrn|rntjdd}t|r^|dd|g}n |d|g}d}n&tjdkrt|rdd d |D}|pd}ytj|||tjtjd d }Wnt k rd 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 NposixZSHELLz/bin/shz-crBFntcss|]}t|VqdS)N) _quote_arg).0rrrr sz _exec_command..T)shellenvstdoutstderrZuniversal_newlines) ) rr(environr _is_sequencer subprocessPopenPIPEEnvironmentErrorZ communicate returncodeprint) ZcommandZ use_shellZuse_teerd executableZshproctexterrrrrr[s>       r[cCsd|krd|krd|S|S)zB Quote the argument for safe use in a shell command line. "rBz"%s"r)rrrrr`sr`cCs8t|ttfrdSy t|dStk r2dSXdS)NFT) isinstancestrbyteslenr))rrrrrl srl)NN))Z __future__rrrZdistutils.ccompilerrrZdistutils.command.build_extrZdistutils.sysconfigrrZr Znumpy.distutils.misc_utilZ misc_utilr; functoolsrrmrZtempfiler r r contextlibr rrpartialZfind_shared_endingZfind_pyext_endingr#r*r2objectr+r0r]r[r`rlrrrrs2       { 3