:j+/6dZddlZddlmZddlmZddlmZmZm Z ddl m Z m Z ddl mZmZmZmZmZmZmZmZmZddlmZGd d Zd Zd Zd ZdZdZedZGddZ eejBejDejDdZ#y)zUtilities for defining a mutable struct. A mutable struct is passed by reference; hence, structref (a reference to a struct). N)create_struct_proxy)njit)typesimputilscgutils)default_managermodels) infer_getattrlower_getattr_genericlower_setattr_generic lower_builtinboxunbox 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 q/home/mcse/projects/srt_converter/srt-converter-venv/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` )rrrrrmeminfo)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_ref6sB,,,,&& 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_pointerAs,,,,&& '', ##;;++GW=++-**73))+ ??8Z-B-B-DErc|j}|j}|j}|j|}|j }t j ||||}|S)zFGet a getter/setter helper for accessing a `StructRefPayload` )ref)rrrr5r+rr)rr&rrrr1r2datavals rget_data_structz_Utils.get_data_structRsd,,,,&& ((-++-6'--g6 W( rN) __name__ __module__ __qualname____doc__rr"r'r5r9rrrrs  '   " 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_resolvels&s~~%- &rN)r:r;r<keyrH)struct_typeclasssrStructAttributerAhs  rrKct|||}|j|}t||}|j|}t j ||||SrC)rr9getattrrDrimpl_ret_borrowed) rrrEr&rFutilsr8ret fieldtypes rstruct_getattr_implz.define_attributes..struct_getattr_implqsQw-'',gt$NN4( ))'7IsKKrcT|j\}}|\}}t|||} | j|} |j|} |j |||| } t | |} |j j||| |j j||| t| || yrC) argsrr9rDcastrMr)increfdecrefsetattr)rrsigrTrF inst_typeval_typeinstancer&rOr8 field_typecasted old_values rstruct_setattr_implz.define_attributes..struct_setattr_implys #H3w3''1))$/ gsHjAGT*  7Hf5 7Hi8v&rN)r rr r )rJrKrRr`s` rdefine_attributesraasU++,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'rrMemInfoPointervoidptrrpyapi unserializeserialize_objectcall_function_objargsrW) rEr&crO 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)rgobject_getattr_stringrrerfrr%rrrr" _getvaluerWr) rEobjrkmi_objrmr rOrlouts runbox_struct_refz'define_boxing..unbox_struct_refs..sJ?'' 6 WWXv & , ,qyy!))S1))"- ""$ v3rN)r StructRef ValueError _numba_box_rr)r obj_classrsr{rrs @r define_boxingrsdeoo%+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 )rJnewctorN)joinappendlistdictrexecr) py_classrJfieldsparamsindentinit_fields_bufk init_fieldssourceglbsrs rdefine_constructorrsYYv F FO 0QCs1#/0vh-$$_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%SrC)r,r+r-get_abi_sizeofr)meminfo_alloc_dtor get_constantruintpmoduler*r.r/storerget_null_value make_helperrrw) rr signaturerTr3r4 alloc_sizer data_pointer inst_structrrZs rcodegenznew..codegen0s**9+B+B+DE))+ ++J7 ++00   j 9 Wgnni 8  {{//A |Z5J5J5LM   g,,Z8,G))'9= % $$&&r) numba.experimental.jitclass.baser instance_type) typingctxrrrYrrZs @@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__``. )_typerucBt||}||_||_|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__rru)clstyr r\ __class__s rr~zStructRefProxy._numba_box_Us'"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|SrCr>)rTrs rrz$StructRefProxy.__new__..ctorvs Dz!r)_StructRefProxy__numba_ctorAttributeErrorr)rrTrs` rrzStructRefProxy.__new__ksL $##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_}s zzr) r:r;r<r= __slots__ classmethodr~rpropertyr __classcell__)rs@rrrJs:&I*$rrc|\}}|j\}}t||||j}t||||j} |jd|| S)z Define the 'is' operator for structrefs by comparing the memory addresses. This is the identity check for structref objects. r$z==)rTrr icmp_unsigned) rrrYrTabatybtya_ptrb_ptrs r structref_isrsg DAqxxHC $  $WgQ ? G GE $  $WgQ ? G GE  ue 44r)$r=operatornumba.core.cgutilsrnumbar numba.corerrrnumba.core.datamodelrr numba.core.extendingr r r r rrrrrnumba.core.typing.templatesrrrarrrrrris_r|rr>rrrs 2//8   :DDN('V6 r@.. F ' 'T99xx||U__eoo> 5? 5r