shxSSKJr SSKJrJr "SS\5r"SS\5rSrSr S r S r S r "S S \5r g))deque)typescgutilsc:\rSrSrSrSrSrS SjrSrSr S r g) DataPackerz A helper to pack a number of typed arguments into a data structure. Omitted arguments (i.e. values with the type `Omitted`) are automatically skipped. cXlX lUVs/sHo1RU5PM snUl/Ul/Ul[ U5HwupC[U[R5(aM&URRU5 UR RURUR55 My gs snfN) _dmm _fe_typeslookup_models _pack_map _be_types enumerate isinstancerOmittedappend get_data_type)selfdmmfe_typestyis o/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/numba/core/datamodel/packer.py__init__DataPacker.__init__s !19:2 2: x(EAb%--00%%a(%%dll1o&C&C&EF) ;sCcURVs/sH$nURURXU5PM& nn[R"X5$s snf)z5 Return the given values packed as a data structure. )rras_datarmake_anonymous_struct)rbuildervaluesrelemss rrDataPacker.as_datasU ..*(Qa(()<( *,,W<<*s+ANc/n[UR5HdupV[R"XSU5nURUR X5nUc"UR URUU45 M`XU'Mf U$Nr)rrr gep_inboundsrload_from_data_pointerrr ) rr!ptr formal_listresri_formalelem_ptrvals r_do_loadDataPacker._do_load#sx$T^^4KA++G!Q?H,,x(??RC" DNN84c:;(+H% 5 c$URX5$)z; Load the packed values and return a (type, value) tuples. r/)rr!r)s rloadDataPacker.load.s}}W**r1c(URXU5 g)zk Load the packed values into a sequence indexed by formal argument number (skipping any Omitted position). Nr3)rr!r)r*s r load_intoDataPacker.load_into4s gK0r1)rr r rrr ) __name__ __module__ __qualname____firstlineno____doc__rrr/r4r7__static_attributes__r1rrrs! G= + 1r1rcJ\rSrSrSrSrSrSrSrS Sjr \ S5r S r g ) ArgPacker<a Compute the position for each high-level typed argument. It flattens every composite argument into primitive types. It maintains a position map for unflattening the arguments. Since struct (esp. nested struct) have specific ABI requirements (e.g. alignment, pointer address-space, ...) in different architecture (e.g. OpenCL, CUDA), flattening composite argument types simplifes the call setup from the Python side. Functions are receiving simple primitive types and there are only a handful of these. c\XlX l[U5Ul/Ul/nUHXnURR U5nURR U5 UR UR55 MZ [U5Ul [[U55Ul gr ) r _fe_argslen_nargs_dm_argsr rget_argument_type _Unflattener _unflattenerlist_flatten_be_args)rrfe_argsargtysrdms rrArgPacker.__init__Is  'l  B!!"%B MM  $ MM"..0 1)0Xf-. r1c4[U5UR:wa#[SUR[U54-5eU(dg[URU5VVs/sHup4UR X5PM nnn[ [U55nU$s snnf)z$Flatten all argument values z+invalid number of args: expected %d, got %dr?)rErF TypeErrorziprG as_argumenttuplerL)rr!r"rPr.argss r as_argumentsArgPacker.as_argumentsWs v;$++ %I#{{CK89: : #4==&99GBw,9 Xd^$ sBcURRU5n[URU5VVs/sHupEUR X5PM nnnU$s snnf)z&Unflatten all argument values )rJ unflattenrTrG from_argument)rr!rWvaltreerPr.r"s rfrom_argumentsArgPacker.from_argumentshs^##--d3!$T]]G!<!M r1c:[SUR55$)zLReturn a list of LLVM types that are results of flattening composite types. c34# UHoS:wdM Uv M g7f)r?Nr?).0rs r +ArgPacker.argument_types..s<-B8RR-s  )rVrM)rs rargument_typesArgPacker.argument_typess <$--<<r?r1rrArA<s4  /" ,?==r1rAc ^U4SjmT"U5$)z+ Flatten nested iterable of (tuple, list). c3># UH7n[U[[45(aT"U5HnUv M M3Uv M9 g7fr )rrVrK)iterablerjrecs rr_flatten..recs;A!eT]++QAG  s?Ar?)rrs @rrLrLs x=r1c*\rSrSrSrSrSrSrSrg)rIz An object used to unflatten nested sequences after a given pattern (an arbitrarily nested sequence). The pattern shows the nested sequence shape desired when unflattening; the values it contains are irrelevant. c0URU5Ulgr )_build_unflatten_code_code)rpatterns rr_Unflattener.__init__s//8 r1c,^^/mUU4SjmT"U5 T$)zjBuild the unflatten opcode sequence for the given *iterable* structure (an iterable of nested sequences). c,>UHn[U[[45(aZ[U5S:a4TR [ 5 T"U5 TR [ 5 MaTR [5 MxTR [5 M gr&) rrVrKrEr _PUSH_LIST_POP_APPEND_EMPTY_TUPLE_APPEND_NEXT_VALUE)rrcoders rr/_Unflattener._build_unflatten_code..recsba%//1vz J/A D) $78KK 23r1r?)rrrrs @@rr"_Unflattener._build_unflatten_codes 4 H  r1c[U5n/nUn/nURHnU[La)URU5 UR/5 USnM5U[La!URUR 55 M_U[ LaURS5 M{U[LdMUR5nM U(aU5eU(aU5eU$)z*Rebuild a nested tuple structure. r?) rrrrrpopleftrrpop)rflatitervalsr+curstackops rr[_Unflattener.unflattensX**BZ S! 2"g)) 4<<>*** 2tiik%yx r1)rN) r9r:r;r<r=rrr[r>r?r1rrIrIs9(r1rIN) collectionsr numba.corerrobjectrrArLrrrrrIr?r1rrsP%2121jM=M=`  666r1