o ۾i@shddlmZddlmZmZGdddeZGdddeZddZd Z d Z d Z d Z Gd ddeZ dS))deque)typescgutilsc@s:eZdZdZddZddZd ddZd 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|tjs5|j ||j |j| qdS)Ncsg|]}|qS)lookup.0tydmmrO/home/ubuntu/.local/lib/python3.10/site-packages/numba/core/datamodel/packer.py sz'DataPacker.__init__..) _dmm _fe_types_models _pack_map _be_types enumerate isinstancerOmittedappend get_data_type)selfr fe_typesir rr r __init__s  zDataPacker.__init__cs$fddjD}t|S)zE Return the given values packed as a data structure. cs"g|] }j||qSr)ras_data)r rbuilderrvaluesrr rsz&DataPacker.as_data..)rrmake_anonymous_struct)rrr elemsrrr rs zDataPacker.as_dataNc Csdg}t|jD](\}}t||d|}|j|||}|dur+||j||fq|||<q|SNr)rrr gep_inboundsrload_from_data_pointerrr) rrptr formal_listresri_formalelem_ptrvalrrr _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_intoN) __name__ __module__ __qualname____doc__rrr,r.r/rrrr rs  rc@sFeZdZdZddZddZddZdd Zdd d Ze d dZ 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_argslen_nargs_dm_argsrrget_argument_type _Unflattener _unflattenerlist_flatten_be_args)rr fe_argsargtysr dmrrr rIs    zArgPacker.__init__csVt||jkrtd|jt|f|sdSfddt|j|D}tt|}|S)z$Flatten all argument values z+invalid number of args: expected %d, got %drcg|] \}}||qSr) as_argumentr rBr+rrr raz*ArgPacker.as_arguments..)r7r8 TypeErrorzipr9tupler>)rrr argsrrFr as_argumentsWs    zArgPacker.as_argumentscs*|j|}fddt|j|D}|S)z&Unflatten all argument values crCr) from_argumentrErFrr rmrGz,ArgPacker.from_arguments..)r< unflattenrIr9)rrrKvaltreer rrFr from_argumentshs   zArgPacker.from_argumentscCs0|j|}t||D] \}}|||q dS)z9Assign names for each flattened argument values. N)r<rNrI _assign_names)rrKnamesrOavalanamerrr assign_namesss zArgPacker.assign_namesrcCsht|ttfrt|D]\}}|j||||fdq dSdtt|}||g}dtt ||_ dS)N)depth.) rrJr=rrQjoinmapstrfilterboolname)r val_or_nestedr]rVposrSpostfixpartsrrr rQ{szArgPacker._assign_namescCstdd|jDS)z\Return a list of LLVM types that are results of flattening composite types. css|] }|dkr|VqdS)rNrrrrr sz+ArgPacker.argument_types..)rJr?)rrrr argument_typesszArgPacker.argument_typesN)r) r1r2r3r4rrLrPrUrQpropertyrcrrrr r5<s   r5csfdd|S)z3 Flatten nested iterable of (tuple, list). c3s:|D]}t|ttfr|D]}|Vqq|VqdSr0)rrJr=)iterablerjrecrr rhs z_flatten..recr)rerrgr r>s r>c@s(eZdZdZddZddZddZdS) 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)rpatternrrr rsz_Unflattener.__init__csgfdd|S)zzBuild the unflatten opcode sequence for the given *iterable* structure (an iterable of nested sequences). cs\|D])}t|ttfr&t|dkr t|tqtqtqdSr#) rrJr=r7r _PUSH_LIST_POP_APPEND_EMPTY_TUPLE_APPEND_NEXT_VALUE)rercoderhrr rhs     z/_Unflattener._build_unflatten_code..recr)rrerrtr rms z"_Unflattener._build_unflatten_codecCst|}g}|}g}|jD]3}|tur"|||g|d}q |tur.||q |tur8|dq |tur@|}q |rGJ||rMJ||S)z*Rebuild a nested tuple structure. r) rrnrprrspopleftrrrqpop)rflatitervalsr(curstackoprrr rNs&       z_Unflattener.unflattenN)r1r2r3r4rrmrNrrrr r;s  r;N) collectionsr numba.corerrobjectrr5r>rprsrrrqr;rrrr s 5P