Index: systrace/profile_chrome/ddms_tracing_agent.py |
diff --git a/systrace/profile_chrome/ddms_tracing_agent.py b/systrace/profile_chrome/ddms_tracing_agent.py |
index 276a3b9fb055a92ef9b11770fed647a8dd82b803..35c29b1187b73d66c103228b1ae57cc83ba921c9 100644 |
--- a/systrace/profile_chrome/ddms_tracing_agent.py |
+++ b/systrace/profile_chrome/ddms_tracing_agent.py |
@@ -10,6 +10,7 @@ import re |
from profile_chrome import util |
from systrace import trace_result |
from systrace import tracing_agents |
+from telemetry.timeline import trace_data as trace_data_module |
_DDMS_SAMPLING_FREQUENCY_US = 100 |
@@ -54,6 +55,11 @@ class DdmsAgent(tracing_agents.TracingAgent): |
trace_data = f.read() |
return trace_result.TraceResult('ddms', trace_data) |
+ def CollectAgentTraceData(self, trace_data_builder): |
+ with open(self._PullTrace(), 'r') as f: |
+ trace_data = f.read() |
+ trace_data_builder(trace_data_module.DDMS_TRACE_PART, trace_data) |
+ |
def _PullTrace(self): |
if not self._output_file: |
return None |