U hd@sddlZddlmZddlmZddlZddlZddlZddl Z ddl Z ddl m Z ddl mZddlmZejjjZejjjZejjjZejjZddd Zd d ZeedZeed Zed dZejd dddZGddde Z!Gddde Z"ddZ#ddZ$dS)N) build_ext) Distribution)mkdtemp)contextmanager)Path)z.dllz.pyd).sor)windefaultcCsttjddtd|S)Nr )_configsgetsysplatformargrL/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/numba/pycc/platform.py get_configssrc cs^z(t}ttj|d|d}|VW5z|t|Wn Yn Xt|XdS)Nztemp%swt)closeosremovermdirropenpathjoin)suffixntfZtmpdirrrr _gentmpfile#s r)maxsizec Cst}t|dD]n}zPt|>}d}|||||j|jgt|jj dW5QRXWqt k r~YdSXqdS)zv Returns True if the "external compiler" bound in numpy.distutil is present and working, False otherwise. )z.cz.cxxzint main(void) { return 0; }) output_dirFT) new_compilercustomize_compilerrwriteflushrcompilenameranchor Exception)compilerrrZsimple_crrrexternal_compiler_works5s  ( r+c@seZdZgZdS)_DummyExtensionN)__name__ __module__ __qualname__ librariesrrrrr,Lsr,c@sjeZdZddZeddZejddZddZdd d Zdd d Z ddZ ddZ ddZ ddZ d S) ToolchaincCsts|d|_t|_t|jtt|_|j |jj |_ |jj |_ dtgi}tjdkrrg|d<n dg|d<||_dS)NF include_dirswin32r0m)r+_raise_external_compiler_error_verboser" _compilerr#rr _build_extfinalize_options library_dirs _py_lib_dirsr2_py_include_dirsnpZ get_includer r _math_info)selfZnp_compile_argsrrr__init__Rs        zToolchain.__init__cCs|jSN)r6r?rrrverbosedszToolchain.verbosecCs ||_t|rtjntjdSrA)r6log set_thresholdINFOWARN)r?valuerrrrChscCsd}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-64r3zCannot find suitable msvc.zUnknown platform %s)r r startswithr r RuntimeError)r?ZbasemsgZ conda_msgZpltZ compilersmsgZwinmsgrrrr5os      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!r2dependsmacros extra_preargs)r7r&)r?sourcesr!r2rOrPZ extra_cflagsobjectsrrrcompile_objectsszToolchain.compile_objectsc Cs2tj|\}}|jjtj|||||||ddS)zu Create a shared library *output* linking the given *objects* and *libraries* (all strings). )export_symbolsrQN)rrsplitr7link CCompilerZ SHARED_OBJECT) r?outputrSr0r:rUZ extra_ldflagsr!Zoutput_filenamerrr link_shareds zToolchain.link_sharedcCs0|jt}tjdkr"|dg}||jdS)zJ Get the library arguments necessary to link with Python. r3msvcrtr0)r8Z get_librariesr,r rr>)r?Zlibsrrrget_python_librariess  zToolchain.get_python_librariescCs t|jS)zL Get the library directories necessary to link with Python. )listr;rBrrrget_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. r2)r]r<r>rBrrrget_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)r?Zext_namerrrr`szToolchain.get_ext_filename)rrrN)rrrN)r-r.r/r@propertyrCsetterr5rTrZr\r^r_r`rrrrr1Ps(     r1cCsd|krd|krd|S|S)zB Quote the argument for safe use in a shell command line. "rJz"%s"rrrrr _quote_argsrdcCs<t|ttfrdSzt|WdStk r6YdSXdS)NFT) isinstancestrbyteslenr)rrrr _is_sequencesri)% setuptoolsZsetuptools.command.build_extrZsetuptools.distrnumpyr= functoolsr subprocessr tempfiler contextlibrpathlibr distutilsZ ccompilerrXr" sysconfigr#rDr rpartialZfind_shared_endingZfind_pyext_endingr lru_cacher+objectr,r1rdrirrrrs8             x