o 1&i;@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 r>C:\wamp64\www\opt\env\Lib\site-packages\numba/core/imputils.py__init__s  zRegistry.__init__cfdd}|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%z Registry.lower..decorater)rrrrrrrlowers zRegistry.lowercCs"||||}||||jf|Sr )r signature)rrtyattrZ impl_list decorator real_implrrr_decorate_attr*s zRegistry._decorate_attrcr)z Decorate an implementation of __getattr__ for type *ty* and the attribute *attr*. The decorated implementation will have the signature (context, builder, typ, val). c|jtSr )r$r _decorate_getattrrr!rr rrr7z(Registry.lower_getattr..decoraterrr r!rrr'r lower_getattr/zRegistry.lower_getattrcC ||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*rr rrrlower_getattr_generic< zRegistry.lower_getattr_genericcr)z Decorate an implementation of __setattr__ for type *ty* and the attribute *attr*. The decorated implementation will have the signature (context, builder, sig, args). cr%r )r$r_decorate_setattrrr'rrrOr(z(Registry.lower_setattr..decoraterr)rr'r lower_setattrGr+zRegistry.lower_setattrcCr,)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)r1r-rrrlower_setattr_genericTr/zRegistry.lower_setattr_genericcr)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)rr4r5rrr3r lower_cast_r+zRegistry.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-rrrsrz)Registry.lower_constant..decorater)rr rrr-rlower_constantlszRegistry.lower_constantcCsd|jdS)NzLowering Registry<>)r rrrr__repr__xzRegistry.__repr__N)r )__name__ __module__ __qualname____doc__rrr$r*r.r1r2r6r7r:rrrrrs      rc@seZdZdZdZdS)RegistryLoaderz6 An incremental loader for a target registry. )r r rrrN)r<r=r>r?Zregistry_itemsrrrrr@|sr@builtin_registrycs8||dur fdd}nfdd}|f|_||_|S)Nc||||Sr rcontextbuildertypvaluer!r#rrresr;z_decorate_getattr..resc|||||Sr rrCrHrrrI)rr!rr r!rIrrHrr&s r&cs<||dur fdd}nfdd}|tjf|_||_|S)NcrBr rrDrEsigargsr!rHrrrIr;z_decorate_setattr..rescrJr rrMrHrrrIrK)rAnyrr!rLrrHrr0s  r0c Cst|jtjrB|jj}|||}t||}|| |j | |||}| ||Wdn1s8wY| |}|Sr ) isinstance return_typerOptionaltypeZmake_optional_nonerZalloca_once_valueif_thennot_Zis_noneZmake_optional_valuestoreload) rDrErNstatusretvalZ value_typeZ optional_noneZ retvalptrZoptional_valuerrrfix_returning_optionals   r[cs"fdd}j|_t||_|S)zC A wrapper inserting code calling Numba-compiled *fndesc*. cs||j}|j||jj|\}}t||j|j ||Wdn1s.wY|j jks;Jt |||||}|j | |j krXd}t||j |j t||j|S)Nz$function returned {0} but expect {1})declare_functionmodule call_conv call_functionrestypeargtypesrZ if_unlikelyZis_errorZreturn_status_propagaterRr[rTZget_value_type TypeErrorformatimpl_ret_new_ref)rDrErNrOrrYrZmsgfndescrrimpszuser_function..imp)rartuplelibs)rgrjrhrrfr user_functions  rkcsfdd}t||_|S)zD A wrapper inserting code calling Numba-compiled *gendesc*. cs2||j}|j||jj|\}}||fSr )r\r]r^r_r`ra)rDrErNrOrrYrZgendescrrrhs zuser_generator..imp)rirj)rmrjrhrrlruser_generators  rncsfdd}|S)zl Decorator a given class as implementing *iterator_type* (by providing an `iternext()` method). cs0jttjfdd}td|S)Ncs |\}|||}||||Sr r)rDrErNrOresultrGZiterobjclsiternextrriternext_wrappers z8iterator_impl..wrapper..iternext_wrapperrr)rr iternext_implRefTypeBORROWED lower_builtin)rqrs iterator_typerprwrappers ziterator_impl..wrapperr) iterable_typeryrzrrxr iterator_impls  r|c@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 r~)rrDrEpairobjrrrrs z_IternextResult.__init__cCs|jtjd|j_dS)z1 Mark the iterator as exhausted. FNrZ get_constantrbooleanrsecondr9rrr set_exhausted sz_IternextResult.set_exhaustedTcCs$|dvr |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_valids z_IternextResult.set_validcCs ||j_dS)zP Mark the iterator as yielding the given *value* (a LLVM inst). Nrfirst)rrGrrryield_s z_IternextResult.yield_cCs|j|jtj|jjS)z> Return whether the iterator is marked valid. )rZget_argument_valuerrrrrr9rrrrs z_IternextResult.is_validcCs|jjS)z> Return the iterator's yielded value, if any. rr9rrr yielded_value&z_IternextResult.yielded_valueN)T) r<r=r>r? __slots__rrrrrrrrrrr}s   r}c@s"eZdZdZ dZ dZ dZdS)ruz& Enumerate the reference type N)r<r=r>r?NEWrv UNTRACKEDrrrrru,srucs*ddtDvr tdfdd}|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|||tjkrt}ntjkr%t}n tjkr-t }nt d||||| S)NzUnknown ref_type encountered) rR make_helperr}rurrdrvimpl_ret_borrowedrimpl_ret_untracked ValueErrorZ _getvalue)rDrErNrO pair_typerZimpl_ret)rref_typerrrzMs      z-iternext_impl..outer..wrapperr)rrzr)rrouterLsziternext_impl..outer)rur)rrrrrrt=s  rtcCs&t|j|}|d|}|||fS)z Call the `getiter()` implementation for the given *iterable_type* of value *val*, and return the corresponding LLVM inst. Zgetiter)rrry get_function)rDrEr{valZ 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. rr) Z yield_typerZPairrrrrrr}) rDrEryrZitemtyrZ iternext_sigrtrrrr call_iternextjs    rcCs<z |tttj|}Wn tyYdSw|||fS)z^ Call len() on the given value. Return None if len() isn't defined on this type. N)rlenrrrZintpNotImplementedError)rDrEr rZlen_implrrrcall_lenys   r _ForIterLoop)rGdo_breakc #s|j}t|||}d}dfdd}||=t|||}j|dd Wdn1sJwYt | |V|Wdn1sfwY |j r}|j ||dSdS)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_endrErrrr;zfor_iter..do_breakF)ZlikelyN)ryrZappend_basic_blockrZ goto_blockrrUrVrrrZposition_at_end enable_nrtnrtZdecref) rDrEr{rryZitervalZbb_bodyrrIrrrfor_iters&       rcC|S)z5 The implementation returns a new reference. rctxrEZrettyretrrrrdrdcCs|jr |j||||S)z The implementation returns a borrowed reference. This function automatically incref so that the implementation is returning a new reference. )rrZincrefrrrrrsrcCr)z. The return type is not a NRT object. rrrrrrrrnumpyccs>ddlm}|j}||||_z dVW||_dS||_w)z7 Temporarily change the context's error model. r)callconvN) numba.corerZ error_modelZcreate_error_model)rDZ model_namerZold_error_modelrrrforce_error_models rcOstd)zPA stub for use internally by Numba when a call is emitted on a TypeRef. z%This function should not be executed.)r)rOkwargsrrrnumba_typeref_ctorrrr )r)/r? collections contextlibinspect functoolsenumrrrrrrZnumba.core.typing.templatesrobjectrr@rArrwr*r.r1r2r6r7r&r0r[rkrnr|r}rurtrrr namedtuplercontextmanagerrrdrrrrrrrrsR  l / #