o 2g@sUdZddlmZddlmZmZmZddlmZm Z m Z ddl m Z ddl mZmZmZmZmZmZmZmZmZmZddlmZmZmZmZmZddlmZm Z dd l!m"Z"m#Z#m$Z$m%Z%dd l&m'Z'm(Z(m)Z)dd l*m+Z+dd l,m-Z-dddZ.dddZ/eee%ege0fZ1ee0ee0e1ffZ2dddfdddfde.fde.fd e.fd!e.fd"efd#efd$Z3d%e4d&<d'efd(efd(efd)efd*Z5d%e4d+<d,d-efiZ6d%e4d.<id/d0efd1d2efd3d4efd5d6efd7d8efd9d:efd;defd?d>efd@dAefdBdAefdCdDefdEdDefdFdGefdHdGefdIdJefdKdJefidLdMefdNdMefdOdPefdQdPefdRdSefdTdSefdUdVefdWdVefdXdYefdZdYefd[d\efd]d\efd^d_efd`d_efdadbefdcdbefdddeefdfefdgefdhefdiefdjefdkefdlefdmefdnefdoefdpefdqefdqefdrefdsZ7d%e4dt<due.fdve.fdwe.fdxZ8d%e4dy<dzd{e5fd|d}e6fd~de7fdde8fgZ9dhZ:dddZ;dddZdddZ?dddZ@dddZAdddZBdddZCdddZDdddZEdddZFdddZGdddZHdddZIdddZJdddZKdddÄZLdddDŽZMdddʄZNddd΄ZOdddЄZPddd҄ZQdddԄZRddd؄ZSdddڄZTddd݄ZUdddZVdddZWdS)z8Code generation for native classes and related wrappers.) annotations)CallableMappingTuple)EmitterHeaderDeclaration ReturnHandler)native_function_header) generate_bin_op_wrappergenerate_bool_wrappergenerate_contains_wrappergenerate_dunder_wrappergenerate_get_wrappergenerate_hash_wrappergenerate_ipow_wrappergenerate_len_wrappergenerate_richcompare_wrappergenerate_set_del_item_wrapper) BITMAP_BITS BITMAP_TYPE NATIVE_PREFIXPREFIX REG_PREFIX)ClassIR VTableEntries)FUNC_CLASSMETHODFUNC_STATICMETHODFuncDeclFuncIR)RTupleRTypeobject_rprimitive) NameGenerator) is_same_typeclrfnremitterrreturnstrcCt||jSN)rcnamenamesr$r%r&r.O/home/garg/my-data/venv/lib/python3.10/site-packages/mypyc/codegen/emitclass.py native_slotr0cCr)r*)rr+r,r-r.r.r/ wrapper_slot!r1r2tp_initcC t|||Sr*)generate_init_for_classcter.r.r/- r:tp_callcCr4r*)generate_call_wrapperr6r.r.r/r:.r;tp_strtp_repr tp_iternexttp_itertp_hash tp_descr_get)__init____call____str____repr____next____iter____hash____get__ SlotTable SLOT_DEFS mp_subscriptmp_ass_subscript mp_length) __getitem__ __setitem__ __delitem____len__AS_MAPPING_SLOT_DEFS __contains__ sq_containsAS_SEQUENCE_SLOT_DEFS__bool__nb_bool__int__nb_int __float__nb_float__neg__ nb_negative__pos__ nb_positive__abs__ nb_absolute __invert__ nb_invert__add__nb_add__radd____sub__ nb_subtract__rsub____mul__ nb_multiply__rmul____mod__ nb_remainder__rmod__ __truediv__nb_true_divide __rtruediv__ __floordiv__nb_floor_divide __rfloordiv__ __divmod__ nb_divmod __rdivmod__ __lshift__ nb_lshift __rlshift__ __rshift__ nb_rshift __rrshift____and__nb_and__rand____or__nb_or__ror____xor__nb_xor__rxor__ __matmul__nb_matrix_multiply __rmatmul____iadd__nb_inplace_addnb_inplace_subtractnb_inplace_multiplynb_inplace_remaindernb_inplace_true_dividenb_inplace_floor_dividenb_inplace_lshiftnb_inplace_rshiftnb_inplace_and nb_inplace_ornb_inplace_xornb_inplace_matrix_multiplynb_powernb_inplace_power)__isub____imul____imod__ __itruediv__ __ifloordiv__ __ilshift__ __irshift____iand____ior____ixor__ __imatmul____pow____rpow____ipow__AS_NUMBER_SLOT_DEFSam_awaitam_aiteram_anext) __await__ __aiter__ __anext__AS_ASYNC_SLOT_DEFS as_mappingPyMappingMethods as_sequencePySequenceMethods as_numberPyNumberMethodsas_asyncPyAsyncMethodsrJcCs|rdSt|||S)NPyVectorcall_Call)use_vectorcallr2r-r.r.r/r=s r=attrcCs&|dr |dks |dkrd|S|S)zuMap dunder method name to sort key. Sort reverse operator methods and __delitem__ after others ('x' > '_'). __rrrSx) startswith)rr.r.r/slot_keysrtabledict[str, str]c Csi}i}t|dddD]2\}\}}||}|r@|d|ks%|tvr@||vr0||||<q|||d|}|||<|||<q|S)NcSs t|dS)Nr)r)rr.r.r/r:r;z generate_slots..)keyr)sorteditemsget_method_and_class ALWAYS_FILL) r$rr&fields generatednameslot generator method_clsr.r.r/generate_slotss  r c_emitterexternal_emitterNonecCs|j}||}td||ddd|j|<|jsdSt|||j o)|j }|rBtt|j |ddd|j| |j <dSdS)NzPyTypeObject *;T) needs_export) contexttype_struct_namer declarations is_ext_classgenerate_object_structis_trait builtin_baser ctornative_function_name)r$rrr&rr generate_fullr.r.r/generate_class_type_decls   rmodulec"s|j}|j}|d}|d}|d}|d}|d} |d} |d} |d} |d } |d }i}d |d |d <|j oL|j }|jpV|j pV|j}|js^||d <|rxd|d|d<d|d|d<d|d|d<|r~||d<| |d<dLfdd }||d}|d}| t |t t D]\}}}t ||}|rt ||||}d||d|<qt|}|r||d<|j}|jrd|jd }n |jrd!}nd|d }|jr't|s'|d"}d#|d$d%|d&d'|d&d(d)||d*<|d+|d,<jd-kr&||d.<||d/<n||d,<|rd0|d1|jd2us?Jt|jd3||d}t||| |||t|| |t|| |t|| | ||jrt||d4| d4d5d6}|nd2}t||| d7d6} ||rt||t||||jrt||t|| |gd8}|r| d9|!d:r"rd;#|j|d<<| d=|$d>sd?|d@<t|r| dAdB%||dC<dD&|dEdF|'D]\} }!dG| dH|!dIqd)dJj#&|dK|rat(|||| |t)||j||| |rkt*|d2Sd2S)MzRGenerate C code for a class. This is the main entry point to the module. _setup_new_members _getseters_vtable _traverse_clear_dealloc_methods_trait_vtable_setup"tp_nametp_newz (destructor) tp_deallocz(traverseproc) tp_traversez (inquiry)tp_clear tp_getset tp_methodsr'rcs dSr*) emit_liner.r&r.r/rs z!generate_class..emit_line__mypyc_defaults_setuprD&tp_tp_richcomparezsizeof()zsizeof(PyObject)z + sizeof(PyObject *)z PyMemberDef [] = {z{"__dict__", T_OBJECT_EX, z , 0, NULL},z{"__weakref__", T_OBJECT_EX, z{0}}; tp_membersz + 2*sizeof(PyObject *) tp_basicsize tp_dictoffsettp_weaklistoffsetstatic PyObject *z(PyTypeObject *type);Nr_shadowT)shadowF)Py_TPFLAGS_DEFAULTPy_TPFLAGS_HEAPTYPEPy_TPFLAGS_BASETYPEPy_TPFLAGS_HAVE_GCrEzoffsetof({}, vectorcall)tp_vectorcall_offset_Py_TPFLAGS_HAVE_VECTORCALL tp_vectorcallrr<Py_TPFLAGS_MANAGED_DICT | tp_flagszstatic PyTypeObject z_template_ = {zPyVarObject_HEAD_INIT(NULL, 0). = ,z3static PyTypeObject *{t}_template = &{t}_template_;)r8)r'r)+r name_prefixr,rrneeds_getseters is_generatedhas_dict get_methodupdaterrM SIDE_TABLESgenerate_side_table_for_classr struct_namehas_managed_dict emit_lines capi_versionrrr generate_new_for_classgenerate_traverse_for_classgenerate_clear_for_classgenerate_dealloc_for_classallow_interpreted_subclassesgenerate_vtablesgenerate_getseter_declarationsgenerate_getseters_tablegenerate_new_for_traitgenerate_methods_tableappend has_methodrformatgetjoinrrgenerate_setup_for_classgenerate_constructor_for_classgenerate_getseters)"r$rr&rr setup_namenew_name members_namegetseters_name vtable_name traverse_name clear_name dealloc_name methods_namevtable_setup_namerrrr defaults_fninit_fn table_nametype slot_defsslotstable_struct_namerichcompare_namer base_size weak_offsetshadow_vtable_nameflagsfieldvaluer.rr/generate_classs                                      rJ attributer,r"cC||j|jd|S)N_get_ private_name module_namerr$rKr,r.r.r/ getter_nametrRcCrL)N_set_rNrQr.r.r/ setter_namexrSrUc CsZt}g}|gd7}|dr|r|dg}t|jD]j}|js|jrY|t |jd|vrYt dt |jt D]}||}||vrX|t d|d||q=|j D]+\}}||f|vr|||||d|||ft|tr||q^q |d||jd|d t|d d |jj||j<dS) N)ztypedef struct { PyObject_HEADzCPyVTableItem *vtable;rEzvectorcallfunc vectorcall;rr rz} T)is_type)setr+rr*reversedbase_mror bitmap_attrs bitmap_fieldlenrangerr attributesr ctype_spacedradd isinstancerdeclare_tuple_structrr,rrr) r$r& seen_attrslinesr]baseirrtyper.r.r/r|s8         rrhr;r6rboolc sdfdd }dfdd }d |td tjd tjjD](\}}d ||d td t|dd||td t|jq,dt|ddj rxsxt|dg} jD]#\}}||} ||} t || gt | || || | fqt j|| dd| s|S|dt| d S)aEmit the vtables and vtable setup functions for a class. This includes both the primary vtable and any trait implementation vtables. The trait vtables go before the main vtable, and have the following layout: { CPyType_T1, // pointer to type object C_T1_trait_vtable, // pointer to array of method pointers C_T1_offset_table, // pointer to array of attribute offsets CPyType_T2, C_T2_trait_vtable, C_T2_offset_table, ... } The method implementations are calculated at the end of IR pass, attribute offsets are {offsetof(native__C, _x1), offsetof(native__C, _y1), ...}. To account for both dynamic loading and dynamic class creation, vtables are populated dynamically at class creation time, so we emit empty array definitions to store the vtables and a function to populate them. If shadow is True, generate "shadow vtables" that point to the shadow glue methods (which should dispatch via the Python C-API). Returns the expression to use to refer to the vtable, which might be different than the name, if there are trait vtables. traitrr'r(cs(dj|jrdSdS)Nz{}_{}_trait_vtable{}rrXr,rr,rlrhr&rr.r/trait_vtable_names  z+generate_vtables..trait_vtable_namecsdj|jS)Nz{}_{}_offset_tablermrn)rhr&r.r/trait_offset_table_namesz1generate_vtables..trait_offset_table_namezstatic CPyVTableItem {}[{}];rrzstatic CPyVTableItem [z];zstatic size_t {}[{}];z static boolz(void){z _shadow();z return 1;}z + N)rlrr'r() rr,maxr_vtable_entries trait_vtablesrrarr$generate_vtablegenerate_offset_tabler*) rhr;r6r&rrprqrlvtable subtablesr offset_namer.ror/r%s@     r%rqrlc Cst|d|d|jD]}|d||jd||dq |js*|d|d|dj|d d S) z0Generate attribute offset row of a trait vtable.zsize_t _scratch[] = {z offsetof(, z),0r/memcpy({name}, {name}_scratch, sizeof({name}));rN)rrarr,rr,)rqr&rlr$rr.r.r/rys (   ryentriesrr{list[tuple[ClassIR, str, str]]c Cs|d|d|r*|d|D]\}}}|d||||q|d|D] }|r6|jr6|jn|j} |d||jjt| |j q,|sT|d|d|d j|d dS) NzCPyVTableItem r}z/* Array of trait vtables */z8(CPyVTableItem){}, (CPyVTableItem){}, (CPyVTableItem){},z/* Start of real vtable */z(CPyVTableItem){}{}{},NULLrrr) rr,r shadow_methodmethodget_group_prefixdeclrr+r,) rr6r&r{rrlr offset_tableentryrr.r.r/rxs.       rx func_namer< FuncIR | NonerF str | Nonec Cs|d||d|d|||jd|d||jd|d|d|r`|d ||d |d |d |d |d |d |dn |d |d tdt|jtD]}||}|d|dqr| dr| r| d |j}|dt |d t|jD]$} | jD]\} } || } | dkr|d|| d| d qq|dur|dt| |jddd|d|ddS)zAGenerate a native function that allocates an instance of a class.rz(PyTypeObject *type)rsz *self;zself = (z *)type->tp_alloc(type, 0);if (self == NULL) return NULL; if (type != ) {zself->vtable = r} else {rtrself->z = 0;rEzself->vectorcall = rrNz#if ({}{}((PyObject *)self) == 0) {{Py_DECREF(self); return NULL;zreturn (PyObject *)self;)rrr,rr`r_r]rr^r+r method_declr+rr[r\rarc_undefined_valuerrr,r) r$rr<r6rFr&rirHrrhrrjrIr.r.r/r/%sL           r/rr=r2c Cs2|t|||d|d|d||d|d|dddgd d |jjD}|d urb|d ||jt | |j ||d |d|d|dn+t |jjdkr|d||d|d|d|d|d|d|d|dd S)zQGenerate a native function that allocates and initializes an instance of a class.rsPyObject *self = (z);rrr~selfcSsg|]}t|jqSr.)rr).0argr.r.r/ lsz2generate_constructor_for_class..Nzchar res = {}{}{}({});zif (res == 2) {rrrtrz int res = z ->tp_init(zif (res < 0) { return self;) rr rr.sigargsr,rrrr+r,r_)r$r%r=r2r6r&rr.r.r/r0^s6              r0cCsv||jd}|d||d|d|js!|jr/|dt||jn|d|d|S)zGenerate an init function suitable for use as tp_init. tp_init needs to be a function that returns an int, and our __init__ methods return a PyObject. Translate NULL to -1, everything else to 0. _init static intz0(PyObject *self, PyObject *args, PyObject *kwds)rsz/return {}{}(self, args, kwds) != NULL ? 0 : -1; return 0;rt)rr,rr$rr,rr+)r$r=r&rr.r.r/r5s      r5cCs|d||d|d|js0|d||d|d|d|d|r<|js<|js<|rF|d |d n)|d |d |d d |dt||jd|dd |d|ddS)Nr4(PyTypeObject *type, PyObject *args, PyObject *kwds)rsrrzUPyErr_SetString(PyExc_TypeError, "interpreted classes cannot inherit from compiled");rrtreturn z(type);rrrzPyObject *ret = z(self, args, kwds);zif (ret == NULL)r) rr$rris_serializablerrr+r,)r$rr6r2r=r&r.r.r/r s(       r cCs||d||d|d|d||d|d|d|d|d |d |d dS) Nrrrsrrz\PyErr_SetString(PyExc_TypeError, "interpreted classes cannot inherit from compiled traits");rzGPyErr_SetString(PyExc_TypeError, "traits may not be directly created");rtr)rr)r$rr&r.r.r/r(s      r(cCs|d||d||jd|dt|jD]}|jD]\}}|d|||q%qt ||rB|dn|j r_||j}|d|dt |d |dt |d |d d S) z>Emit function that performs cycle GC traversal of an instance.rrz# *self, visitproc visit, void *arg)rsrz8PyObject_VisitManagedDict((PyObject *)self, visit, arg);&*((PyObject **)((char *)self + sizeof()));*((PyObject **)((char *)self + sizeof(PyObject *) + sizeof(rrtN) rrr,r[r\rar emit_gc_visitrrrr!r$rr&rhrrjrr.r.r/r!s,        r!cCs|d||d||jd|dt|jD]}|jD]\}}|d|||q%qt ||rB|dn|j r_||j}|d|dt |d |dt |d |d dS) Nrr *self)rsrz,PyObject_ClearManagedDict((PyObject *)self);rrrrrt) rrr,r[r\rar emit_gc_clearrrrr!rr.r.r/r"s(        r"dealloc_func_nameclear_func_namecCs|d||d||jd|d|d|d|d||d|d |d |d dS) Nz static voidrrrszPyObject_GC_UnTrack(self);zCPy_TRASHCAN_BEGIN(self, rz(self);z)Py_TYPE(self)->tp_free((PyObject *)self);zCPy_TRASHCAN_END(self)rt)rrr,)r$rrr&r.r.r/r#s     r#rcCs|d|d|jD]J}|jjs|jjrq|d|jd|dt||j dddg}|jj t krB| d n |jj t krM| d |d d |q|d si|dsi|dd|d|ddS)Nzstatic PyMethodDef r{"",z (PyCFunction)r METH_FASTCALL METH_KEYWORDS METH_STATIC METH_CLASSz {}, NULL}},r __setstate__ __getstate__z@{"__setstate__", (PyCFunction)CPyPickle_SetState, METH_O, NULL},zE{"__getstate__", (PyCFunction)CPyPickle_GetState, METH_NOARGS, NULL},{NULL} /* Sentinel */r)rmethodsvaluesris_prop_setteris_prop_getterrrr+r,kindrr*rr,r.r+r)r$rr&r%rGr.r.r/r)s&     r)r?rAcCsf||jd|}|d|d|d|D]\}}|d|d|dq|d|S) N_zstatic rWz = {rrrr)rr,rr)r$rr?rAr&rHrIr.r.r/r0s  rc Cs|js7|jD]0}|d|dt|||j||j|d|dt|||j||jq|j D];\}\}}|j j rGq<|d|dt|||j||j|rw|d|dt|||j||jqrrr)rrr,rUr,r is_deletablerrrrr emit_dec_ref is_unboxed emit_unboxrr#r! emit_castrr error_overlapemit_attr_bitmap_setremit_attr_bitmap_clear)r$rrjr&r deletablerrr.r.r/rsX                  rfunc_irc Cs|d|dt|||j||j|d|jrF|d||t||j| d|d|j dd|dd |d n|d t||jd |d dS)Nrrrsz#{}retval = {}{}((PyObject *) self);rrretboxTrzreturn retbox;rz((PyObject *) self);rt) rr,rRr,rrrbrr+emit_error_checkr)r$rrjrr&r.r.r/rs(   rrcCs|d|dt|||j||j|d|jr:|jdd|tddd|t| |jd n|t| |jd |d |d dS) NrrrsrIrrTrz((PyObject *) self, tmp);z((PyObject *) self, value);rrt) rr,rUr,rrrrrr+)r$rrrr&r.r.r/rs"   rcCs|jdko |jo |jdkS)z6Should the class get the Py_TPFLAGS_MANAGED_DICT flag?rPyBaseExceptionObject)rrr)r$r&r.r.r/rs rN)r$rr%rr&rr'r()rr(r'r()r$rrrLr&rr'r) r$rrrrrr&rr'r)r$rrr(r&rr'r)r$rrKr(r,r"r'r()r$rr&rr'r) rhrr;r(r6r(r&rrrkr'r() rqr(r&rrlrr$rr'r) rrr6r(r&rr{rrrkr'r)r$rrr(r<rr6r(rFrr&rr'r)r$rr%rr=rr2r(r6r(r&rr'r)r$rr=rr&rr'r()r$rrr(r6r(r2r(r=rr&rr'r)r$rrr(r&rr'r) r$rrr(rr(r&rr'r)r$rrr(r&rr'r) r$rrr(r?r(rArr&rr'r) r$rrr(rjr r&rr'r) r$rrr(rjr rrr&rr'r) r$rrr(rr rrr&rr'r)r$rr&rr'rk)X__doc__ __future__rtypingrrrmypyc.codegen.emitrrrmypyc.codegen.emitfuncr mypyc.codegen.emitwrapperr r r r rrrrrr mypyc.commonrrrrrmypyc.ir.class_irrrmypyc.ir.func_irrrrrmypyc.ir.rtypesrr r! mypyc.namegenr"mypyc.sametyper#r0r2r( SlotGeneratorrLrM__annotations__rUrXrrrrr=rrrrJrRrUrr%ryrxr/r0r5r r(r!r"r#r)rr&r'r1rrrrrr.r.r.r/s6  0             !"#%&9    0   V  # 9 *  %      &    9