xKg;a<dZddlZddlZddlZddlmZddlZddlm Z m Z m Z m Z m Z ddlmZmZmZddlmZeegZGddZeZiej2d ej4d ej6d ej8d ej:d ej<dej>dej@dejBdejDdejFdejHdejJdejLdejNdejPdejRdejTdejVdejXdejZdej\dej^dej`d iZ1d!e1ejd<d"e1ejf<ejhd#ejjd$ejld%ejnd&ejpd'ejrd(ejtd)ejvd*ejxd+i Z=Gd,d-eZ>y).z* Lowering implementation for object mode. N)cached_property)typesutilsir generatorscgutils)ForbiddenConstruct LoweringErrorNumbaNotImplementedError) BaseLowerceZdZdZdZy) _UndefinedzH A sentinel value for undefined variable created by Expr.undef. cy)Nz selfs Y/home/alanp/www/video.onchill/myenv/lib/python3.12/site-packages/numba/core/pylowering.py__repr__z_Undefined.__repr__sN)__name__ __module__ __qualname____doc__rrrrrrs rr) number_addF)number_subtractF)number_multiplyF)number_truedivideF)number_floordivideF)number_remainderF) number_powerF) number_lshiftF) number_rshiftF) number_andF) number_orF) number_xorF)rT)rT)rT)rT)rT)rT)r T)r!T)r"T)r#T)r$T)r%T)number_matrix_multiplyF)r&T==!=z>=iszis notinceZdZejZdZfdZdZdZ dZ e dZ dZ dZd d Zd Zd Zd Zd ZdZdZdZdZd!dZdZdZdZd"dZdZdZd dZdZ d"dZ!dZ"dZ#dZ$dZ%xZ&S)#PyLowerc@t|_t|_yN)set_frozen_strings _live_varsrs rinitz PyLower.initSs"u%rcJtt| |jyr0)superr. pre_lower init_pyapi)r __class__s rr7zPyLower.pre_lowerYs gt&( rcyr0rrs r post_lowerzPyLower.post_lower]s rc&|j|yr0) init_varsrblocks r pre_blockzPyLower.pre_block`s urc. t|tjr8|j|}|j ||j j yt|tjr|j|j j }|j|jj }|j|jj }|jj|||}|j|yt|tjrx|j|j j }|j|jj }|jj||}|j|yt|tj r|j|j j }|j|jj }|jj#||j%|j&|}|j|yt|tj(rl|j|j j }|jj+||j%|j&}|j|yt|tj,r|j|j.j }|j|j0j }|j|jj }|jj3|||}|j|yt|tj4r|j|jj }|j6r7|jj9||j:j=|y|j>jA|jB|yt|tjDr|j|jFj } | jHtJjjMdk(r| } n|jjO| } tJjjQ| jHd} |jBjSd| | } |jT|jV} |jT|jX}|jBj[| | |yt|tj\r5|jT|j }|jBj_|yt|tj`r|jc|jyt|tjdryt|tjfrq|jh7|j|jhj }|jk|nd}|jjm||joytI|d|}tq|)Nr(z, )9 isinstancerAssign lower_assignstorevartargetnameSetItemloadvarindexvaluepyapiobject_setitemcheck_int_statusDelItemobject_delitemSetAttrobject_setattr_freeze_stringattrDelAttrobject_delattrStoreMapdctkey dict_setitemReturngenerator_infodecrefgenlowerreturn_from_generator call_conv return_valuebuilderBranchcondtypellvmliteIntType object_istrueConstant icmp_unsignedblkmaptruebrfalsebrcbranchJumpbranchDeldelvarPopBlockRaise exceptionincref raise_objectreturn_exception_raisedr )rinstrLrGrKokrYrZretvalreistruezeropredtrflexcmsgs r lower_instzPyLower.lower_instcso dBII &%%d+E MM%!1!1 2 bjj )\\$++"2"23FLL1ELL1E**65%@B  ! !" % bjj )\\$++"2"23FLL1E**659B  ! !" % bjj )\\$++"2"23FLL1E**6+/+>+>tyy+I+02B  ! !" % bjj )\\$++"2"23F**6+/+>+>tyy+IKB  ! !" % bkk *,,txx}}-C,,txx}}-CLL1E((c59B  ! !" % bii (\\$**//2F"" !!&) 33D9 NN ' ' f = bii (<< /DyyHKK//2211$7;;'' T:D<<--dFDADT[[)BT\\*B LL r2 . bgg &[[-F LL   ' bff % KK # bkk *  bhh '~~)ll4>>#6#67 C  JJ # #C (  ( ( *$ZL4&)C*3/ /rcvddlm}|jj|jj |S)zeReturn a `OmittedArg` type instance as a LLVM value suitable for testing at runtime. r) OmittedArg)numba.core.dispatcherrrM unserializeserialize_object)rrs r_omitted_typobjzPyLower._omitted_typobjs0 5zz%% JJ ' ' 35 5rc|j}t|tjtjfr|j |jSt|tj r.|j|j}|j||St|tjr|j|St|tjr&|j|j|jSt|tjr|j|St|tj r|j"j$j&j(j+|j}|j,|j.}t1j2|j4|}|T|j6t8j:j<ur.|j||j4j?||n|j@jC|}|j4jEd||jF}|j4jI|d5\} } | 5|j||j4j?||ddd| 5|j@jK|d}|j4j?||dddddd|j4jM|StOtQ||#1swYxYw#1swYMxYw#1swYQxYw)z? The returned object must have a new reference Nr'F)likelyrL))rLrCrConstFreeVar lower_constVarrJrHrwExpr lower_exprGlobal lower_globalYield lower_yieldArgfunc_irfunc_idpysig parametersgetfnargsrKralloca_once_valuercdefaultinspect ParameteremptystorerMget_typerkrif_elseobject_getattr_stringloadNotImplementedErrorrf) rrzrLvalparamobjslottypobj is_omittedomittedpresents rrEzPyLower.lower_assignsg  ebhh 3 4##EKK0 0 rvv &,,uzz*C KK J rww '??5) ) ryy )$$UZZ= = rxx (##E* * rvv &LL((..99==ejjIE++ekk*C,,T\\3?D U]]g6G6G6M6M%M C  ""3-,,S1!\\77f8<8L8LN \\))*U)CGYQX  C( **35!!"jj>>sGL **35! D<<$$T* *%d5k59 9!! DCs<<M .L30 M :9L?3M 3L< 8M ?M M  Mc:|jj|j}|j|usJ|jj |t j|||j|jz}|j|j|jj}|jj||j j#|j$||j'|jj)Sr0)r] yield_pointsrKrzr_init_generator_stater LowerYield live_varsweak_live_varslower_yield_suspendrJrLrHrMrwrarbrclower_yield_resume make_none)rrzypyrs rrzPyLower.lower_yields  - -djj 9ww$ **40  ! !$BLL2;L;L,L M ll4::??+ # ##DLL#6 zz##%%rc|j|jj}|j|jj}t |t rJ|t vr.t |\}}t|j|}||||}nPtj|j|j}|dk(r||}}|jj|||}|j||S)Ninplacer,)rJlhsrHrhsrCstrPYTHON_BINOPMAPgetattrrMPYTHON_COMPAREOPMAPrfnobject_richcompare check_error) rexproprrrfnamerress r lower_binopzPyLower.lower_binopsll488==)ll488==)b#&&&  ,R0NE7U+BS#w/C%(($'':BTzS**//S"=C  rc|jdk(r|j||jdS|jdk(r|j||jdS|jdk(rS|j|jj }|jt jk(r|jj|}n|jt jk(r|jj|}n|jt jk(rH|jj|}|j||jj|}nD|jt j k(r|jj#|}n t%||j'||S|jdk(r|j(Dcgc]}|j|j }}|j|j*j }|jj-|}|j.r~|jj1|j|j.j }|jj3||} |j5||j5|| }|j6s|jj9||d} n|j6D cgc]"\} } | |j| j f$} } } |jj;| }|jj9|||} |j5||j5||j'| | S|jdk(rm|j|jj }|jj=||j?|j@}|j'||S|jd k(r`|jBDcgc]}|j|j }}|jj-|}|j'||S|jd k(r`|jBDcgc]}|j|j }}|jjE|}|j'||S|jd k(r|jjG|jH}|j'||jBD]i\} } |j| j }|j| j }|jjK|||}|j|k|S|jd k(r|jBDcgc]}|j|j }}|jjM}|j'||D]/}|jjO||}|j|1|S|jd k(rS|j|jj }|jjQ|}|j'||S|jdk(rR|j|jj }|jjS|}tUjV|jX|}|jj[d}|jXj]|5\}}|5|jj_|d|ddd|5|ja|jj_|d|jjcdddddd|jj_|d|jj||S|jdk(rT|j|jj }|jje|d}|jg||S|jdk(rT|j|jj }|jje|d}|jg||S|jdk(r|j|jj }|jj1|}|j'||jji|}|jjjmtnjp|jr}|jXjud||}tUjv|jX|5|jytzddd|S|jdk(ry|j|jj }|j|j|j } |jj|| }|j'||S|jdk(r|j|jj }|jjjmtnjp|j|} |jj| }!|j'|!|jj||!}|j5|!|j'||S|jdk(r|j|jj }"|j|jj }#|j|jj }$|jd}%|jj|%|#|$f}&|j5|%|j'|&|jj|"|&}|j'||S|jdk(r8|j|jj }'|jg|'|'S|jdk(r td|jdk(r)tUj|jjS|jdk(r|jtSt%|cc}wcc} } wcc}wcc}wcc}w#1swYxYw#1swYexYw#1swYjxYw#1swY|SxYw)NbinopFr inplace_binopTunarycallr build_tuple build_list build_map build_setgetiteriternextrrB pair_first pair_second exhaust_iterr(getitemstatic_getitemgetsliceslicecastphizPHI not strippednullundef)KrrrrJrLrHoperatornegrMnumber_negativeposnumber_positivenot_ object_notrObool_from_boolinvert number_invertrrargsfunc tuple_packvarargsequence_tuplesequence_concatr^kwsr dict_packobject_getattrrTrUitems list_packdict_newsizer[set_newset_addobject_getiter iter_nextr is_not_nullrc tuple_newr tuple_setitemcheck_occurredr tuple_getitemrw tuple_sizecontext get_constantrintpcountrk if_unlikelyreturn_exception ValueErrorrKobject_getitemlong_from_ssize_trGstartstopget_builtin_objcall_function_objargsr get_null_valuepyobjr _UNDEFINED)(rrrLraargvalsrrvarargsnew_argsretkv keyvaluesrritrrZr{iterobjitemis_validpairthen otherwisefirstsecondtuptup_size expected_sizehas_wrong_sizerKindexobjrGrrslicefnsliceobjrs( rrzPyLower.lower_exprs 77g ##D$''5#A A WW '##D$''4#@ @ WW LL1Eww(,,&jj007HLL(jj007HMM)jj++E2%%c*jj//4HOO+jj..u5)$//   S !J WW 59YY?Yt||AFF+YG?diinn-B::((1D{{**33 $ T[[-=-= >@::55dGD G$ D!88jjoob$5DH88L841aaaff!568 Ljj**95jjoob$4 C KK    S !J WW !,,tzz/C**++C1D1DTYY1OPC   S !J WW %59ZZ@ZrT\\"''*ZE@**''.C   S !J WW $59ZZ@ZrT\\"''*ZE@**&&u-C   S !J WW #**%%dii0C   S ! 1ll166* QVV,ZZ,,S#u=%%b) # J WW #59ZZ@ZrT\\"''*ZE@**$$&C   S !ZZ''R0%%b)J WW !,,tzz/C**++C0C   S !J WW "ll4::??3G::''0D**4<<>H::''*D%%h/3DD)JJ,,T1d;'')JJ,,T1djj6J6J6LM 0 JJ $ $T1djj.G.G.Q RK WW $<< 0DJJ,,T15E KK L WW %<< 0DZZ--dA6F KK M WW &ll4::??3G**++G4C   S !zz,,S1H LL55ejj$**MM!\\77/7HN$$T\\>B%%j1CJ WW !LL1ELL1E**++E59C   S !J WW( (LL1ELL--ejj$**EEzz33E:H   X &**++E8 9uu uc|jj|}|jj|}|j||j ||Sr0) env_manager add_const read_constrrw)rconstrKrs rrzPyLower.lower_constsM  **51))%0  C rc|j}|jj||j|}|j | |t vrt d|z|j tt|r|j|}|jj}|jj|5|jj||jd}|j||}|jj} ddd|jj!|jj"} | j%||| j% | S|} t'j(|j|j| 5|jj+||j-ddd| S#t$rYvwxYw#1swYxYw#1swY| SxYw)z 1) Check global scope dictionary. 2) Check __builtins__. 2a) is it a dictionary (for non __main__ module) 2b) is it a module (for __main__ module) zbuiltins %s() is not supported)loc __builtins__N)get_module_dictrM dict_getitemrTrw_unsupported_builtinsr r3 TypeErrorhasattrbuiltinsis_nullrc basic_blockif_thenbuiltin_lookuprr add_incomingrr raise_missing_global_errorry) rrHrLmoddictr obj_is_nullbbelsemodbuiltinbbifr|s rrzPyLower.lower_globals&&(jj%%gt/B/B4/HI C --()I+/*059XX??. 8T ",,s+K\\--F%%k2jj--g*.*=*=n*MO--c48||// 3 \\%%djj&6&67F   V ,    . F$$T\\4<<3GH 55d;,,.I 1   32I s+"GAG+%,G7 G('G(+G47Hc.|jjSr0)env_bodyglobalsrs rr5zPyLower.get_module_dicts}}$$$rc|j}|jj||jd}|j ||S)Nr4)r5rMr6rTr>)rrHrArDs rrzPyLower.get_builtin_objsH&&(jj%%g&*&9&9.&IK""3--rc8|jj||j|}|j||jj }t j|j|j|5|jj||j|}t j|j|j|5|jj||jddd|jj }ddd|jj|jj}|j|||j|S#1swY|xYw#1swYjxYw)z Args ---- mod: The __builtins__ dictionary or module, as looked up in a module's globals. name: str The object to lookup N)rMr6rTrwrcr<rr r;rr@ryrrr?)rrDrHfromdictbbifdictfrommodbbifmodrEs rr>zPyLower.builtin_lookups/::**30C0CD0IJ H<<++  t||H/E Fjj//T5H5H5NOG$$T\\4<<3HI 55d;,,.Jll..GG,,""4::#3#34Xx0Wg.JI G Fs%AF,F FF FFctj|j|jj }tj |j|5|j dddy#1swYyxYw)z2 Return if an exception occurred. N)rrrcrM err_occurredr ry)rrQs rrzPyLower.check_occurredsY**4<<+/::+B+B+DF  | <  ( ( *= < s rr=zPyLower.init_vars>sdll??FGrc|||jvr |j|||j|<|j|S)Nltype)varmapalloca)rrHr`s r_getvarzPyLower._getvarDs8 t{{ " $ D >DKK {{4  rcd||jvsJ||j|}|jj|}t j |j|j |5|jj||jddd|S#1swY|SxYw)zC Load the llvm value of the variable named *name*. N) r3rarcrrr r;rMraise_missing_name_errorry)rrHptrrs rrJzPyLower.loadvarIs t&,,&kk$ll$  t||C/@ A JJ / / 5  ( ( *B B s /,B%%B/c6|jj||j|}|j|jj ||jj tj|jj|y)zy Delete the variable slot with the given name. This will decref the corresponding Python object. N) r3removercr^rcrrrrrfpointeerrHrfs rrszPyLower.delvarVsj t$ll4  DLL%%c*+ 711#((2B2BCSIrc||jvxr| }|j||j}|r|jj |}n|jj ||j|jj k(s9Jt|jt|jj f|jj|||r|jyy)z| Stores a llvm value and allocate stack slot if necessary. The llvm value can be of arbitrary type. r_N) r3rcrfrcraddrirrr^)rrLrHclobber is_redefinerfolds rrFzPyLower.storevards doo-=g+ ll4uzzl2 ,,##C(C OO   %zzSXX--- GEJJ03CHH4D4D0E0G G- 5#&  KK  rc|jD]=}|j|}|j|jj |?y)z) Cleanup live variables. N)r3rcr^rcrrjs rrZzPyLower.cleanup_varsvs;OOD,,t$C KK ))#. /$rcj|)|jjtj}|jj |j 5|jj||}|jjtj||ddd|S#1swYSxYw)z Allocate a stack slot and initialize it to NULL. The default is to allocate a pyobject pointer. Use ``ltype`` to override. N)rH) rget_value_typerpyobjectrc goto_block entry_blockrbrrr)rrHr`rfs rrbzPyLower.alloca~s =LL//?E \\ $ $T%5%5 6,,%%e$%7C LL  w55e Freeze a Python string object into the code. )r)rstrings rrTzPyLower._freeze_strings''r)F)rr0)'rrrrPyGeneratorLowerGeneratorLowerr4r7r;r@rrrrErrrrrr5rr>rrrOr;ryr=rcrJrsrFrZrbrxrwr^rT __classcell__)r9s@rr.r.Os00N  V0p55+:Z&("e,N'V%.<++20H !  J$0 ! !%(rr.)?rr:rr functoolsr llvmlite.irrg numba.corerrrrrnumba.core.errorsr r r numba.core.loweringr r1localsr7rrrlsubmultruedivfloordivrDpowlshiftrshiftand_or_xoriaddisubimulitruediv ifloordivimodipowilshiftirshiftiandiorixorrmatmulimatmuleqneltlegtgeis_is_notcontainsrr.rrrrsX %<<99) VH \  LL' LL, LL, 2   4   LL-  LL) OO- OO- MM( LL& LL' MM' MM,  MM,!" 2#$ 4%& MM- MM) - - MM' LL% MM'38$E $D  ! KK KK KK KK KK KK LL$ OOX t Q (iQ (r