o wi? @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*d4ddZ+d4ddZ,ddZ-eddZ.iZ/e.dZ0d5ddZ1d d!Z2e+ej34ed"ddd#d#dddf d$d%Z5d&Z6dddej7d'fd(d)Z8de5_9e:d*j;Zd.d/Z?d0d1Z@Gd2d3d3ZAdS)6N) 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)namepathspecmodulerP/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/Cython/Build/Inline.py load_dynamics  r!cs0eZdZfddZddZfddZZS)UnboundSymbolscstt|jddt|_dS)N)context)superr __init__setunbound)self __class__rr r%$s zUnboundSymbols.__init__cCs"||js|j|j|SN) current_envlookuprr'addr(noderrr visit_NameNode'szUnboundSymbols.visit_NameNodecst||jSr+)r$__call__r'r/r)rr r2+s zUnboundSymbols.__call__)__name__ __module__ __qualname__r%r1r2 __classcell__rrr)r r"#s r"cCs|durtgtttd}ddlm}td|}t|dD]}|dur&q||}t ||r1nqddl }t t |t t|S)Noptionsr)AnalyseDeclarationsTransformz(tree fragment)pyxr)rr rrCompiler.ParseTreeTransformsr9r rcreate_pipeline isinstancebuiltinstupler"r&dir)coder#r9treephaser>rrr unbound_symbols0s   rDcCst|}|tur dSt||S)Nlong)typeint safe_type)argr#py_typerrr unsafe_typeAs rKcCst|}|ttttfvr|jS|turdS|turdS|tur!dSdt j vr8t |t j dj r8d|j j|jfS|jD]'}|jdvrEdS|j|jdd }|rb||j}|jrbd |j|jfSq;dS) Nzdouble complexdoublebintnumpyz"numpy.ndarray[numpy.%s_t, ndim=%s]) __builtin__r>objectF)need_pxdz%s.%s)rFlistr?dictstrr3complexfloatboolsysmodulesr=ndarraydtyperndim__mro__r4 find_moduler-is_type)rIr#rJ base_typerentryrrr rHIs*   rHcCs,t}|}||t|}||Sr+)rfind_config_filesparse_config_filesrfinalize_options)dist config_filesbuild_extensionrrr _get_build_extensionas  rhcCstt|tttdS)Nr7)rrRr rr)cython_include_dirsrrr _create_contextls rj).cCs|D]H}||vrJ|dus|dur%tjjj}|dur|j}|dur%|j}t|ts.t|}||vr9||||<q||vrD||||<qtd|qdS)NzCouldn't find %r)inspect currentframef_backf_locals f_globalsr=rSprint)kwdsrDlocalsglobalssymbol calling_framerrr _populate_unboundys"  rwcCs.||tjtj|tjf}tt|d S)Nzutf-8) rX version_info executableCython __version__hashlibsha256rTencode hexdigest) orig_codearg_sigslanguage_levelkeyrrr _inline_keysrinlineFc , sjdurdd|rtt|nt|rt|ni}| dur$d|vr$d} | dur,| |d<d} t|} | durmt| ||t} tfdd| D}t ||| } t||| f}|durmdd| D}||S|}t |\}}t |}|durt jjj}|durt jjj}zt|t|<} t| ||Wnty|stdYnwg}tD]\}}|tur|d ||=qt}tfd d|D}| durt ||| } d | }|tjvrtj|}n)d}tjdurt}|d t_tj|}tj ||tj}tj!|s#t"||s-tj#|sg}g}g}t$%d }|D]1\}} |&|}!|!rj|d|!'d|!'ddkrjddl(}"||")|dq:t*|\}#}$d dd|D}%dd ||#|%|$d}&|D] \}'}(|&+|'|(}&qtj ||d})t,|)d}*z |*-|&W|*.n|*.wt/||)g|pd|pd|pdd}+|durt}t0|+g|pdg||d|_1tj2|)|_3||_4|5tj6dkrtj7dkrt8tj| t9||}Wdn 1swYnt9||}|j:t||| f<fdd|D}|j:|S)NcSsdS)NrPr)xrrr szcython_inline..r3csg|] \}}||fqSrr).0rIvalue)ctxget_typerr z!cython_inline..cSsg|]}|dqS)rrrrIrrr rz>Could not parse code as a string (to extract unbound symbols).z cimport cython as %scsg|] }||fqSrrrrrrrrr rr_cython_inline_z ([.\w]+)[.]z cimport %srrN)NPY_NO_DEPRECATED_APINPY_1_7_API_VERSIONz, cSsg|]}d|qS)z%s %sr)rarrr rrzf %(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_macrosrk) include_pathcompiler_directivesquietwin32)csg|]}|qSrrr)rrrr rr);rjr?_cython_inline_default_contextrS_cython_inline_cachegetrwsorteditemsrrstrip_common_indentrlrmrnrorprDAssertionErrorrqrR cython_moduleappendrXrY cython_inlineso_extrhget_ext_filenameosrabspathjoinexistsmakedirsisfilerecompilematchgroupsrN get_includeextract_func_codereplaceopenwritecloserr extensionsdirname build_temp build_librunplatformrxadd_dll_directoryr!__invoke),rArlib_dirricython_compiler_directivesforcerrsrtrrrkey_hash_unbound_symbolsargsrinvokearg_listrliteralsrrrI arg_names module_namerrg module_pathcflagsrc_include_dirs qualifiedrF_mrNrrr module_coderrpyx_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}|r t|nd}|rt|nd}|rt|jnd}tj||d||dd}t|||d} | d | d fd d t t rSd 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_codebenchmark_code)rrs__PYX_make_benchmark__PYX_repeat_benchmarkcsF}t}tz||}W|rt|S|r"twwr+)gc isenableddisableenable)number benchmarkgcoldtiming)make_benchmarkrepeat_benchmarktimerrr timeit\s zcymeit..timeitiʚ;g?csrd} dD].}||}|}t|rtntsJ|kr"|Sr3|dkr3|dkr3tdq|d9}q)NrT)rrr z&Timer seems to return non-ns timings: )r=rGrV RuntimeError)ijr 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=rGrange) rAr import_module directivesrrepeatrmodule_namespace cymeit_code namespace one_secondrtimingsr)rrrrrrrrr cymeit1s<   r z\ScCsd}|}|D]}t|}|sq|}||dkrq|dus$||kr&|}qt|D]\}}t|}|rA|rA|||ddkrBq+||d||<q+d|S)N#rr) splitlines_find_non_spacestart enumerater)rA min_indentlineslinerrixrrr rs$  rzI^((cdef +(extern|class))|cimport|(from .+ cimport)|(from .+ import +[*]))cCslg}g}|}|dd}|d}|D]}|ds$t|r"|}n|}||qd|dd|fS)N  rrz )rsplit startswithmodule_statementrrr)rArfunctioncurrentrrrrr rs     rcCs>|d}|dddkrd||ddS||ddS)N:rlambdaz return %sr)index)sourcerrrr get_bodys r"c@seZdZddZddZdS)RuntimeCompiledFunctioncCs||_tt||_dSr+)_fr"rl getsource_body)r(frrr r%sz RuntimeCompiledFunction.__init__cOs:tj|jg|Ri|}t|jf|jj|jjd|S)N)rsrt)rl getcallargsr$rr& __globals__)r(rrrallrrr r2s z RuntimeCompiledFunction.__call__N)r3r4r5r%r2rrrr r#s r#r+)NN)Brr|rlrrrXtimedistutils.corerrdistutils.command.build_extrrz Compiler.MainrCompiler.Optionsrrr Compiler.Visitorr r r;r Compiler.TreeFragmentr DependenciesrrrCacherCompilerrcythonrimportlib.utilrimportlib.machineryrr!r"rDrKrHrhrjrrrwrrrrr perf_counterr rrsearchrrrrr"r#rrrr sb               b