tools/cygprofile/cygprofile_utils.py

   1 #!/usr/bin/python
   2 # Copyright 2015 The Chromium Authors. All rights reserved.
   3 # Use of this source code is governed by a BSD-style license that can be
   4 # found in the LICENSE file.
   5
   6 """Common utilites used by cygprofile scripts.
   7 """
   8
   9 import logging
  10 import os
  11 import re
  12
  13 class WarningCollector(object):
  14   """Collects warnings, but limits the number printed to a set value."""
  15   def __init__(self, max_warnings):
  16     self._warnings = 0
  17     self._max_warnings = max_warnings
  18
  19   def Write(self, message):
  20     """Print a warning if fewer than max_warnings have already been printed."""
  21     if self._warnings < self._max_warnings:
  22       logging.warning(message)
  23     self._warnings += 1
  24
  25   def WriteEnd(self, message):
  26     """Once all warnings have been printed, use this to print the number of
  27     elided warnings."""
  28     if self._warnings > self._max_warnings:
  29       logging.warning('%d more warnings for: %s' % (
  30           self._warnings - self._max_warnings, message))
  31
  32
  33 def DetectArchitecture(default='arm'):
  34   """Detects the architecture by looking for target_arch in GYP_DEFINES.
  35   If not not found, returns default.
  36   """
  37   gyp_defines = os.environ.get('GYP_DEFINES', '')
  38   match = re.match('target_arch=(\S+)', gyp_defines)
  39   if match and len(match.groups()) == 1:
  40     return match.group(1)
  41   else:
  42     return default