| # Copyright 2016 The Chromium Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| from systrace.tracing_agents import atrace_agent |
| from telemetry.internal.platform import tracing_agent |
| from tracing.trace_data import trace_data |
| |
| from devil.android.sdk import version_codes |
| |
| |
| class AtraceTracingAgent(tracing_agent.TracingAgent): |
| def __init__(self, platform_backend): |
| super(AtraceTracingAgent, self).__init__(platform_backend) |
| self._device = platform_backend.device |
| self._categories = None |
| self._atrace_agent = atrace_agent.AtraceAgent( |
| platform_backend.device.build_version_sdk) |
| self._config = None |
| |
| @classmethod |
| def IsSupported(cls, platform_backend): |
| return (platform_backend.GetOSName() == 'android' and |
| platform_backend.device.build_version_sdk > |
| version_codes.JELLY_BEAN_MR1) |
| |
| def StartAgentTracing(self, config, timeout): |
| if not config.enable_atrace_trace: |
| return False |
| |
| app_name = (','.join(config.atrace_config.app_name) if |
| isinstance(config.atrace_config.app_name, list) else |
| config.atrace_config.app_name) |
| self._config = atrace_agent.AtraceConfig( |
| config.atrace_config.categories, |
| trace_buf_size=None, kfuncs=None, app_name=app_name, |
| compress_trace_data=True, from_file=True, |
| device_serial_number=str(self._device), trace_time=None, |
| target='android') |
| return self._atrace_agent.StartAgentTracing(self._config, timeout) |
| |
| def StopAgentTracing(self): |
| self._atrace_agent.StopAgentTracing() |
| |
| def SupportsExplicitClockSync(self): |
| return self._atrace_agent.SupportsExplicitClockSync() |
| |
| def RecordClockSyncMarker(self, sync_id, |
| record_controller_clock_sync_marker_callback): |
| return self._atrace_agent.RecordClockSyncMarker(sync_id, |
| lambda t, sid: record_controller_clock_sync_marker_callback(sid, t)) |
| |
| def CollectAgentTraceData(self, trace_data_builder, timeout=None): |
| raw_data = self._atrace_agent.GetResults(timeout).raw_data |
| trace_data_builder.AddTraceFor(trace_data.ATRACE_PART, raw_data) |
