U hd):@sfdZddlZddlZddlZddlZddlmZddlmZm Z m Z m Z ddl m Z GdddeZGdd d e Zed ZejZejZejZejZejZejZejZd d Zd dZddZddZddZddZGdddeZ GdddeZ!d2ddZ"ddZ#dd Z$d!d"Z%e&d#d$Z'ej(d%d&Z)d'd(Z*d)d*Z+d+d,Z,ej(d3d.d/Z-d0d1Z.dS)4z< Utilities to simplify the boilerplate for native lowering. N)Enum)typingtypesutilscgutils)BaseRegistryLoaderc@sbeZdZdZdddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ dS)Registryz? A registry of function and attribute implementations. unspecifiedcCs(||_g|_g|_g|_g|_g|_dSN)name functionsgetattrssetattrscasts constants)selfr rL/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/numba/core/imputils.py__init__s zRegistry.__init__csfdd}|S)a1 Decorate an implementation of *func* for the given argument types. *func* may be an actual global function object, or any pseudo-function supported by Numba, such as "getitem". The decorated implementation has the signature (context, builder, sig, args). csj|f|Sr )r appendimplargtysfuncrrrdecorate%sz Registry.lower..decorater)rrrrrrrlowers zRegistry.lowercCs"||||}||||jf|Sr )r signature)rrtyattrZ impl_list decorator real_implrrr_decorate_attr*s zRegistry._decorate_attrcsfdd}|S)z Decorate an implementation of __getattr__ for type *ty* and the attribute *attr*. The decorated implementation will have the signature (context, builder, typ, val). cs|jtSr )r"r _decorate_getattrrrrrrrr7sz(Registry.lower_getattr..decoraterrrrrrr$r lower_getattr/szRegistry.lower_getattrcCs ||dS)a: Decorate the fallback implementation of __getattr__ for type *ty*. The decorated implementation will have the signature (context, builder, typ, val, attr). The implementation is called for attributes which haven't been explicitly registered with lower_getattr(). N)r&rrrrrlower_getattr_generic<s zRegistry.lower_getattr_genericcsfdd}|S)z Decorate an implementation of __setattr__ for type *ty* and the attribute *attr*. The decorated implementation will have the signature (context, builder, sig, args). cs|jtSr )r"r_decorate_setattrrr$rrrOsz(Registry.lower_setattr..decoraterr%rr$r lower_setattrGszRegistry.lower_setattrcCs ||dS)a; Decorate the fallback implementation of __setattr__ for type *ty*. The decorated implementation will have the signature (context, builder, sig, args, attr). The implementation is called for attributes which haven't been explicitly registered with lower_setattr(). N)r*r'rrrlower_setattr_genericTs zRegistry.lower_setattr_genericcsfdd}|S)z Decorate the implementation of implicit conversion between *fromty* and *toty*. The decorated implementation will have the signature (context, builder, fromty, toty, val). csj|ff|Sr )rrrfromtyrtotyrrrgsz%Registry.lower_cast..decorater)rr-r.rrr,r lower_cast_szRegistry.lower_castcsfdd}|S)z Decorate the implementation for creating a constant of type *ty*. The decorated implementation will have the signature (context, builder, ty, pyval). csj|ff|Sr )rrrr'rrrssz)Registry.lower_constant..decorater)rrrrr'rlower_constantlszRegistry.lower_constantcCsd|jdS)NzLowering Registry<>)r rrrr__repr__xszRegistry.__repr__N)r )__name__ __module__ __qualname____doc__rrr"r&r(r*r+r/r0r3rrrrrs       rc@seZdZdZdZdS)RegistryLoaderz6 An incremental loader for a target registry. )r r rrrN)r4r5r6r7Zregistry_itemsrrrrr8|sr8builtin_registrycs8||dk rfdd}n fdd}|f|_||_|S)Ncs||||Sr rcontextbuildertypvaluerr!rrressz_decorate_getattr..rescs|||||Sr rr:r?rrr@s)rrrrrr@rr?rr#s r#cs<||dk rfdd}n fdd}|tjf|_||_|S)Ncs||||Sr rr;r<sigargsrr?rrr@sz_decorate_setattr..rescs|||||Sr rrBr?rrr@s)rAnyrrrArr?rr)s  r)c Cstt|jtjrp|jj}|||}t||}|| |j  | |||}| ||W5QRX| |}|Sr ) isinstance return_typerOptionaltypeZmake_optional_nonerZalloca_once_valueif_thennot_Zis_noneZmake_optional_valuestoreload) r;r<rCstatusretvalZ value_typeZ optional_noneZ retvalptrZoptional_valuerrrfix_returning_optionals   rPcs"fdd}j|_t||_|S)zC A wrapper inserting code calling Numba-compiled *fndesc*. c s||j}|j||jj|\}}t||j|j ||W5QRX|j jksbt t |||||}|j ||j krd}t||j |j t||j|S)Nz$function returned {0} but expect {1})declare_functionmodule call_conv call_functionrestypeargtypesrZ if_unlikelyZis_errorZreturn_status_propagaterGAssertionErrorrPrIZget_value_type TypeErrorformatimpl_ret_new_ref)r;r<rCrDrrNrOmsgfndescrrimps zuser_function..imp)rVrtuplelibs)r]r`r^rr\r user_functions  racsfdd}t||_|S)zD A wrapper inserting code calling Numba-compiled *gendesc*. cs2||j}|j||jj|\}}||fSr )rQrRrSrTrUrV)r;r<rCrDrrNrOgendescrrr^szuser_generator..imp)r_r`)rcr`r^rrbruser_generators  rdcsfdd}|S)zl Decorator a given class as implementing *iterator_type* (by providing an `iternext()` method). cs0jttjfdd}td|S)Ncs |\}|||}||||Sr r)r;r<rCrDresultr>Ziterobjclsiternextrriternext_wrappers z8iterator_impl..wrapper..iternext_wrapperrh)rh iternext_implRefTypeBORROWED lower_builtin)rgri iterator_typerfrwrappers ziterator_impl..wrapperr) iterable_typerorprrnr iterator_impls rrc@sFeZdZdZdZddZddZddd Zd d Zd d Z ddZ dS)_IternextResultzb A result wrapper for iteration, passed by iternext_impl() into the wrapped function. _context_builder_pairobjcCs||_||_||_dSr rt)rr;r<pairobjrrrrsz_IternextResult.__init__cCs|jtjd|j_dS)z1 Mark the iterator as exhausted. FNruZ get_constantrbooleanrwsecondr2rrr set_exhausted sz_IternextResult.set_exhaustedTcCs$|dkr|jtj|}||j_dS)z Mark the iterator as valid according to *is_valid* (which must be either a Python boolean or a LLVM inst). )FTNry)ris_validrrr set_validsz_IternextResult.set_validcCs ||j_dS)zP Mark the iterator as yielding the given *value* (a LLVM inst). Nrwfirst)rr>rrryield_sz_IternextResult.yield_cCs|j|jtj|jjS)z> Return whether the iterator is marked valid. )ruZget_argument_valuervrrzrwr{r2rrrr}s z_IternextResult.is_validcCs|jjS)z> Return the iterator's yielded value, if any. rr2rrr yielded_value&sz_IternextResult.yielded_valueN)T) r4r5r6r7 __slots__rr|r~rr}rrrrrrss rsc@seZdZdZdZdZdZdS)rkz& Enumerate the reference type N)r4r5r6r7NEWrl UNTRACKEDrrrrrk,srkcs*ddtDkrtdfdd}|S)a Wrap the given iternext() implementation so that it gets passed an _IternextResult() object easing the returning of the iternext() result pair. ref_type: a numba.targets.imputils.RefType value, the reference type used is that specified through the RefType enum. The wrapped function will be called with the following signature: (context, builder, sig, args, iternext_result) cSsg|]}|qSrr).0xrrr Isz!iternext_impl..z3ref_type must be an enum member of imputils.RefTypecsfdd}|S)Nc st|j}|||}||||t|||tjkr:t}n(tjkrJt}ntjkrZt }nt d||||| S)NzUnknown ref_type encountered) rG make_helperrsrkrrZrlimpl_ret_borrowedrimpl_ret_untracked ValueErrorZ _getvalue)r;r<rCrD pair_typerxZimpl_ret)rref_typerrrpMs       z-iternext_impl..outer..wrapperr)rrpr)rrouterLsziternext_impl..outer)rkr)rrrrrrj=s  rjcCs&t|j|}|d|}|||fS)z Call the `getiter()` implementation for the given *iterable_type* of value *val*, and return the corresponding LLVM inst. Zgetiter)rrro get_function)r;r<rqvalZ getiter_sigZ getiter_implrrr call_getiter`s rc CsR|j}t|tj}t||}|d|}|||f}||||}t|||S)z Call the `iternext()` implementation for the given *iterator_type* of value *val*, and return a convenience _IternextResult() object reflecting the results. rh) Z yield_typerZPairrzrrrrrs) r;r<rorZitemtyrZ iternext_sigrjrxrrr call_iternextjs   rcCsBz|tttj|}Wntk r0YdSX|||fSdS)z^ Call len() on the given value. Return None if len() isn't defined on this type. N)rlenrrrZintpNotImplementedError)r;r<rrZlen_implrrrcall_lenys r _ForIterLoop)r>do_breakc #s|j}t|||}d}dfdd}||\t|||}j|ddW5QRXt | |V|W5QRX |j r|j ||dS)z Simulate a for loop on the given iterable. Yields a namedtuple with the given members: - `value` is the value being yielded - `do_break` is a callable to early out of the loop z for_iter.bodyz for_iter.endcsdSr )branchrZbb_endr<rrrszfor_iter..do_breakF)ZlikelyN)rorZappend_basic_blockrZ goto_blockrrJrKr}rrZposition_at_end enable_nrtnrtZdecref) r;r<rqrroZitervalZbb_bodyrr@rrrfor_iters     rcCs|S)z5 The implementation returns a new reference. rctxr<ZrettyretrrrrZsrZcCs|jr|j||||S)z The implementation returns a borrowed reference. This function automatically incref so that the implementation is returning a new reference. )rrZincrefrrrrrsrcCs|S)z. The return type is not a NRT object. rrrrrrsrnumpyccs8ddlm}|j}||||_z dVW5||_XdS)z7 Temporarily change the context's error model. r)callconvN) numba.corerZ error_modelZcreate_error_model)r;Z model_namerZold_error_modelrrrforce_error_models   rcOs tddS)zPA stub for use internally by Numba when a call is emitted on a TypeRef. z%This function should not be executed.N)r)rDkwargsrrrnumba_typeref_ctorsr)N)r)/r7 collections contextlibinspect functoolsenumrrrrrrZnumba.core.typing.templatesrobjectrr8r9rrmr&r(r*r+r/r0r#r)rPrardrrrsrkrjrrr namedtuplercontextmanagerrrZrrrrrrrrsP  l / #