io_scene_gltf2/io/com/gltf2_io_debug.py

   1 # SPDX-License-Identifier: Apache-2.0
   2 # Copyright 2018-2021 The glTF-Blender-IO authors.
   3
   4 #
   5 # Imports
   6 #
   7
   8 import time
   9 import logging
  10
  11 #
  12 # Globals
  13 #
  14
  15 OUTPUT_LEVELS = ['ERROR', 'WARNING', 'INFO', 'PROFILE', 'DEBUG', 'VERBOSE']
  16
  17 g_current_output_level = 'DEBUG'
  18 g_profile_started = False
  19 g_profile_start = 0.0
  20 g_profile_end = 0.0
  21 g_profile_delta = 0.0
  22
  23 #
  24 # Functions
  25 #
  26
  27
  28 def set_output_level(level):
  29     """Set an output debug level."""
  30     global g_current_output_level
  31
  32     if OUTPUT_LEVELS.index(level) < 0:
  33         return
  34
  35     g_current_output_level = level
  36
  37
  38 def print_console(level, output):
  39     """Print to Blender console with a given header and output."""
  40     global OUTPUT_LEVELS
  41     global g_current_output_level
  42
  43     if OUTPUT_LEVELS.index(level) > OUTPUT_LEVELS.index(g_current_output_level):
  44         return
  45
  46     print(get_timestamp() + " | " + level + ': ' + output)
  47
  48
  49 def print_newline():
  50     """Print a new line to Blender console."""
  51     print()
  52
  53
  54 def get_timestamp():
  55     current_time = time.gmtime()
  56     return time.strftime("%H:%M:%S", current_time)
  57
  58
  59 def print_timestamp(label=None):
  60     """Print a timestamp to Blender console."""
  61     output = 'Timestamp: ' + get_timestamp()
  62
  63     if label is not None:
  64         output = output + ' (' + label + ')'
  65
  66     print_console('PROFILE', output)
  67
  68
  69 def profile_start():
  70     """Start profiling by storing the current time."""
  71     global g_profile_start
  72     global g_profile_started
  73
  74     if g_profile_started:
  75         print_console('ERROR', 'Profiling already started')
  76         return
  77
  78     g_profile_started = True
  79
  80     g_profile_start = time.time()
  81
  82
  83 def profile_end(label=None):
  84     """Stop profiling and printing out the delta time since profile start."""
  85     global g_profile_end
  86     global g_profile_delta
  87     global g_profile_started
  88
  89     if not g_profile_started:
  90         print_console('ERROR', 'Profiling not started')
  91         return
  92
  93     g_profile_started = False
  94
  95     g_profile_end = time.time()
  96     g_profile_delta = g_profile_end - g_profile_start
  97
  98     output = 'Delta time: ' + str(g_profile_delta)
  99
 100     if label is not None:
 101         output = output + ' (' + label + ')'
 102
 103     print_console('PROFILE', output)
 104
 105
 106 # TODO: need to have a unique system for logging importer/exporter
 107 # TODO: this logger is used for importer, but in io and in blender part, but is written here in a _io_ file
 108 class Log:
 109     def __init__(self, loglevel):
 110         self.logger = logging.getLogger('glTFImporter')
 111         self.hdlr = logging.StreamHandler()
 112         formatter = logging.Formatter('%(asctime)s %(levelname)s %(message)s')
 113         self.hdlr.setFormatter(formatter)
 114         self.logger.addHandler(self.hdlr)
 115         self.logger.setLevel(int(loglevel))