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

Side by Side Diff: tools/profile_chrome/chrome_controller.py

Issue 1738563002: Remove //tools/profile_chrome from Chromium repo (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Created 4 years, 10 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
« no previous file with comments | « tools/profile_chrome/__init__.py ('k') | tools/profile_chrome/chrome_controller_unittest.py » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
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 json
6 import os
7 import re
8 import time
9
10 from profile_chrome import controllers
11
12 from devil.android import device_errors
13 from devil.android.sdk import intent
14
15
16 _HEAP_PROFILE_MMAP_PROPERTY = 'heapprof.mmap'
17
18 class ChromeTracingController(controllers.BaseController):
19 def __init__(self, device, package_info,
20 categories, ring_buffer, trace_memory=False):
21 controllers.BaseController.__init__(self)
22 self._device = device
23 self._package_info = package_info
24 self._categories = categories
25 self._ring_buffer = ring_buffer
26 self._logcat_monitor = self._device.GetLogcatMonitor()
27 self._trace_file = None
28 self._trace_interval = None
29 self._trace_memory = trace_memory
30 self._is_tracing = False
31 self._trace_start_re = \
32 re.compile(r'Logging performance trace to file')
33 self._trace_finish_re = \
34 re.compile(r'Profiler finished[.] Results are in (.*)[.]')
35
36 def __repr__(self):
37 return 'chrome trace'
38
39 @staticmethod
40 def GetCategories(device, package_info):
41 with device.GetLogcatMonitor() as logmon:
42 device.BroadcastIntent(intent.Intent(
43 action='%s.GPU_PROFILER_LIST_CATEGORIES' % package_info.package))
44 try:
45 json_category_list = logmon.WaitFor(
46 re.compile(r'{"traceCategoriesList(.*)'), timeout=5).group(0)
47 except device_errors.CommandTimeoutError:
48 raise RuntimeError('Performance trace category list marker not found. '
49 'Is the correct version of the browser running?')
50
51 record_categories = set()
52 disabled_by_default_categories = set()
53 json_data = json.loads(json_category_list)['traceCategoriesList']
54 for item in json_data:
55 for category in item.split(','):
56 if category.startswith('disabled-by-default'):
57 disabled_by_default_categories.add(category)
58 else:
59 record_categories.add(category)
60
61 return list(record_categories), list(disabled_by_default_categories)
62
63 def StartTracing(self, interval):
64 self._trace_interval = interval
65 self._logcat_monitor.Start()
66 start_extras = {'categories': ','.join(self._categories)}
67 if self._ring_buffer:
68 start_extras['continuous'] = None
69 self._device.BroadcastIntent(intent.Intent(
70 action='%s.GPU_PROFILER_START' % self._package_info.package,
71 extras=start_extras))
72
73 if self._trace_memory:
74 self._device.EnableRoot()
75 self._device.SetProp(_HEAP_PROFILE_MMAP_PROPERTY, 1)
76
77 # Chrome logs two different messages related to tracing:
78 #
79 # 1. "Logging performance trace to file"
80 # 2. "Profiler finished. Results are in [...]"
81 #
82 # The first one is printed when tracing starts and the second one indicates
83 # that the trace file is ready to be pulled.
84 try:
85 self._logcat_monitor.WaitFor(self._trace_start_re, timeout=5)
86 self._is_tracing = True
87 except device_errors.CommandTimeoutError:
88 raise RuntimeError(
89 'Trace start marker not found. Possible causes: 1) Is the correct '
90 'version of the browser running? 2) Is the browser already launched?')
91
92 def StopTracing(self):
93 if self._is_tracing:
94 self._device.BroadcastIntent(intent.Intent(
95 action='%s.GPU_PROFILER_STOP' % self._package_info.package))
96 self._trace_file = self._logcat_monitor.WaitFor(
97 self._trace_finish_re, timeout=120).group(1)
98 self._is_tracing = False
99 if self._trace_memory:
100 self._device.SetProp(_HEAP_PROFILE_MMAP_PROPERTY, 0)
101
102 def PullTrace(self):
103 # Wait a bit for the browser to finish writing the trace file.
104 time.sleep(self._trace_interval / 4 + 1)
105
106 trace_file = self._trace_file.replace('/storage/emulated/0/', '/sdcard/')
107 host_file = os.path.join(os.path.curdir, os.path.basename(trace_file))
108 try:
109 self._device.PullFile(trace_file, host_file)
110 except device_errors.AdbCommandFailedError:
111 raise RuntimeError(
112 'Cannot pull the trace file. Have you granted Storage permission to '
113 'the browser? (Android Settings -> Apps -> [the browser app] -> '
114 'Permissions -> Storage)')
115 return host_file
OLDNEW
« no previous file with comments | « tools/profile_chrome/__init__.py ('k') | tools/profile_chrome/chrome_controller_unittest.py » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698