o ØÝ2g¹-ã@s4UdZddlmZddlmZmZddlmZmZm Z m Z ddlmZm Z mZmZmZddlmZmZmZmZmZmZmZmZddlmZmZmZddlmZGd d „d ƒZ Gdd„dƒZ!d(dd„Z"dZ#de$d<dZ%de$d<dZ&de$d<Gdd„dƒZ'Gdd„dƒZ(e dge gƒƒZ)de$d<d)d#d$„Z*d)d%d&„Z+d'S)*z)Intermediate representation of functions.é)Úannotations)ÚFinalÚSequence)ÚARG_POSÚArgKindÚBlockÚFuncDef)ÚBITMAP_BITSÚJsonDictÚbitmap_nameÚget_id_from_nameÚshort_id_from_name)ÚAssignÚAssignMultiÚ BasicBlockÚ ControlOpÚ DeserMapsÚLoadAddressÚRegisterÚValue)ÚRTypeÚbitmap_rprimitiveÚdeserialize_type)Ú NameGeneratorc@sPeZdZdZedfdd d„Zeddd„ƒZddd„Zddd„Z e d dd„ƒZdS)!Ú RuntimeArgzrDescription of a function argument in IR. Argument kind is one of ARG_* constants defined in mypy.nodes. FÚnameÚstrÚtyprÚkindrÚpos_onlyÚboolÚreturnÚNonecCs||_||_||_||_dS©N©rÚtyperr)Úselfrrrr©r'úH/home/garg/my-data/venv/lib/python3.10/site-packages/mypyc/ir/func_ir.pyÚ__init__s zRuntimeArg.__init__cCs |j ¡Sr#)rÚis_optional©r&r'r'r(Úoptional%ó zRuntimeArg.optionalcCsd |j|j|j|j¡S)Nz:RuntimeArg(name={}, type={}, optional={!r}, pos_only={!r}))Úformatrr%r,rr+r'r'r(Ú__repr__)sÿzRuntimeArg.__repr__r cCs |j|j ¡t|jjƒ|jdœS)Nr$)rr%Ú serializeÚintrÚvaluerr+r'r'r(r0.s üzRuntimeArg.serializeÚdataÚctxrcCs(t|dt|d|ƒt|dƒ|dƒS)Nrr%rr)rrr©Úclsr3r4r'r'r(Údeserialize6s üzRuntimeArg.deserializeN) rrrrrrrr r!r")r!r ©r!r©r!r )r3r r4rr!r)Ú__name__Ú __module__Ú__qualname__Ú__doc__rr)Úpropertyr,r/r0Úclassmethodr7r'r'r'r(rsÿ rc@sPeZdZdZddd „Zddd„Zdd d„Zddd„Zddd„Ze d dd„ƒZ dS)!Ú FuncSignaturezSignature of a function in IR.ÚargsúSequence[RuntimeArg]Úret_typerr!r"cCsRt|ƒ|_||_t|jƒ|_|jr'dd„t|jƒDƒ}|jtt|ƒƒ|_dSdS)NcSsg|]}tt|ƒtdd‘qS)T)r)rrr)Ú.0Úir'r'r(Ú Lsÿÿz*FuncSignature.__init__..)ÚtuplerArCÚnum_bitmap_argsÚrangeÚreversed)r&rArCÚextrar'r'r(r)Es þûzFuncSignature.__init__útuple[RuntimeArg, ...]cCs|jr|jd|j…S|jS)z8Return arguments without any synthetic bitmap arguments.N)rHrAr+r'r'r(Ú real_argsRszFuncSignature.real_argscCs6|jrt|jd|j…|jƒSt|jdd…|jƒS)Né)rHr@rArCr+r'r'r(Ú bound_sigXszFuncSignature.bound_sigrcCsd|j›d|j›dS)NzFuncSignature(args=z, ret=ú)©rArCr+r'r'r(r/^szFuncSignature.__repr__r cCs:|jr |jd|j…}n|j}dd„|Dƒ|j ¡dœS)NcSsg|]}| ¡‘qSr')r0)rDÚtr'r'r(rFfsz+FuncSignature.serialize..rQ)rHrArCr0)r&rAr'r'r(r0aszFuncSignature.serializer3r4rcs&t‡fdd„|dDƒt|dˆƒƒS)Ncsg|]}t |ˆ¡‘qSr')rr7)rDÚarg©r4r'r(rFksz-FuncSignature.deserialize..rArC)r@rr5r'rTr(r7hsþzFuncSignature.deserializeN)rArBrCrr!r")r!rL©r!r@r8r9)r3r r4rr!r@)r:r;r<r=r)rMrOr/r0r?r7r'r'r'r(r@@s r@rArLr!r1cCs8d}|D]}|jjr|j ¡r|d7}q|tdtS)NrrN)r%Ú error_overlaprr*r )rAÚnrSr'r'r(rHps€rHrÚFUNC_NORMALrNÚFUNC_STATICMETHODéÚFUNC_CLASSMETHODc@sªeZdZdZedddfd0dd„Zed1dd„ƒZejd2dd„ƒZed3dd„ƒZ e d4dd„ƒZed3dd„ƒZed3dd„ƒZ d5d"d#„Zd6d%d&„Ze d7d(d)„ƒZed8d-d.„ƒZd/S)9ÚFuncDeclzÊDeclaration of a function in IR (without body or implementation). A function can be a regular module-level function, a method, a static method, a class method, or a property getter/setter. FrrÚ class_nameú str | NoneÚmodule_nameÚsigr@rr1Úis_prop_setterr Úis_prop_getterÚimplicitr!r"c Csd||_||_||_||_||_||_||_|durd|_n |tkr%||_n| ¡|_||_ d|_ dSr#)rr]r_r`rrarbrOrYrcÚ_line) r&rr]r_r`rrarbrcr'r'r(r)„s zFuncDecl.__init__cCs|jdusJ‚|jSr#©rdr+r'r'r(Úline¦sz FuncDecl.linerfcCs ||_dSr#re)r&rfr'r'r(rf«r-cCs |jdusJ‚t|j|j|jƒSr#)rfrrÚfullnamer+r'r'r(Úid¯szFuncDecl.idcCs|r|d|S|S©NÚ.r'©r]rr'r'r(Úcompute_shortname´szFuncDecl.compute_shortnamecCst |j|j¡Sr#)r\rlr]rr+r'r'r(Ú shortname¸ózFuncDecl.shortnamecCs|jd|jSri)r_rmr+r'r'r(rg¼rnzFuncDecl.fullnameÚnamesrcCs t|j|j|jƒ}| |j|¡Sr#)r rrmrdÚprivate_namer_)r&roÚpartial_namer'r'r(ÚcnameÀszFuncDecl.cnamer c Cs*|j|j|j|j ¡|j|j|j|jdœS)N)rr]r_r`rrarbrc) rr]r_r`r0rrarbrcr+r'r'r(r0ÄsøzFuncDecl.serializeÚfunc_ircCs@|d}t |d|d¡}|dd|}t|d||dƒS)zCGet the id from the serialized FuncIR associated with this FuncDeclÚdeclr]rr_rjrf)r\rlr)rsrtrmrgr'r'r(Úget_id_from_jsonÑszFuncDecl.get_id_from_jsonr3r4rc Cs>t|d|d|dt |d|¡|d|d|d|dƒS) Nrr]r_r`rrarbrc)r\r@r7r5r'r'r(r7ÙsøzFuncDecl.deserializeN)rrr]r^r_rr`r@rr1rar rbr rcr r!r"©r!r1)rfr1r!r"r8)r]r^rrr!r©rorr!rr9)rsr r!r)r3r r4rr!r\)r:r;r<r=rXr)r>rfÚsetterrhÚstaticmethodrlrmrgrrr0rur?r7r'r'r'r(r\}s2÷" r\c@s¼eZdZdZ d4d5dd„Zed6dd„ƒZed7dd„ƒZed8dd„ƒZed9dd„ƒZ ed:dd„ƒZ ed;d d!„ƒZed;d"d#„ƒZed;d$d%„ƒZ dd2d3„ƒZdS)?ÚFuncIRz’Intermediate representation of a function with contextual information. Unlike FuncDecl, this includes the IR of the body (basic blocks). éÿÿÿÿNrtr\Úarg_regsúlist[Register]Úblocksúlist[BasicBlock]rfr1Útraceback_namer^r!r"cCs$||_||_||_||j_||_dSr#)rtr|r~rfr€)r&rtr|r~rfr€r'r'r(r)ís zFuncIR.__init__cCó|jjSr#)rtrfr+r'r'r(rfózFuncIR.linerBcCó |jjjSr#)rtr`rAr+r'r'r(rAr-zFuncIR.argsrcCrƒr#)rtr`rCr+r'r'r(rC r-zFuncIR.ret_typecCrr#)rtr]r+r'r'r(r] r‚zFuncIR.class_namer@cCrr#)rtr`r+r'r'r(r`r‚z FuncIR.sigrcCrr#)rtrr+r'r'r(rr‚zFuncIR.namecCrr#)rtrgr+r'r'r(rgr‚zFuncIR.fullnamecCrr#)rtrhr+r'r'r(rhr‚z FuncIR.idrorcCs|j |¡Sr#)rtrr)r&ror'r'r(rr!szFuncIR.cnamecCs*|jrd|j›d|j›dSd|j›dS)Nzrkr+r'r'r(r/$szFuncIR.__repr__r cCs|j ¡|j|jdœS)N)rtrfr€)rtr0rfr€r+r'r'r(r0*sýzFuncIR.serializer3r4rcCs$tt |d|¡gg|d|dƒS)Nrtrfr€)rzr\r7r5r'r'r(r72sÿzFuncIR.deserialize)r{N)rtr\r|r}r~rrfr1r€r^r!r"rv)r!rB)r!r)r!r^rUr8rwr9)r3r r4rr!rz)r:r;r<r=r)r>rfrArCr]r`rrgrhrrr/r0r?r7r'r'r'r(rzçs4 ú rzzÚINVALID_FUNC_DEFr}r~rúlist[Value]cCs°t|ƒ}t|ƒ}|D]K}|jD]E}t|tƒsTt|ttfƒr/|j|vr.| |j¡| |j¡q|j r3qt|tƒrOt|jt ƒrO|j|vrO| |j¡| |j¡| |¡qq |S)zvReturn the set of all values that may be initialized in the blocks. This omits registers that are only read. )ÚlistÚsetÚopsÚ isinstancerrrÚdestÚappendÚaddÚis_voidrÚsrcr)rAr~ÚvaluesÚseen_registersÚblockÚopr'r'r(Ú all_values<s. €ÿ þ €ïr”cCs¬t|ƒ}t|ƒ}|D]I}|jD]C}| ¡D]}t|tƒr*||vr*| |¡| |¡qt|tƒsRt|t t fƒrI|j|vrH| |j¡| |j¡q|jrMq| |¡qq |S)z:Return set of all values that are initialized or accessed.) r‡rˆr‰ÚsourcesrŠrrŒrrrrr‹rŽ)rAr~rr‘r’r“Úsourcer'r'r(Úall_values_full[s* € € €ñr—N)rArLr!r1)rAr}r~rr!r†),r=Ú __future__rÚtypingrrÚ mypy.nodesrrrrÚmypyc.commonr r rrr Úmypyc.ir.opsrrrrrrrrÚmypyc.ir.rtypesrrrÚ mypyc.namegenrrr@rHrXÚ__annotations__rYr[r\rzr…r”r—r'r'r'r(Ús&( ) 0jR