o tBhc@sUdZddlmZddlZddlmZddlmZddlm Z ddl m Z m Z m Z mZmZmZmZddlmZmZdd lmZdd lmZmZdd lmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4dd l5m6Z6m7Z7dd l8m9Z9dZ:de;d<GdddZGddde>Z?Gddde>Z@Gddde>ZAGddde>ZBGdd d ZCdS)!zUtilities for emitting C code.) annotationsN)Callable)Final)Literals) ATTR_PREFIXFAST_ISINSTANCE_MAX_SUBCLASSES NATIVE_PREFIX REG_PREFIX STATIC_PREFIX TYPE_PREFIXuse_vectorcall)ClassIRall_concrete_classes)FuncDecl) BasicBlockValue) RInstance RPrimitiveRTupleRTypeRUnionint_rprimitiveis_bit_rprimitiveis_bool_rprimitiveis_bytes_rprimitiveis_dict_rprimitiveis_fixed_width_rtypeis_float_rprimitiveis_int32_rprimitiveis_int64_rprimitiveis_int_rprimitiveis_list_rprimitiveis_none_rprimitiveis_object_rprimitiveis_optional_typeis_range_rprimitiveis_set_rprimitiveis_short_int_rprimitiveis_str_rprimitiveis_tuple_rprimitiveobject_rprimitiveoptional_value_type) NameGenerator exported_name) is_same_typeFr DEBUG_ERRORSc@s(eZdZdZ ddddddddZdS)HeaderDeclarationanA representation of a declaration in C. This is used to generate declarations in header files and (optionally) definitions in source files. Attributes: decl: C source code for the declaration. defn: Optionally, C source code for a definition. dependencies: The names of any objects that must be declared prior. is_type: Whether the declaration is of a C type. (C types will be declared in external header files and not marked 'extern'.) needs_export: Whether the declared object needs to be exported to other modules in the linking table. NF) dependenciesis_type needs_exportdeclstr | list[str]defnlist[str] | Noner1set[str] | Noner2boolr3returnNonecCs8t|tr|gn||_||_|pt|_||_||_dSN) isinstancestrr4r6setr1r2r3)selfr4r6r1r2r3rAi/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/mypyc/codegen/emit.py__init__Js    zHeaderDeclaration.__init__r<) r4r5r6r7r1r8r2r9r3r9r:r;__name__ __module__ __qualname____doc__rCrArArArBr0:sr0c@s eZdZdZ  d dd d ZdS)EmitterContextz-Shared emitter state for a compilation group.Nnamesr, group_name str | None group_mapdict[str, str | None] | Noner:r;cCs6d|_||_||_|p i|_t|_i|_t|_dS)zSetup shared emitter state. Args: names: The name generator to use group_map: Map from module names to group name group_name: Current group name rN) temp_counterrJrKrMr? group_deps declarationsrliterals)r@rJrKrMrArArBrC]s   zEmitterContext.__init__NN)rJr,rKrLrMrNr:r;rDrArArArBrIZs rIc@eZdZdZdS) ErrorHandlerz3Describes handling errors in unbox/cast operations.NrErFrGrHrArArArBrU}rUc@rT) AssignHandlerzAssign an error value on error.NrVrArArArBrXrWrXc@eZdZdZd ddZdS) GotoHandlerzGoto label on error.labelr>r:r;cC ||_dSr<r[r@r[rArArBrC zGotoHandler.__init__N)r[r>r:r;rDrArArArBrZrZc@seZdZdZd d d Zd S)TracebackAndGotoHandlerz+Add traceback item and goto label on error.r[r> source_path module_nametraceback_entrytuple[str, int]r:r;cCs||_||_||_||_dSr<)r[rbrcrd)r@r[rbrcrdrArArBrCs z TracebackAndGotoHandler.__init__N) r[r>rbr>rcr>rdrer:r;rDrArArArBrar`rac@rY) ReturnHandlerz!Return a constant value on error.valuer>r:r;cCr\r<)rg)r@rgrArArBrCr_zReturnHandler.__init__N)rgr>r:r;rDrArArArBrfr`rfc@seZdZdZ  ddd d Zdd dZdddZdddZdddZdddZ dddd Z dd"d#Z dd%d&Z dd(d)Z dd,d-Zdd.d/Zdd0d1Zdd3d4Zdd7d8Zefdd=d>ZddAdBZddEdFZddGdHZddIdJZddKdLZddOdPZddTdUZddWdXZ Yddd]d^ZddbdcZddddeZddfdgZ ddidjZ!dYdkddndoZ"dYdYdpddrdsZ#ddudvZ$dYddwdYddwdxdddZ%dddZ&dddZ'dddZ(dddZ)dYddwdYdYddddZ* YddddZ+dddZ,dddZ-dddZ.dddZ/dddZ0  ddddZ1dS)EmitterzHelper for C code generation.NcontextrI value_namesdict[Value, str] | None capi_versiontuple[int, int] | Noner:r;cCs<||_|p tjdd|_|j|_|pi|_g|_d|_dS)Nr)risys version_inforlrJrj fragments_indent)r@rirjrlrArArBrCs   zEmitter.__init__cCs|jd7_dS)Nrrr@rArArBindentzEmitter.indentcCs |jd8_|jdksJdS)NrsrrtrurArArBdedentszEmitter.dedentr[rr>cCs d|jS)NzCPyL%sr]r^rArArBr[r_z Emitter.labelregrcCst|j|Sr<)r rj)r@ryrArArBrysz Emitter.regnamecCst|Sr<)r)r@rzrArArBattrsz Emitter.attrlinecCsF|dr ||j|jd|d|dr!|dSdS)N}  {) startswithrxrqappendrrendswithrv)r@r}rArArB emit_lines   zEmitter.emit_linelinescGs|D]}||qdSr<)r)r@rr}rArArB emit_liness zEmitter.emit_linesBasicBlock | strcCs0t|tr|}n||}|j|ddS)Nz: ; )r=r>r[rqr)r@r[textrArArB emit_labels  zEmitter.emit_labelemittercCs|j|jdSr<)rqextend)r@rrArArBemit_from_emitterrwzEmitter.emit_from_emitterfmtargscGs<|dd}|ddd|gt||ddS)Nrz\nz printf(%s);, z"%s"zfflush(stdout);)replacerjoinlist)r@rrrArArB emit_printfs "zEmitter.emit_printfcC|jjd7_d|jjS)Nz__tmp%drirOrurArArB temp_name zEmitter.temp_namecCr)Nrz__LL%drrurArArB new_labelrzEmitter.new_labelrccCsD|jj}||}|r ||jjkr |jj|dt|dSdS)aGet the group prefix for a module (relative to the current group). The prefix should be prepended to the object name whenever accessing an object from this module. If the module lives is in the current compilation group, there is no prefix. But if it lives in a different group (and hence a separate extension module), we need to access objects from it indirectly via an export table. For example, for code in group `a` to call a function `bar` in group `b`, it would need to do `exports_b.CPyDef_bar(...)`, while code that is also in group `b` can simply do `CPyDef_bar(...)`. Thus the prefix for a module in group `b` is 'exports_b.' if the current group is *not* b and just '' if it is. exports_.r|)rirMgetrKrPaddr-)r@rcgroupstarget_group_namerArArBget_module_group_prefixs  zEmitter.get_module_group_prefixobjClassIR | FuncDeclcCs ||jS)z#Get the group prefix for an object.)rrc)r@rrArArBget_group_prefixs zEmitter.get_group_prefixidmodulerLprefixcCsD|sdn||}|r dnd}|j|pd|}||||S)aCreate name of a C static variable. These are used for literals and imported modules, among other things. The caller should ensure that the (id, module) pair cannot overlap with other calls to this method within a compilation group. r|*)rrJ private_name)r@rrr lib_prefix star_maybesuffixrArArB static_names zEmitter.static_nameclr cCs|j|j|jtdS)N)r)rrzrcr )r@rrArArBtype_struct_namezEmitter.type_struct_namertypercCs|jSr<)_ctyper@rrArArBctypesz Emitter.ctypecCs"||}|ddkr |S|dS)z*Adds a space after ctype for non-pointers.rr)r)r@rrrArArB ctype_spaceds  zEmitter.ctype_spacedcCs6|jsdSt|tr |jSt|tr||SJ|)NNULL) is_unboxedr=r c_undefinedrtuple_undefined_valuerrArArBc_undefined_value"s   zEmitter.c_undefined_valuecCs ||Sr<)rrrArArB c_error_value+r_zEmitter.c_error_valuefnrcCst||jSr<)rcnamerJ)r@rrArArBnative_function_name.rzEmitter.native_function_namertupler list[str]c Csd|jd|jd|jdg}t|jdkr|dnd}|jD]}|||d|d|d 7}q$|d |jd||}|d ||||d ||d |d |S)Nz#ifndef MYPYC_DECLARED_z#define MYPYC_DECLARED_ztypedef struct  {rzint empty_struct_error_flag;f;rz} zstatic {} {} = {{ {} }};r|z#endif) struct_namelentypesrrtuple_undefined_value_helperformatrrr)r@rresultitypvaluesrArArBtuple_c_declaration1s(         zEmitter.tuple_c_declarationr9cCs t|jSr<)r rlrurArArBr Lr_zEmitter.use_vectorcallF attr_exprcompareunlikelycCsht|trd||||j|}nd|d|d||d}|r)d|d}|d|ddS)Nz({})(r)z (unlikelyzif r)r=rrtuple_undefined_check_condrr)r@rrrrcheckrArArBemit_undefined_attr_checkOs  z!Emitter.emit_undefined_attr_checktuple_expr_in_cc_type_compare_valCallable[[RType], str]cCs`t|jdkrd|||tS|jd}t|tr$|||d||S|d|d||S)Nrz {}.empty_struct_error_flag {} {}z.f0z.f0 r)rrrrr=rr)r@rrrr item_typerArArBr\s    z"Emitter.tuple_undefined_check_condcCs d|jS)Ntuple_undefined_) unique_id)r@rrArArBrpr_zEmitter.tuple_undefined_valuecCsg}t|jdkr|tgS|jD])}t|ts"|||n||}|d|||d|dq|ddS)Nrz{ z }rr) rrrrr=rrrr)r@rresitemsub_listrArArBrss         z$Emitter.tuple_undefined_value_helper tuple_typecCs\|j|jjvr,t}|jD] }t|tr||jq t| ||dd|jj|j<dSdS)NT)r1r2) rrirQr?rr=rrr0r)r@rr1rrArArBdeclare_tuple_structs    zEmitter.declare_tuple_struct)raredestrcCst|r|r|d|dS|d|dSt|tr4t|jD]\}}||d||q"dS|js@|d|dSdS)zIncrement reference count of C expression `dest`. For composite unboxed structures (e.g. tuples) recursively increment reference counts for each component. If rare is True, optimize for code size and compilation speed. zCPyTagged_IncRef(%s);zCPyTagged_INCREF(%s);.fzCPy_INCREF(%s);N)r rr=r enumerater emit_inc_refr)r@rrrrrrArArBrs zEmitter.emit_inc_refis_xdecrrcCs|rdnd}t|r(|r|d|d|dd S|d|d|dd St|trGt|jD]\}}|j|d||||dq2d S|jsh|rZ|d |d|dd S|d |d|dd Sd S) zDecrement reference count of C expression `dest`. For composite unboxed structures (e.g. tuples) recursively decrement reference counts for each component. If rare is True, optimize for code size and compilation speed. Xr| CPyTagged_zDecRef();zDECREF(rrCPy_N)r rr=rrr emit_dec_refr)r@rrrrxrrrArArBrs  zEmitter.emit_dec_refrcCs&t|}|durd||St|S)Nz %s or None)r+ pretty_namer>)r@r value_typerArArBrszEmitter.pretty_nameT) declare_desterrorraise_exceptionoptionalsrc_typelikelysrcrrErrorHandler | Nonerrr RType | Nonerc s|pt}|r\t|r\t|s\t|} | dusJt| |r\|r)d|dd} | r3d| d} |||| ||d|d|dd  |||||d dSt |st |st |st |st|st|st|st|st|st|r |rd|dt |rd } n@t |rd } n9t |rd } n2t |rd} n+t|rd} n$t|rd} nt|st|rd} nt|st|rd} nJd|d} | rd| d} |||| | ||d|d|dd  |||||d dSt|rQ|rd|dd} | r'd| d} |||| |||d|d|dd  |||||d dSt|r|rg|d|dd} | rrd| d} |||| ||d|d|dd  |||||d dSt|tr(|rd|dt|j} | rt| tdkrd||j} n,d}tdt| D] }|d|7}qt| dkrd|}|j|fdd| Dd } | rd| d} |||| |d|d|d||d  |||||d dSt|rm|r9d|dd!} | rDd| d} |||| ||d|d|dd  |||||d dSt|r|r~d|d|||d"||d|d|rd dSdSt|t r!||||||||dSt|t"r|rJ#||||||dSJd#|)$auEmit code for casting a value of given type. Somewhat strangely, this supports unboxed types but only operates on boxed versions. This is necessary to properly handle types such as Optional[int] in compatibility glue. By default, assign NULL (error value) to dest if the value has an incompatible type and raise TypeError. These can be customized using 'error' and 'raise_exception'. Always copy/steal the reference in 'src'. Args: src: Name of source C variable dest: Name of target C variable typ: Type of value declare_dest: If True, also declare the variable 'dest' error: What happens on error raise_exception: If True, also raise TypeError on failure likely: If the cast is likely to succeed (can be False for unions) N PyObject *rz({} != Py_None)z(likelyr  = else {r~PyListPyDictPySet PyUnicodePyRangeCPyFloatPyLongPyBoolFzunexpected primitive type: z({}_Check({}))z,(PyBytes_Check({}) || PyByteArray_Check({}))rz(PyTuple_Check({}))rz(PyObject_TypeCheck({}, {}))z (Py_TYPE({src}) == {targets[0]})z% || (Py_TYPE({src}) == {targets[%d]})z(%s)csg|]}|qSrA)r).0irrurArB Lsz%Emitter.emit_cast..)rtargetsz({} == Py_None)r|zCast not implemented: %s)$rXr$r#r+r.remit_arg_checkrremit_cast_error_handlerr!rr&r(r%rr rrrrr)rr=rrclass_irrrrranger"remit_union_castremit_tuple_cast)r@rrrrrrrrrrrrconcretefull_strrrArurB emit_casts "                "       zEmitter.emit_castrUcCs|r-t|tr|j|j|j|j||d|d|jdS|d||d|dt|t r;|d|dSt|t rJ|d|jdSt|trj|d|| |j|j|j|d|jdSt|t sqJ|d|j dS)N)rrgoto %s;CPy_TypeError("", ); z %s = NULL; return %s;)r=raemit_type_error_tracebackrbrcrdrr[rrXrZemit_tracebackrfrg)r@rrrrrrArArBr os$    zEmitter.emit_cast_error_handlerrc Cs|r |d|d|} |r6|d|d||d||d|d| d|d|jD]} |j||| ddddd |d|d | dq9|||||||| d S) zTEmit cast to a union type. The arguments are similar to emit_cast. rrif ( == NULL) {rgoto r~F)rrrrz != NULL) goto N)rrritemsrr r) r@rrrrrrrr good_labelrrArArBrs*   zEmitter.emit_union_castc Cs|r |d|d|}|dj|t|jd|dd|ddt|jD]"\}} |jd|d |d || d d d d |d |d|dq+||d|d||dS)zTEmit cast to a tuple type. The arguments are similar to emit_cast. rrzJif (unlikely(!(PyTuple_Check({r}) && PyTuple_GET_SIZE({r}) == {size}))) {{)rsizez = NULL;rr~zPyTuple_GET_ITEM(rrF)rrrrz == NULL) goto rN) rrrrrrrrr) r@rrrrrr out_labelrrrArArBrs.   zEmitter.emit_tuple_castrcCsn|r|d|d||d||d|dkr,|d|r%dnd|dS|r5|ddSdS) Nrrrrr|z{}if {}z} else r)rrr)r@rrrrrrArArBr szEmitter.emit_arg_check)rrrrborrowr"c Cs|pt}t|tr|d||d} nt|tr!d|j} n t|ts(Jd|j} |r?d||d|d} | | } t|sGt |r|rR| d|d| |||d |d ||rn| d |d |d n | d |d|d | d| | | dd"St |st |r|r| d|d| |||d|d|| | | d|d} | d |d| dd"St|r|r| d|d| |||d|d|| | | d| d |dd"St|r|r | d|d| |d|d d"St|r5|r(| d|d| |d|d d"St|tr|||rQ| ||d|d|rp| d|d| |d||d| d |} |j|| |d!|d"d#| d$| d%| | | d |js| |d&t|jD]M\} }|}| d'|d(|d)| d |}|jr|j|||||d!|d*n|s||t|j|||d!d+| |d,| d|dq| d|r| dd"Sd"SJd.|)/aEmit code for unboxing a value of given type (from PyObject *). By default, assign error value to dest if the value has an incompatible type and raise TypeError. These can be customized using 'error' and 'raise_exception'. Generate a new reference unless 'borrow' is True. Args: src: Name of source C variable dest: Name of target C variable typ: Type of value declare_dest: If True, also declare the variable 'dest' error: What happens on error raise_exception: If True, also raise TypeError on failure borrow: If True, create a borrowed reference rrrrrrrz CPyTagged z(likely(PyLong_Check(z)))rz = CPyTagged_BorrowFromObject(rz = CPyTagged_FromObject(rr~zchar z(unlikely(!PyBool_Check(z))) {z} elsez == Py_Truez (unlikely(z != Py_None)) {z = 1;zint64_t z = CPyLong_AsInt64(zint32_t z = CPyLong_AsInt32(rrrz} else {TN)rrr if (unlikely(z == NULL)) {z.empty_struct_error_flag = 0;rz = PyTuple_GET_ITEM(r)rrrr"rrFzUnboxing not implemented: %s)rXr=rrZr[rfrgrr r'rr rrr"rrrrrrrrrr emit_unboxrr*r)r@rrrrrrrr"failure raise_exc conversion cast_temprrtemptemp2rArArBr%s                        zEmitter.emit_unbox can_borrowc Cs|rd}nd}t|st|r|||d|ddSt|s&t|r?|||d|d|s=||tdSdSt|rY|||d|sW||tdSdSt |rl|||d|ddSt |r|||d |ddSt |t r| ||||d t|jd|d |d |d tdt|jD];}|js|d|d|d|d|q|}|j|d|||j|dd|d|d|d|dqdS|jrJ|||d|ddS)zEmit code for boxing a value of given type. Generate a simple assignment if no boxing is needed. The source reference count is stolen for the result (no need to decref afterwards). rr|z = CPyTagged_StealAsObject(rrz ? Py_True : Py_False;z = Py_None;z = PyLong_FromLong(z = PyLong_FromLongLong(z = PyTuple_New(r#z == NULL))z CPyError_OutOfMemory();rzPyTuple_SET_ITEM(rrTr$rN)r r'rrrrrr*r"rrr=rrrrr rremit_box) r@rrrrr, declarationr inner_namerArArBr-isB     $"  zEmitter.emit_boxrgr&cCspt|ts|d|d||dnt|jdkrdS||||jd}|d|d||ddS)zMEmit code for checking a native function return value for uncaught exception.rz == z) {rNz==r~)r=rrrrrrr)r@rgrr&condrArArBemit_error_checks  zEmitter.emit_error_checktargetcCs|jsdSt|tr(|jdkr(|d|d|d|d|ddSt|trDt|jD]\}}||d||q2dS| |d krV|d |d dSJd t |)zEmit code for GC visiting a C variable reference. Assume that 'target' represents a C expression that refers to a struct member, such as 'self->x'. N builtins.intif (CPyTagged_CheckLong()) {z Py_VISIT(CPyTagged_LongAsObject(z));r~rrz Py_VISIT(rFz&emit_gc_visit() not implemented for %s) is_refcountedr=rrzrrrr emit_gc_visitrreprr@r2rrrrArArBr7s zEmitter.emit_gc_visitcCs|jsdSt|tr;|jdkr;|d|d|d|d||d||d|d|d dSt|trWt|jD]\}}| |d ||qEdS| |d krp||d krp|d |ddSJdt |)zEmit code for clearing a C attribute reference for GC. Assume that 'target' represents a C expression that refers to a struct member, such as 'self->x'. Nr3r4r5zCPyTagged __tmp = rrz*Py_XDECREF(CPyTagged_LongAsObject(__tmp));r~rrrz Py_CLEAR(rFz&emit_gc_clear() not implemented for %s) r6r=rrzrrrrr emit_gc_clearrr8r9rArArBr:s  zEmitter.emit_gc_clearrbrdrecCs|d|||S)NCPy_AddTraceback)_emit_traceback)r@rbrcrdrArArBrszEmitter.emit_tracebackcCs,d}d||d}|j||||||dS)NCPy_TypeErrorTraceback")type_strr)rr<)r@rbrcrdrrfuncr?rArArBrs   z!Emitter.emit_type_error_tracebackr@r?c Csv|d|}d||dd|d|d|f}|r'|sJ|d|d|7}|d7}||tr9|d dSdS) Nglobalsz%s("%s", "%s", %d, %s\z\\rrrrz7assert(PyErr_Occurred() != NULL && "failure w/o err!");)rrrr/) r@r@rbrcrdr?rglobals_staticr}rArArBr<s   zEmitter._emit_tracebackrS)rirIrjrkrlrmr:r;)r:r;)r[rr:r>)ryrr:r>)rzr>r:r>)r|)r}r>r:r;)rr>r:r;)r[rr:r;)rrhr:r;)rr>rr>r:r;)r:r>)rcr>r:r>)rrr:r>)rr>rrLrr>r:r>)rr r:r>)rrr:r>)rrr:r>)rrr:r)r:r9)F) rrrr>rr>rr9r:r;) rrrr>rrrr>r:r>)rrr:r>)rrr:r;)rr>rrrr9r:r;) rr>rrrr9rr9r:r;)rrr:r>)rr>rr>rrrr9rrrr9rr9rrrr9r:r;) rrUrr>rr>rrrr9r:r;)rr>rr>rrrr9rrUrr9rrrr9r:r;)rr>rr>rrrr9rrUrrr:r;) rr>rr>rrrr>rr9r:r;)rr>rr>rrrr9rrrr9rr9r"r9r:r;)FF) rr>rr>rrrr9r,r9r:r;)rgr>rrr&r>r:r;)r2r>rrr:r;)rbr>rcr>rdrer:r;) rbr>rcr>rdrerrrr>r:r;)r|r|)r@r>rbr>rcr>rdrer?r>rr>r:r;)2rErFrGrHrCrvrxr[ryr{rrrrrrrrrr rrrrrrrrr rrrrrrrrrr rrr r%r-r1r7r:rrr<rArArArBrhs                            (  & * 5   rh)DrH __future__rrotypingrtyping_extensionsrmypyc.codegen.literalsr mypyc.commonrrrr r r r mypyc.ir.class_irr rmypyc.ir.func_irr mypyc.ir.opsrrmypyc.ir.rtypesrrrrrrrrrrrrrrr r!r"r#r$r%r&r'r(r)r*r+ mypyc.namegenr,r-mypyc.sametyper.r/__annotations__r0rIrUrXrZrarfrhrArArArBs,    $ p   #