1@j$1dZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z m Z ddlmZddlmZmZGddejZegd Zd ZGd d ZeeZd ZdZdZGddejZGddeZ GddeZ!e dZ"e dZ#e dZ$ddZ%d dZ&e ddZ'de!fdZ(dZ)ej*r e)dSdS)!ag The ``numba.core.event`` module provides a simple event system for applications to register callbacks to listen to specific compiler events. The following events are built in: - ``"numba:compile"`` is broadcast when a dispatcher is compiling. Events of this kind have ``data`` defined to be a ``dict`` with the following key-values: - ``"dispatcher"``: the dispatcher object that is compiling. - ``"args"``: the argument types. - ``"return_type"``: the return type. - ``"numba:compiler_lock"`` is broadcast when the internal compiler-lock is acquired. This is mostly used internally to measure time spent with the lock acquired. - ``"numba:llvm_lock"`` is broadcast when the internal LLVM-lock is acquired. This is used internally to measure time spent with the lock acquired. - ``"numba:run_pass"`` is broadcast when a compiler pass is running. - ``"name"``: pass name. - ``"qualname"``: qualified name of the function being compiled. - ``"module"``: module name of the function being compiled. - ``"flags"``: compilation flags. - ``"args"``: argument types. - ``"return_type"`` return type. Applications can register callbacks that are listening for specific events using ``register(kind: str, listener: Listener)``, where ``listener`` is an instance of ``Listener`` that defines custom actions on occurrence of the specific event. N) default_timer)contextmanager ExitStack) defaultdict)configutilscNeZdZdZejZejZdS) EventStatuszStatus of an event. N)__name__ __module__ __qualname____doc__enumautoSTARTENDAC:\Projects\Historical\venv\Lib\site-packages\numba/core/event.pyr r 2s2 DIKKE $)++CCCrr )znumba:compiler_lockz numba:compileznumba:llvm_locknumba:run_passcj|dr|tvr|d}t||S)a>Guard to ensure that an event kind is valid. All event kinds with a "numba:" prefix must be defined in the pre-defined ``numba.core.event._builtin_kinds``. Custom event kinds are allowed by not using the above prefix. Parameters ---------- kind : str Return ------ res : str znumba:zG is not a valid event kind, it starts with the reserved prefix 'numba:') startswith_builtin_kinds ValueError)kindmsgs r _guard_kindrBsG x  T%?%?===oo KrceZdZdZd dZedZedZedZedZ edZ ed Z d Z e Z dS) EventzAn event. Parameters ---------- kind : str status : EventStatus data : any; optional Additional data for the event. exc_details : 3-tuple; optional Same 3-tuple for ``__exit__``. Ncpt||_||_||_||ddn||_dSNr)r_kind_status_data _exc_details)selfrstatusdata exc_detailss r__init__zEvent.__init__dsF &&   + 3{1~7M"T"- rc|jS)zFEvent kind Returns ------- res : str )r"r&s rrz Event.kindl zrc|jS)zPEvent status Returns ------- res : EventStatus )r#r,s rr'z Event.statusvs |rc|jS)zIEvent data Returns ------- res : object )r$r,s rr(z Event.datar-rc,|jtjkS)zSIs it a *START* event? Returns ------- res : bool )r#r rr,s ris_startzEvent.is_starts|{000rc,|jtjkS)zRIs it an *END* event? Returns ------- res : bool )r#r rr,s ris_endz Event.is_ends|{..rc|jduS)zIs the event carrying an exception? This is used for *END* event. This method will never return ``True`` in a *START* event. Returns ------- res : bool N)r%r,s r is_failedzEvent.is_faileds D((rct|jt|jjnd}d|jd|jd|dS)NNonezEvent(z, z, data: ))r(typer r"r#)r&r(s r__str__z Event.__str__sM9( ??/11.4 C CCdlCCDCCCCrNN)r r r rr*propertyrr'r(r1r3r5r:__repr__rrrrrXs  ////XXX11X1//X/ ) )X )DDD HHHrrct|tsJt|}t||dS)zvRegister a listener for a given event kind. Parameters ---------- kind : str listener : Listener N) isinstanceListenerr _registeredappendrlisteners rregisterrEsF h ) ))) ) t  DX&&&&&rct|tsJt|}t|}||dS)zxUnregister a listener for a given event kind. Parameters ---------- kind : str listener : Listener N)r?r@rrAremove)rrDlsts r unregisterrIsI h ) ))) ) t  D d CJJxrcZt|jD]}||dS)zeBroadcast an event to all registered listeners. Parameters ---------- event : Event N)rArnotify)eventrDs r broadcastrMs9  +rcXeZdZdZejdZejdZdZdS)r@z(Base class for all event listeners. cdS)zkCalled when there is a *START* event. Parameters ---------- event : Event Nrr&rLs ron_startzListener.on_start  rcdS)ziCalled when there is a *END* event. Parameters ---------- event : Event NrrPs ron_endzListener.on_endrRrc|jr||dS|jr||dSt d)zpNotify this Listener with the given Event. Parameters ---------- event : Event unreachableN)r1rQr3rTAssertionErrorrPs rrKzListener.notifysV > 0 MM% \ 0 KK      // /rN) r r r rabcabstractmethodrQrTrKrrrr@r@si        0 0 0 0 0rr@cPeZdZdZdZdZdZedZedZ dS)TimingListenerzA listener that measures the total time spent between *START* and *END* events during the time this listener is active. cd|_dSr!)_depthr,s rr*zTimingListener.__init__s  rcb|jdkrt|_|xjdz c_dS)Nr)r]timer_tsrPs rrQzTimingListener.on_start s. ;!  wwDH q rc|xjdzc_|jdkr1t|dd}t|jz |z|_dSdS)Nr_r _duration)r]getattrr`rarc)r&rLlasts rrTzTimingListener.on_endsS q ;!  4a00D#gg0D8DNNN  rc"t|dS)zReturns a ``bool`` indicating whether a measurement has been made. When this returns ``False``, the matching event has never fired. If and only if this returns ``True``, ``.duration`` can be read without error. rc)hasattrr,s rdonezTimingListener.donest[)))rc|jS)zReturns the measured duration. This may raise ``AttributeError``. Users can use ``.done`` to check that a measurement has been made. )rcr,s rdurationzTimingListener.durations ~rN) r r r rr*rQrTr<rhrjrrrr[r[s 999 **X*Xrr[c$eZdZdZdZdZdZdS)RecordingListeneraA listener that records all events and stores them in the ``.buffer`` attribute as a list of 2-tuple ``(float, Event)``, where the first element is the time the event occurred as returned by ``time.time()`` and the second element is the event. cg|_dSN)bufferr,s rr*zRecordingListener.__init__.s  rc`|jtj|fdSrnrorBtimerPs rrQzRecordingListener.on_start1) DIKK/00000rc`|jtj|fdSrnrqrPs rrTzRecordingListener.on_end4rsrN)r r r rr*rQrTrrrrlrl(sK 11111111rrlc#~Kt|| |Vt||dS#t||wxYw)aInstall a listener for event "kind" temporarily within the duration of the context. Returns ------- res : Listener The *listener* provided. Examples -------- >>> with install_listener("numba:compile", listener): >>> some_code() # listener will be active here. >>> other_code() # listener will be unregistered by this point. N)rErIrCs rinstall_listenerrv8sR$ T8#4""""" 4""""s*<c#Kt}t||5|Vdddn #1swxYwY|jr||jdSdS)aInstall a TimingListener temporarily to measure the duration of an event. If the context completes successfully, the *callback* function is executed. The *callback* function is expected to take a float argument for the duration in seconds. Returns ------- res : TimingListener Examples -------- This is equivalent to: >>> with install_listener(kind, TimingListener()) as res: >>> ... N)r[rvrhrj)rcallbacktls r install_timerrzQs*   B $ # # ws 266c#Kt}t||5|VddddS#1swxYwYdS)anInstall a RecordingListener temporarily to record all events. Once the context is closed, users can use ``RecordingListener.buffer`` to access the recorded events. Returns ------- res : RecordingListener Examples -------- This is equivalent to: >>> with install_listener(kind, RecordingListener()) as res: >>> ... N)rlrv)rrls rinstall_recorderr}ns&   B $ # #s 377c\t|tj|}t|dS)zTrigger the start of an event of *kind* with *data*. Parameters ---------- kind : str Event kind. data : any; optional Extra event data. )rr'r(N)rr rrM)rr(evts r start_eventrs* T+"3$ ? ? ?C cNNNNNrc^t|tj||}t|dS)aTrigger the end of an event of *kind*, *exc_details*. Parameters ---------- kind : str Event kind. data : any; optional Extra event data. exc_details : 3-tuple; optional Same 3-tuple for ``__exit__``. Or, ``None`` if no error. )rr'r(r)N)rr rrM)rr(r)rs r end_eventrs4  +/+   CcNNNNNrc#Kt5}|jfd}tdVddddS#1swxYwYdS)a;A context manager to trigger the start and end events of *kind* with *data*. The start event is triggered when entering the context. The end event is triggered when exiting the context. Parameters ---------- kind : str Event kind. data : any; optional Extra event data. c,t|dS)N)r(r))r)r)r(rs ron_exitztrigger_event..on_exits d; ? ? ? ? ? ?r)r(N)rpushr)rr(scopers`` r trigger_eventrs   @ @ @ @ @  @ Dt$$$$  s)A  A A rDc 0tj}tj}g}|jD]e\}}|j}t |j}|dz}|jrdnd} |d} |} t||||| | | } | | f|S)zGPrepare events in `listener` for serializing as chrome trace data. i@BBEname)catpidtidtsphrargs) osgetpid threading get_native_idror(strrr1dictrB) rDrrevsrrecr(r ts_scaledrrrevs r_prepare_chrome_trace_datars )++C  ! # #C C?  Cx#(mmN L )SScF| #)    2 Jrcttdtjtjfd}dS)z\Setup a RecordingListener and an exit handler to write the captured events to file. rct}td5}tj||tjddddS#1swxYwYdS)Nw)cls)ropenjsondumpr_LazyJSONEncoder)routfilenamerDs r_write_chrome_tracez=_setup_chrome_trace_exit_handler.._write_chrome_traces)22 (C  rs!!F ))))))00000000######$$$$$$$$$),XXXXXXXXvk$  ' ' '   #0#0#0#0#0sw#0#0#0L#####X###L 1 1 1 1 1 1 1 1 ###08.    $*):0 < < <  '$$&&&&&''r