xKg,dZddlmZddlmZmZmZddlmZm Z ddl m Z m Z m Z mZmZmZmZmZddlmZGddZd Zd Zd Zd Zd ZedZGddZy)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)AttributeTemplatec.eZdZdZdZdZdZdZdZy)_Utilsz;Internal builder-code utils for structref definitions. c.||_||_||_y)z Parameters ---------- context : a numba target context builder : a llvmlite IRBuilder struct_type : numba.core.types.StructRef N)contextbuilder struct_type)selfrrrs `/home/alanp/www/video.onchill/myenv/lib/python3.12/site-packages/numba/experimental/structref.py__init__z_Utils.__init__s  &c|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(sF,,,,&& 5W ( ( 5gw G  rc|j}|j}|j}tj||||S)z6Return a helper for accessing a StructRefType )value)rrrrr)rvalrrrs rget_struct_refz_Utils.get_struct_ref3sB,,,,&& 7w**;7 WC  rc^|j}|j}|j}|j|}|j}|j 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-DErc|j}|j}|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_structOsd,,,,&& ((-++-6'--g6 W( rN) __name__ __module__ __qualname____doc__rr!r%r3r7rrrrs  '   " rrc~tGfddt}td}td}y)zDefine attributes on `struct_typeclass`. Defines both setters and getters in jit-code. This is called directly in `register()`. ceZdZWZdZy)*define_attributes..StructAttributecB||jvr|j|}|SyN) field_dict)rtypattrattrtys rgeneric_resolvez:define_attributes..StructAttribute.generic_resolveis&s~~%- &rN)r8r9r:keyrF)struct_typeclasssrStructAttributer?es  rrIct|||}|j|}t||}|j|}t j ||||SrA)rr7getattrrBrimpl_ret_borrowed) rrrCr$rDutilsr6ret fieldtypes rstruct_getattr_implz.define_attributes..struct_getattr_implnsQw-'',gt$NN4( ))'7IsKKrcT|j\}}|\}}t|||} | j|} |j|} |j |||| } t | |} |j j||| |j j||| t| || yrA) argsrr7rBcastrKr'increfdecrefsetattr)rrsigrRrD inst_typeval_typeinstancer$rMr6 field_typecasted old_values rstruct_setattr_implz.define_attributes..struct_setattr_implvs #H3w3''1))$/ gsHjAGT*  7Hf5 7Hi8v&rN)r rr r )rHrIrPr^s` rdefine_attributesr_^sU++,L-L+,'-'rc|tjurtdtj|jt |fd}t |d}y)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 ct|j|j|}|j|}|j}t j t j}|j||}|jj|jj }|jj|jj|} |jj|| |g} |jj||jj| |jj|| 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 qyy!))S1))#. $$'' 6h0 gg))!''*B*B8*LM 77&&qww'?'?'DEgg++ (M2  {# x  }% rc|jj|d}tjtj}|j ||j }t|j|j|}|j|}|j}|jj|t|S)N_meminfo)reobject_getattr_stringrrcrdr r#rrrr! _getvaluerUr) rCobjrimi_objrkrrMrjouts runbox_struct_refz'define_boxing..unbox_struct_refs..sJ?'' 6 WWXv & , ,qyy!))S1))"- ""$ v3rN)r StructRef ValueError _numba_box_r r )r obj_classrqryrps @r define_boxingr~sdeoo%+EOO+<=>>$$H, ;    rc .dj|}d}g}|D]}|jd|d|d|j|}d|dt|d|d |d |d }t|t } t || | d } t || y)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_constructorrsYYv F FO QCs1#/vh-$$_5K --1&\N#fXFX   F !1s ;D >[&*?*?@k" rcNddlm|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}|j}|j|}|jj ||j tj| ||j }|jj||}|j||j}|jtj|||j!| } || _| j%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+DE))+ ++J7 ++00   j 9 Wgnni 8  {{//A |Z5J5J5LM   g,,Z8,G))'9= % $$&&r) numba.experimental.jitclass.baser instance_type) typingctxrrrWrrXs @@rrrs.:))I', K C <rcFeZdZdZdZefdZdZedZ xZ S)StructRefProxyzA PyObject proxy to the Numba allocated structref data structure. Notes ----- * Subclasses should not define ``__init__``. * Subclasses can override ``__new__``. )_typerscBt||}||_||_|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'"7?3'rcp j}||S#t$rtfd}|_Y%wxYw)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__..ctorss Dz!r)_StructRefProxy__numba_ctorAttributeErrorr)rrRrs` rrzStructRefProxy.__new__hsL $##DT{ $  " " $C   $s 55c|jS)znReturns the Numba type instance for this structref instance. Subclasses should NOT override. )r)rs r _numba_type_zStructRefProxy._numba_type_zs zzr) r8r9r:r; __slots__ classmethodr|rpropertyr __classcell__)rs@rrrGs:&I*$rrN)r;numbar numba.corerrrnumba.core.datamodelrrnumba.core.extendingr r r r r rrrnumba.core.typing.templatesrrr_r~rrrrrr<rrrsu //8   :DDN('V6 r@.. F ' 'T99r