| OLD | NEW |
| 1 # Copyright 2016 The Chromium Authors. All rights reserved. | 1 # Copyright 2016 The Chromium Authors. All rights reserved. |
| 2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
| 3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
| 4 | 4 |
| 5 from systrace.tracing_agents import atrace_agent | 5 from systrace.tracing_agents import atrace_agent |
| 6 from telemetry.internal.platform import tracing_agent | 6 from telemetry.internal.platform import tracing_agent |
| 7 from telemetry.timeline import trace_data | 7 from tracing.trace_data import trace_data |
| 8 | 8 |
| 9 from devil.android.sdk import version_codes | 9 from devil.android.sdk import version_codes |
| 10 | 10 |
| 11 | 11 |
| 12 class AtraceTracingAgent(tracing_agent.TracingAgent): | 12 class AtraceTracingAgent(tracing_agent.TracingAgent): |
| 13 def __init__(self, platform_backend): | 13 def __init__(self, platform_backend): |
| 14 super(AtraceTracingAgent, self).__init__(platform_backend) | 14 super(AtraceTracingAgent, self).__init__(platform_backend) |
| 15 self._device = platform_backend.device | 15 self._device = platform_backend.device |
| 16 self._categories = None | 16 self._categories = None |
| 17 self._atrace_agent = atrace_agent.AtraceAgent( | 17 self._atrace_agent = atrace_agent.AtraceAgent( |
| (...skipping 28 matching lines...) Expand all Loading... |
| 46 return self._atrace_agent.SupportsExplicitClockSync() | 46 return self._atrace_agent.SupportsExplicitClockSync() |
| 47 | 47 |
| 48 def RecordClockSyncMarker(self, sync_id, | 48 def RecordClockSyncMarker(self, sync_id, |
| 49 record_controller_clock_sync_marker_callback): | 49 record_controller_clock_sync_marker_callback): |
| 50 return self._atrace_agent.RecordClockSyncMarker(sync_id, | 50 return self._atrace_agent.RecordClockSyncMarker(sync_id, |
| 51 lambda t, sid: record_controller_clock_sync_marker_callback(sid, t)) | 51 lambda t, sid: record_controller_clock_sync_marker_callback(sid, t)) |
| 52 | 52 |
| 53 def CollectAgentTraceData(self, trace_data_builder, timeout=None): | 53 def CollectAgentTraceData(self, trace_data_builder, timeout=None): |
| 54 raw_data = self._atrace_agent.GetResults(timeout).raw_data | 54 raw_data = self._atrace_agent.GetResults(timeout).raw_data |
| 55 trace_data_builder.AddTraceFor(trace_data.ATRACE_PART, raw_data) | 55 trace_data_builder.AddTraceFor(trace_data.ATRACE_PART, raw_data) |
| OLD | NEW |