g,dZddlmZddlmZmZmZddlmZm Z ddl m Z m Z m Z mZmZmZmZmZddlmZGddZd Zd Zd Zd Zd ZedZGddZdS)zUtilities for defining a mutable struct. A mutable struct is passed by reference; hence, structref (a reference to a struct). )njit)typesimputilscgutils)default_managermodels) infer_getattrlower_getattr_genericlower_setattr_genericboxunbox NativeValue intrinsicoverload)AttributeTemplatec0eZdZdZdZdZdZdZdZdS)_Utilsz;Internal builder-code utils for structref definitions. c0||_||_||_dS)z Parameters ---------- context : a numba target context builder : a llvmlite IRBuilder struct_type : numba.core.types.StructRef N)contextbuilder struct_type)selfrrrs R/granary/edev/quantum/lib/python3.11/site-packages/numba/experimental/structref.py__init__z_Utils.__init__s   &cz|j}|j}|j}tj|||}||_|S)zKEncapsulate the MemInfo from a `StructRefPayload` in a `StructRef` )rrrrcreate_struct_proxymeminfo)rmirrrsts rnew_struct_refz_Utils.new_struct_ref(sB,,& 5W ( 5 5gw G G  rcl|j}|j}|j}tj||||S)z6Return a helper for accessing a StructRefType )value)rrrrr)rvalrrrs rget_struct_refz_Utils.get_struct_ref3sE,,& 7w*;77 WC    rcX|j}|j}|j}||}|j}|j||}|}|j|} | } | || }|S)zDGet the data pointer to the payload from a `StructRefType`. ) rrrr%rnrt meminfo_data get_data_typedata_model_managerget_value_typebitcast as_pointer) rr$rrr structvalrdata_ptrvaltypemodel alloc_types rget_data_pointerz_Utils.get_data_pointer>s,,& '',, #;++GW==++--*73))++ ??8Z-B-B-D-DEErc|j}|j}|j}||}|}t j||||}|S)zFGet a getter/setter helper for accessing a `StructRefPayload` )ref)rrrr3r)rr)rr$rrrr/r0datavals rget_data_structz_Utils.get_data_structOsm,,& ((--++--6'-g66 W(   rN) __name__ __module__ __qualname____doc__rr!r%r3r7rrrrsi ' ' '       "     rrctGfddt}td}td}dS)zDefine attributes on `struct_typeclass`. Defines both setters and getters in jit-code. This is called directly in `register()`. ceZdZZdZdS)*define_attributes..StructAttributec6||jvr|j|}|SdSN) field_dict)rtypattrattrtys rgeneric_resolvez:define_attributes..StructAttribute.generic_resolveis(s~%%- &%rN)r8r9r:keyrF)struct_typeclasssrStructAttributer?es)     rrIct|||}||}t||}|j|}t j||||SrA)rr7getattrrBrimpl_ret_borrowed) rrrCr$rDutilsr6ret fieldtypes rstruct_getattr_implz.define_attributes..struct_getattr_implnsXw--'',,gt$$N4( )'7IsKKKrcl|j\}}|\}}t|||} | |} |j|} ||||| } t | |} |j||| |j||| t| || dSrA) argsrr7rBcastrKr'increfdecrefsetattr)rrsigrRrD inst_typeval_typeinstancer$rMr6 field_typecasted old_values rstruct_setattr_implz.define_attributes..struct_setattr_implvs #H3w33''11)$/ gsHjAAGT**  7Hf555 7Hi888v&&&&&rN)r rr r )rHrIrPr^s` rdefine_attributesr_^s+]+,,LL-,L+,,''-,'''rc|tjurtdtj|jt |fd}t |d}dS)aDefine the boxing & unboxing logic for `struct_type` to `obj_class`. Defines both boxing and unboxing. - boxing turns an instance of `struct_type` into a PyObject of `obj_class` - unboxing turns an instance of `obj_class` into an instance of `struct_type` in jit-code. Use this directly instead of `define_proxy()` when the user does not want any constructor to be defined. cannot register c|t|j|j|}||}|j}t jt j}|||}|j |j }|j |j |} |j || |g} |j ||j | |j || S)z8 Convert a raw pointer to a Python int. )rrrr%rrMemInfoPointervoidptrr pyapi unserializeserialize_objectcall_function_objargsrU) rCr$crM struct_refrmip_type boxed_meminfo ctor_pyfuncty_pyobjresobj_ctors rbox_struct_refz%define_boxing..box_struct_refs qy!)S11))#.. $' 66h00 g))!'*B*B8*L*LMM 7&&qw'?'?'D'DEEg++ (M2   {### x    }%%% rc|j|d}tjtj}|||j}t|j|j |}| |}| }|j |t|S)N_meminfo)reobject_getattr_stringrrcrdr r#rrrr! _getvaluerUr) rCobjrimi_objrkrrMrjouts runbox_struct_refz'define_boxing..unbox_struct_refs..sJ??' 66 WWXv & & ,qy!)S11))"-- ""$$ v3rN)r StructRef ValueError _numba_box_r r )r obj_classrqryrps @r define_boxingr~seo%%=EO==>>>$H, ;       rc rd|}d}g}|D]}|d|d|d||}d|dt|d|d |d |d }t|t } t || | d } t || dS)zDefine the jit-code constructor for `struct_typeclass` using the Python type `py_class` and the required `fields`. Use this instead of `define_proxy()` if the user does not want boxing logic defined. z, z zst.z =  z def ctor(z/): struct_type = struct_typeclass(list(zip(z, [z]))) def impl(z)): st = new(struct_type) z# return st return impl )rHnewctorN)joinappendlistdictrexecr) py_classrHfieldsparamsindentinit_fields_bufk init_fieldssourceglbsrs rdefine_constructorrs YYv  F FO 00Q1////v--$$_55K -1&\\>D  F !1s ; ; ;D >>[&*?@@@k""" rcJddlm|jfd}|}||fS)anew(struct_type) A jit-code only intrinsic. Used to allocate an **empty** mutable struct. The fields are zero-initialized and must be set manually after calling the function. Example: instance = new(MyStruct) instance.field = field_value r)imp_dtorc `|j }|}||}|j||tj| ||j }|j ||}| || }| tj|||| } || _| SrA)r*r)r+get_abi_sizeofr'meminfo_alloc_dtor get_constantruintpmoduler(r,r-storerget_null_value make_helperrru) rr signaturerRr1r2 alloc_sizer data_pointer inst_structrrXs rcodegenznew..codegen-s *9+B+B+D+DE))++ ++J77 +00   j 9 9 HWgni 8 8   {//AA |Z5J5J5L5LMM   g,Z88,GGG))'9== % $$&&&r) numba.experimental.jitclass.baser instance_type) typingctxrrrWrrXs @@rrrsY:99999)I'''''', )K C <rcReZdZdZdZefdZdZedZ xZ S)StructRefProxyzA PyObject proxy to the Numba allocated structref data structure. Notes ----- * Subclasses should not define ``__init__``. * Subclasses can override ``__new__``. )_typerscft|}||_||_|S)aOCalled by boxing logic, the conversion of Numba internal representation into a PyObject. Parameters ---------- ty : a Numba type instance. mi : a wrapped MemInfoPointer. Returns ------- instance : a StructRefProxy instance. )super__new__rrs)clstyrrZ __class__s rr|zStructRefProxy._numba_box_Rs."77??3''rcr j}n)#t$rtfd}|_YnwxYw||S)zConstruct a new instance of the structref. This takes positional-arguments only due to limitation of the compiler. The arguments are mapped to ``cls(*args)`` in jit-code. c|SrAr<)rRrs rrz$StructRefProxy.__new__..ctorsssDz!r)_StructRefProxy__numba_ctorAttributeErrorr)rrRrs` rrzStructRefProxy.__new__hsn $#DD $ $ $  " " " "T " $C     $tT{s  #11c|jS)znReturns the Numba type instance for this structref instance. Subclasses should NOT override. )r)rs r _numba_type_zStructRefProxy._numba_type_zs zr) r8r9r:r; __slots__ classmethodr|rpropertyr __classcell__)rs@rrrGs}&I[*$XrrN)r;numbar numba.corerrrnumba.core.datamodelrrnumba.core.extendingr r r r r rrrnumba.core.typing.templatesrrr_r~rrrrrr<rrrs //////////88888888                    :99999DDDDDDDDN('('('V6 6 6 r@....   F '' 'T9999999999r