B \E5@sRdZddlmZmZmZddlZddlZddlZddlZddl m Z m Z m Z m Z ddlmZGdddeZGd d d eZeZejZejZejZejZejZejZejZd d Zd dZddZddZddZ ddZ!GdddeZ"ddZ#ddZ$ddZ%dd Z&e'd!d"Z(ej)d#d$Z*d%d&Z+d'd(Z,d)d*Z-ej)d.d,d-Z.dS)/z< Utilities to simplify the boilerplate for native lowering. )print_functionabsolute_importdivisionN)typingcgutilstypesutils)BaseRegistryLoaderc@sXeZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ dS)Registryz? A registry of function and attribute implementations. cCs"g|_g|_g|_g|_g|_dS)N) functionsgetattrssetattrscasts constants)selfr5lib/python3.7/site-packages/numba/targets/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|S)N)r append)impl)argtysfuncrrrdecorate$sz Registry.lower..decorater)rrrrr)rrrrlowers zRegistry.lowercCs"||||}||||jf|S)N)r signature)rrtyattrZ impl_listZ 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|jtS)N)rr _decorate_getattr)r)rrrrrr6sz(Registry.lower_getattr..decorater)rrrrr)rrrr 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|jtS)N)rr_decorate_setattr)r)rrrrrrNsz(Registry.lower_setattr..decorater)rrrrr)rrrr lower_setattrFszRegistry.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$)rrrrrlower_setattr_genericSs 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|S)N)rr)r)fromtyrtotyrrrfsz%Registry.lower_cast..decorater)rr&r'rr)r&rr'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|S)N)rr)r)rrrrrrsz)Registry.lower_constant..decorater)rrrr)rrrlower_constantkszRegistry.lower_constantN) __name__ __module__ __qualname____doc__rrrr!r"r$r%r(r)rrrrr s     r c@seZdZdZdZdS)RegistryLoaderz6 An incremental loader for a target registry. )r r rrrN)r*r+r,r-Zregistry_itemsrrrrr.xsr.cs8||dk rfdd}n fdd}|f|_||_|S)Ncs||||S)Nr)contextbuildertypvaluer)rrrressz_decorate_getattr..rescs|||||S)Nr)r/r0r1r2r)rrrr3s)rr)rrrr3r)rrr s r cs<||dk rfdd}n fdd}|tjf|_||_|S)Ncs||||S)Nr)r/r0sigargsr)rrrr3sz_decorate_setattr..rescs|||||S)Nr)r/r0r4r5r)rrrr3s)rZAnyrr)rrrr3r)rrr#s  r#c Cstt|jtjrp|jj}|||}t||}|| |j  | |||}| ||WdQRX| |}|S)N) isinstance return_typerZOptionaltypeZmake_optional_nonerZalloca_once_valueif_thennot_Zis_noneZmake_optional_valueZstoreload) r/r0r4statusretvalZ value_typeZ optional_noneZ retvalptrZoptional_valuerrrfix_returning_optionals    r>cs"fdd}j|_t||_|S)zC A wrapper inserting code calling Numba-compiled *fndesc*. c s||j}|j||jj|\}}t||j|j ||WdQRX|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_propagater7AssertionErrorr>r8Zget_value_type TypeErrorformatimpl_ret_new_ref)r/r0r4r5rr<r=msg)fndescrrimpszuser_function..imp)rDrtuplelibs)rJrMrKr)rJr user_functions  rNcsfdd}t||_|S)zD A wrapper inserting code calling Numba-compiled *gendesc*. cs2||j}|j||jj|\}}||fS)N)r?r@rArBrCrD)r/r0r4r5rr<r=)gendescrrrKszuser_generator..imp)rLrM)rOrMrKr)rOruser_generators  rPcsfdd}|S)zl Decorator a given class as implementing *iterator_type* (by providing an `iternext()` method). cs*jtfdd}td|S)Ncs |\}|||}||||S)Nr)r/r0r4r5resultr2Ziterobj)clsiternextrriternext_wrappers z8iterator_impl..wrapper..iternext_wrapperrS)rS iternext_impl lower_builtin)rRrT) iterator_type)rRrSrwrappersziterator_impl..wrapperr) iterable_typerWrXr)rWr iterator_impls rZc@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||_||_||_dS)N)r\r]r^)rr/r0pairobjrrrrsz_IternextResult.__init__cCs|jtjd|j_dS)z1 Mark the iterator as exhausted. FN)r\ get_constantrbooleanr^second)rrrr set_exhaustedsz_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). )FTN)r\r`rrar^rb)ris_validrrr set_valid sz_IternextResult.set_validcCs ||j_dS)zP Mark the iterator as yielding the given *value* (a LLVM inst). N)r^first)rr2rrryield_sz_IternextResult.yield_cCs|j|jtj|jjS)z> Return whether the iterator is marked valid. )r\Zget_argument_valuer]rrar^rb)rrrrrds z_IternextResult.is_validcCs|jjS)z> Return the iterator's yielded value, if any. )r^rf)rrrr yielded_value"sz_IternextResult.yielded_valueN)T) r*r+r,r- __slots__rrcrergrdrhrrrrr[s r[csfdd}|S)a Wrap the given iternext() implementation so that it gets passed an _IternextResult() object easing the returning of the iternext() result pair. The wrapped function will be called with the following signature: (context, builder, sig, args, iternext_result) c s<|j}|||}||||t|||t||||S)N)r7 make_helperr[impl_ret_borrowedZ _getvalue)r/r0r4r5 pair_typer_)rrrrX3s   ziternext_impl..wrapperr)rrXr)rrrU)s rUcCs&t|j|}|d|}|||fS)z Call the `getiter()` implementation for the given *iterable_type* of value *val*, and return the corresponding LLVM inst. Zgetiter)rrrW get_function)r/r0rYvalZ getiter_sigZ getiter_implrrr call_getiter=s roc 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. rS) Z yield_typerZPairrarrrmrjr[) r/r0rWrnZitemtyrlZ iternext_sigrUr_rrr call_iternextGs   rpcCs@y|tttj|}Wntk r.dSX|||fSdS)z^ Call len() on the given value. Return None if len() isn't defined on this type. N)rmlenrrrZintpNotImplementedError)r/r0rrnZlen_implrrrcall_lenVs rs _ForIterLoop)r2do_breakc #s|j}t|||}d}dfdd}||\t|||}j|ddWdQRXt | |V|WdQRX |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.endcsdS)N)branchr)bb_endr0rrruuszfor_iter..do_breakF)ZlikelyN)rWroZappend_basic_blockrvZ goto_blockrpr9r:rdrtrhZposition_at_end enable_nrtnrtZdecref) r/r0rYrnrWZitervalZbb_bodyrur3r)rwr0rfor_itergs     rzcCs|S)z5 The implementation returns a new reference. r)ctxr0rettyretrrrrHsrHcCs|jr|j||||S)z The implementation returns a borrowed reference. This function automatically incref so that the implementation is returning a new reference. )rxryZincref)r{r0r|r}rrrrksrkcCs|S)z. The return type is not a NRT object. r)r{r0r|r}rrrimpl_ret_untrackedsr~numpyccs8ddlm}|j}||||_z dVWd||_XdS)z7 Temporarily change the context's error model. )callconvN)rZ error_modelZcreate_error_model)r/Z model_namerZold_error_modelrrrforce_error_models   r)r)/r-Z __future__rrr collections contextlibinspect functoolsrrrrr Ztyping.templatesr objectr r.Zbuiltin_registryrrVr!r"r$r%r(r)r r#r>rNrPrZr[rUrorprs namedtuplertcontextmanagerrzrHrkr~rrrrrsH h 0