re-land: Use browser compositor rendering stats in smoothness

tools/perf/metrics/rendering_stats.py

 # Copyright 2013 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 operator import attrgetter
 
 # These are LatencyInfo component names indicating the various components
 # that the input event has travelled through.
 # This is when the input event first reaches chrome.
 UI_COMP_NAME = 'INPUT_EVENT_LATENCY_UI_COMPONENT'
 # This is when the input event was originally created by OS.
 ORIGINAL_COMP_NAME = 'INPUT_EVENT_LATENCY_ORIGINAL_COMPONENT'
 # This is when the input event was sent from browser to renderer.
 BEGIN_COMP_NAME = 'INPUT_EVENT_LATENCY_BEGIN_RWH_COMPONENT'
 # This is when the input event has reached swap buffer.
 END_COMP_NAME = 'INPUT_EVENT_LATENCY_TERMINATED_FRAME_SWAP_COMPONENT'
 
+
 def GetScrollInputLatencyEvents(scroll_type, browser_process, timeline_range):
   """Get scroll events' LatencyInfo from the browser process's trace buffer
      that are within the timeline_range.
 
   Scroll events (MouseWheel, GestureScrollUpdate or JS scroll on TouchMove)
   dump their LatencyInfo into trace buffer as async trace event with name
   "InputLatency". The trace event has a memeber 'step' containing its event
   type and a memeber 'data' containing its latency history.
 
   """
   scroll_events = []
   if not browser_process:
     return scroll_events
   for event in browser_process.IterAllAsyncSlicesOfName("InputLatency"):
     if event.start >= timeline_range.min and event.end <= timeline_range.max:
       for ss in event.sub_slices:
         if 'step' not in ss.args:
           continue
         if 'data' not in ss.args:
           continue
         if ss.args['step'] == scroll_type:
           scroll_events.append(ss)
   return scroll_events
 
+
 def ComputeMouseWheelScrollLatency(mouse_wheel_events):
   """ Compute the mouse wheel scroll latency.
 
   Mouse wheel scroll latency is the time from when mouse wheel event is sent
   from browser RWH to renderer (the timestamp of component
   'INPUT_EVENT_LATENCY_BEGIN_RWH_COMPONENT') to when the scrolled page is
   buffer swapped (the timestamp of component
   'INPUT_EVENT_LATENCY_TERMINATED_FRAME_SWAP_COMPONENT')
 
   """
   mouse_wheel_latency = []
   for event in mouse_wheel_events:
     data = event.args['data']
     if BEGIN_COMP_NAME in data and END_COMP_NAME in data:
       latency = data[END_COMP_NAME]['time'] - data[BEGIN_COMP_NAME]['time']
       mouse_wheel_latency.append(latency / 1000.0)
   return mouse_wheel_latency
 
+
 def ComputeTouchScrollLatency(touch_scroll_events):
   """ Compute the touch scroll latency.
 
   Touch scroll latency is the time from when the touch event is created to
   when the scrolled page is buffer swapped.
   Touch event on differnt platforms uses different LatencyInfo component to
   record its creation timestamp. On Aura, the creation time is kept in
   'INPUT_EVENT_LATENCY_UI_COMPONENT' . On Android, the creation time is kept
   in 'INPUT_EVENT_LATENCY_ORIGINAL_COMPONENT'.
 
   """
   touch_scroll_latency = []
   for event in touch_scroll_events:
     data = event.args['data']
     if END_COMP_NAME in data:
       end_time = data[END_COMP_NAME]['time']
       if UI_COMP_NAME in data and ORIGINAL_COMP_NAME in data:
         raise ValueError, 'LatencyInfo has both UI and ORIGINAL component'
       if UI_COMP_NAME in data:
         latency = end_time - data[UI_COMP_NAME]['time']
         touch_scroll_latency.append(latency / 1000.0)
       elif ORIGINAL_COMP_NAME in data:
         latency = end_time - data[ORIGINAL_COMP_NAME]['time']
         touch_scroll_latency.append(latency / 1000.0)
   return touch_scroll_latency
 
