DŽgxddlmZddlmZmZGddeZGddeZdZdZ d Z d Z d Z Gd d eZ y))deque)typescgutilsc0eZdZdZdZdZddZdZdZy) 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. c||_||_|Dcgc]}|j|c}|_g|_g|_t |D]q\}}t|tjr!|jj||j j|j|jsycc}wN) _dmm _fe_typeslookup_models _pack_map _be_types enumerate isinstancerOmittedappend get_data_type)selfdmmfe_typestyis c/home/mcse/projects/flask_80/flask-venv/lib/python3.12/site-packages/numba/core/datamodel/packer.py__init__zDataPacker.__init__s !19:2 2: x( GEArb%--0%%a(%%dll1o&C&C&EF G ;sCc|jDcgc]$}|j|j|||&}}tj||Scc}w)zE Return the given values packed as a data structure. )rr as_datarmake_anonymous_struct)rbuildervaluesrelemss rrzDataPacker.as_datasV ..*a((&)<**,,We<<*s)ANcg}t|jD]d\}}tj||d|}|j|j ||}|!|j |j||f`|||<f|SNr)rrr gep_inboundsr load_from_data_pointerrr ) rrptr formal_listresri_formalelem_ptrvals r_do_loadzDataPacker._do_load#s$T^^4 ,KAx++GS!Q?H,,x(??RC" DNN84c:;(+ H%  , c&|j||S)zK Load the packed values and return a (type, value) tuples. r,)rrr&s rloadzDataPacker.load.s}}Wc**r-c*|j|||y)z Load the packed values into a sequence indexed by formal argument number (skipping any Omitted position). Nr/)rrr&r's r load_intozDataPacker.load_into4s gsK0r-r ) __name__ __module__ __qualname____doc__rrr,r0r2r-rrrs! G= + 1r-rc@eZdZdZdZdZdZdZd dZe dZ y) ArgPackera 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\||_||_t||_g|_g}|D]W}|jj |}|jj ||j |jYt||_ tt||_ yr ) r _fe_argslen_nargs_dm_argsr rget_argument_type _Unflattener _unflattenerlist_flatten_be_args)rrfe_argsargtysrdms rrzArgPacker.__init__Is  'l   2B!!"%B MM  $ MM"..0 1 2)0Xf-. r-c&t||jk7r#td|jt|fz|syt|j|Dcgc]\}}|j ||}}}t t|}|Scc}}w)z$Flatten all argument values z+invalid number of args: expected %d, got %dr7)r<r= TypeErrorzipr> as_argumenttuplerC)rrr rGr+argss r as_argumentszArgPacker.as_argumentsWs v;$++ %I#{{CK89: : #4==&9Bw,Xd^$ sB c|jj|}t|j|Dcgc]\}}|j ||}}}|Scc}}w)z&Unflatten all argument values )rA unflattenrJr> from_argument)rrrMvaltreerGr+r s rfrom_argumentszArgPacker.from_argumentshs_##--d3!$T]]G!<b#""7C0 sAc|jj|}t||D]\}}|j||y)z9Assign names for each flattened argument values. N)rArPrJ _assign_names)rrMnamesrRavalanames r assign_nameszArgPacker.assign_namesssD##--d3w. ,KD%   tU + ,r-ct|ttfr,t|D]\}}|j ||||fzydj t t|}||g}dj tt||_ y)N)depth.) rrLrBrrUjoinmapstrfilterboolname)r val_or_nestedrbr[posrWpostfixpartss rrUzArgPacker._assign_names{s| meT] 3&}5 E T""4UcV^"D Ehhs3/G7OE!$&u*=!>M r-c:td|jDS)z\Return a list of LLVM types that are results of flattening composite types. c3,K|] }|dk7s |yw)r7Nr7).0rs r z+ArgPacker.argument_types..s.recsC A!eT]+QAG  s58r7)rorqs @rrCrCs x=r-c"eZdZdZdZdZdZy)r@z 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. c0|j||_yr )_build_unflatten_code_code)rpatterns rrz_Unflattener.__init__s//8 r-c*gfd|S)zzBuild the unflatten opcode sequence for the given *iterable* structure (an iterable of nested sequences). c|D]}t|ttfrWt|dkDr3j t |j t Zj tpj tyr#) rrLrBr<r _PUSH_LIST_POP_APPEND_EMPTY_TUPLE_APPEND_NEXT_VALUE)rorcoderqs rrqz/_Unflattener._build_unflatten_code..recsd 4a%/1vz J/A D) $78KK 23 4r-r7)rrorrqs @@rrxz"_Unflattener._build_unflatten_codes 4 H  r-ct|}g}|}g}|jD]}|tur(|j||jg|d}3|tur |j|j [|t ur|jdu|tus~|j}|rJ||rJ||S)z*Rebuild a nested tuple structure. r7) rryr}rrpopleftrr~pop)rflatitervalsr(curstackops rrPz_Unflattener.unflattensX** "BZ S! 2"g)) 4<<>*** 2tiik "%yx r-N)r3r4r5r6rrxrPr7r-rr@r@s9(r-r@N) collectionsr numba.corerrobjectrr9rCr}rrr~r@r7r-rrsP%2121jM=M=`  666r-