piUdZddlmZmZmZmZddlmZddlm Z m Z ddl m Z m Z ddlmZmZ ddlZdZed ZgZeged fed <d ed fdZGdd e ZGddeZej4dZer6e dGddej8j:Zdeded efdZ yy#e$rd ZYwxYw)zw Implement code coverage support. Currently contains logic to extend ``coverage`` with lines covered by the compiler. )OptionalSequenceCallable no_type_check)Mapping) dataclassfield)ABCabstractmethod)irconfigNTFcRd}trtjj}|S)z>Get active coverage instance or return None if not found. N)coverage_availablecoverageCoveragecurrent)covs ]/opt/services/ai/voice_agent/venv/lib/python3.12/site-packages/numba/misc/coverage_support.pyget_active_coveragers% C'') J NotifyLocBase _the_registryreturnchtjsgSttddtDS)zC Returns a list of the registered NotifyLocBase instances. c |duSN)xs rz+get_registered_loc_notify..,s $rc3*K|] }| ywrr).0factorys r z,get_registered_loc_notify..-s?g ?s)r JIT_COVERAGElistfilterrrrrget_registered_loc_notifyr'%s4    .??A BBrcPeZdZdZedej ddfdZeddZy)rzr?r@rArBrC NumbaTracer_tracertracersappend)r+ collectorrFs r__init__zNotifyCompilerCoverage.__init__Ds#"4499;  ^^#--%11$++&33#,#?#?^^%.%C%C(77%NN66  #3]3   .rr)c|j}|jjdr4|jd|j ||jdyy)Nz.pynumba_compiled)rLfilenameendswithrBtrace)r+r)tracers rr,zNotifyCompilerCoverage.notifyZsK <<  '  ! !"2 3 LL   ! !$ ' (rcyrrr/s rr0zNotifyCompilerCoverage.closeas rN) r2r3r4r5rPr r6r,r0rrrr8r8<s!/,("&&( rr8cdt}|$|j}|jr t|Syyr)rrDrMr8)rcols r_register_coverage_notifierrZes4  C nn ;;)#. . r)kw_onlyc2eZdZUdZej j ed<eed<ej jed<e e ej jdzfed<ej jdzed<ee dzgdfdzed<egdfed <egdfed <ej jed <eed <ed Zeed<dej j$dzfdZddZdefdZddZdee efdzfdZdej6ddfdZy)rKa  Not actually a tracer as in the coverage implementation, which will setup a Python trace function. This implementation pretends to trace but instead receives fake trace events for each line the compiler has visited. See coverage.PyTracer r:r=r>Nr?rArBr;r<r@rCF)default _activityrcy)zPStart this tracer, return a trace function if based on sys.settrace.Nrr/s rstartzNumbaTracer.startsrcy)zStop this tracer.Nrr/s rstopzNumbaTracer.stoprc|jS)zHas there been any activity?r^r/s ractivityzNumbaTracer.activitys >> !rcd|_y)zReset the activity() flag.FNrer/s rreset_activityzNumbaTracer.reset_activitys "DNrcy)z+Return a dictionary of statistics, or None.Nrr/s r get_statszNumbaTracer.get_statsrcrr)c|jj|j}|5|j|jd}||j|j<|jsy|j }|j |jj|t}|jrd|jr0|jt|j|jnC|j|j|jfn|j|j|jd|_y)z8Insert coverage data given source location. NT)r?getrSr>rUsource_filenamer;r: setdefaultsetr=rCadd _pack_arcsliner<r^)r+r)disp tracename cur_file_datas rrUzNumbaTracer.traces**..s||>22T9    "d " # tCH~4  "RVV " "rrKl1l2cTd}|dkr|dz}| }|dkr|dz}| }||dz|zz}|S)zPack arcs into a single integer for compatibility with .packed_arcs option. See https://github.com/nedbat/coveragepy/blob/e7c05fe91ee36c0c94e144bb88d25db4fc3d02fd/coverage/ctracer/tracer.c#L171 rllr)rrpackeds rrqrqsP 6 g FB 6 g FB28r/! r)!r5typingrrrrcollections.abcr dataclassesrr abcr r numba.corer r rr ImportErrorrrrxr'rr8rNrZrvTracerrKrrqrrrrs ?>#(#!:< xH_556;B8O#<B  C  & ]& R//tF"hnn++F"F"PsW{sB66C?C