+
+def HasRenderingStats(process):
+  """ Returns True if the process contains at least one
+      BenchmarkInstrumentation::*RenderingStats event with a frame.
+  """
+  if not process:
+    return False
+  for event in process.IterAllSlicesOfName(
+      'BenchmarkInstrumentation::MainThreadRenderingStats'):
+    if 'data' in event.args and event.args['data']['frame_count'] == 1:
+      return True
+  for event in process.IterAllSlicesOfName(
+      'BenchmarkInstrumentation::ImplThreadRenderingStats'):
+    if 'data' in event.args and event.args['data']['frame_count'] == 1:
+      return True
+  return False
+
+
 class RenderingStats(object):
   def __init__(self, renderer_process, browser_process, timeline_ranges):
     """
     Utility class for extracting rendering statistics from the timeline (or
     other loggin facilities), and providing them in a common format to classes
     that compute benchmark metrics from this data.
 
     Stats are lists of lists of numbers. The outer list stores one list per
     timeline range.
 
     All *_time values are measured in milliseconds.
     """
     assert(len(timeline_ranges) > 0)
-    self.renderer_process = renderer_process
+    # Find the top level process with rendering stats (browser or renderer).
+    if HasRenderingStats(browser_process):
+      self.top_level_process = browser_process
+    else:
+      self.top_level_process  = renderer_process
 
     self.frame_timestamps = []
     self.frame_times = []
     self.paint_times = []
     self.painted_pixel_counts = []
     self.record_times = []
     self.recorded_pixel_counts = []
     self.rasterize_times = []
     self.rasterized_pixel_counts = []
     # End-to-end latency for MouseWheel scroll - from when mouse wheel event is
@@ skipping 36 matching lines @@
     self.touch_scroll_latency = ComputeTouchScrollLatency(touch_scroll_events)
 
     js_touch_scroll_events = GetScrollInputLatencyEvents(
         "TouchMove", browser_process, timeline_range)
     self.js_touch_scroll_latency = ComputeTouchScrollLatency(
         js_touch_scroll_events)
 
   def initMainThreadStatsFromTimeline(self, timeline_range):
     event_name = 'BenchmarkInstrumentation::MainThreadRenderingStats'
     events = []
-    for event in self.renderer_process.IterAllSlicesOfName(event_name):
+    for event in self.top_level_process.IterAllSlicesOfName(event_name):
       if event.start >= timeline_range.min and event.end <= timeline_range.max:
         if 'data' not in event.args:
           continue
         events.append(event)
     events.sort(key=attrgetter('start'))
 
     first_frame = True
     for event in events:
       frame_count = event.args['data']['frame_count']
       if frame_count > 1:
@@ skipping 10 matching lines @@
       self.painted_pixel_counts[-1].append(
           event.args['data']['painted_pixel_count'])
       self.record_times[-1].append(1000.0 *
           event.args['data']['record_time'])
       self.recorded_pixel_counts[-1].append(
           event.args['data']['recorded_pixel_count'])
 
   def initImplThreadStatsFromTimeline(self, timeline_range):
     event_name = 'BenchmarkInstrumentation::ImplThreadRenderingStats'
     events = []
-    for event in self.renderer_process.IterAllSlicesOfName(event_name):
+    for event in self.top_level_process.IterAllSlicesOfName(event_name):
       if event.start >= timeline_range.min and event.end <= timeline_range.max:
         if 'data' not in event.args:
           continue
         events.append(event)
     events.sort(key=attrgetter('start'))
 
     first_frame = True
     for event in events:
       frame_count = event.args['data']['frame_count']
       if frame_count > 1:
         raise ValueError, 'trace contains multi-frame render stats'
       if frame_count == 1:
         self.frame_timestamps[-1].append(
             event.start)
         if not first_frame:
           self.frame_times[-1].append(round(self.frame_timestamps[-1][-1] -
                                             self.frame_timestamps[-1][-2], 2))
         first_frame = False
       self.rasterize_times[-1].append(1000.0 *
           event.args['data']['rasterize_time'])
       self.rasterized_pixel_counts[-1].append(
           event.args['data']['rasterized_pixel_count'])