U |e):@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 rP/var/www/website-v5/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)r4r5r6r7registry_itemsrrrrr8|sr8builtin_registrycs8||dk rfdd}n fdd}|f|_||_|S)Ncs||||Sr rcontextbuildertypvaluerr!rrressz_decorate_getattr..rescs|||||Sr rr;r@rrrAs)rrrrrrArr@rr#s r#cs<||dk rfdd}n fdd}|tjf|_||_|S)Ncs||||Sr rr<r=sigargsrr@rrrAsz_decorate_setattr..rescs|||||Sr rrCr@rrrAs)rAnyrrrBrr@rr)s  r)c Cstt|jtjrp|jj}|||}t||}|| |j  | |||}| ||W5QRX| |}|Sr ) isinstance return_typerOptionaltypeZmake_optional_noneralloca_once_valueif_thennot_Zis_noneZmake_optional_valuestoreload) r<r=rDstatusretval value_typeZ optional_noneZ retvalptrZoptional_valuerrrfix_returning_optionals   rScs"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_functionrestypeargtypesr if_unlikelyZis_errorZreturn_status_propagaterHAssertionErrorrSrJget_value_type TypeErrorformatimpl_ret_new_ref)r<r=rDrErrPrQmsgfndescrrimps zuser_function..imp)rYrtuplelibs)rbrercrrar user_functions  rfcsfdd}t||_|S)zD A wrapper inserting code calling Numba-compiled *gendesc*. cs2||j}|j||jj|\}}||fSr )rTrUrVrWrXrY)r<r=rDrErrPrQgendescrrrcszuser_generator..imp)rdre)rhrercrrgruser_generators  ricsfdd}|S)zl Decorator a given class as implementing *iterator_type* (by providing an `iternext()` method). cs0jttjfdd}td|S)Ncs |\}|||}||||Sr r)r<r=rDrEresultr?Ziterobjclsiternextrriternext_wrappers z8iterator_impl..wrapper..iternext_wrapperrm)rm iternext_implRefTypeBORROWED lower_builtin)rlrn iterator_typerkrwrappers ziterator_impl..wrapperr) iterable_typertrurrsr iterator_impls rwc@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 ry)rr<r=pairobjrrrrsz_IternextResult.__init__cCs|jtjd|j_dS)z1 Mark the iterator as exhausted. FNrzZ get_constantrbooleanr|secondr2rrr 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). )FTNr~)ris_validrrr set_validsz_IternextResult.set_validcCs ||j_dS)zP Mark the iterator as yielding the given *value* (a LLVM inst). Nr|first)rr?rrryield_sz_IternextResult.yield_cCs|j|jtj|jjS)z> Return whether the iterator is marked valid. )rzZget_argument_valuer{rrr|rr2rrrrs z_IternextResult.is_validcCs|jjS)z> Return the iterator's yielded value, if any. rr2rrr yielded_value&sz_IternextResult.yielded_valueN)T) r4r5r6r7 __slots__rrrrrrrrrrrxs rxc@seZdZdZdZdZdZdS)rpz& Enumerate the reference type N)r4r5r6r7NEWrq UNTRACKEDrrrrrp,srpcs*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) rH make_helperrxrprr_rqimpl_ret_borrowedrimpl_ret_untracked ValueError _getvalue)r<r=rDrE pair_typer}Zimpl_ret)rref_typerrruMs       z-iternext_impl..outer..wrapperr)rrur)rrouterLsziternext_impl..outer)rpr)rrrrrro=s  rocCs&t|j|}|d|}|||fS)z Call the `getiter()` implementation for the given *iterable_type* of value *val*, and return the corresponding LLVM inst. getiter)rrrt get_function)r<r=rvvalZ 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. rm) yield_typerPairrrrrrrx) r<r=rtritemtyrZ iternext_sigror}rrr 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)rlenrrrintpNotImplementedError)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)likelyN)rtrappend_basic_blockr goto_blockrrLrMrrrposition_at_end enable_nrtnrtZdecref) r<r=rvrrtZitervalZbb_bodyrrArrrfor_iters     rcCs|S)z5 The implementation returns a new reference. rctxr=rettyretrrrr_sr_cCs|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< 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)rEkwargsrrrnumba_typeref_ctorsr)N)r)/r7 collections contextlibinspect functoolsenumrrrrrrnumba.core.typing.templatesrobjectrr8r:rrrr&r(r*r+r/r0r#r)rSrfrirwrxrprorrr namedtuplercontextmanagerrr_rrrrrrrrsP  l / #