o oÇ hBã@sRddlmZddlmZddlmZGdd„dƒZdedd fd d „Z dd d „Z d S)é)Úprotoné)Úget_profiling_on©Úwrapsc@s<eZdZdZdeddfdd„Zdd„Zd d d „Zd d „ZdS)Ústateaz A context manager and decorator for entering and exiting a state. Usage: context manager: ```python with proton.state("test0"): foo[1,](x, y) ``` decorator: ```python @proton.state("test0") def foo(x, y): ... ``` Args: name (str): The name of the state. ÚnameÚreturnNcCs ||_dS©N©r)Úselfr©r úi/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/triton/profiler/state.pyÚ__init__s zstate.__init__cCstƒs|St |j¡|Sr )rÚ libprotonÚ enter_stater)r r r rÚ __enter__s zstate.__enter__cCstƒsdSt ¡dSr )rrÚ exit_state)r Úexc_typeÚ exc_valueÚ tracebackr r rÚ__exit__%s zstate.__exit__cstˆƒ‡‡fdd„ƒ}|S)Ncs2tƒr t ˆj¡ˆ|i|¤Ž}tƒrt ¡|Sr )rrrrr)ÚargsÚkwargsÚret©Úfuncr r rÚwrapper,s  zstate.__call__..wrapperr)r rrr rrÚ__call__*szstate.__call__©r N) Ú__name__Ú __module__Ú __qualname__Ú__doc__Ústrrrrrr r r rrs   rrr NcCst |¡dSr )rrr r r rr8srcCs t ¡dSr )rrr r r rr<s rr) Útriton._C.libprotonrrÚflagsrÚ functoolsrrr$rrr r r rÚs  2