oPjdZddlmZddlZddlmZmZddlmZddl m Z m Z m Z GddZ dd Zdd Zed d d ZeddZedZej&j)ddZej&dZeej0eej2zZeDchc]}d ej6|vs|c}ZGddZycc}w)z!Bytecode analysis for coverage.py) annotationsN)IterableMapping)CodeType)TArcTLineNoTOffsetcPeZdZdZdddd d dZd dZd dZd dZd dZy) ByteParserz3Parse bytecode to understand the structure of code.N)codetextfilenamecD||Jt||xsddd}||_y)NzexecT) dont_inherit)compiler )selfr r rs V/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/coverage/bytecode.py__init__zByteParser.__init__s1 <# ##4!7ZdSD c0d|jDS)a|Iterate over all the code objects nested within this one. The iteration includes `self` as its first value. We skip code objects named `__annotate__` since they are deferred annotations that usually are never run. If there are errors in the annotations, they will be caught by type checkers or other tools that use annotations. c3TK|] }|jdk7st|"yw) __annotate__)r N)co_namer ).0cs r z,ByteParser._child_parsers..)s$_q199P^C^ ""_s(() code_objects)rs r_child_parserszByteParser._child_parserss`D,=,=,?__rc#K|jg}|rK|j}|jD]$}t|ts|j |&||rJyyw)z,Iterate over all the code objects in `code`.N)r pop co_consts isinstancerappend)rstackr rs rrzByteParser.code_objects+sU 99;D^^ $a*LLO $Js?AAAc#`K|jjD] \}}}|s |yw)zyYield the line numbers possible in this code object. Uses co_lines() to produce a sequence: l0, l1, ... N)r co_lines)r_lines r _line_numberszByteParser._line_numbers7s2 )),,. JAq$  s$..c#jK|jD]}|jEd{y7w)zFind the statements in `self.code`. Produce a sequence of line numbers that start statements. Recurses into all code objects reachable from `self.code`. N)rr*)rbps rfind_statementszByteParser.find_statements@s5%%' *B'') ) ) * )s '313)r zCodeType | Noner str | Nonerr.returnNone)r/zIterable[ByteParser])r/zIterable[CodeType])r/zIterable[TLineNo]) __name__ __module__ __qualname____doc__rrrr*r-rrr r sR= !%#        `  *rr cpi}|jD] \}}}| t||dD]}|||< "|S)z6Make a dict mapping byte code offsets to line numbers.)r'range)r b2lbstartbendlinenoboffsets rbytes_to_linesr>LsQ C $ &f   q1 &%G  && Jrc|Dchc]&}tjj|x}s%|(}}|s Jd||Scc}w)zMake a set of opcodes from instruction names. The names might not exist in this version of Python, skip those if not. z At least one opcode must exist: )disopmapget)op_namesnameopopss rop_setrGVsN # B$SYY]]4-@'@r'@2 BC B =28*==3 J Cs &AA JUMP_BACKWARDJUMP_BACKWARD_NO_INTERRUPT JUMP_FORWARD RETURN_VALUERETURN_GENERATORRERAISECACHE EXTENDED_ARGc8eZdZdZ ddZddZddZy) BranchArcResolveraDResolve branch events to line arcs, one (source, dest) pair at a time. Branch events are one-shot (they are DISABLEd after firing), so each (source offset, destination offset) pair is resolved at most a couple of times per code object. Resolving pairs on demand is much cheaper than precomputing trails for every branch in the code object, most of which never fire. We walk the raw bytecode bytes so that we never need to disassemble whole code objects with `dis`. To resolve one pair: starting from the destination, follow the trail of instructions (through unconditional jumps) until we reach an instruction on a new source line (giving us the arc), a return (an arc to leaving the code object), or another branch possibility (no arc: that branch will produce its own events). cP||_|j|_||_||_y)N)r co_code byte_to_line multiline_map)rr rUrVs rrzBranchArcResolver.__init__s&  || (*rcx|jj|}||jj||}|S)z>The source line of the instruction at `offset`, de-multilined.)rUrBrV)roffsetr)s rline_atzBranchArcResolver.line_ats;  $$V,  %%))$5D rc|j|}|y|j}t|}|j}|j}|}d} t } d|cxkr|kr$ny|| vr| j |||} | tk(r|dz }9| tk(r| ||dzzdz} |dz }V|j|} | |j| | } | r | |k7r|| fS| tvr]| tvrT| ||dzz} |dz}||kr#||tk(r|dz }||kr ||tk(r| tvr |d| zz }n|d| zz}d} y| tvr||jj fS| t vryd} |dz }d|cxkr|kr ny|| vry)zBTurn a branch event's (source, dest) offsets into an arc, or None.Nrr7)rYrTlenrUrVsetadd_CACHE _EXTENDED_ARGrBJUMPS ALWAYS_JUMPSBACKWARD_JUMPSRETURNSr co_firstlinenoNO_FALL_THROUGH)rsourcedest from_linerT max_offsetrUrVrXext_argseenrEr)arg next_offsets rresolvezBranchArcResolver.resolvesLL(  ,,\ (( **  U6&J&LM,2+= HHV BV|! ]""WVaZ%88Q>! ##F+D$((t4DI-%t,,U{%!GFQJ$77C#)1*K% 2w{7Kv7U#q( & 2w{7Kv7U^+!,q3w!6!,q3w!6GW}!DII$<$<#<==_$G aKFK6&J&LM,2+=LrN)r rrUzMapping[TOffset, TLineNo]rVzMapping[TLineNo, TLineNo]r/r0)rXr r/zTLineNo | None)rhr rir r/z TArc | None)r1r2r3r4rrYrpr5rrrRrR~s<$ + +0 +1 +  +2rrR)r rr/zdict[TOffset, TLineNo])rCstrr/zset[int])r4 __future__rr@collections.abcrrtypesrcoverage.typesrrr r r>rGrcrergrArBr`rar^hasjrelhasjabsrbopnamerdrR)rEs0rrys(" -11:*:*z    # w # .)  CKK3s{{++ %J3::b>(I"JXXKs &C=C