o gµ„iõã@shddlmZddlmZmZGdd„deƒZGdd„deƒZdd„Zd Z d Z d Z d Z Gd d„deƒZ dS)é)Údeque)ÚtypesÚcgutilsc@s:eZdZdZdd„Zdd„Zd dd„Zd d „Zd d „ZdS)Ú 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. cspˆ|_||_‡fdd„|Dƒ|_g|_g|_t|ƒD]\}}t|tjƒs5|j  |¡|j  |j|  ¡¡qdS)Ncsg|]}ˆ |¡‘qS©)Úlookup©Ú.0Úty©ÚdmmrúN/home/kuhnn/.local/lib/python3.10/site-packages/numba/core/datamodel/packer.pyÚ sz'DataPacker.__init__..) Ú_dmmÚ _fe_typesÚ_modelsÚ _pack_mapÚ _be_typesÚ enumerateÚ isinstancerÚOmittedÚappendÚ get_data_type)Úselfr Úfe_typesÚir rr r Ú__init__s  €ýzDataPacker.__init__cs$‡‡‡fdd„ˆjDƒ}t ˆ|¡S)zE Return the given values packed as a data structure. cs"g|] }ˆj| ˆˆ|¡‘qSr)rÚas_data)r r©ÚbuilderrÚvaluesrr rsÿz&DataPacker.as_data..)rrÚmake_anonymous_struct)rrr Úelemsrrr rsÿ zDataPacker.as_dataNc Csdg}t|jƒD](\}}t ||d|¡}|j| ||¡}|dur+| |j||f¡q|||<q|S©Nr)rrrÚ gep_inboundsrÚload_from_data_pointerrr) rrÚptrÚ formal_listÚresrÚi_formalÚelem_ptrÚvalrrr Ú_do_load#s zDataPacker._do_loadcCs | ||¡S)zK Load the packed values and return a (type, value) tuples. ©r,)rrr&rrr Úload.s zDataPacker.loadcCs| |||¡dS)zƒ Load the packed values into a sequence indexed by formal argument number (skipping any Omitted position). Nr-)rrr&r'rrr Ú load_into4szDataPacker.load_into©N) Ú__name__Ú __module__Ú __qualname__Ú__doc__rrr,r.r/rrrr rs  rc@sFeZdZdZdd„Zdd„Zdd„Zdd „Zdd d „Ze d d„ƒZ dS)Ú 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. cCsl||_||_t|ƒ|_g|_g}|D]}|j |¡}|j |¡| | ¡¡qt|ƒ|_ t t |ƒƒ|_ dSr0) rÚ_fe_argsÚlenÚ_nargsÚ_dm_argsrrÚget_argument_typeÚ _UnflattenerÚ _unflattenerÚlistÚ_flattenÚ_be_args)rr Úfe_argsÚargtysr Údmrrr rIs    zArgPacker.__init__csVt|ƒ|jkrtd|jt|ƒfƒ‚|sdS‡fdd„t|j|ƒDƒ}tt|ƒƒ}|S)z$Flatten all argument values z+invalid number of args: expected %d, got %drcóg|] \}}| ˆ|¡‘qSr)Ú as_argument©r rBr+©rrr raóÿz*ArgPacker.as_arguments..)r7r8Ú TypeErrorÚzipr9Útupler>)rrr ÚargsrrFr Ú as_argumentsWs ÿ  ÿ zArgPacker.as_argumentscs*|j |¡}‡fdd„t|j|ƒDƒ}|S)z&Unflatten all argument values crCr)Ú from_argumentrErFrr rmrGz,ArgPacker.from_arguments..)r<Ú unflattenrIr9)rrrKÚvaltreer rrFr Úfrom_argumentshs   ÿzArgPacker.from_argumentscCs0|j |¡}t||ƒD] \}}| ||¡q dS)z9Assign names for each flattened argument values. N)r<rNrIÚ _assign_names)rrKÚnamesrOÚavalÚanamerrr Ú assign_namesss ÿzArgPacker.assign_namesrcCsht|ttfƒrt|ƒD]\}}|j||||fdq dSd tt|ƒ¡}||g}d tt |ƒ¡|_ dS)N)ÚdepthÚ.) rrJr=rrQÚjoinÚmapÚstrÚfilterÚboolÚname)rÚ val_or_nestedr]rVÚposrSÚpostfixÚpartsrrr rQ{sÿzArgPacker._assign_namescCstdd„|jDƒƒS)z\Return a list of LLVM types that are results of flattening composite types. css|] }|dkr|VqdS)rNrrrrr Ú ‰s€z+ArgPacker.argument_types..)rJr?)rrrr Úargument_types„szArgPacker.argument_typesN)r) r1r2r3r4rrLrPrUrQÚpropertyrcrrrr r5<s   r5cs‡fdd„‰ˆ|ƒS)z3 Flatten nested iterable of (tuple, list). c3s:|D]}t|ttfƒrˆ|ƒD]}|Vqq|VqdSr0)rrJr=)ÚiterablerÚj©Úrecrr rhs€ ÿûz_flatten..recr)rerrgr r>Œs r>ééééc@s(eZdZdZdd„Zdd„Zdd„ZdS) 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. cCs| |¡|_dSr0)Ú_build_unflatten_codeÚ_code)rÚpatternrrr r§sz_Unflattener.__init__csg‰‡‡fdd„‰ˆ|ƒˆS)zzBuild the unflatten opcode sequence for the given *iterable* structure (an iterable of nested sequences). cs\|D])}t|ttfƒr&t|ƒdkr ˆ t¡ˆ|ƒˆ t¡qˆ t¡qˆ t¡qdSr#) rrJr=r7rÚ _PUSH_LISTÚ_POPÚ_APPEND_EMPTY_TUPLEÚ_APPEND_NEXT_VALUE)rer©Úcoderhrr rh¯s     ÷z/_Unflattener._build_unflatten_code..recr)rrerrtr rmªs z"_Unflattener._build_unflatten_codecCsžt|ƒ}g}|}g}|jD]3}|tur"| |¡| g¡|d}q |tur.| | ¡¡q |tur8| d¡q |tur@| ¡}q |rGJ|ƒ‚|rMJ|ƒ‚|S)z*Rebuild a nested tuple structure. éÿÿÿÿr) rrnrprrsÚpopleftrrrqÚpop)rÚflatiterÚvalsr(ÚcurÚstackÚoprrr rN¾s&     €  z_Unflattener.unflattenN)r1r2r3r4rrmrNrrrr r;Ÿs  r;N)Ú collectionsrÚ numba.corerrÚobjectrr5r>rprsrrrqr;rrrr Ús 5P