o 1&i$1@sjdZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z m Z ddlmZddlmZmZGdddejZegd Zd d ZGd d d ZeeZddZddZddZGdddejZGdddeZ GdddeZ!e ddZ"e ddZ#e ddZ$d+d d!Z%d,d"d#Z&e d+d$d%Z'd&e!fd'd(Z(d)d*Z)ej*re)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)configutilsc@s eZdZdZeZeZdS) EventStatuszStatus of an event. N)__name__ __module__ __qualname____doc__enumautoSTARTENDrr;C:\wamp64\www\opt\env\Lib\site-packages\numba/core/event.pyr2s r)znumba:compiler_lockz numba:compileznumba:llvm_locknumba:run_passcCs(|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)kindmsgrrr _guard_kindBs rc@sneZdZdZdddZeddZeddZed d Zed d Z ed dZ eddZ ddZ 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__``. NcCs>t||_||_||_|dus|ddurd|_dS||_dSNr)r_kind_status_data _exc_details)selfrstatusdata exc_detailsrrr__init__ds  zEvent.__init__cC|jS)zFEvent kind Returns ------- res : str )rr rrrrlz Event.kindcCr%)zPEvent status Returns ------- res : EventStatus )rr&rrrr!vr'z Event.statuscCr%)zIEvent data Returns ------- res : object )rr&rrrr"r'z Event.datacC |jtjkS)zSIs it a *START* event? Returns ------- res : bool )rrrr&rrris_start zEvent.is_startcCr()zRIs it an *END* event? Returns ------- res : bool )rrrr&rrris_endr*z Event.is_endcCs |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)rr&rrr is_faileds zEvent.is_failedcCs8|jdur t|jjnd}d|jd|jd|dS)NNonezEvent(z, z, data: ))r"typer rr)r r"rrr__str__s z Event.__str__NN)r r r r r$propertyrr!r"r)r+r,r0__repr__rrrrrXs"        rcCs(t|tsJt|}t||dS)zvRegister a listener for a given event kind. Parameters ---------- kind : str listener : Listener N) isinstanceListenerr _registeredappendrlistenerrrrregistersr:cCs,t|tsJt|}t|}||dS)zxUnregister a listener for a given event kind. Parameters ---------- kind : str listener : Listener N)r4r5rr6remove)rr9lstrrr unregistersr=cCst|jD]}||qdS)zeBroadcast an event to all registered listeners. Parameters ---------- event : Event N)r6rnotify)eventr9rrr broadcasts r@c@s4eZdZdZejddZejddZddZdS) r5z(Base class for all event listeners. cCdS)zkCalled when there is a *START* event. Parameters ---------- event : Event Nrr r?rrron_startzListener.on_startcCrA)ziCalled when there is a *END* event. Parameters ---------- event : Event NrrBrrron_endrDzListener.on_endcCs0|jr ||dS|jr||dStd)zpNotify this Listener with the given Event. Parameters ---------- event : Event Z unreachableN)r)rCr+rEAssertionErrorrBrrrr>s zListener.notifyN) r r r r abcabstractmethodrCrEr>rrrrr5s   r5c@s@eZdZdZddZddZddZedd Zed d Z d S) TimingListenerzA listener that measures the total time spent between *START* and *END* events during the time this listener is active. cCs d|_dSr)_depthr&rrrr$ zTimingListener.__init__cCs$|jdkr t|_|jd7_dS)Nr)rJtimer_tsrBrrrrC s zTimingListener.on_startcCs>|jd8_|jdkrt|dd}t|j||_dSdS)NrLr _duration)rJgetattrrMrNrO)r r?lastrrrrEs   zTimingListener.on_endcCs 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. rO)hasattrr&rrrdones zTimingListener.donecCr%)zReturns the measured duration. This may raise ``AttributeError``. Users can use ``.done`` to check that a measurement has been made. )rOr&rrrdurationszTimingListener.durationN) r r r r r$rCrEr2rSrTrrrrrIs  rIc@s(eZdZdZddZddZddZdS) 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. cCs g|_dSN)bufferr&rrrr$.rKzRecordingListener.__init__cC|jt|fdSrVrWr7timerBrrrrC1zRecordingListener.on_startcCrXrVrYrBrrrrE4r[zRecordingListener.on_endN)r r r r r$rCrErrrrrU(s  rUc cs0t||z |VWt||dSt||w)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)r:r=r8rrrinstall_listener8s  r\ccsPt}t|| |VWdn1swY|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)rIr\rSrT)rcallbacktlrrr install_timerQs r_ccs>t}t|| |VWddS1swYdS)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)rUr\)rZrlrrrinstall_recorderns  "r`cCst|tj|d}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)rrrr@)rr"evtrrr start_events rbcCst|tj||d}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)rrrr@)rr"r#rarrr end_events  rcc#sTt}|jfdd}tddVWddS1s#wYdS)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. cst|ddS)N)r"r#)rc)r#r"rrron_exitsztrigger_event..on_exit)r"N)rpushrb)rr"scopererrdr trigger_events  "rhr9c Cszt}t}g}|jD]-\}}|j}t|j}|d}|jr"dnd} |d} |} t ||||| | | d} | | q |S)zGPrepare events in `listener` for serializing as chrome trace data. i@BBEname)catpidtidtsphrkargs) osgetpid threading get_native_idrWr"strrr)dictr7) r9rmrnevsroZrecr"rlZ ts_scaledrprkrqZevrrr_prepare_chrome_trace_datas    rycs.ttdtjtjfdd}dS)z\Setup a RecordingListener and an exit handler to write the captured events to file. rcsJt}td}tj||tjdWddS1swYdS)Nw)cls)ryopenjsondumprZ_LazyJSONEncoder)rxoutfilenamer9rr_write_chrome_traces "z=_setup_chrome_trace_exit_handler.._write_chrome_traceN)rUr:r CHROME_TRACEatexit)rrrr _setup_chrome_trace_exit_handlers  rrVr1)+r rrr}rrGr rZrtZtimeitrrM contextlibrr collectionsrZ numba.corerrEnumr frozensetrrrlistr6r:r=r@ABCr5rIrUr\r_r`rbrcrhryrrrrrrsL#   [  &&