Teach mojo_benchmark --save-traces to pull results from Android devices.

Fixes domokit/devtools#48.

R=etiennej@chromium.org

Review URL: https://codereview.chromium.org/1376603007 .

Cr-Mirrored-From: https://github.com/domokit/mojo
Cr-Mirrored-Commit: 7ee3c9f45ef8313b91d2be987cdc22cb18891551

devtoolslib/android_shell.py

diff --git a/devtoolslib/android_shell.py b/devtoolslib/android_shell.py
index 2833138..275a3dc 100644
--- a/devtoolslib/android_shell.py
+++ b/devtoolslib/android_shell.py
@@ -220,6 +220,21 @@
     return len(subprocess.check_output(self._adb_command([
         'shell', 'pm', 'list', 'packages', _MOJO_SHELL_PACKAGE_NAME]))) > 0
 
+  @staticmethod
+  def get_tmp_dir_path():
+    """Returns a path to a cache directory owned by the shell where temporary
+    files can be stored.
+    """
+    return '/data/data/%s/cache/tmp/' % _MOJO_SHELL_PACKAGE_NAME
+
+  def pull_file(self, device_path, destination_path, remove_original=False):
+    """Copies or moves the specified file on the device to the host."""
+    subprocess.check_call(self._adb_command([
+        'pull', device_path, destination_path]))
+    if remove_original:
+      subprocess.check_call(self._adb_command([
+          'shell', 'rm', device_path]))
+
   def check_device(self, require_root=False):
     """Verifies if the device configuration allows adb to run.
 

mojo_benchmark

diff --git a/mojo_benchmark b/mojo_benchmark
index 9b24268..476b8e3 100755
--- a/mojo_benchmark
+++ b/mojo_benchmark
@@ -9,6 +9,7 @@
 import logging
 import sys
 import time
+import os.path
 
 from devtoolslib import shell_arguments
 from devtoolslib import shell_config
@@ -53,8 +54,16 @@
 _EXTRA_TIMEOUT = 20
 
 
+def _get_output_file(shell, name, cold_start):
+    file_name = 'benchmark-%s-%s-%s.trace' % (
+        name.replace(' ', '_'),
+        'cold_start' if cold_start else 'warm_start',
+        time.strftime('%Y%m%d%H%M%S'))
+    return file_name
+
+
 def _run_benchmark(shell, shell_args, name, app, duration_seconds, measurements,
-                   cold_start, verbose, save_traces):
+                   cold_start, verbose, android, save_traces):
   """Runs `benchmark.mojo` in shell with correct arguments, parses and
   presents the benchmark results.
   """
@@ -62,12 +71,16 @@
   benchmark_args = []
   benchmark_args.append('--app=' + app)
   benchmark_args.append('--duration=' + str(duration_seconds))
+
+  output_file = None
+  device_output_file = None
   if save_traces:
-    trace_output_file = 'benchmark-%s-%s-%s.trace' % (
-        name.replace(' ', '_'),
-        'cold_start' if cold_start else 'warm_start',
-        time.strftime('%Y%m%d%H%M%S'))
-    benchmark_args.append('--trace-output=' + trace_output_file)
+    output_file = _get_output_file(shell, name, cold_start)
+    if android:
+      device_output_file = os.path.join(shell.get_tmp_dir_path(), output_file)
+      benchmark_args.append('--trace-output=' + device_output_file)
+    else:
+      benchmark_args.append('--trace-output=' + output_file)
 
   for measurement in measurements:
     benchmark_args.append(measurement)
@@ -102,6 +115,9 @@
   for line in output_lines:
     if line.strip().startswith('measurement:') or 'WARNING' in line:
       print line
+
+  if device_output_file:
+    shell.pull_file(device_output_file, output_file, remove_original=True)
   return True
 
 
@@ -139,9 +155,11 @@
     measurements = benchmark_spec['measurements']
     _run_benchmark(shell, shell_args, name, app, duration, measurements,
                    cold_start=True, verbose=script_args.verbose,
+                   android=script_args.android,
                    save_traces=script_args.save_traces)
     _run_benchmark(shell, shell_args, name, app, duration, measurements,
                    cold_start=False, verbose=script_args.verbose,
+                   android=script_args.android,
                    save_traces=script_args.save_traces)
 
   return 0 if succeeded else 1