U ?h@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]6\}}t|tjs4|j ||j |j| q4dS)Ncsg|]}|qS)lookup.0tydmmrM/var/www/html/venv/lib/python3.8/site-packages/numba/core/datamodel/packer.py sz'DataPacker.__init__..) _dmm _fe_types_models _pack_mapZ _be_types enumerate isinstancerZOmittedappendZ get_data_type)selfr Zfe_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..)rrZmake_anonymous_struct)rrrZelemsrrr rszDataPacker.as_dataNc Csdg}t|jD]P\}}t||d|}|j|||}|dkrV||j||fq|||<q|SNr)rrrZ gep_inboundsrZload_from_data_pointerrr) rrptr formal_listresrZi_formalZelem_ptrvalrrr _do_load#s zDataPacker._do_loadcCs |||S)zK Load the packed values and return a (type, value) tuples. r")rrrrrr load.szDataPacker.loadcCs||||dS)z Load the packed values into a sequence indexed by formal argument number (skipping any Omitted position). Nr#)rrrrrrr load_into4szDataPacker.load_into)N) __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|||q$t||_ t t ||_ dSN) rZ_fe_argslen_nargs_dm_argsrrZget_argument_type _Unflattener _unflattenerlist_flatten_be_args)rr Zfe_argsZargtysr dmrrr rIs    zArgPacker.__init__csVt||jkr$td|jt|f|s,dSfddt|j|D}tt|}|S)z$Flatten all argument values z+invalid number of args: expected %d, got %drcsg|]\}}||qSr)Z as_argumentr r4r!rrr rasz*ArgPacker.as_arguments..)r,r- TypeErrorzipr.tupler2)rrrargsrr6r as_argumentsWs    zArgPacker.as_argumentscs*|j|}fddt|j|D}|S)z&Unflatten all argument values csg|]\}}||qSr)Z from_argumentr5r6rr rmsz,ArgPacker.from_arguments..)r0 unflattenr8r.)rrr:valtreerrr6r from_argumentshs    zArgPacker.from_argumentscCs0|j|}t||D]\}}|||qdS)z9Assign names for each flattened argument values. N)r0r<r8 _assign_names)rr:namesr=avalZanamerrr assign_namesss zArgPacker.assign_namesrcCsft|ttfr8t|D]\}}|j||||fdqn*dtt|}||g}dtt ||_ dS)N)depth.) rr9r1rr?joinmapstrfilterboolname)rZ val_or_nestedrJrCposrAZpostfixpartsrrr r?{s zArgPacker._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..)r9r3)rrrr argument_typesszArgPacker.argument_typesN)r) r&r'r(r)rr;r>rBr?propertyrNrrrr r*<s   r*csfdd|S)z3 Flatten nested iterable of (tuple, list). c3s8|D].}t|ttfr,|D] }|Vqq|VqdSr+)rr9r1)iterablerjrecrr rSs   z_flatten..recr)rPrrRr r2s r2c@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|||_dSr+)_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]R}t|ttfrLt|dkr@t|tqVtqtqdSr) rr9r1r,r _PUSH_LIST_POP_APPEND_EMPTY_TUPLE_APPEND_NEXT_VALUE)rPrcoderSrr rSs    z/_Unflattener._build_unflatten_code..recr)rrPrr_r rXs z"_Unflattener._build_unflatten_codecCst|}g}|}g}|jD]f}|tkrD|||g|d}q|tkr\||q|tkrp|dq|tkr|}q|rt ||rt ||S)z*Rebuild a nested tuple structure. r) rrYr[rr^popleftr]r\popAssertionError)rZflatitervalsr curstackoprrr r<s$        z_Unflattener.unflattenN)r&r'r(r)rrXr<rrrr r/sr/N) collectionsrZ numba.corerrobjectrr*r2r[r^r]r\r/rrrr s 5P