telemetry: Add GetRendererProcessFromTab to timeline model.

tools/telemetry/telemetry/core/timeline/model.py

 # Copyright (c) 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.
 '''A container for timeline-based events and traces and can handle importing
 raw event data from different sources. This model closely resembles that in the
 trace_viewer project:
 https://code.google.com/p/trace-viewer/
 '''
 
 from operator import attrgetter
 
 import telemetry.core.timeline.process as tracing_process
+from telemetry.core import web_contents
+from telemetry.core import browser
 
 # Register importers for data
 from telemetry.core.timeline import inspector_importer
 from telemetry.core.timeline import bounds
 from telemetry.core.timeline import trace_event_importer
 
 _IMPORTERS = [
     inspector_importer.InspectorTimelineImporter,
     trace_event_importer.TraceEventTimelineImporter
 ]
@@ skipping 11 matching lines @@
         'Overlapping timeline markers found')
 
 
 class TimelineModel(object):
   def __init__(self, event_data=None, shift_world_to_zero=True):
     self._bounds = bounds.Bounds()
     self._processes = {}
     self._frozen = False
     self.import_errors = []
     self.metadata = []
+    self._core_object_to_timeline_container_map = {}
 
     if event_data is not None:
       self.ImportTraces([event_data], shift_world_to_zero=shift_world_to_zero)
 
   @property
   def bounds(self):
     return self._bounds
 
   @property
   def processes(self):
@@ skipping 107 matching lines @@
     for (i, event) in enumerate(events):
       if event.name != labels[i]:
         raise MarkerMismatchError()
     for i in xrange(0, len(events)):
       for j in xrange(i+1, len(events)):
         if (events[j].start < events[i].start + events[i].duration):
           raise MarkerOverlapError()
 
     return events
 
+  def GetRendererProcessFromTab(self, tab):
+    return self._core_object_to_timeline_container_map[tab]
+
+  def AddCoreObjectToContainerMapping(self, core_object, container):
+    """ Add a mapping from a core object to a timeline container.
+
+    Used for example to map a Tab to its renderer process in the timeline model.
+    """
+    assert(isinstance(core_object, web_contents.WebContents) or
+           isinstance(core_object, browser.Browser))
+    self._core_object_to_timeline_container_map[core_object] = container
+
   def _CreateImporter(self, event_data):
     for importer_class in _IMPORTERS:
       if importer_class.CanImport(event_data):
         return importer_class(self, event_data)
     raise ValueError("Could not find an importer for the provided event data")