X-PhKdZddlmZddlmZddlmZmZddlm Z m Z ddl m Z m Z mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm 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/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9GddZ:d d d?dZ;d@dZe?e@eeAfeBe>fZCGdde e>ZDe@eBe>eBe>fZEGdde,eEe>ZFGddeFe9ZGdBd!ZH dCdDd%ZIGd&d'eFe9ZJdEd)ZKGd*d+eFe9ZLdFd.ZMGd/d0eFe9ZNdGd1ZOdZPd2ZQ dHdId>ZRd3S)JzData-flow analyses.) annotations)abstractmethod)IterableIterator)GenericTypeVar).Assign AssignMulti BasicBlockBoxBranchCallCallCCast ComparisonOp ControlOpDecRefExtendFloatFloatComparisonOpFloatNegFloatOpGetAttr GetElementPtrGotoIncRef InitStaticIntegerIntOp KeepAlive LoadAddressLoadErrorValue LoadGlobal LoadLiteralLoadMem LoadStatic MethodCallOp OpVisitor PrimitiveOpRaiseStandardError RegisterOpReturnSetAttrSetMemTruncateTupleGetTupleSetUnborrowUnbox UnreachableValuec"eZdZdZd d Zdd Zd S)CFGzyControl-flow graph. Node 0 is always assumed to be the entry point. There must be a non-empty set of exits. succ"dict[BasicBlock, list[BasicBlock]]predexitsset[BasicBlock]returnNonec8|sJ||_||_||_dSN)r9r;r<)selfr9r;r<s W/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/mypyc/analysis/dataflow.py__init__z CFG.__init__Bs&     strcZt|jd}d|d|jd|jS)Nc*t|jSrA)intlabel)es rCzCFG.__str__..NsQWrE)keyzexits: z succ: z pred: )sortedr<r9r;)rBr<s rC__str__z CFG.__str__Ms<tz'='=>>>FFF FF49FFFrEN)r9r:r;r:r<r=r>r?r>rF)__name__ __module__ __qualname____doc__rDrOrErCr8r8;sL    GGGGGGrEr8F) use_yieldsblockslist[BasicBlock]rVboolr>cti}i}t}|D]}td|jddDr Jd|r4t|jt r|jjr|jjg}n&t|j}|s| ||g|zD]#}|j r| |j $|||<g||<| D]%\}} | D]} ||  |&t|||S)zCalculate basic block control-flow graph. If use_yields is set, then we treat returns inserted by yields as gotos instead of exits. c3@K|]}t|tVdSrA) isinstancer).0ops rC zget_cfg..\s=  *,Jr9 % %      rENz-Control-flow ops must be at the end of blocks)setanyopsr\ terminatorr- yield_targetlisttargetsadd error_handlerappenditemsr8) rWrVsucc_mappred_mapr<blockr9 error_pointprevnxtrJs rCget_cfgrrRs H35H EEE  05 #2#      ; ; : ; ; ;  4*U%5v>> 45CSC` 4$12DD(002233D  IIe   "7T> 7 7K( 7 K5666^^%%)) c ) )E UO " "4 ( ( ( ( ) x5 ) ))rErJr ct|jdkr2t|jdtr|jdj}|S)Nr`)lenrcr\rrJ)rJs rCget_real_targetrvs= 59~~z%)B->> " # LrEr?cd}|r|D]V}t|jD]-\}}|j|t |.Wd}t |}|}|t|D]0\}}|dks |j|r| |.d}1|dSdS)zCleanup the control flow graph. This eliminates obviously dead basic blocks and eliminates blocks that contain nothing but a single jump. There is a lot more that could be done. TFrN) enumeraterdrg set_targetrvrrcopyclearr;rj)rWchangedrnitgtcfg orig_blockss rC cleanup_cfgrs G  E EE#E$4$<$<$>$>?? E E3 ++As/C/CDDDD Efookkmm  !+..  HAuAvv%v e$$$$ rETceZdZd dZd dZd S) AnalysisResultbeforeAnalysisDict[T]afterr>r?c"||_||_dSrArr)rBrrs rCrDzAnalysisResult.__init__s  rErFc(d|jd|jdS)Nzbefore: z after:  r)rBs rCrOzAnalysisResult.__str__s>$+>> >>>>rEN)rrrrr>r?rP)rQrRrSrDrOrUrErCrrs<??????rErc^eZdZdIdZedJdZedKd ZedLd ZedMd ZdNdZ dOdZ dPdZ dQdZ dRdZ dSdZdTdZdUdZdVdZdWd!ZdXd#ZdYd%ZdZd'Zd[d)Zd\d+Zd]d-Zd^d/Zd_d1Zd`d3Zdad5Zdbd7Zdcd9Zddd;Zded=Z dfd?Z!dgdAZ"dhdCZ#didEZ$djdGZ%dHS)kBaseAnalysisVisitorr^rr> GenAndKill[T]c:ttfSrArarBr^s rC visit_gotozBaseAnalysisVisitor.visit_gotouucee|rEr,ctrANotImplementedErrorrs rCvisit_register_opz%BaseAnalysisVisitor.visit_register_op!!rEr ctrArrs rC visit_assignz BaseAnalysisVisitor.visit_assignrrEr ctrArrs rCvisit_assign_multiz&BaseAnalysisVisitor.visit_assign_multirrEr/ctrArrs rC visit_set_memz!BaseAnalysisVisitor.visit_set_memrrErc,||SrArrs rC visit_callzBaseAnalysisVisitor.visit_call%%b)))rEr'c,||SrArrs rCvisit_method_callz%BaseAnalysisVisitor.visit_method_callrrEr"c,||SrArrs rCvisit_load_error_valuez*BaseAnalysisVisitor.visit_load_error_valuerrEr$c,||SrArrs rCvisit_load_literalz&BaseAnalysisVisitor.visit_load_literalrrErc,||SrArrs rCvisit_get_attrz"BaseAnalysisVisitor.visit_get_attrrrEr.c,||SrArrs rCvisit_set_attrz"BaseAnalysisVisitor.visit_set_attrrrEr&c,||SrArrs rCvisit_load_staticz%BaseAnalysisVisitor.visit_load_staticrrErc,||SrArrs rCvisit_init_staticz%BaseAnalysisVisitor.visit_init_staticrrEr1c,||SrArrs rCvisit_tuple_getz#BaseAnalysisVisitor.visit_tuple_getrrEr2c,||SrArrs rCvisit_tuple_setz#BaseAnalysisVisitor.visit_tuple_setrrEr c,||SrArrs rC visit_boxzBaseAnalysisVisitor.visit_boxrrEr4c,||SrArrs rC visit_unboxzBaseAnalysisVisitor.visit_unboxrrErc,||SrArrs rC visit_castzBaseAnalysisVisitor.visit_castrrEr+c,||SrArrs rCvisit_raise_standard_errorz.BaseAnalysisVisitor.visit_raise_standard_errorrrErc,||SrArrs rC visit_call_cz BaseAnalysisVisitor.visit_call_crrEr*c,||SrArrs rCvisit_primitive_opz&BaseAnalysisVisitor.visit_primitive_oprrEr0c,||SrArrs rCvisit_truncatez"BaseAnalysisVisitor.visit_truncaterrErc,||SrArrs rC visit_extendz BaseAnalysisVisitor.visit_extendrrEr#c,||SrArrs rCvisit_load_globalz%BaseAnalysisVisitor.visit_load_globalrrErc,||SrArrs rC visit_int_opz BaseAnalysisVisitor.visit_int_oprrErc,||SrArrs rCvisit_float_opz"BaseAnalysisVisitor.visit_float_oprrErc,||SrArrs rCvisit_float_negz#BaseAnalysisVisitor.visit_float_negrrErc,||SrArrs rCvisit_comparison_opz'BaseAnalysisVisitor.visit_comparison_oprrErc,||SrArrs rCvisit_float_comparison_opz-BaseAnalysisVisitor.visit_float_comparison_op rrEr%c,||SrArrs rCvisit_load_memz"BaseAnalysisVisitor.visit_load_mem rrErc,||SrArrs rCvisit_get_element_ptrz)BaseAnalysisVisitor.visit_get_element_ptrrrEr!c,||SrArrs rCvisit_load_addressz&BaseAnalysisVisitor.visit_load_addressrrEr c,||SrArrs rCvisit_keep_alivez$BaseAnalysisVisitor.visit_keep_aliverrEr3c,||SrArrs rCvisit_unborrowz"BaseAnalysisVisitor.visit_unborrowrrEN)r^rr>r)r^r,r>r)r^r r>r)r^r r>r)r^r/r>r)r^rr>r)r^r'r>r)r^r"r>r)r^r$r>r)r^rr>r)r^r.r>r)r^r&r>r)r^rr>r)r^r1r>r)r^r2r>r)r^r r>r)r^r4r>r)r^rr>r)r^r+r>r)r^rr>r)r^r*r>r)r^r0r>r)r^rr>r)r^r#r>r)r^rr>r)r^rr>r)r^rr>r)r^rr>r)r^rr>r)r^r%r>r)r^rr>r)r^r!r>r)r^r r>r)r^r3r>r)&rQrRrSrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrUrErCrrs"""^""""^""""^""""^"**********************************************************************************************************************rErcTeZdZdZdddZdd Zdd ZddZddZddZ d dZ d!dZ dS)"DefinedVisitoraVisitor for finding defined registers. Note that this only deals with registers and not temporaries, on the assumption that we never access temporaries when they might be undefined. If strict_errors is True, then we regard any use of LoadErrorValue as making a register undefined. Otherwise we only do if `undefines` is set on the error value. This lets us only consider the things we care about during uninitialized variable checking while capturing all possibly undefined things for refcounting. F strict_errorsrYr>r?c||_dSrAr)rBrs rCrDzDefinedVisitor.__init__-s*rEr^r GenAndKill[Value]c:ttfSrArrs rC visit_branchzDefinedVisitor.visit_branch0rrEr-c:ttfSrArrs rC visit_returnzDefinedVisitor.visit_return3rrEr5c:ttfSrArrs rCvisit_unreachablez DefinedVisitor.visit_unreachable6rrEr,c:ttfSrArrs rCrz DefinedVisitor.visit_register_op9rrEr ct|jtr)|jjs|jrt |jhfS|jht fSrA)r\srcr" undefinesrradestrs rCrzDefinedVisitor.visit_assign<sR bfn - - $263C $tGY $5527)# #G9cee# #rEr c:ttfSrArrs rCrz!DefinedVisitor.visit_assign_multiCsuucee|rEr/c:ttfSrArrs rCrzDefinedVisitor.visit_set_memGrrENF)rrYr>r?r^r r>rr^r-r>rr^r5r>rr^r,r>rr^r r>rr^r r>rr^r/r>r) rQrRrSrTrDrrrrrrrrUrErCrrs  +++++$$$$rErrinitial_defined set[Value]AnalysisResult[Value]cNt||t|dtS)zCalculate potentially defined registers at each CFG location. A register is defined if it has a value along some path from the initial location. FrWr gen_and_killinitialbackwardkind) run_analysisrMAYBE_ANALYSIS)rWrrs rCanalyze_maybe_defined_regsrKs3  #%%     rEregsIterable[Value]rc nt||t||dtt|S)aICalculate always defined registers at each CFG location. This analysis can work before exception insertion, since it is a sound assumption that registers defined in a block might not be initialized in its error handler. A register is defined if it has a value along all paths from the initial location. rFrWrrrrruniverse)rr MUST_ANALYSISra)rWrrrrs rCanalyze_must_defined_regsr\sA  #-@@@ T   rEcNeZdZddZdd Zdd Zdd ZddZddZddZ ddZ dS)BorrowedArgumentsVisitorargsrr>r?c||_dSrA)r )rBr s rCrDz!BorrowedArgumentsVisitor.__init__xs  rEr^r rc:ttfSrArrs rCrz%BorrowedArgumentsVisitor.visit_branch{rrEr-c:ttfSrArrs rCrz%BorrowedArgumentsVisitor.visit_return~rrEr5c:ttfSrArrs rCrz*BorrowedArgumentsVisitor.visit_unreachablerrEr,c:ttfSrArrs rCrz*BorrowedArgumentsVisitor.visit_register_oprrEr c|j|jvrt|jhfSttfSrA)rr rars rCrz%BorrowedArgumentsVisitor.visit_assigns6 7di  5527)# #uucee|rEr c:ttfSrArrs rCrz+BorrowedArgumentsVisitor.visit_assign_multirrEr/c:ttfSrArrs rCrz&BorrowedArgumentsVisitor.visit_set_memrrEN)r rr>r?rrrrrrr) rQrRrSrDrrrrrrrrUrErCr r ws rEr borrowedc Rt||t||dt|S)zCalculate arguments that can use references borrowed from the caller. When assigning to an argument, it no longer is borrowed. Fr)rr r)rWrrs rCanalyze_borrowed_argumentsrs8  -h77    rEcFeZdZddZddZdd Zdd Zdd ZddZddZ dS)UndefinedVisitorr^r r>rc:ttfSrArrs rCrzUndefinedVisitor.visit_branchrrEr-c:ttfSrArrs rCrzUndefinedVisitor.visit_returnrrEr5c:ttfSrArrs rCrz"UndefinedVisitor.visit_unreachablerrEr,cNt|js|hn tfSrA)rais_voidrs rCrz"UndefinedVisitor.visit_register_ops"uu"*7rdd#%%77rEr c.t|jhfSrArarrs rCrzUndefinedVisitor.visit_assignuurwirEr c.t|jhfSrArrs rCrz#UndefinedVisitor.visit_assign_multir rEr/c:ttfSrArrs rCrzUndefinedVisitor.visit_set_memrrENrrrrrrr) rQrRrSrrrrrrrrUrErCrrs8888        rErr^r(ct}|D]3}t|ttfs||4|SrA)rasourcesr\rrrh)r^resultsources rCnon_trivial_sourcesr'sQ UUF**,,&7E"233  JJv    MrEcVeZdZddZddZdd Zdd Zdd ZddZddZ ddZ ddZ dS) LivenessVisitorr^r r>rc<t|tfSrAr'rars rCrzLivenessVisitor.visit_branch"2&&--rEr-ct|jttfs|jht fSt t fSrA)r\valuerrrars rCrzLivenessVisitor.visit_returns@"(We$455 H:suu$ $55#%%< rEr5c:ttfSrArrs rCrz!LivenessVisitor.visit_unreachablerrEr,cXt|}|js||hfS|tfSrA)r'rra)rBr^gens rCrz!LivenessVisitor.visit_register_ops3!"%%z 9 : rEr c0t||jhfSrAr'rrs rCrzLivenessVisitor.visit_assign"2&& 11rEr c0t||jhfSrAr3rs rCrz"LivenessVisitor.visit_assign_multir4rEr/c<t|tfSrAr+rs rCrzLivenessVisitor.visit_set_memr,rErc:ttfSrArrs rC visit_inc_refzLivenessVisitor.visit_inc_refrrErc:ttfSrArrs rC visit_dec_refzLivenessVisitor.visit_dec_refrrENrrrrrrr)r^rr>r)r^rr>r) rQrRrSrrrrrrrr8r:rUrErCr)r)s....    22222222....rEr)cft||ttdtS)zCalculate live registers at each CFG location. A register is live at a location if it can be read along some CFG path starting from the location. Tr)rr)rar)rWrs rCanalyze_live_regsr<s7  $&&     rErtNrOpVisitor[GenAndKill[T]]rset[T]rrIrr set[T] | NoneAnalysisResult[T]ci}i}|D]z} t} t} | j} |rtt| } | D]*} | |\}}| |z |z} | |z |z} +| || <| || <{t|}|s|t|}i}i}|D]`} |t kr#t|| <t|| <0| Jdt||| <t||| <a|r|j}|j}n|j}|j}|r| }| |||rLd}||D]<}|t||}|t kr |||z}1|||z}=|Jnt|}|||<|||z ||z}|||kr9||D]0}||vr*| || |1|||<|i}i}|D]t} | }||}t| j}|rtt|}|D]3\}} ||||f<| |\}}||z |z}||||f<4u|r||}}t||S)aPRun a general set-based data flow analysis. Args: blocks: All basic blocks cfg: Control-flow graph for the code gen_and_kill: Implementation of gen and kill functions for each op initial: Value of analysis for the entry points (for a forward analysis) or the exit points (for a backward analysis) kind: MUST_ANALYSIS or MAYBE_ANALYSIS backward: If False, the analysis is a forward analysis; it's backward otherwise universe: For a must analysis, the set of all possible values. This is the starting value for the work list algorithm, which will narrow this down until reaching a fixed point. For a maybe analysis the iteration always starts from an empty set and this argument is ignored. Return analysis results: (before, after) Nz,Universe must be defined for a must analysis)rarcrfreversedacceptreverserr9r;popremoverjrhrxr) rWrrrrrr block_gen block_killrnr1killrcr^opgenopkillworklistworksetrrrmrlrJ new_beforer; new_afterr9 op_beforeop_aftercurops_enumidxs rCrrs{4IJ ! !eeuui  &x}}%%C + +BIIl33ME6<5(C5LF*DD % 5F||H (mmG')F&(E)) > ! !EEF5M55E%LL'')W'''MMF5Mx==E%LL8888 ! u E? &(,J  . .%!$U4[!1!1JJ^++%+-JJ%+-JJ))))WJ"u *U"33y7GG e $ $  & &w&&OOD)))KK%%% e - !279I57H ' 'Um-6uy-A-A  0X//H ' 'GC$'IeSj !IIl33ME6<5(C#&HUCZ  ' 2') )X . ..rE)rWrXrVrYr>r8)rJr r>r )rWrXr>r?)rWrXrr8rrr>rr) rWrXrr8rrrrrrYr>r)rWrXrr8rrr>r)r^r(r>r)rWrXrr8r>rrA)rWrXrr8rr=rr>rrIrrYrr?r>r@)SrT __future__rabcrcollections.abcrrtypingrr mypyc.ir.opsr r r r r rrrrrrrrrrrrrrrrrrr r!r"r#r$r%r&r'r(r)r*r+r,r-r.r/r0r1r2r3r4r5r6r8rrrvrrdicttuplerIra AnalysisDictr GenAndKillrrrrr rrr'r)r<rrrrUrErCr^s""""""........########////////////////////////////////////////////////////////////////////////////////////////////////dGGGGGGGG.=B************Z 6 GCLLE*c/*CF23 ?????WQZ???3q63q6> " i*i*i*i*i*)JqM2i*i*i*X+++++(/+++\, 62598$*51.!!!!!)%0!!!H    " #l/l/l/l/l/l/l/rE