1 files changed, 182 insertions, 0 deletions
diff --git a/tensorflow/python/platform/default/_logging.py b/tensorflow/python/platform/default/_logging.py
new file mode 100644
index 0000000000..2e289b1abe
--- /dev/null
+++ b/tensorflow/python/platform/default/_logging.py
@@ -0,0 +1,182 @@
+"""Logging utilities."""
+# pylint: disable=unused-import
+# pylint: disable=g-bad-import-order
+# pylint: disable=invalid-name
+import os
+import sys
+import time
+import thread
+from logging import getLogger
+from logging import log
+from logging import debug
+from logging import error
+from logging import fatal
+from logging import info
+from logging import warn
+from logging import warning
+from logging import DEBUG
+from logging import ERROR
+from logging import FATAL
+from logging import INFO
+from logging import WARN
+
+# Controls which methods from pyglib.logging are available within the project
+# Do not add methods here without also adding to platform/default/_logging.py
+__all__ = ['log', 'debug', 'error', 'fatal', 'info', 'warn', 'warning',
+           'DEBUG', 'ERROR', 'FATAL', 'INFO', 'WARN',
+           'flush', 'log_every_n', 'log_first_n', 'vlog',
+           'TaskLevelStatusMessage', 'get_verbosity', 'set_verbosity']
+
+warning = warn
+
+_level_names = {
+    FATAL: 'FATAL',
+    ERROR: 'ERROR',
+    WARN: 'WARN',
+    INFO: 'INFO',
+    DEBUG: 'DEBUG',
+}
+
+# Mask to convert integer thread ids to unsigned quantities for logging
+# purposes
+_THREAD_ID_MASK = 2 * sys.maxint + 1
+
+_log_prefix = None  # later set to google2_log_prefix
+
+# Counter to keep track of number of log entries per token.
+_log_counter_per_token = {}
+
+
+def TaskLevelStatusMessage(msg):
+  error(msg)
+
+
+def flush():
+  raise NotImplementedError()
+
+
+# Code below is taken from pyglib/logging
+def vlog(level, msg, *args, **kwargs):
+  log(level, msg, *args, **kwargs)
+
+
+def _GetNextLogCountPerToken(token):
+  """Wrapper for _log_counter_per_token.
+
+  Args:
+    token: The token for which to look up the count.
+
+  Returns:
+    The number of times this function has been called with
+    *token* as an argument (starting at 0)
+  """
+  global _log_counter_per_token  # pylint: disable=global-variable-not-assigned
+  _log_counter_per_token[token] = 1 + _log_counter_per_token.get(token, -1)
+  return _log_counter_per_token[token]
+
+
+def log_every_n(level, msg, n, *args):
+  """Log 'msg % args' at level 'level' once per 'n' times.
+
+  Logs the 1st call, (N+1)st call, (2N+1)st call,  etc.
+  Not threadsafe.
+
+  Args:
+    level: The level at which to log.
+    msg: The message to be logged.
+    n: The number of times this should be called before it is logged.
+    *args: The args to be substituted into the msg.
+  """
+  count = _GetNextLogCountPerToken(_GetFileAndLine())
+  log_if(level, msg, not (count % n), *args)
+
+
+def log_first_n(level, msg, n, *args):  # pylint: disable=g-bad-name
+  """Log 'msg % args' at level 'level' only first 'n' times.
+
+  Not threadsafe.
+
+  Args:
+    level: The level at which to log.
+    msg: The message to be logged.
+    n: The number of times this should be called before it is logged.
+    *args: The args to be substituted into the msg.
+  """
+  count = _GetNextLogCountPerToken(_GetFileAndLine())
+  log_if(level, msg, count < n, *args)
+
+
+def log_if(level, msg, condition, *args):
+  """Log 'msg % args' at level 'level' only if condition is fulfilled."""
+  if condition:
+    vlog(level, msg, *args)
+
+
+def _GetFileAndLine():
+  """Returns (filename, linenumber) for the stack frame."""
+  # Use sys._getframe().  This avoids creating a traceback object.
+  # pylint: disable=protected-access
+  f = sys._getframe()
+  # pylint: enable=protected-access
+  our_file = f.f_code.co_filename
+  f = f.f_back
+  while f:
+    code = f.f_code
+    if code.co_filename != our_file:
+      return (code.co_filename, f.f_lineno)
+    f = f.f_back
+  return ('<unknown>', 0)
+
+
+def google2_log_prefix(level, timestamp=None, file_and_line=None):
+  """Assemble a logline prefix using the google2 format."""
+  # pylint: disable=global-variable-not-assigned
+  global _level_names
+  global _logfile_map, _logfile_map_mutex
+  # pylint: enable=global-variable-not-assigned
+
+  # Record current time
+  now = timestamp or time.time()
+  now_tuple = time.localtime(now)
+  now_microsecond = int(1e6 * (now % 1.0))
+
+  (filename, line) = file_and_line or _GetFileAndLine()
+  basename = os.path.basename(filename)
+
+  # Severity string
+  severity = 'I'
+  if level in _level_names:
+    severity = _level_names[level][0]
+
+  s = '%c%02d%02d %02d:%02d:%02d.%06d %5d %s:%d] ' % (
+      severity,
+      now_tuple[1],  # month
+      now_tuple[2],  # day
+      now_tuple[3],  # hour
+      now_tuple[4],  # min
+      now_tuple[5],  # sec
+      now_microsecond,
+      _get_thread_id(),
+      basename,
+      line)
+
+  return s
+
+
+def get_verbosity():
+  """Return how much logging output will be produced."""
+  return getLogger().getEffectiveLevel()
+
+
+def set_verbosity(verbosity):
+  """Sets the threshold for what messages will be logged."""
+  getLogger().setLevel(verbosity)
+
+
+def _get_thread_id():
+  """Get id of current thread, suitable for logging as an unsigned quantity."""
+  thread_id = thread.get_ident()
+  return thread_id & _THREAD_ID_MASK
+
+
+_log_prefix = google2_log_prefix