U {h? @sddlZddlZddlZddlZddlZddlZddlZddlmZm Z ddl m Z ddl Z ddl mZddlmZmZmZddlmZmZddlmZdd lmZd d lmZmZmZd d lmZdd l m!Z!ddl"Z#ddl$Z%ddl&m'Z'ddZ(GdddeeZ)ed4ddZ*d5ddZ+d6ddZ,ddZ-eddZ.iZ/e.dZ0d7ddZ1d d!Z2e+ej34ed"ddd#d#dddf d$d%Z5d&Z6dddej7d'fd(d)Z8de5_9e:d*j;Zd.d/Z?d0d1Z@Gd2d3d3ZAdS)8N) Distribution Extension) build_ext)Context)default_optionsCompilationOptionsget_directive_defaults)CythonTransform EnvTransform)SkipDeclarations)parse_from_strings)strip_string_literals cythonizecached_function)get_cython_cache_dir)Pipeline)ExtensionFileLoadercCs2tjj|t||d}tj|}|j||S)N)loader) importlibutilspec_from_file_locationrmodule_from_specr exec_module)namepathspecmoduler7/tmp/pip-unpacked-wheel-fhl22ezh/Cython/Build/Inline.py load_dynamics  r!cs0eZdZfddZddZfddZZS)UnboundSymbolscstt|jddt|_dS)N)context)superr __init__setunbound)self __class__rr r%$szUnboundSymbols.__init__cCs"||js|j|j|SN)Z current_envlookuprr'addr(noderrr visit_NameNode'szUnboundSymbols.visit_NameNodecst||jSr+)r$__call__r'r.r)rr r1+s zUnboundSymbols.__call__)__name__ __module__ __qualname__r%r0r1 __classcell__rrr)r r"#s r"cCs|dkrtgtttd}ddlm}td|}t|dD]$}|dkrLq>||}t ||r>qdq>ddl }t t |t t|S)Noptionsr)AnalyseDeclarationsTransformz(tree fragment)Zpyxr)rr rrCompiler.ParseTreeTransformsr8r rZcreate_pipeline isinstancebuiltinstupler"r&dir)coder#r8treephaser;rrr unbound_symbols0s   rAcCs"t|}|tkrdSt||SdS)Nlong)typeint safe_type)argr#py_typerrr unsafe_typeAsrHcCst|}|ttttfkr|jS|tkr*dS|tkr6dS|tkrBdSdt j krpt |t j dj rpd|j j|jfS|jD]N}|jdkrdS|j|jdd }|rv||j}|jrvd |j|jfSqvdSdS) Nzdouble complexdoubleZbintnumpyz"numpy.ndarray[numpy.%s_t, ndim=%s]) __builtin__r;objectF)Zneed_pxdz%s.%s)rClistr<dictstrr2complexfloatboolsysmodulesr:ZndarrayZdtyperndim__mro__r3 find_moduler,Zis_type)rFr#rGZ base_typerentryrrr rEIs(   rEcCs,t}|}||t|}||Sr+)rfind_config_filesparse_config_filesrfinalize_options)dist config_filesbuild_extensionrrr _get_build_extensionas  r_cCstt|tttdS)Nr6)rrMr rr)cython_include_dirsrrr _create_contextls ra).cCs|D]}||kr|dks |dkrJtjjj}|dkr<|j}|dkrJ|j}t|ts\t|}||krr||||<q||kr||||<qtd|qdS)NzCouldn't find %r)inspect currentframef_backf_locals f_globalsr:rNprint)kwdsrAlocalsglobalssymbolZ calling_framerrr _populate_unboundys rmcCs.||tjtj|tjf}tt|d S)Nzutf-8) rS version_info executableCython __version__hashlibsha256rOencode hexdigest) orig_codearg_sigslanguage_levelkeyrrr _inline_keysrzinlineFc , s^dkrdd|r tt|nt|r0t|ni}| dkrHd|krHd} | dk rX| |d<d} t|} | dk rt| ||t} tfdd| D}t ||| } t||| f}|dk rdd| D}||S|}t |\}}t |}|dkr t jjj}|dkr"t jjj}z"t|t|<} t| ||Wn$tk rh|sdtdYnXg}tD](\}}|tkrz|d ||=qzt}tfd d|D}| dkrt ||| } d | }|tjkrtj|}n4d}tjdkr t}|d t_tj|}tj ||tj}tj!|sXt"||sltj#|sg}g}g}t$%d }|D]b\}} |&|}!|!r|d|!'d|!'ddkrddl(}"||")|dqt*|\}#}$d dd|D}%dd ||#|%|$d}&|D]\}'}(|&+|'|(}&q*tj ||d})t,|)d}*z|*.|&W5|*-Xt/||)g|pd|pd|pdd}+|dkrt}t0|+g|pdg||d|_1tj2|)|_3||_4|5tj6dkr(tj7dkr(t8tj|t9||}W5QRXn t9||}|j:t||| f<fdd|D}|j:|S)NcSsdS)NrLr)xrrr zcython_inline..rx3csg|]\}}||fqSrr).0rFvalue)ctxget_typerr sz!cython_inline..cSsg|] }|dqS)rrrrFrrr rsz>Could not parse code as a string (to extract unbound symbols).z cimport cython as %scsg|]}||fqSrrrrrrirr rsZ_cython_inline_z ([.\w]+)[.]z cimport %srrJ)ZNPY_NO_DEPRECATED_APIZNPY_1_7_API_VERSIONz, cSsg|] }d|qS)z%s %sr)rarrr rszf %(module_body)s %(cimports)s def __invoke(%(params)s): %(func_body)s return locals()  )cimports module_bodyparams func_bodyz.pyxw)rsources include_dirsextra_compile_args define_macrosrb)Z include_pathZcompiler_directivesquietwin32)csg|] }|qSrrr)rirr rs);rar<_cython_inline_default_contextrN_cython_inline_cachegetrmsorteditemsrzrstrip_common_indentrcrdrerfrgrAAssertionErrorrhrM cython_moduleappendrSrT cython_inlineso_extr_Zget_ext_filenameosrabspathjoinexistsmakedirsisfilerecompilematchgroupsrJZ get_includeextract_func_codereplaceopenclosewriterr extensionsdirname build_temp build_librunplatformrnZadd_dll_directoryr!Z__invoke),r>rlib_dirr`cython_compiler_directivesforcerrjrkrxriZkey_hashZ_unbound_symbolsargsrwZinvokearg_listrvliteralsrrrF arg_names module_namerr^ module_pathcflagsrZc_include_dirsZ qualifiedrC_mrJrrrZ module_coderyrZpyx_filefh extensionrrr rs                         ra= def __PYX_repeat_benchmark(benchmark, timer, size_t number): cdef size_t i t0 = timer() for i in range(number): benchmark() t1 = timer() return t1 - t0 def __PYX_make_benchmark(): {setup_code} def __PYX_run_benchmark(): {benchmark_code} return __PYX_run_benchmark c sddl}|rt|nd}|r(t|nd}|r:t|jnd}tj||d||dd}t|||d} | d | d fd d t t rd nd} r| dn| dfdd} | | fddt |D} dfdd| D} | fS)aBenchmark a Cython code string similar to 'timeit'. 'setup_code': string of setup code that will be run before taking the timings. 'import_module': a module namespace to run the benchmark in (usually a compiled Cython module). 'directives': Cython directives to use when compiling the benchmark code. 'timer': The timer function. Defaults to 'time.perf_counter', returning float seconds. Nanosecond timers are detected (and can only be used) if they return integers. 'repeat': The number of timings to take and return. Returns a tuple: (list of single-loop timings, number of loops run for each) rNrpass z ) setup_codeZbenchmark_code)rrjZ__PYX_make_benchmarkZ__PYX_repeat_benchmarkcs:}t}tz||}W5|r4tX|Sr+)gc isenableddisableenable)numberZ benchmarkZgcoldtiming)make_benchmarkrepeat_benchmarktimerrr timeit\s zcymeit..timeitiʚ;g?cstd}dD]\}||}|}t|r(tnts2t|krB|Sr|dkr|dkrtdq|d9}qdS)Nr)rrr z&Timer seems to return non-ns timings: )r:rDrQr RuntimeError)ijrZ time_taken) min_runtimerrtimer_returns_nanosecondsrr autorangepszcymeit..autorangecsg|] }qSrr)rr)rrrr rszcymeit..rcs$g|]}r|n|qSrr)rr)halfrrrr rs) textwraprstrip __import____dict___CYMEIT_TEMPLATEformatindentrr:rDrange) r>r import_moduleZ directivesrrepeatrZmodule_namespaceZ cymeit_code namespaceZ one_secondrZtimingsr)rrrrrrrrr cymeit1s<   rz\ScCsd}|}|D]<}t|}|s"q|}||dkr8q|dksH||kr|}qt|D]>\}}t|}|rV|rV|||ddkrqV||d||<qVd|S)N#rr) splitlines_find_non_spacestart enumerater)r>Z min_indentlineslinerrixrrr rs" rzI^((cdef +(extern|class))|cimport|(from .+ cimport)|(from .+ import +[*]))cCslg}g}|}|dd}|d}|D],}|dsHt|rD|}n|}||q&d|dd|fS)N  rrz )rsplit startswithmodule_statementrrr)r>rfunctioncurrentrrrrr rs     rcCsB|d}|dddkr.d||ddS||ddSdS)N:rlambdaz return %sr)index)sourcerrrr get_bodys rc@seZdZddZddZdS)RuntimeCompiledFunctioncCs||_tt||_dSr+)_frrc getsource_body)r(frrr r%sz RuntimeCompiledFunction.__init__cOs4tj|jf||}t|jf|jj|jjd|S)N)rjrk)rc getcallargsrrr __globals__)r(rriallrrr r1sz RuntimeCompiledFunction.__call__N)r2r3r4r%r1rrrr rsr)N)N)N)NN)BrrrrcrrrStimedistutils.corerrZdistutils.command.build_extrrpZ Compiler.MainrZCompiler.Optionsrrr ZCompiler.Visitorr r r9r ZCompiler.TreeFragmentr Z DependenciesrrrCacherZCompilerrZcythonrimportlib.utilrZimportlib.machineryrr!r"rArHrEr_rarrrmrzrrrr perf_counterrrrsearchrrrrrrrrrr sj                b