xKg):.dZddlZddlZddlZddlZddlmZddlmZm Z m Z m Z ddl m Z GddeZGdd e Zed Zej$Zej(Zej*Zej,Zej.Zej0Zej2Zd Zd Zd ZdZdZdZGddeZ GddeZ!d!dZ"dZ#dZ$dZ%ejLddZ'ejPdZ)dZ*dZ+dZ,ejPd"dZ-d Z.y)#z< Utilities to simplify the boilerplate for native lowering. N)Enum)typingtypesutilscgutils)BaseRegistryLoadercNeZdZdZd dZdZdZdZdZdZ dZ d Z d Z d Z y )Registryz? A registry of function and attribute implementations. cX||_g|_g|_g|_g|_g|_yN)name functionsgetattrssetattrscasts constants)selfr s W/home/alanp/www/video.onchill/myenv/lib/python3.12/site-packages/numba/core/imputils.py__init__zRegistry.__init__s,    cfd}|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). cDjj|f|Sr )rappend)implargtysfuncrs rdecoratez Registry.lower..decorate%s! NN ! !4v"6 7Kr)rrrrs``` rlowerzRegistry.lowers rcV||||}|j|||jf|Sr )r signature)rrtyattr impl_list decorator real_impls r_decorate_attrzRegistry._decorate_attr*s0dB- )T9+>+>?@ rcfd}|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). cJj|jtSr )r'r_decorate_getattrrr#rr"s rrz(Registry.lower_getattr..decorate7%&&tRt}}'8: :rrrr"r#rs``` r lower_getattrzRegistry.lower_getattr/ :rc&|j|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"s rlower_getattr_genericzRegistry.lower_getattr_generic<!!"d++rcfd}|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). cJj|jtSr )r'r_decorate_setattrr+s rrz(Registry.lower_setattr..decorateOr,rrr-s``` r lower_setattrzRegistry.lower_setattrGr/rc&|j|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)r7r1s rlower_setattr_genericzRegistry.lower_setattr_genericTr3rcfd}|S)z Decorate the implementation of implicit conversion between *fromty* and *toty*. The decorated implementation will have the signature (context, builder, fromty, toty, val). cFjj|ff|Sr )rr)rfromtyrtotys rrz%Registry.lower_cast..decorategs# JJ  tfd^4 5Krr)rr<r=rs``` r lower_castzRegistry.lower_cast_s rcfd}|S)z Decorate the implementation for creating a constant of type *ty*. The decorated implementation will have the signature (context, builder, ty, pyval). cDjj|ff|Sr )rr)rrr"s rrz)Registry.lower_constant..decoratess NN ! !4"- 0Krr)rr"rs`` rlower_constantzRegistry.lower_constantls rc"d|jdS)NzLowering Registry<>)r rs r__repr__zRegistry.__repr__xs#DII;a00rN) unspecified)__name__ __module__ __qualname____doc__rrr'r.r2r7r9r>rArErrrr r s9   ,  ,  1rr ceZdZdZdZy)RegistryLoaderz6 An incremental loader for a target registry. )rrrrrN)rGrHrIrJregistry_itemsrrrrLrL|sQNrrLbuiltin_registrycD||fd}nfd}|f|_||_|S)Nc||||Sr rcontextbuildertypvaluer#r&s rresz_decorate_getattr..ressWgsE: :rc|||||Sr rrQs rrVz_decorate_getattr..ressWgsE4@ @r)r!r#rr"r#rVr&s @rr*r*s0I  ; AECMCH Jrcb||fd}nfd}|tjf|_||_|S)Nc||||Sr rrRrSsigargsr#r&s rrVz_decorate_setattr..ressWgsD9 9rc|||||Sr rr[s rrVz_decorate_setattr..ressWgsD$? ?r)rAnyr!r#rXs @rr6r6s6I  : @OCMCH Jrct|jtjr|jj}|j ||}t j||}|j|j|j5|j|||}|j||ddd|j|}|S#1swYxYwr ) isinstance return_typerOptionaltypemake_optional_noneralloca_once_valueif_thennot_is_nonemake_optional_valuestoreload) rRrSr\statusretval value_type optional_none retvalptroptional_values rfix_returning_optionalrss#//5>>2__)) 227JG --g}E __W\\&..9 :$88VN MM.) 4 ; i( M ; :s &CCcTfd}j|_t||_|S)zC A wrapper inserting code calling Numba-compiled *fndesc*. c|j|j}|jj||jj |\}}t j||j5|jj||ddd|jjk(sJt|||||}|j|j|jk7r1d}t|j|j|jt!||j|S#1swYxYw)Nz$function returned {0} but expect {1})declare_functionmodule call_conv call_functionrestypeargtypesr if_unlikelyis_errorreturn_status_propagaterbrsrdget_value_type TypeErrorformatimpl_ret_new_ref) rRrSr\r]rrmrnmsgfndescs rimpzuser_function..imps''? **88 T6>>6??DB  &// :    5 5gv F;&..000'#vvN ;;'00A A8CCJJv{{COODE E&..&II; :s 4D44D=)r{r!tuplelibs)rrrs` r user_functionrs& J"OOCMT{CH Jrc2fd}t||_|S)zD A wrapper inserting code calling Numba-compiled *gendesc*. c|j|j}|jj||jj |\}}||fSr )rvrwrxryrzr{)rRrSr\r]rrmrngendescs rrzuser_generator..impsU''@ **88 T7??G,<,.wrapper..iternext_wrappers)HU'7E2GGWgv> >rr)r iternext_implRefTypeBORROWED lower_builtin)rrr iterator_types` @rwrapperziterator_impl..wrappersB<< w'' ( ? ) ? 1 j-01AB rr) iterable_typerrs ` r iterator_implrs  Nrc:eZdZdZdZdZdZd dZdZdZ dZ y ) _IternextResultzb A result wrapper for iteration, passed by iternext_impl() into the wrapped function. _context_builder_pairobjc.||_||_||_yr r)rrRrSpairobjs rrz_IternextResult.__init__s   rcv|jjtjd|j_y)z1 Mark the iterator as exhausted. FNr get_constantrbooleanrsecondrDs r set_exhaustedz_IternextResult.set_exhausted s% $}}99%--O rc|dvr*|jjtj|}||j_y)z Mark the iterator as valid according to *is_valid* (which must be either a Python boolean or a LLVM inst). )FTNr)ris_valids r set_validz_IternextResult.set_valids2 } $}}11%--JH' rc&||j_y)zP Mark the iterator as yielding the given *value* (a LLVM inst). Nrfirst)rrUs ryield_z_IternextResult.yield_s$ rc|jj|jtj|j j S)z> Return whether the iterator is marked valid. )rget_argument_valuerrrrrrDs rrz_IternextResult.is_valids7}}// 05 04 0D0DF Frc.|jjS)z> Return the iterator's yielded value, if any. rrDs r yielded_valuez_IternextResult.yielded_value&s}}"""rN)T) rGrHrIrJ __slots__rrrrrrrrrrrs.5I P ($ F#rrc"eZdZdZ dZ dZ dZy)rz& Enumerate the reference type N)rGrHrIrJNEWr UNTRACKEDrrrrr,s+ CHIrrc^tDcgc]}|c}vr tdfd}|Scc}w)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) z3ref_type must be an enum member of imputils.RefTypecfd}|S)Nc R|j}|j||}||||t|||tjk(rt }n?tj k(rt}n%tjk(rt}n td|||||jS)NzUnknown ref_type encountered) rb make_helperrrrrrimpl_ret_borrowedrimpl_ret_untracked ValueError _getvalue) rRrSr\r] pair_typerimpl_retrref_types rrz-iternext_impl..outer..wrapperMsI))'9=G '3': <7;;&+W---,W...- !?@@GW$-w/@/@/BD Drr)rrrs` routerziternext_impl..outerLs Dr)rr)rxrs` rrr=s97+7a7++NOO" L),s *cztj|j|}|jd|}|||fS)z Call the `getiter()` implementation for the given *iterable_type* of value *val*, and return the corresponding LLVM inst. getiter)rr!r get_function)rRrSrval getiter_sig getiter_impls r call_getiterr`s= ""=#>#> NK'' ;?L # ((rc|j}tj|tj}t j ||}|j d|}|||f}|j|||}t|||S)z Call the `iternext()` implementation for the given *iterator_type* of value *val*, and return a convenience _IternextResult() object reflecting the results. r) yield_typerPairrrr!rrr) rRrSrritemtyr iternext_sigrrs r call_iternextrjsx  % %F 65==1I##I}=L((\BM # (C!!'9c:G 7GW 55rc |jttjtj |}|||fS#t $rYywxYw)z^ Call len() on the given value. Return None if len() isn't defined on this type. N)rlenrr!rintpNotImplementedError)rRrSr"rlen_impls rcall_lenrysO )''V-=-=ejj"-NO#(( s8A AA _ForIterLoop)rUdo_breakc# K|j}t|||}jd}jd fd}j|j |5t |||}j j|jd5j dddt|j|j|dddj |jr|jj||yy#1swYzxYw#1swYQxYww)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.endc(jyr )branch)bb_endrSsrrzfor_iter..do_breaksvrF)likelyN)rrappend_basic_blockr goto_blockrrgrhrrrposition_at_end enable_nrtnrtdecref) rRrSrrritervalbb_bodyrrVrs ` @rfor_iterrs"//M7G]C@G((9G  ' ' 7F NN7   G $GWmWE __W\\#,,.9%_ H NN6 "I3,,.99w % F# 7M7; I H % $s7A%E )?D?(D3:5D?/AE 3D< 8D??EE c|S)z5 The implementation returns a new reference. rctxrSrettyrets rrr  JrcX|jr|jj||||S)z The implementation returns a borrowed reference. This function automatically incref so that the implementation is returning a new reference. )rrincrefrs rrrs$  ~~ ws+ Jrc|S)z. The return type is not a NRT object. rrs rrrrrc#Kddlm}|j}|j|||_ d||_y#||_wxYww)z7 Temporarily change the context's error model. r)callconvN) numba.corer error_modelcreate_error_model)rR model_namerold_error_models rforce_error_modelrsC $))O"55j'JG. -os*A9A AActd)zPA stub for use internally by Numba when a call is emitted on a TypeRef. z%This function should not be executed.)r)r]kwargss rnumba_typeref_ctorrs E FFrr )numpy)/rJ collections contextlibinspect functoolsenumrrrrrrnumba.core.typing.templatesrobjectr rLrNrrr.r2r7r9r>rAr*r6rsrrrrrrrrr namedtuplercontextmanagerrrrrrrrrrr sM 44:i1vi1XQ'Q./ &&  .. (>> .. (>>  ( ( !00   6",-#f-#^d" F) 6 )&{%%n&;=  <<<  . .Gr