blob: 97008a2047f86aa91f99ebec0472bafaf0a5cc8b (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
|
#!/usr/bin/python
# Copyright 2007 Google Inc. All rights reserved.
"""Extract UMA histogram strings from the Chrome source.
This program generates the list of known histograms we expect to see in
the user behavior logs. It walks the Chrome source, looking for calls
to UMA histogram macros.
Run it from the chrome/browser directory like:
extract_histograms.py > histogram_list
"""
# TODO(evanm): get all the jankometer histogram names.
__author__ = 'evanm (Evan Martin)'
import os
import re
import sys
def GrepForHistograms(path, histograms):
"""Grep a source file for calls to histogram macros functions.
Arguments:
path: path to the file
histograms: set of histograms to add to
"""
histogram_re = re.compile(r'HISTOGRAM_\w+\(L"(.*)"')
for line in open(path):
match = histogram_re.search(line)
if match:
histograms.add(match.group(1))
def WalkDirectory(root_path, histograms):
for path, dirs, files in os.walk(root_path):
if '.svn' in dirs:
dirs.remove('.svn')
for file in files:
ext = os.path.splitext(file)[1]
if ext == '.cc':
GrepForHistograms(os.path.join(path, file), histograms)
def main(argv):
histograms = set()
# Walk the source tree to process all .cc files.
WalkDirectory('..', histograms)
# Print out the histograms as a sorted list.
for histogram in sorted(histograms):
print histogram
if '__main__' == __name__:
main(sys.argv)
|
