# 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.
import collections

from metrics import Metric

TRACING_MODE = 'tracing-mode'
TIMELINE_MODE = 'timeline-mode'

class TimelineMetric(Metric):
  def __init__(self, mode):
    assert mode in (TRACING_MODE, TIMELINE_MODE)
    super(TimelineMetric, self).__init__()
    self._mode = mode
    self._model = None
    self._thread_for_tab = None

  def Start(self, page, tab):
    self._model = None
    self._thread_for_tab = None

    if self._mode == TRACING_MODE:
      if not tab.browser.supports_tracing:
        raise Exception('Not supported')
      tab.browser.StartTracing()
    else:
      assert self._mode == TIMELINE_MODE
      tab.StartTimelineRecording()

  def Stop(self, page, tab):
    if self._mode == TRACING_MODE:
      # This creates an async trace event in the render process for tab that
      # will allow us to find that tab during the AddTracingResultsForTab
      # function.
      tab.ExecuteJavaScript("""
          console.time("__loading_measurement_was_here__");
          console.timeEnd("__loading_measurement_was_here__");
          """)

      trace_result = tab.browser.StopTracing()
      self._model = trace_result.AsTimelineModel()
      events = [s for
                s in self._model.GetAllEventsOfName(
                    '__loading_measurement_was_here__')
                if s.parent_slice == None]
      assert len(events) == 1
      self._thread_for_tab = events[0].start_thread
    else:
      tab.StopTimelineRecording()
      self._model = tab.timeline_model
      self._thread_for_tab = self._model.GetAllThreads()[0]

  def AddResults(self, tab, results):
    assert self._model

    events = self._thread_for_tab.all_slices

    events_by_name = collections.defaultdict(list)
    for e in events:
      events_by_name[e.name].append(e)

    for event_name, event_group in events_by_name.iteritems():
      times = [event.self_time for event in event_group]
      total = sum(times)
      biggest_jank = max(times)
      results.Add(event_name, 'ms', total)
      results.Add(event_name + '_max', 'ms', biggest_jank)
      results.Add(event_name + '_avg', 'ms', total / len(times))