sandwich: Implement SandwichTaskBuilder

tools/android/loading/sandwich_metrics.py

 # 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.
 
 """Pull a sandwich run's output directory's metrics from traces into a CSV.
 
 python pull_sandwich_metrics.py -h
 """
 
 import collections
 import logging
 import os
 import shutil
 import sys
 import tempfile
 
 _SRC_DIR = os.path.abspath(os.path.join(
     os.path.dirname(__file__), '..', '..', '..'))
 
 sys.path.append(os.path.join(_SRC_DIR, 'tools', 'perf'))
 from chrome_telemetry_build import chromium_config
 
 sys.path.append(chromium_config.GetTelemetryDir())
 from telemetry.internal.image_processing import video
 from telemetry.util import image_util
 from telemetry.util import rgba_color
 
 import loading_trace as loading_trace_module
+import sandwich_runner
 import tracing
 
 
-# List of selected trace event categories when running chrome.
-CATEGORIES = [
-    # Need blink network trace events for prefetch_view.PrefetchSimulationView
-    'blink.net',
-
-    # Need to get mark trace events for _GetWebPageTrackedEvents()
-    'blink.user_timing',
-
-    # Need to memory dump trace event for _GetBrowserDumpEvents()
-    'disabled-by-default-memory-infra']
-
 CSV_FIELD_NAMES = [
     'id',
     'url',
     'total_load',
     'onload',
     'browser_malloc_avg',
     'browser_malloc_max',
     'speed_index']
 
 _TRACKED_EVENT_NAMES = set(['requestStart', 'loadEventStart', 'loadEventEnd'])
@@ skipping 187 matching lines @@
   assert os.path.isdir(output_directory_path)
   metrics = []
   for node_name in os.listdir(output_directory_path):
     if not os.path.isdir(os.path.join(output_directory_path, node_name)):
       continue
     try:
       page_id = int(node_name)
     except ValueError:
       continue
     run_path = os.path.join(output_directory_path, node_name)
-    trace_path = os.path.join(run_path, 'trace.json')
+    trace_path = os.path.join(run_path, sandwich_runner.TRACE_FILENAME)
     if not os.path.isfile(trace_path):
       continue
     logging.info('processing \'%s\'' % trace_path)
     loading_trace = loading_trace_module.LoadingTrace.FromJsonFile(trace_path)
     row_metrics = {key: 'unavailable' for key in CSV_FIELD_NAMES}
     row_metrics.update(_PullMetricsFromLoadingTrace(loading_trace))
     row_metrics['id'] = page_id
     row_metrics['url'] = loading_trace.url
-    video_path = os.path.join(run_path, 'video.mp4')
+    video_path = os.path.join(run_path, sandwich_runner.VIDEO_FILENAME)
     if os.path.isfile(video_path):
       logging.info('processing \'%s\'' % video_path)
       completeness_record = _ExtractCompletenessRecordFromVideo(video_path)
       row_metrics['speed_index'] = ComputeSpeedIndex(completeness_record)
     metrics.append(row_metrics)
   assert len(metrics) > 0, ('Looks like \'{}\' was not a sandwich ' +
                             'run directory.').format(output_directory_path)
   return metrics