Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(243)

Side by Side Diff: build/android/chrome_profiler/profiler.py

Issue 290013006: adb_profile_chrome: Refactor into multiple modules and add tests (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: Review comments. Created 6 years, 7 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch | Annotate | Revision Log
OLDNEW
(Empty)
1 # Copyright 2014 The Chromium Authors. All rights reserved.
2 # Use of this source code is governed by a BSD-style license that can be
3 # found in the LICENSE file.
4
5 import gzip
6 import os
7 import shutil
8 import sys
9 import zipfile
10
11 from chrome_profiler import ui
12 from chrome_profiler import util
13
14 from pylib import constants
15
16 sys.path.append(os.path.join(constants.DIR_SOURCE_ROOT,
17 'third_party',
18 'trace-viewer'))
19 # pylint: disable=F0401
20 from trace_viewer.build import trace2html
21
22
23 def _CompressFile(host_file, output):
24 with gzip.open(output, 'wb') as out:
25 with open(host_file, 'rb') as input_file:
26 out.write(input_file.read())
27 os.unlink(host_file)
28
29
30 def _ArchiveFiles(host_files, output):
31 with zipfile.ZipFile(output, 'w', zipfile.ZIP_DEFLATED) as z:
32 for host_file in host_files:
33 z.write(host_file)
34 os.unlink(host_file)
35
36
37 def _PackageTracesAsHtml(trace_files, html_file):
38 with open(html_file, 'w') as f:
39 trace2html.WriteHTMLForTracesToFile(trace_files, f)
40 for trace_file in trace_files:
41 os.unlink(trace_file)
42
43
44 def _StartTracing(controllers, interval):
45 for controller in controllers:
46 controller.StartTracing(interval)
47
48
49 def _StopTracing(controllers):
50 for controller in controllers:
51 controller.StopTracing()
52
53
54 def _PullTraces(controllers, output, compress, write_json):
55 ui.PrintMessage('Downloading...', eol='')
56 trace_files = []
57 for controller in controllers:
58 trace_files.append(controller.PullTrace())
59
60 if not write_json:
61 html_file = os.path.splitext(trace_files[0])[0] + '.html'
62 _PackageTracesAsHtml(trace_files, html_file)
63 trace_files = [html_file]
64
65 if compress and len(trace_files) == 1:
66 result = output or trace_files[0] + '.gz'
67 _CompressFile(trace_files[0], result)
68 elif len(trace_files) > 1:
69 result = output or 'chrome-combined-trace-%s.zip' % util.GetTraceTimestamp()
70 _ArchiveFiles(trace_files, result)
71 elif output:
72 result = output
73 shutil.move(trace_files[0], result)
74 else:
75 result = trace_files[0]
76
77 ui.PrintMessage('done')
78 ui.PrintMessage('Trace written to file://%s' % os.path.abspath(result))
79 return result
80
81
82 def GetSupportedBrowsers():
83 """Returns the package names of all supported browsers."""
84 # Add aliases for backwards compatibility.
85 supported_browsers = {
86 'stable': constants.PACKAGE_INFO['chrome_stable'],
87 'beta': constants.PACKAGE_INFO['chrome_beta'],
88 'dev': constants.PACKAGE_INFO['chrome_dev'],
89 'build': constants.PACKAGE_INFO['chrome'],
90 }
91 supported_browsers.update(constants.PACKAGE_INFO)
92 unsupported_browsers = ['content_browsertests', 'gtest', 'legacy_browser']
93 for browser in unsupported_browsers:
94 del supported_browsers[browser]
95 return supported_browsers
96
97
98 def CaptureProfile(controllers, interval, output=None, compress=False,
99 write_json=False):
100 """Records a profiling trace saves the result to a file.
101
102 Args:
103 controllers: List of tracing controllers.
104 interval: Time interval to capture in seconds. An interval of None (or 0)
105 continues tracing until stopped by the user.
106 output: Output file name or None to use an automatically generated name.
107 compress: If True, the result will be compressed either with gzip or zip
108 depending on the number of captured subtraces.
109 write_json: If True, prefer JSON output over HTML.
110
111 Returns:
112 Path to saved profile.
113 """
114 trace_type = ' + '.join(map(str, controllers))
115 try:
116 _StartTracing(controllers, interval)
117 if interval:
118 ui.PrintMessage('Capturing %d-second %s. Press Enter to stop early...' % \
119 (interval, trace_type), eol='')
120 ui.WaitForEnter(interval)
121 else:
122 ui.PrintMessage('Capturing %s. Press Enter to stop...' % \
123 trace_type, eol='')
124 raw_input()
125 finally:
126 _StopTracing(controllers)
127 if interval:
128 ui.PrintMessage('done')
129
130 return _PullTraces(controllers, output, compress, write_json)
OLDNEW
« no previous file with comments | « build/android/chrome_profiler/main.py ('k') | build/android/chrome_profiler/profiler_unittest.py » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698