o Æ µiG ã@s¾ddlmZddlmZmZmZmZmZmZm Z m Z Gdd„deƒZ de fdd„Z de dd fd d „Zde fd d „Zde dd fdd„Zdefdd„Zdedd fdd„Zde de dd fdd„Zd S)é)ÚEnum)ÚLogLevelÚ_get_is_colored_output_onÚ_get_logging_prefixÚ_get_reportable_log_levelÚ_logÚ_set_is_colored_output_onÚ_set_logging_prefixÚ_set_reportable_log_levelc@sJeZdZdZejZejZej Z ej Z ej ZejZedddefdd„ƒZdS)ÚLevelzKEnum to set the minimum required logging level to print a message to stdoutÚexternalÚreturncCsp|tjkrtjS|tjkrtjS|tjkrtjS|tjkr tj S|tj kr(tj S|tj kr0tj St|ƒtdƒ‚)NzUnknown log severity)r Ú InternalErrorrÚINTERNAL_ERRORÚErrorÚERRORÚWarningÚWARNINGÚInfoÚINFOÚDebugÚDEBUGÚGraphÚGRAPHÚprintÚ ValueError)r ©rúM/home/ubuntu/.local/lib/python3.10/site-packages/torch_tensorrt/ts/logging.pyÚ_to_internal_levels      zLevel._to_internal_levelN)Ú__name__Ú __module__Ú __qualname__Ú__doc__rrrrrrrrrrrrrÚ staticmethodrrrrrr sr r cCó ttƒƒS)z^Get the prefix set for logging messages Returns: str: Prefix used for logger )ÚstrrrrrrÚget_logging_prefix,ó r&ÚprefixNcCó t|ƒdS)zrSet the prefix used when logging messages Args: prefix (str): Prefix to use for logging messages N)r )r(rrrÚset_logging_prefix5ó r*cCr$)z¨Get the level required for a message to be printed in the log Returns: torch_tensorrt.logging.Level: The enum representing the level required to print )r rrrrrÚget_reportable_log_level>r'r,ÚlevelcCstt |¡ƒdS)z­Set the level required for a message to be printed to the log Args: level (torch_tensorrt.logging.Level): The enum representing the level required to print N)r r r)r-rrrÚset_reportable_log_levelGsr.cCr$)zfGet if colored output is enabled for logging Returns: bool: If colored output is one )ÚboolrrrrrÚget_is_colored_output_onPr'r0Úcolored_output_oncCr))z†Enable or disable color in the log output Args: colored_output_on (bool): If colored output should be enabled or not N)r)r1rrrÚset_is_colored_output_onYr+r2ÚmsgcCs8tt |¡|ƒtj}tj}tj}tj}tj}tj }dS)aAdd a new message to the log Adds a new message to the log at a specified level. The message will only get printed out if Level > reportable_log_level Args: level (torch_tensorrt.logging.Level): Severity of the message msg (str): Actual message text N) rr rrrrrrrr)r-r3rrrrrrrrrÚlogbs  r4)ÚenumrÚtorch_tensorrt._Crrrrrrr r r r%r&r*r,r.r/r0r2r4rrrrÚs (